code optimize

src/background/actions.ts

@@ -19,14 +19,14 @@ export function redirectTab(tab: chrome.tabs.Tab, url: string) {
                 let queryErr: any;
                 let newURL = await queryUrl(tab).catch(e => queryErr = e);
                 if (queryErr) {
-                    return Promise.reject(queryErr);
+                    throw queryErr;
                 }
                 if (newURL == url) return url;
                 if (
                     tryCount % 1 == 0 &&
                     !confirm('Cannot navigate to target url. \nPress OK to continue, Cancel to stop.')
                 ) {
-                    return Promise.reject("Tasks stopped by user.");
+                    throw "Tasks stopped by user.";
                 }
                 return undefined;
             }

src/background/messaging.ts

@@ -50,46 +50,57 @@ export function sendMessage<T>(
             chrome.tabs.sendMessage(tab.id, req, async (r: Response<T>) => {
                 // check error but do nothing until dataChecker.
                 let err = chrome.runtime.lastError;
-                let result: T;
+                let [result, error] = await checkResponse(dataChecker, r, err, count);
-                // r could be undefined if the content script is interrupted.
+                if (error) {
-                if (r) {
-                    result = r.result;
-
-                    if (dataChecker) {
-                        let pms: T | Promise<T>;
-                        try {
-                            pms = dataChecker(r, err, count);
-                        } catch (error) {
                     reject(error);
                     return;
                 }
+                let flag = result !== undefined;
+                if (log) logger.info(log, flag ? '(OK)' : '(failed)');
+                if (flag) {
+                    resolve(result);
+                } else {
+                    setTimeout(() => {
+                        logger.debug('Invalid response', r, 'retry...');
+                        loop();
+                    }, interval);
+                }
+            });
+        }
+    });
+}
+
+async function checkResponse<T>(
+    dataChecker: ResponseChecker<T>,
+    response: Response<T>,
+    error: chrome.runtime.LastError,
+    tryCount: number
+): Promise<[T, string]> {
+    // response could be undefined if the content script is interrupted.
+    // don't check, tell sendMessage to retry.
+    if (!response) return [undefined, undefined];
+    if (!dataChecker) {
+        return [response.result, response.error];
+    }
+    let result: T;
+    let pms: T | Promise<T>;
+    try {
+        pms = dataChecker(response, error, tryCount);
+    } catch (err) {
+        return [undefined, err];
+    }
     // don't catch if it's not a Promise
     if (pms instanceof Promise) {
         let checkerError: any;
         pms = pms.catch(e => checkerError = e);
         result = await pms;
         if (checkerError) {
-                                reject(checkerError);
+            return [undefined, checkerError];
-                                return;
         }
     } else {
         result = pms;
     }
-                    }
+    return [result, undefined];
-                }
-
-                let flag = result !== undefined && result !== null;
-                if (log) logger.info(log, flag ? '(OK)' : '(failed)');
-                if (flag) {
-                    resolve(result);
-                } else {
-                    setTimeout(() => {
-                        loop();
-                    }, interval);
-                }
-            });
-        }
-    });
 }
 
 export type ActionSubscriberSync = (request: Request, sender: chrome.runtime.MessageSender, sendResponse: (response?: any) => void) => void;

src/background/task.ts

@@ -103,8 +103,8 @@ export class Task {
         messageSubscribers.addListener(Actions.REPORT_NEW_PAGE, listener);
     }
     async execute(tab: chrome.tabs.Tab, upstreamData?: ExtractResult): Promise<void> {
-        if (!tab) return Promise.reject("No tab to execute the task.");
+        if (!tab) throw "No tab to execute the task.";
-        if (this._running) return Promise.reject("The task is running. Please wait...");
+        if (this._running) throw "The task is running. Please wait...";
         this._running = true;
         let urls = this._urls
         if (!urls.length) {
@@ -140,7 +140,7 @@ export class Task {
         ).catch(
             e => {
                 this._running = false;
-                return Promise.reject(e);
+                throw e;
             }
         );
     }
@@ -152,7 +152,7 @@ export class Task {
         return pm;
     }
     private runningCheck(fn: () => Promise<any>): Promise<any> {
-        if (!this._running) return Promise.reject("The task is stopped by user.");
+        if (!this._running) throw "The task is stopped by user.";
         return fn();
     }
     private saveResult(results, key) {