Vite 整合 Electron 总结-iColak's Blog

Vite 是面向未来几年的构建工具，很有必要在各个场景下都试试集成进来
Electron 作为前端标配的桌面开发工具，官方并有脚手架也没和哪个框架整合
@vue/cli 官方有给出模板；但是 Vite 这块并没提供，毕竟人家定位是和 webpack 那样的通用构建工具甚至连 Vue 都没集成那么我们尝试来做下这件事儿
按照 Vue 在 Vite 中的集成风格 Electron 这块应当写一个插件！
这里假定你简单的知道一些 Vite 的工作原理，这种文章网上有好多的
同时也假定你使用或者上手过 Electron；上手非常简单，直接看官网即可
项目整体所有代码在这 electron-vue-vite 可以直接 用于生产 (亲点个 start 呗 )

·
├── script           项目脚本目录
├── src
|  ├── main          Electron 主进程代码
|  ├── preload       Electron 预加载目录
|  ├── render        Electron 渲染进程 - 既 Vite 代码
|
├── vite.config.ts   Vite 配置文件

Electron 支持全量的 NodeJs API 渲染进程难免会用到
Vite 基于的 Rollup 构建工程，所以我们改造 Rollup 部分配置，输出 CommonJs 格式
目录结构相比于 Vite 提供的默认结构有所不同，也需要配置一番才能按预期工作

import { defineConfig } from 'vite'
import vue from '@vitejs/plugin-vue'
import { join } from 'path' export default defineConfig((env) => ({ plugins: [ vue(), // 开启 Vue 支持
 ], root: join(__dirname, 'src/render'), // 指向渲染进程目录
 base: './', // index.html 中静态资源加载位置
 build: { outDir: join(__dirname, 'dist/render'), assetsDir: '', // 相对路径 加载问题
 rollupOptions: { output: { format: 'cjs', // 配置 Rollup 打包输出 CommonJs 格式
 }, external: ['electron'], // 告诉 Rollup 不要去打包 electron
 }, }, optimizeDeps: { exclude: ['electron'], // 告诉 Vite 不要转换 electron 模块
 }, // 其他配置略...
}))

我们先出个结论 - Electron 的启动与 NodeJs 相比行为几乎是一致的 - 可执行程序 + 入口文件

# 全局目录安装的 NodeJs
node path/filename.js

# 项目目录安装的 Electron
node_modules/.bin/electron path/filename.js

如果我们把 Electron 的启动设计到 npm 的 scripts 中，它还可以更简单 npm run electron

{ "scripts": { "electron": "electron path/filename.js" }
}

思考一下， npm 只有一个启动命令 Vite、Electron 各需要一个，加一起就是两个启动命令了这里我们借助 concurrently 来同时启动 Vite、Electron

{ "scripts": { "dev": "concurrently npm run vite npm run electron", "vite": "vite", "electron": "electron path/filename.js" }
}

看起来不错！

不过我们再思考下关于 Electron 启动的问题
开发环境下 Electron 应该加载 Vite 启动的开发服务器，生产环境下启动一个具体的文件
那么这里就会出现一个等待的情况，就是 Electron 要等待 Vite 启动后再启动

启动脚本设计

我们需要监听 Vite 的启动后拉起 Electron，这块我们考虑用轮训监听端口的方式
监听到 Vite 启动后 Electron 我们用 NodeJs 的子进程 API child_process.spawn() 拉起
我们将 npm scripts 做下改动，方便我们知道脚本是干啥的， 相对上面的脚本这里重命名一下concurrently 也加入一些命令行参数，能得好更友好的控制台输出

{ "scripts": { "dev": "concurrently -n=vue,ele -c=green,blue npm run dev:vue npm run dev:ele", "dev:vite": "vite", "dev:electron": "node -r ts-node/register script/build-main --env=development --watch" }
}

由于我们要控制的是 Electron 的启动，为此我们单独写个脚本(script/build-main.ts)来控制，包括下面两个功能点
启动时机控制 - 监听 Vite 已经启动
主进程代码使用使用 typescript 开发 - 加入 Rollup 编译、打包

script/build-main.ts

