vite-electron/script/dev-runner.ts

// @ts-nocheck

import { main } from "../package.json"
import { join } from "path"
import * as electron from "electron"
import * as dotenv from "dotenv"
import * as ora from "ora"
import * as chalk from "chalk"
import * as minimist from "minimist"

import { exec, spawn, ChildProcess } from "child_process"
import { watch, rollup, OutputOptions } from "rollup"
import { waitOn } from "./utils"
import options from "./rollup.config"

const {startMain} = require("./webpack/runMain")

// 注入环境变量
dotenv.config({ path: join(__dirname, "../.env") })
// 解析命令行参数
const argv = minimist(process.argv.slice(2))
// rollup配置
const opts = options(argv.env)
const TAG = "[build-main.ts]"
// 加载图
const spinner = ora(`${TAG} Electron build...`)
console.log(join(__dirname, "../node_modules/.bin/vite"))

function runVite() {
  return new Promise((resolve, reject) => {
    let viteProcess = spawn("node", [`D:/1XYX/pro/vite-electron/node_modules/vite/bin/vite.js`], {
      stdio: "pipe",
      env: Object.assign(process.env, { NODE_ENV: argv.env }),
    })
    if (viteProcess) {
      let isReady = false
      viteProcess.stdout.on("data", data => {
        // console.log(`${data}`)
        if (!isReady && data.indexOf("ready") != -1) {
          resolve(viteProcess)
        }
      })

      viteProcess.stderr.on("data", data => {
        console.error(`[vite err]: ${data}`)
        reject()
      })

      viteProcess.on("close", code => {
        console.log(`[vite close]: exited with code ${code}`)
        reject()
      })
    } else {
      reject()
    }
  })
}

function runElectron() {
  return new Promise((resolve, reject) => {
    let electronProcess = spawn(electron as unknown as string, [join(__dirname, `../${main}`), '--inspect=5858'], {
      stdio: "pipe",
      env: Object.assign(process.env, { NODE_ENV: argv.env }),
    })
    if (electronProcess) {
      resolve(electronProcess)
    } else {
      reject()
    }
  })
}

;(async () => {
  let vitePorcess = await runVite()
  console.log('[vite]', chalk.green(`vite ready.`))
  let child: ChildProcess | null
  let manualRestart = false
  startMain(async ()=>{
    if (child && child.kill) {
      manualRestart = true
      child.kill()
      child = null
      setTimeout(() => {
        manualRestart = false
      }, 5000)
    }
    try {
      child = await runElectron()
      console.log('[electron]', chalk.green(`electron ready.`))
      child.on("close", () => {
        if (!manualRestart) {
          // https://juejin.cn/post/6844904071682326535
          vitePorcess.kill()
          child.kill()
          process.exit()
        }
      })
    } catch (ev) {
      console.log(ev.error)
      vitePorcess.kill()
      child.kill()
      process.exit()
    }
  })
  
  // const watcher = watch(opts)
  // let child: ChildProcess | null
  // let manualRestart = false
  // watcher.on("change", filename => {
  //   const log = chalk.green(`change -- ${filename}`)
  //   console.log(TAG, log)
  // })
  // watcher.on("event",async ev => {
  //   if (ev.code === "END") {
  //     if (child && child.kill) {
  //       manualRestart = true
  //       child.kill()
  //       child = null
  //       setTimeout(() => {
  //         manualRestart = false
  //       }, 5000)
  //     }
  //     try {
  //       child = await runElectron()
  //       console.log('[electron]', chalk.green(`electron ready.`))
  //       child.on("close", () => {
  //         if (!manualRestart) {
  //           vitePorcess.kill()
  //           child.kill()
  //           process.exit()
  //         }
  //       })
  //     } catch (ev) {
  //       console.log(ev.error)
  //       vitePorcess.kill()
  //       child.kill()
  //       process.exit()
  //     }
  //   } else if (ev.code === "ERROR") {
  //     console.log(ev.error)
  //   }
  // })
})()