import type { ForgeConfig } from '@electron-forge/shared-types';
import { MakerSquirrel } from '@electron-forge/maker-squirrel';
import { MakerZIP } from '@electron-forge/maker-zip';
import { MakerDeb } from '@electron-forge/maker-deb';
import { MakerRpm } from '@electron-forge/maker-rpm';
import { VitePlugin } from '@electron-forge/plugin-vite';
import { FusesPlugin } from '@electron-forge/plugin-fuses';
import { FuseV1Options, FuseVersion } from '@electron/fuses';
import * as fs from 'fs-extra';
import * as path from 'path';

const config: ForgeConfig = {
  packagerConfig: {
    asar: true,
  },
  rebuildConfig: {},
  makers: [
    new MakerSquirrel({}),
    new MakerZIP({}, ['darwin']),
    new MakerRpm({}),
    new MakerDeb({}),
  ],
  plugins: [
    new VitePlugin({
      // `build` can specify multiple entry builds, which can be Main process, Preload scripts, Worker process, etc.
      // If you are familiar with Vite configuration, it will look really familiar.
      build: [
        {
          // `entry` is just an alias for `build.lib.entry` in the corresponding file of `config`.
          entry: 'src/electron/main/main.ts',
          config: 'vite.main.config.ts',
          target: 'main',
        },
        {
          entry: 'src/electron/preload/preload.ts',
          config: 'vite.preload.config.ts',
          target: 'preload',
        },
      ],
      renderer: [
        {
          name: 'main_window',
          config: 'vite.renderer.config.ts',
        },
      ],
    }),
    // Fuses are used to enable/disable various Electron functionality
    // at package time, before code signing the application
    new FusesPlugin({
      version: FuseVersion.V1,
      [FuseV1Options.RunAsNode]: false,
      [FuseV1Options.EnableCookieEncryption]: true,
      [FuseV1Options.EnableNodeOptionsEnvironmentVariable]: false,
      [FuseV1Options.EnableNodeCliInspectArguments]: false,
      [FuseV1Options.EnableEmbeddedAsarIntegrityValidation]: true,
      [FuseV1Options.OnlyLoadAppFromAsar]: true,
    }),
  ],
  hooks: {
    async prePackage() {
      const outDir = path.resolve(process.cwd(), 'out');
      fs.rmSync(outDir, { recursive: true, force: true });
    },

    async preMake() {
      const outDir = path.resolve(process.cwd(), 'out');
      fs.rmSync(outDir, { recursive: true, force: true });
    },

    async postPackage(_forgeConfig, options) {
      const electronVersion = require('electron/package.json').version;
      const nodeVersion = process.version;
      const versionData = {
        node: nodeVersion,
        electron: electronVersion,
        platform: process.platform,
        arch: process.arch,
        buildTime: new Date().toISOString(),
      };
      const outDir = path.resolve(process.cwd(), 'out');
      fs.ensureDirSync(outDir);

      const outputs = (options && (options as any).outputPaths) || [];

      if (Array.isArray(outputs) && outputs.length > 0) {
        for (const p of outputs) {
          try {
            const versionFile = path.join(p, 'version');
            fs.writeFileSync(versionFile, JSON.stringify(versionData, null, 2));
          } catch {}
        }
      }
    },

    async postMake(_forgeConfig, outputs) {
      const electronVersion = require('electron/package.json').version;
      const nodeVersion = process.version;
      const versionData = {
        node: nodeVersion,
        electron: electronVersion,
        platform: process.platform,
        arch: process.arch,
        buildTime: new Date().toISOString(),
      };

      const outDir = path.resolve(process.cwd(), 'out');
      fs.ensureDirSync(outDir);

      if (Array.isArray(outputs) && outputs.length > 0) {
        for (const p of outputs) {
          try {
            const versionFile = path.join(p as any, 'version');
            fs.writeFileSync(versionFile, JSON.stringify(versionData, null, 2));
          } catch {}
        }
      }
    },
  },
};

export default config;