import { join } from 'path'
import { get } from 'http'
import { spawn, ChildProcess } from 'child_process'
import { watch } from 'rollup'
import minimist from 'minimist'
import electron from 'electron'
import options from './rollup.config'
import { main } from '../package.json' /**
 * 1. 监听 vite 启动
 */
function waitOn(arg0: { port: string | number; interval?: number; }) { return new Promise(resolve => { const { port, interval = 149 } = arg0 const url = `http://localhost:${port}` // 通过定时器轮训向 Vite 服务器请求
 const timer: NodeJS.Timer = setInterval(() => { get( `http://localhost:${port}`, // 指向 Vite 开发服务器
 res => { clearInterval(timer) resolve(res.statusCode) } ) }, interval) })
} /**
 * 2. 控制 Electron 启动时机，编译 typescript
 */
waitOn({ port: '3000' }).then(msg => { // 解析 npm script 的命令行参数
 const argv = minimist(process.argv.slice(2)) // 加载 rollup 配置
 const opts = options(argv.env) // Vite 启动后以监听模式开启 Rollup 编译 Electron 主进程代码
 const watcher = watch(opts) let child: ChildProcess watcher.on('event', ev => { if (ev.code === 'END') { // 保证只启动一个 Electron 个程序
 if (child) child.kill() // 使用 NodeJs 子进程能力拉起 Electron 程序
 child = spawn( // 这里 electron 本质上只是一个字符串；指向 Electron 可执行程序的绝对路径
 electron as any, // 指定 Electron 主进程入口文件；既 Rollup 编译后输出文件的路径
 [join(__dirname, `../${main}`)], { stdio: 'inherit' } ) } })
})

script/rollup.config

import { builtinModules } from 'module'
import { join } from 'path'
import { RollupOptions } from 'rollup'
import nodeResolve from '@rollup/plugin-node-resolve'
import commonjs from '@rollup/plugin-commonjs'
import typescript from '@rollup/plugin-typescript'
import json from '@rollup/plugin-json' /** node.js builtins module */
const builtins = () => builtinModules.filter(x => !/^_|^(internal|v8|node-inspect)\/|\//.test(x)) export default (env = 'production') => { const options: RollupOptions = { input: join(__dirname, '../src/main/index.ts'), output: { file: join(__dirname, '../dist/main/index.js'), format: 'cjs', // 使用 CommonJs 模块化
 }, plugins: [ nodeResolve(), // 支持 node_modules 下面的包查找
 commonjs(), // 支持 CommonJs 模块
 json(), // 支持引入 json 文件
 typescript({ module: 'ESNext', // 支持 typescript
 }), ], external: [ // 打包避开内置模块
 ...builtins(), 'electron', ], } return options
}

到这为止，项目应该是能跑起来的状态了；但是还不能使用 Electron、NodeJs 相关的 API

这里简单的改了下 App.vue、HelloWorld.vue 中的一些文案，没做逻辑修改；就不贴出代码了

加入 Electron API

渲染和主进程通信是个十分常用的功能；我试着从 electron 导出 ipcRenderer

// src/render/main.ts
import { createApp } from 'vue'
import App from './App.vue'
import { ipcRenderer } from 'electron' console.log('ipcRenderer:', ipcRenderer) createApp(App).mount('#app')

报错了！默认情况下直接用 import 语法会被 Rollup 编译
事实上在 'electron' 在 Electron 运行环境中是一个 内置模块 你可以在控制台中试试下这段代码

注意这里先不要引入 Electron 相关的包，保障项目能跑起来

require.resolve('electron')
"electron" // 将会输出

既然 Electron 本就支持全量的 NodeJs API 我们不妨直接在在代码中直接写成

// src/render/main.ts
import { createApp } from 'vue'
import App from './App.vue'
- import { ipcRenderer } from 'electron'
+ const { ipcRenderer } = require('electron')

console.log('ipcRenderer:', ipcRenderer)

createApp(App).mount('#app')

项目确实跑起来了，我们还可以用这个办法进一步验证其他的模块

插件设计分析

