huanghs/electron-vite-vue
1
0
Fork 0
mirror of https://github.com/electron-vite/electron-vite-vue synced 2025-02-26 13:00:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: better hot reload preload script

Browse Source
This commit is contained in:
草鞋没号 2021-11-09 09:01:52 +08:00
parent fc9abe80bc
commit 9eda22f353
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
scripts/build-preload.ts
View File

@ -1,36 +1,43 @@
import { watch, rollup, OutputOptions } from 'rollup'
import { watch, rollup, OutputOptions, Plugin } from 'rollup'
import minimist from 'minimist'
import chalk from 'chalk'
import ora from 'ora'
import WebSocket from 'ws'
import options from './rollup.config'
import { createWsServer, formatWsSendData } from './ws'
import { createWsServer, formatWsSendData, WssServer } from './ws'
const argv = minimist(process.argv.slice(2))
const opt = options({ proc: 'preload', env: argv.env })
const opts = options({ proc: 'preload', env: argv.env })
const TAG = '[build-preload.ts]'
const spinner = ora(`${TAG} Electron preload build...`)
function hotReloadPreload(wssServer: WssServer): Plugin {
return {
name: 'hot-reload-preload',
writeBundle() {
// Hot reload preload script !!!
if (wssServer.instance?.readyState === WebSocket.OPEN) {
console.log(TAG, 'Hot reload preload script')
wssServer.instance.send(formatWsSendData({ cmd: 'reload', data: Date.now() }))
}
},
}
}
; (async () => {
if (argv.watch) {
const watcher = watch(opt)
const wssObj = createWsServer({ TAG })
const wssServer = createWsServer()
opts.plugins = opts.plugins?.concat(hotReloadPreload(wssServer))
const watcher = watch(opts)
watcher.on('change', filename => {
const log = chalk.yellow(`change -- ${filename}`)
console.log(TAG, log)
// Hot reload renderer process !!!
if (wssObj.instance?.readyState === WebSocket.OPEN) {
console.log(TAG, 'Hot reload renderer process')
wssObj.instance.send(formatWsSendData({ cmd: 'reload', data: Date.now() }))
}
console.log(TAG, chalk.yellow(`change -- ${filename}`))
})
} else {
spinner.start()
try {
const build = await rollup(opt)
await build.write(opt.output as OutputOptions)
const build = await rollup(opts)
await build.write(opts.output as OutputOptions)
spinner.succeed()
process.exit()
} catch (error) {