refactor(webetension): curve out textlint backgroun page

textlint · Aug 1, 2020 · 22fd569 · 22fd569
1 parent ee55ff1
commit 22fd569
Show file tree

Hide file tree

Showing 2 changed files with 118 additions and 85 deletions.
diff --git a/packages/webextension/app/scripts/background.ts b/packages/webextension/app/scripts/background.ts
@@ -1,8 +1,7 @@
 import { browser } from "webextension-polyfill-ts";
-import { TextlintFixResult, TextlintMessage, TextlintResult } from "@textlint/types";
-import { TextlintWorkerCommandFix, TextlintWorkerCommandLint, TextlintWorkerCommandResponse } from "@textlint/compiler";
 import { createBackgroundEndpoint, isMessagePort } from "comlink-extension";
 import * as Comlink from "comlink";
+import { createTextlintPage } from "./background/textlint";
 
 browser.runtime.onInstalled.addListener((details) => {
     console.log("previousVersion", details.previousVersion);
@@ -12,96 +11,17 @@ browser.tabs.onUpdated.addListener(async (tabId) => {
     browser.pageAction.show(tabId);
 });
 
-const worker = new Worker("download/textlint.js");
-const waiterForInit = (worker: Worker) => {
-    let initialized = false;
-    let _resolve: null | ((init: boolean) => void) = null;
-    const deferred = new Promise((resolve) => {
-        _resolve = resolve;
-    });
-    worker.addEventListener(
-        "message",
-        function (event) {
-            const data: TextlintWorkerCommandResponse = event.data;
-            if (data.command === "init") {
-                initialized = true;
-                _resolve && _resolve(initialized);
-            }
-        },
-        {
-            once: true
-        }
-    );
-    return {
-        ready() {
-            return deferred;
-        }
-    };
-};
-
-const workerStatus = waiterForInit(worker);
-
-const lintText = async ({ text, ext }: { text: string; ext: string }): Promise<TextlintResult> => {
-    return new Promise((resolve, _reject) => {
-        worker.addEventListener(
-            "message",
-            function (event) {
-                const data: TextlintWorkerCommandResponse = event.data;
-                if (data.command === "lint:result") {
-                    resolve(data.result);
-                }
-            },
-            {
-                once: true
-            }
-        );
-        return worker.postMessage({
-            command: "lint",
-            text,
-            ext
-        } as TextlintWorkerCommandLint);
-    });
-};
-const fixText = async ({
-    text,
-    ext,
-    message
-}: {
-    text: string;
-    ext: string;
-    message: TextlintMessage;
-}): Promise<TextlintFixResult> => {
-    return new Promise((resolve, _reject) => {
-        worker.addEventListener(
-            "message",
-            function (event) {
-                const data: TextlintWorkerCommandResponse = event.data;
-                if (data.command === "fix:result") {
-                    resolve(data.result);
-                }
-            },
-            {
-                once: true
-            }
-        );
-        return worker.postMessage({
-            command: "fix",
-            text,
-            ruleId: message.ruleId,
-            ext: ext
-        } as TextlintWorkerCommandFix);
-    });
-};
+const textlint = createTextlintPage();
 // receive
 const backgroundExposedObject = {
-    lintText: lintText,
-    fixText: fixText
+    lintText: textlint.lintText,
+    fixText: textlint.fixText
 };
 export type backgroundExposedObject = typeof backgroundExposedObject;
 browser.runtime.onConnect.addListener(async (port) => {
     if (isMessagePort(port)) {
         return;
     }
-    await workerStatus.ready();
+    await textlint.ready();
     Comlink.expose(backgroundExposedObject, createBackgroundEndpoint(port));
 });
diff --git a/packages/webextension/app/scripts/background/textlint.ts b/packages/webextension/app/scripts/background/textlint.ts
@@ -0,0 +1,113 @@
+import { browser } from "webextension-polyfill-ts";
+import { TextlintFixResult, TextlintMessage, TextlintResult } from "@textlint/types";
+import { TextlintWorkerCommandFix, TextlintWorkerCommandLint, TextlintWorkerCommandResponse } from "@textlint/compiler";
+
+browser.runtime.onInstalled.addListener((details) => {
+    console.log("previousVersion", details.previousVersion);
+});
+
+browser.tabs.onUpdated.addListener(async (tabId) => {
+    browser.pageAction.show(tabId);
+});
+const waiterForInit = (worker: Worker) => {
+    let initialized = false;
+    let _resolve: null | ((init: boolean) => void) = null;
+    const deferred = new Promise((resolve) => {
+        _resolve = resolve;
+    });
+    worker.addEventListener(
+        "message",
+        function (event) {
+            const data: TextlintWorkerCommandResponse = event.data;
+            if (data.command === "init") {
+                initialized = true;
+                _resolve && _resolve(initialized);
+            }
+        },
+        {
+            once: true
+        }
+    );
+    return {
+        ready() {
+            return deferred;
+        }
+    };
+};
+const createWorkerRef = (worker: Worker) => {
+    let currentWorker: Worker = worker;
+    let currentWorkerStatus = waiterForInit(worker);
+    return {
+        current: currentWorker,
+        ready: currentWorkerStatus.ready,
+        updateWorker(worker: Worker) {
+            currentWorker = worker;
+            currentWorkerStatus = waiterForInit(worker);
+        }
+    };
+};
+export const createTextlintPage = (defaultWorkerUrl: string | URL = "download/textlint.js") => {
+    const defaultWorker = new Worker(defaultWorkerUrl);
+    const workerRef = createWorkerRef(defaultWorker);
+    const lintText = async ({ text, ext }: { text: string; ext: string }): Promise<TextlintResult> => {
+        return new Promise((resolve, _reject) => {
+            workerRef.current.addEventListener(
+                "message",
+                function (event) {
+                    const data: TextlintWorkerCommandResponse = event.data;
+                    if (data.command === "lint:result") {
+                        resolve(data.result);
+                    }
+                },
+                {
+                    once: true
+                }
+            );
+            return workerRef.current.postMessage({
+                command: "lint",
+                text,
+                ext
+            } as TextlintWorkerCommandLint);
+        });
+    };
+    const fixText = async ({
+        text,
+        ext,
+        message
+    }: {
+        text: string;
+        ext: string;
+        message: TextlintMessage;
+    }): Promise<TextlintFixResult> => {
+        return new Promise((resolve, _reject) => {
+            workerRef.current.addEventListener(
+                "message",
+                function (event) {
+                    const data: TextlintWorkerCommandResponse = event.data;
+                    if (data.command === "fix:result") {
+                        resolve(data.result);
+                    }
+                },
+                {
+                    once: true
+                }
+            );
+            return workerRef.current.postMessage({
+                command: "fix",
+                text,
+                ruleId: message.ruleId,
+                ext: ext
+            } as TextlintWorkerCommandFix);
+        });
+    };
+    return {
+        lintText,
+        fixText,
+        ready() {
+            return workerRef.ready();
+        },
+        updateWorker(workerUrl: string | URL) {
+            workerRef.updateWorker(new Worker(workerUrl));
+        }
+    };
+};