可以打包票说，这个可以用！(仅限开发期)；但是这个会带来两个问题
编码风格不统一，人家都在用 ESModule 混入 CommonJs 确实不好看
require('xxxx') 在打包期间如果不做些处理，并不会被 Rollup 处理 (这里只 .ts 文件,有大神知道怎么对付这种情况的请指点下小弟) 如果你引入的是 node_modules 中的包那可就惨了；比如 require('electron-store') 这种会原样输出；打包后的程序开起来会找不到 'electron-store' 这个模块，铁定报错！
我们知道 ESModule 写法在开发期运行会报错，但是还是要写；如果我们在运行的前一刻将 ESModule 转换成 NodeJs 内置的 CommonJs 岂不是两全其美的好事；
甚至只要有关 NodeJs API 的包我们都可以转化，毕竟开发期项目根目录是有 node_modules 这个 NodeJs包仓库 给你用的！

分析至此，我们该动动手写个插件了；让插件去自动化完成 - ESModule to CommonJs

vitejs-plugin-electron

import * as acorn from 'acorn'
import { Plugin as VitePlugin } from 'vite' const extensions = ['.js', '.jsx', '.ts', '.tsx', '.vue'] // 需要处理的文件后缀

export interface Esm2cjsOptions { excludes?: string[] // 需要被转换的模块
} export default function esm2cjs(options?: Esm2cjsOptions): VitePlugin { const opts: Esm2cjsOptions = { // 默认我们转换 electron、electron-store 两个模块
 excludes: [ 'electron', 'electron-store', ], ...options } return { name: 'vitejs-plugin-electron', // 这个 name 就是插件名字
 transform(code, id) { const parsed = path.parse(id) // 解析引入模块的路径，id 即引入文件完整路径
 if (!extensions.includes(parsed.ext)) return // 只处理需要处理的文件后缀
 const node: any = acorn.parse(code, { // 使用 acorn 解析 ESTree
 ecmaVersion: 'latest', // 指定按照最新的 es 模块标准解析
 sourceType: 'module', // 指定按照模块进行解析
 }) let codeRet = code node.body.reverse().forEach((item) => { if (item.type !== 'ImportDeclaration') return // 跳过非 import 语句
 if (!opts.excludes.includes(item.source.value)) return // 跳过不要转换的模块
 /**
 * 下面这些 const 声明用来确定 import 的写法
 */ const statr = codeRet.substring(0, item.start) const end = codeRet.substring(item.end) const deft = item.specifiers.find(({ type }) => type === 'ImportDefaultSpecifier') const deftModule = deft ? deft.local.name : '' const nameAs = item.specifiers.find(({ type }) => type === 'ImportNamespaceSpecifier') const nameAsModule = nameAs ? nameAs.local.name : '' const modules = item. specifiers .filter((({ type }) => type === 'ImportSpecifier')) .reduce((acc, cur) => acc.concat(cur.imported.name), []) /**
 * 这里开始根据各种 import 语法做转换
 */ if (nameAsModule) { // import * as name from
 codeRet = `${statr}const ${nameAsModule} = require(${item.source.raw})${end}` } else if (deftModule && !modules.length) { // import name from 'mod'
 codeRet = `${statr}const ${deftModule} = require(${item.source.raw})${end}` } else if (deftModule && modules.length) { // import name, { name2, name3 } from 'mod'
 codeRet = `${statr}const ${deftModule} = require(${item.source.raw})
 const { ${modules.join(', ')} } = ${deftModule}${end}` } else { // import { name1, name2 } from 'mod'
 codeRet = `${statr}const { ${modules.join(', ')} } = require(${item.source.raw})${end}` } }) return codeRet }, }
}

在 vite.config.ts 中使用 vitejs-plugin-electron

import { defineConfig } from 'vite'
import vue from '@vitejs/plugin-vue'
import electron from 'vitejs-plugin-electron' export default defineConfig((env) => ({ plugins: [ vue(), electron(), ], // 其他配置略...
}))

It\'s Worked！
好了，这个插件可以用了；再想想上面的关于 Rollup 的配置其实我们完全可以集成到 vitejs-plugin-electron 中的，这样会使 vite.config.ts 文件更少、更清晰；
还有打包部分，就不一一展示了...具体代码就不演示了，自己拉代码看看吧
vitejs-plugins-electron