lens/src/main/electron-app/before-application-is-ready/setup-runnables-before-closing-of-application.injectable.ts

/**
 * Copyright (c) OpenLens Authors. All rights reserved.
 * Licensed under MIT License. See LICENSE in root directory for more information.
 */
import { getInjectable } from "@ogre-tools/injectable";
import { beforeApplicationIsReadyInjectionToken } from "../../start-main-application/before-application-is-ready/before-application-is-ready-injection-token";
import { beforeQuitOfFrontEndInjectionToken } from "../../start-main-application/before-quit-of-front-end/before-quit-of-front-end-injection-token";
import { runManyFor } from "../../start-main-application/run-many-for";
import { beforeQuitOfBackEndInjectionToken } from "../../start-main-application/before-quit-of-back-end/before-quit-of-back-end-injection-token";
import electronAppInjectable from "../electron-app.injectable";
import isIntegrationTestingInjectable from "../../../common/vars/is-integration-testing.injectable";
import autoUpdaterInjectable from "../features/auto-updater.injectable";

const setupRunnablesBeforeClosingOfApplicationInjectable = getInjectable({
  id: "setup-closing-of-application",

  instantiate: (di) => {
    const runMany = runManyFor(di);

    const runRunnablesBeforeQuitOfFrontEnd = runMany(
      beforeQuitOfFrontEndInjectionToken,
    );

    const runRunnablesBeforeQuitOfBackEnd = runMany(
      beforeQuitOfBackEndInjectionToken,
    );

    return {
      run: () => {
        const app = di.inject(electronAppInjectable);

        const isIntegrationTesting = di.inject(isIntegrationTestingInjectable);
        const autoUpdater = di.inject(autoUpdaterInjectable);

        let isAutoUpdating = false;

        autoUpdater.on("before-quit-for-update", () => {
          isAutoUpdating = true;
        });

        app.on("will-quit", async ({ preventDefault: cancelNativeQuit }) => {
          await runRunnablesBeforeQuitOfFrontEnd();

          const shouldQuitBackEnd = isIntegrationTesting || isAutoUpdating;

          if (shouldQuitBackEnd) {
            await runRunnablesBeforeQuitOfBackEnd();
          } else {
            cancelNativeQuit();
          }
        });
      },
    };
  },

  injectionToken: beforeApplicationIsReadyInjectionToken,
});

export default setupRunnablesBeforeClosingOfApplicationInjectable;