Replace global state usages of ExtensionLoader with DI

Signed-off-by: Janne Savolainen <janne.savolainen@live.fi>

src/main/protocol-handler/__test__/router.test.ts

@@ -25,11 +25,15 @@ import { broadcastMessage } from "../../../common/ipc";
 import { ProtocolHandlerExtension, ProtocolHandlerInternal } from "../../../common/protocol-handler";
 import { delay, noop } from "../../../common/utils";
 import { LensExtension } from "../../../extensions/main-api";
-import { ExtensionLoader } from "../../../extensions/extension-loader";
 import { ExtensionsStore } from "../../../extensions/extensions-store";
-import { LensProtocolRouterMain } from "../router";
+import type { LensProtocolRouterMain } from "../lens-protocol-router-main/lens-protocol-router-main";
 import mockFs from "mock-fs";
 import { AppPaths } from "../../../common/app-paths";
+import { getDiForUnitTesting } from "../../getDiForUnitTesting";
+import extensionLoaderInjectable
+  from "../../../extensions/extension-loader/extension-loader.injectable";
+import lensProtocolRouterMainInjectable
+  from "../lens-protocol-router-main/lens-protocol-router-main.injectable";
 
 jest.mock("../../../common/ipc");
 
@@ -58,14 +62,22 @@ function throwIfDefined(val: any): void {
 }
 
 describe("protocol router tests", () => {
+  // TODO: This test suite is using any to access protected property.
+  // Unit tests are allowed to only public interfaces.
+  let extensionLoader: any;
+  let lpr: LensProtocolRouterMain;
+
   beforeEach(() => {
+    const di = getDiForUnitTesting();
+
+    extensionLoader = di.inject(extensionLoaderInjectable);
+
     mockFs({
       "tmp": {},
     });
     ExtensionsStore.createInstance();
-    ExtensionLoader.createInstance();
 
-    const lpr = LensProtocolRouterMain.createInstance();
+    lpr = di.inject(lensProtocolRouterMainInjectable);
 
     lpr.rendererLoaded = true;
   });
@@ -74,15 +86,11 @@ describe("protocol router tests", () => {
     jest.clearAllMocks();
 
     ExtensionsStore.resetInstance();
-    ExtensionLoader.resetInstance();
-    LensProtocolRouterMain.resetInstance();
     mockFs.restore();
   });
 
   it("should throw on non-lens URLS", async () => {
     try {
-      const lpr = LensProtocolRouterMain.getInstance();
-
       expect(await lpr.route("https://google.ca")).toBeUndefined();
     } catch (error) {
       expect(error).toBeInstanceOf(Error);
@@ -91,8 +99,6 @@ describe("protocol router tests", () => {
 
   it("should throw when host not internal or extension", async () => {
     try {
-      const lpr = LensProtocolRouterMain.getInstance();
-
       expect(await lpr.route("lens://foobar")).toBeUndefined();
     } catch (error) {
       expect(error).toBeInstanceOf(Error);
@@ -113,14 +119,13 @@ describe("protocol router tests", () => {
       isCompatible: true,
       absolutePath: "/foo/bar",
     });
-    const lpr = LensProtocolRouterMain.getInstance();
 
     ext.protocolHandlers.push({
       pathSchema: "/",
       handler: noop,
     });
 
-    (ExtensionLoader.getInstance() as any).instances.set(extId, ext);
+    extensionLoader.instances.set(extId, ext);
     (ExtensionsStore.getInstance() as any).state.set(extId, { enabled: true, name: "@mirantis/minikube" });
 
     lpr.addInternalHandler("/", noop);
@@ -143,7 +148,6 @@ describe("protocol router tests", () => {
   });
 
   it("should call handler if matches", async () => {
-    const lpr = LensProtocolRouterMain.getInstance();
     let called = false;
 
     lpr.addInternalHandler("/page", () => { called = true; });
@@ -159,7 +163,6 @@ describe("protocol router tests", () => {
   });
 
   it("should call most exact handler", async () => {
-    const lpr = LensProtocolRouterMain.getInstance();
     let called: any = 0;
 
     lpr.addInternalHandler("/page", () => { called = 1; });
@@ -178,7 +181,6 @@ describe("protocol router tests", () => {
   it("should call most exact handler for an extension", async () => {
     let called: any = 0;
 
-    const lpr = LensProtocolRouterMain.getInstance();
     const extId = uuid.v4();
     const ext = new LensExtension({
       id: extId,
@@ -202,7 +204,7 @@ describe("protocol router tests", () => {
         handler: params => { called = params.pathname.id; },
       });
 
-    (ExtensionLoader.getInstance() as any).instances.set(extId, ext);
+    extensionLoader.instances.set(extId, ext);
     (ExtensionsStore.getInstance() as any).state.set(extId, { enabled: true, name: "@foobar/icecream" });
 
     try {
@@ -217,7 +219,6 @@ describe("protocol router tests", () => {
   });
 
   it("should work with non-org extensions", async () => {
-    const lpr = LensProtocolRouterMain.getInstance();
     let called: any = 0;
 
     {
@@ -241,7 +242,7 @@ describe("protocol router tests", () => {
           handler: params => { called = params.pathname.id; },
         });
 
-      (ExtensionLoader.getInstance() as any).instances.set(extId, ext);
+      extensionLoader.instances.set(extId, ext);
       (ExtensionsStore.getInstance() as any).state.set(extId, { enabled: true, name: "@foobar/icecream" });
     }
 
@@ -266,7 +267,7 @@ describe("protocol router tests", () => {
           handler: () => { called = 1; },
         });
 
-      (ExtensionLoader.getInstance() as any).instances.set(extId, ext);
+      extensionLoader.instances.set(extId, ext);
       (ExtensionsStore.getInstance() as any).state.set(extId, { enabled: true, name: "icecream" });
     }
 
@@ -286,13 +287,10 @@ describe("protocol router tests", () => {
   });
 
   it("should throw if urlSchema is invalid", () => {
-    const lpr = LensProtocolRouterMain.getInstance();
-
     expect(() => lpr.addInternalHandler("/:@", noop)).toThrowError();
   });
 
   it("should call most exact handler with 3 found handlers", async () => {
-    const lpr = LensProtocolRouterMain.getInstance();
     let called: any = 0;
 
     lpr.addInternalHandler("/", () => { called = 2; });
@@ -311,7 +309,6 @@ describe("protocol router tests", () => {
   });
 
   it("should call most exact handler with 2 found handlers", async () => {
-    const lpr = LensProtocolRouterMain.getInstance();
     let called: any = 0;
 
     lpr.addInternalHandler("/", () => { called = 2; });

src/main/protocol-handler/index.ts

@@ -19,4 +19,4 @@
  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  */
 
-export * from "./router";
+export * from "./lens-protocol-router-main/lens-protocol-router-main";

src/main/protocol-handler/lens-protocol-router-main/lens-protocol-router-main.injectable.ts

@@ -0,0 +1,40 @@
+/**
+ * Copyright (c) 2021 OpenLens Authors
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of
+ * this software and associated documentation files (the "Software"), to deal in
+ * the Software without restriction, including without limitation the rights to
+ * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+ * the Software, and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+ * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+ * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+ * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+ * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+import type { Injectable } from "@ogre-tools/injectable";
+import { lifecycleEnum } from "@ogre-tools/injectable";
+import extensionLoaderInjectable from "../../../extensions/extension-loader/extension-loader.injectable";
+import type { Dependencies } from "./lens-protocol-router-main";
+import { LensProtocolRouterMain } from "./lens-protocol-router-main";
+
+const lensProtocolRouterMainInjectable: Injectable<
+  LensProtocolRouterMain,
+  Dependencies
+> = {
+  getDependencies: di => ({
+    extensionLoader: di.inject(extensionLoaderInjectable),
+  }),
+
+  instantiate: dependencies => new LensProtocolRouterMain(dependencies),
+
+  lifecycle: lifecycleEnum.singleton,
+};
+
+export default lensProtocolRouterMainInjectable;

src/main/protocol-handler/lens-protocol-router-main/lens-protocol-router-main.ts

@@ -19,15 +19,16 @@
  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  */
 
-import logger from "../logger";
-import * as proto from "../../common/protocol-handler";
+import logger from "../../logger";
+import * as proto from "../../../common/protocol-handler";
 import URLParse from "url-parse";
-import type { LensExtension } from "../../extensions/lens-extension";
-import { broadcastMessage } from "../../common/ipc";
+import type { LensExtension } from "../../../extensions/lens-extension";
+import { broadcastMessage } from "../../../common/ipc";
 import { observable, when, makeObservable } from "mobx";
-import { ProtocolHandlerInvalid, RouteAttempt } from "../../common/protocol-handler";
-import { disposer, noop } from "../../common/utils";
-import { WindowManager } from "../window-manager";
+import { ProtocolHandlerInvalid, RouteAttempt } from "../../../common/protocol-handler";
+import { disposer, noop } from "../../../common/utils";
+import { WindowManager } from "../../window-manager";
+import type { ExtensionLoader } from "../../../extensions/extension-loader";
 
 export interface FallbackHandler {
   (name: string): Promise<boolean>;
@@ -50,6 +51,10 @@ function checkHost<Query>(url: URLParse<Query>): boolean {
   }
 }
 
+export interface Dependencies {
+  extensionLoader: ExtensionLoader
+}
+
 export class LensProtocolRouterMain extends proto.LensProtocolRouter {
   private missingExtensionHandlers: FallbackHandler[] = [];
 
@@ -57,8 +62,8 @@ export class LensProtocolRouterMain extends proto.LensProtocolRouter {
 
   protected disposers = disposer();
 
-  constructor() {
-    super();
+  constructor(protected dependencies: Dependencies) {
+    super(dependencies);
 
     makeObservable(this);
   }