1.4.4:修复样式scoped的范围

Reviewed-on: #9

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2018 
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

index.js

@@ -7,7 +7,7 @@
  */
 
 import fs from 'iofs'
-import { join, normalize } from 'path'
+import { join, normalize } from 'node:path'
 import { red, blue } from 'kolorist'
 
 import createServer from './lib/dev.js'
@@ -18,16 +18,18 @@ const IS_WINDOWS = process.platform === 'win32'
 
 const CONFIG_FILE = normalize(join(WORK_SPACE, 'fite.config.js'))
 const PROTOCOL = IS_WINDOWS ? 'file://' : ''
-const NODE_VERSION = +process.versions.node.split('.').slice(0, 2).join('.')
+const NODE_VERSION = process.versions.node.split('.').map(n => +n)
+const ABS_CONFIG_FILEPATH = PROTOCOL + CONFIG_FILE
 
 let args = process.argv.slice(2)
 let mode = args.shift() || 'prod'
-let clean = args.shift() !== '--no-clean'
+let clean = !args.includes('--no-clean') //
+let verbose = args.includes('--verbose') //
 
-if (NODE_VERSION < 16.6) {
+if (NODE_VERSION[0] < 16 || (NODE_VERSION[0] === 16 && NODE_VERSION[1] < 6)) {
-  console.log(red('Error: 你当前的环境不满足 Vue-live 构建工具的要求'))
+  console.log(red('Error: 你当前的环境不满足 fite 构建工具的要求'))
   console.log(
-    'Vue-live 需要Node.js版本在 %s 以上, \n你当前的Node.js版本为: %s',
+    'fite 需要Node.js版本在 %s 以上, \n你当前的Node.js版本为: %s',
     blue('v16.6.0'),
     red(process.version),
     '\n\n'
@@ -39,8 +41,9 @@ switch (mode) {
   case 'dev':
     process.env.NODE_ENV = 'development'
 
-    import(PROTOCOL + CONFIG_FILE)
+    import(ABS_CONFIG_FILEPATH)
       .then(function (conf) {
+        conf.default.ABS_CONFIG_FILEPATH = ABS_CONFIG_FILEPATH
         createServer(WORK_SPACE, conf.default)
       })
       .catch(err => {
@@ -51,13 +54,15 @@ switch (mode) {
   case 'build':
     process.env.NODE_ENV = 'production'
 
-    import(PROTOCOL + CONFIG_FILE)
+    import(ABS_CONFIG_FILEPATH)
       .then(function (conf) {
         let dist = conf.buildDir || 'dist'
         if (clean && fs.isdir(dist)) {
-          fs.mkdir(dist)
+          console.log('清除dist目录...')
+          fs.rm(dist)
         }
-        compile(WORK_SPACE, dist, conf.default)
+        conf.default.ABS_CONFIG_FILEPATH = ABS_CONFIG_FILEPATH
+        compile(WORK_SPACE, dist, conf.default, verbose)
       })
       .catch(err => {
         console.log(err)

lib/compile-vue.js

@@ -6,42 +6,32 @@
 
 import fs from 'iofs'
 import scss from '@bytedo/sass'
-import { createHash, randomUUID } from 'crypto'
 import Es from 'esbuild'
-import { join } from 'path'
+
 import { compile } from '@vue/compiler-dom'
 import { red, cyan, blue } from 'kolorist'
 
-function uuid() {
-  return randomUUID().slice(-8)
-}
-
 import {
   JS_EXP,
   STYLE_EXP,
   HTML_EXP,
   CSS_SHEET_EXP,
-  HMR_SCRIPT,
   V_DEEP,
-  PERCENT_EXP
+  PERCENT_EXP,
+  SHEETS_DEF,
+  LEGACY_POLYFILL
 } from './constants.js'
+import { createHmrScript, md5, uuid, urlJoin } from './utils.js'
 
 const OPTIONS = {
-  indentType: 'space',
+  style: 'compressed'
-  indentWidth: 2
 }
 
-// 修正路径合并 避免在windows下被转义
+function minify(code) {
-function urlJoin(...args) {
+  return Es.transformSync(code, { minify: true }).code.trim()
-  return join(...args).replace(/\\/g, '/')
-}
-
-function md5(str = '') {
-  let sum = createHash('md5')
-  sum.update(str, 'utf8')
-  return sum.digest('hex').slice(0, 8)
 }
 
+// 处理css中的 :deep()
 function parseVDeep(curr, val, scoped) {
   let res = V_DEEP.exec(curr)
   if (res) {
@@ -52,6 +42,7 @@ function parseVDeep(curr, val, scoped) {
   }
 }
 
+// 处理style中的scoped属性
 function scopeCss(css = '', hash) {
   return css.replace(CSS_SHEET_EXP, (m, selector) => {
     if (!selector.startsWith('@')) {
@@ -73,7 +64,7 @@ function scopeCss(css = '', hash) {
               if (last.startsWith(':')) {
                 output = parseVDeep(last, output, true)
               } else {
-                output = `${last} ${output}`
+                output = `${last}[data-${hash}] ${output}`
               }
             } else {
               if (last.includes(':')) {
@@ -102,18 +93,16 @@ function scopeCss(css = '', hash) {
 /**
  * 编译scss为css
  * @param file <String> 文件路径或scss代码
- * @param mini <Boolean> 是否压缩
+ * @param inject <String> 要注入的scss代码
  */
-export function compileScss(file, mini = true) {
+export function compileScss(file, inject = '') {
-  let style = mini ? 'compressed' : 'expanded'
   try {
     if (fs.isfile(file)) {
-      return scss.compile(file, { style, ...OPTIONS }).css.trim()
+      return scss.compile(file, OPTIONS).css.trim()
     } else {
-      return scss.compileString(file, { style, ...OPTIONS }).css.trim()
+      return scss.compileString(inject + file, OPTIONS).css.trim()
     }
   } catch (err) {
-    console.log('compile scss: ', file)
     console.error(err)
   }
 }
@@ -126,10 +115,11 @@ export function compileScss(file, mini = true) {
 export function parseJs(
   code = '',
   imports,
-  { IS_MPA, currentPage, IS_ENTRY, DEPLOY_PATH } = {},
+  { IS_MPA, currentPage, IS_ENTRY, DEPLOY_PATH, LEGACY_MODE, define } = {},
-  filename
+  filename,
+  linePatch = 1
 ) {
-  let fixedStyle = '\n\n'
+  let fixedStyle = ''
   let ASSETS_DIR = '/@/'
   let isBuild = process.env.NODE_ENV === 'production'
 
@@ -137,35 +127,35 @@ export function parseJs(
     ASSETS_DIR = '/assets/'
   }
 
-  if (IS_ENTRY) {
-    fixedStyle += 'const __sheets__ = [...document.adoptedStyleSheets];\n'
-  }
-
   try {
     code = Es.transformSync(code).code || ''
   } catch (e) {
     let err = e.errors.pop()
-    console.log('%s: %s', red('Uncaught SyntaxError'), err.text)
+    let lines = code.split('\n')
+
+    console.log('%s: %s', red('Uncaught SyntaxError'), red(err.text))
+    // 将上下文几行都打印出来
+    for (let i = err.location.line - 3; i <= err.location.line + 1; i++) {
       console.log(
         '  @ line %d: %s',
-      err.location.line,
+        i + linePatch,
-      cyan(err.location.lineText)
+        err.location.line === i + 1 ? red(lines[i]) : lines[i]
       )
+    }
     console.log(
       '  @ %s:%d:%d',
       blue(filename),
-      err.location.line,
+      err.location.line + linePatch - 1,
       err.location.column
     )
   }
 
-  return (
+  code = code
-    code
     .replace(/\r\n/g, '\n')
     .replace(/process\.env\.NODE_ENV/g, `'${process.env.NODE_ENV}'`)
     .replace(
-        /import ([\w\W]*?) from (["'])(.*?)\2/g,
+      /(import|export) ([^'"]*?) from (["'])(.*?)\3/g,
-        function (m, alias, q, name) {
+      function (m, t, alias, q, name) {
         if (name.startsWith('@/')) {
           name = name.replace('@/', urlJoin(DEPLOY_PATH, ASSETS_DIR))
         }
@@ -186,7 +176,11 @@ export function parseJs(
             name = name.replace(/^\//, urlJoin(DEPLOY_PATH, ASSETS_DIR))
           }
 
-            if (!name.endsWith('.js') && !name.endsWith('.vue')) {
+          if (
+            !name.endsWith('.js') &&
+            !name.endsWith('.mjs') &&
+            !name.endsWith('.vue')
+          ) {
             if (name.includes('components')) {
               name += '.vue'
             } else {
@@ -197,7 +191,19 @@ export function parseJs(
         if (isBuild) {
           name = name.replace(/\.vue$/, '.js')
         }
-          return `import ${alias} from '${name}'`
+        if (alias.trim() === '*') {
+          return `${t} ${alias} from '${name}'`
+        }
+        let _alias = alias
+        let _import = ''
+        if (alias.includes('* as')) {
+          _alias = ' default ' + alias.replace('* as', '').trim()
+        }
+
+        _import = `import ${alias} from '${name}'`
+        _import += t === 'export' ? `\nexport ${_alias}` : ''
+
+        return _import
       }
     )
     .replace(/import\((['"])(.*?)\1\)/g, function (m, q, name) {
@@ -207,25 +213,49 @@ export function parseJs(
       if (name.startsWith('@/')) {
         name = name.replace('@/', urlJoin(DEPLOY_PATH, ASSETS_DIR))
       }
+      if (name.endsWith('.json')) {
+        name += '.js'
+      }
       return `import('${name}')`
     })
     .replace(/import (["'])(.*?)\1/g, function (m, q, name) {
       if (name.endsWith('.css') || name.endsWith('.scss')) {
         if (name.startsWith('@/')) {
-            name = name.replace('@/', '/')
+          name = name.replace('@/', urlJoin(DEPLOY_PATH, ASSETS_DIR))
         }
 
         if (isBuild) {
           name = name.replace(/\.scss/, '.css')
         }
-          let tmp = `style_${uuid()}`
-          fixedStyle += `__sheets__.push(${tmp})\n`
-
         // 修正那反人类的windows路径
-          return `import ${tmp} from '${name}' assert { type: 'css' }\n${tmp}.path = '${name.replace(
+        let _name = name.replace(/\\/g, '/').replace('@/', '')
-            /\\/g,
+        let tmp = `style_${uuid()}`
-            '/'
+
-          )}'`
+        // 因为esm语法的变更, 原先的 import xx from xx assets {type: css} 变为了 with
+        // 而这个语法的变化, 构建工具无法做版本判断, 故, 统一降级到fetch()加载
+        if (LEGACY_MODE) {
+          fixedStyle +=
+            `{\n` +
+            `  let stylesheet = document.createElement('style');\n` +
+            `  stylesheet.setAttribute('name', '${_name}');\n` +
+            `  stylesheet.textContent = ${tmp};\n` +
+            `  document.head.appendChild(stylesheet);\n` +
+            `}\n`
+
+          return `let ${tmp};\n!(async function(){\n  ${tmp} = await __fite_import('${name}', import.meta.url);\n})()`
+        } else {
+          // CSSStyleSheet.replaceSync 需要FF v101, Safari 16.4才支持
+          fixedStyle +=
+            `{\n` +
+            `  let stylesheet = new CSSStyleSheet();\n` +
+            `  stylesheet.path = '${_name}';\n` +
+            `  stylesheet.replaceSync(${tmp} );\n` +
+            `  __sheets__.push(stylesheet);\n` +
+            `  document.adoptedStyleSheets = __sheets__;\n` +
+            `}\n`
+
+          return `const ${tmp} = await __fite_import('${name}', import.meta.url)`
+        }
       } else {
         if (name.startsWith('@/')) {
           name = name.replace('@/', urlJoin(DEPLOY_PATH, ASSETS_DIR))
@@ -247,17 +277,32 @@ export function parseJs(
             name = name.replace(/^\//, urlJoin(DEPLOY_PATH, ASSETS_DIR))
           }
 
-            if (!name.endsWith('.js') && !name.endsWith('.vue')) {
+          if (
+            !name.endsWith('.js') &&
+            !name.endsWith('.mjs') &&
+            !name.endsWith('.vue')
+          ) {
             name += '.js'
           }
         }
 
         return `import '${name}'`
       }
-      }) +
+    })
-    fixedStyle +
+
-    (IS_ENTRY ? '\ndocument.adoptedStyleSheets = __sheets__' : '')
+  for (let key in define) {
-  )
+    code = code.replaceAll(key, define[key])
+  }
+
+  if (fixedStyle) {
+    code += '\n\n' + (IS_ENTRY ? SHEETS_DEF : '') + fixedStyle
+
+    if (IS_ENTRY && !LEGACY_MODE) {
+      code += '\ndocument.adoptedStyleSheets = __sheets__'
+    }
+  }
+
+  return code
 }
 
 /**
@@ -265,16 +310,19 @@ export function parseJs(
  * @param file <String> 文件路径
  * @return <String> 返回转换后的js代码
  */
-export function compileVue(file, imports, options = {}) {
+export async function compileVue(file, imports, options = {}) {
+  // 修正那反人类的windows路径
   let filename = file.slice(options.SOURCE_DIR.length).replace(/\\/g, '/')
   let code = (fs.cat(file) || '').toString().replace(/\r\n/g, '\n')
   let CACHE = options.CACHE || {}
+  let { isCustomElement } = options
   let output = '',
     scoped = false
 
   let js = code.match(JS_EXP)
   let scss = [...code.matchAll(STYLE_EXP)]
   let html = code.match(HTML_EXP) || ['', '']
+  let linePatch = code.slice(0, js?.index || 0).split('\n').length // js起始行数修正
 
   let hash = md5(file)
   let scopeId = 'data-' + hash
@@ -283,20 +331,25 @@ export function compileVue(file, imports, options = {}) {
     .map(it => {
       let css
       if (it.length > 2) {
-        css = compileScss(it[2])
+        css = compileScss(it[2], options.INJECT_SCSS)
 
         if (it[1].includes('scoped')) {
           scoped = true
           css = scopeCss(css, hash)
         }
       } else {
-        css = compileScss(it[1])
+        css = compileScss(it[1], options.INJECT_SCSS)
       }
+
       return css
     })
     .join(' ')
 
-  js = js ? js[1] : ''
+  for (let fn of options.plugin) {
+    scss = await fn('css', scss)
+  }
+
+  js = js ? js[1] : 'export default {}'
 
   try {
     html = compile(html[1], {
@@ -306,11 +359,11 @@ export function compileVue(file, imports, options = {}) {
       cacheHandlers: true,
       scopeId: scoped ? scopeId : void 0,
       sourceMap: false,
-      isCustomElement: tag => tag.startsWith('wc-')
+      isCustomElement
     }).code.replace('export function render', 'function render')
   } catch (err) {
-    let tmp = html[1].split('\n')
+    let lines = html[1].split('\n')
-    let line = tmp[err.loc.start.line - 1]
+    let line = lines[err.loc?.start.line - 1]
 
     console.log('%s: %s', red('SyntaxError'), red(err.message))
     console.log(
@@ -351,15 +404,26 @@ function render(_ctx, _cache) {
     CACHE[file] = { changed: false, js, html }
   }
 
-  output += parseJs(js, imports, options, file).replace(
+  output += parseJs(js, imports, options, file, linePatch).replace(
     'export default {',
-    `\nconst __sheets__ = [...document.adoptedStyleSheets]\n${html}\n\nconst __sfc__ = {\n  render,\n`
+    `\n${
+      options.LEGACY_MODE ? '' : SHEETS_DEF
+    }${html}\n\nconst __sfc__ = {\n  render,\n`
   )
 
   if (scss) {
     CACHE[file].css = scss
 
-    // 修正那反人类的windows路径
+    if (options.LEGACY_MODE) {
+      output += `
+{
+  let stylesheet = document.createElement('style')
+  stylesheet.setAttribute('name', '${filename}')
+  stylesheet.textContent = \`${scss}\`
+  document.head.appendChild(stylesheet)
+}
+`
+    } else {
       output += `
 {
   let stylesheet = new CSSStyleSheet()
@@ -370,6 +434,7 @@ function render(_ctx, _cache) {
 document.adoptedStyleSheets = __sheets__
 `
     }
+  }
   if (scoped) {
     output += `__sfc__.__scopeId = '${scopeId}'\n`
   }
@@ -381,17 +446,15 @@ document.adoptedStyleSheets = __sheets__
 /**
  * 解析模板html
  */
-export function parseHtml(html, { page, imports, entry }) {
+export function parseHtml(html, { page, imports, entry, LEGACY_MODE }) {
   return html
     .replace(/\r\n/g, '\n')
     .replace(
       '</head>',
       `${
         process.env.NODE_ENV === 'development'
-          ? `  <script>${
+          ? `  <script>${minify(createHmrScript(LEGACY_MODE))}</script>\n`
-              Es.transformSync(HMR_SCRIPT, { minify: true }).code
+          : `  <script>${minify(LEGACY_POLYFILL)}</script>\n`
-            }</script>`
-          : ''
       }</head>`
     )
     .replace('{{title}}', page.title || '')

lib/compile.js

@@ -0,0 +1,113 @@
+/**
+ * {}
+ * @author yutent<yutent.io@gmail.com>
+ * @date 2023/06/14 18:36:15
+ */
+
+import { join, dirname, parse, normalize } from 'node:path'
+import fs from 'iofs'
+import Es from 'esbuild'
+import { compileScss, parseJs, compileVue, parseHtml } from './compile-vue.js'
+
+const template = fs.cat(join(process.cwd(), 'index.html')).toString()
+
+export async function compileFiles(
+  currentPage,
+  page,
+  files,
+  options,
+  { verbose, imports, dist } = {}
+) {
+  let pageDir = options.IS_MPA && currentPage ? `pages/${currentPage}` : ''
+  options.currentPage = currentPage
+
+  for (let [path, it] of files) {
+    // 入口文件, 特殊处理
+    if (page && path === page.entry) {
+      let entry = fs.cat(page.entry).toString()
+
+      entry = parseJs(entry, imports, { ...options, IS_ENTRY: true })
+
+      let code = parseHtml(template, {
+        page,
+        imports,
+        entry,
+        LEGACY_MODE: options.LEGACY_MODE
+      })
+
+      fs.echo(code, join(dist, `${currentPage}.html`))
+      continue
+    }
+
+    verbose && console.log('  解析 %s ...', it.name)
+
+    switch (it.ext) {
+      case '.vue':
+        {
+          let code = await compileVue(path, imports, options)
+
+          await Es.transform(code, { minify: true }).then(async ({ code }) => {
+            for (let fn of options.plugin) {
+              code = await fn('js', code)
+            }
+            fs.echo(
+              code,
+              join(dist, 'assets/', pageDir, it.name.replace(/\.vue$/, '.js'))
+            )
+          })
+        }
+        break
+
+      case '.js':
+        {
+          let code = fs.cat(path)
+
+          code = parseJs(code + '', imports, options)
+
+          await Es.transform(code, { minify: true }).then(async ({ code }) => {
+            for (let fn of options.plugin) {
+              code = await fn('js', code)
+            }
+            fs.echo(code, join(dist, 'assets/', pageDir, it.name))
+          })
+        }
+        break
+      // es2024之后esm的语法的assets 变成了with, 对构建工具来说无法适配到具体的浏览器
+      // 故把json文件改成js文件
+      case '.json':
+        {
+          let code = fs.cat(path)
+
+          code = 'export default ' + JSON.stringify(JSON.parse(code + ''))
+
+          fs.echo(code, join(dist, 'assets/', pageDir, it.name + '.js'))
+        }
+        break
+
+      case '.scss':
+      case '.css':
+        {
+          let target = join(
+            dist,
+            'assets/',
+            pageDir,
+            it.name.replace(/\.scss$/, '.css')
+          )
+          if (it.ext === '.css') {
+            fs.cp(path, target)
+          } else {
+            let code = compileScss(path)
+            for (let fn of options.plugin) {
+              code = await fn('css', code)
+            }
+            fs.echo(code, target)
+          }
+        }
+        break
+
+      default:
+        fs.cp(path, join(dist, 'assets/', pageDir, it.name))
+        break
+    }
+  }
+}

lib/constants.js

@@ -4,9 +4,9 @@
  * @date 2022/09/06 11:54:56
  */
 
-export const JS_EXP = /<script[^>]*?>([\w\W]*?)<\/script>/
+export const JS_EXP = /(?<=\n|^)<script[^>]*?>([\w\W]*?)<\/script>/
-export const STYLE_EXP = /<style([^>]*?)>([\w\W]*?)<\/style>/g
+export const STYLE_EXP = /(?<=\n|^)<style([^>]*?)>([\w\W]*?)<\/style>/g
-export const HTML_EXP = /<template[^>]*?>([\w\W]*?)\n<\/template>/
+export const HTML_EXP = /(?<=\n|^)<template[^>]*?>([\w\W]*?)\n<\/template>/
 export const V_DEEP = /:deep\(([^)]*?)\)(.*)/
 
 export const CSS_SHEET_EXP = /([%@\w\.,#\-:>\+\~\|\(\)\[\]"'\=\s]+)\{/g
@@ -16,47 +16,35 @@ export const COMMON_HEADERS = {
   'Cache-Control': 'no-store'
 }
 
-export const HMR_SCRIPT = `
+export const SHEETS_DEF =
-!(function vue_live_hmr(){
+  'const __sheets__ = [...document.adoptedStyleSheets];\n'
-  var ws = new WebSocket(\`ws\${location.protocol === 'https:' ? 's' : ''}://\${location.host}/ws-vue-live\`)
 
-  ws.addEventListener('open', function (r) {
+export const LEGACY_POLYFILL = `!(function(){
-    if(vue_live_hmr.closed){
+  function join(p1, p2) {
-      delete vue_live_hmr.closed
+    let tmp1 = p1.split('/')
-      location.reload()
+    let tmp2 = p2.split('/')
+    if (tmp1.at(-1) === '') {
+      tmp1.pop()
     }
-    console.log('vue-live hmr ready...')
+    while (tmp2.length) {
-  })
+      let tmp = tmp2.shift()
-
+      if (tmp === '.' || tmp === '') {
-  ws.addEventListener('close', function(){
+        continue
-    vue_live_hmr.closed = true
+      } else if (tmp === '..') {
-    setTimeout(vue_live_hmr, 2000)
+        tmp1.pop()
-  })
+      } else {
-  
+        tmp1.push(tmp)
-  ws.addEventListener('message', function (ev) {
-    var { action, data } = JSON.parse(ev.data)
-  
-    switch (action) {
-      case 'reload':
-        location.reload()
-        break
-  
-      case 'render':
-        {
-          let tmp = [...document.adoptedStyleSheets]
-          for (let i = -1, it; (it = tmp[++i]); ) {
-            if (it.path === data.path) {
-              let stylesheet = new CSSStyleSheet()
-              stylesheet.path = data.path
-              stylesheet.replaceSync(data.content)
-              tmp[i] = stylesheet
-              document.adoptedStyleSheets = tmp
-              break
       }
     }
+    return tmp1.join('/')
   }
-        break
+  
+  window.__fite_import = function(url,relPath){
+    let absPath = relPath.split('/').slice(0, -1).join('/')
+    let req
+    if(url.startsWith('./') || url.startsWith('../')) {
+      url = join(absPath, url)
     }
-  })
+    return window.fetch(url).then(r =>  r.text())
-})()
+  }
-`
+})()`

lib/dev.js

@@ -1,13 +1,13 @@
-import http from 'http'
+import http from 'node:http'
-import https from 'https'
+import https from 'node:http2'
 import fs from 'iofs'
-import { join, dirname } from 'path'
+import { join, dirname } from 'node:path'
-import { parse } from 'url'
+import { parse } from 'node:url'
 import socket from './ws.js'
 import chokidar from 'chokidar'
 import { red } from 'kolorist'
 
-import { friendlyErrors } from './utils.js'
+import { friendlyErrors, defaultCustomElement, gzip } from './utils.js'
 
 import { compileScss, parseJs, compileVue, parseHtml } from './compile-vue.js'
 
@@ -15,9 +15,13 @@ import MIME_TYPES from './mime-tpyes.js'
 import { COMMON_HEADERS } from './constants.js'
 
 const noc = Buffer.from('')
-const SERVER_OPTIONS = {}
+const SERVER_OPTIONS = { allowHTTP1: true }
 const CACHE = {} //文件缓存, 用于hmr
 
+function readFile(file) {
+  return (file && fs.cat(file)?.toString()) || ''
+}
+
 export default async function createServer(root = '', conf = {}) {
   const SOURCE_DIR = join(root, 'src')
   const PUBLIC_DIR = join(root, 'public')
@@ -26,6 +30,22 @@ export default async function createServer(root = '', conf = {}) {
   const PORT = conf.devServer.port || 8080
   const USE_HTTPS = conf.devServer.https
   const DOMAIN = conf.devServer.domain || 'localhost'
+  const INJECT_SCSS = readFile(conf.inject?.scss)
+  const LEGACY_MODE = !!conf.legacy
+  const ENABLE_GZIP = !!conf.devServer.gzip
+  const { isCustomElement = defaultCustomElement } = conf.compileOptions || {}
+  const { plugin = [], define = {} } = conf
+
+  if (conf.imports['vue-dev']) {
+    conf.imports.vue = conf.imports['vue-dev']
+  }
+  if (conf.imports['vue-router-dev']) {
+    conf.imports['vue-router'] = conf.imports['vue-router-dev']
+  }
+
+  if (conf.devServer.headers) {
+    Object.assign(COMMON_HEADERS, conf.devServer.headers)
+  }
 
   if (USE_HTTPS) {
     Object.assign(SERVER_OPTIONS, conf.devServer.ssl)
@@ -37,7 +57,7 @@ export default async function createServer(root = '', conf = {}) {
   }
 
   const server = (USE_HTTPS ? https : http)
-    .createServer(SERVER_OPTIONS)
+    [USE_HTTPS ? 'createSecureServer' : 'createServer'](SERVER_OPTIONS)
     .listen(PORT)
 
   const ws = socket(server)
@@ -54,7 +74,7 @@ export default async function createServer(root = '', conf = {}) {
     currentPage = ''
 
   server
-    .on('request', function (req, res) {
+    .on('request', async function (req, res) {
       let prefix = DEPLOY_PATH ? DEPLOY_PATH.replace(/\/$/, '') : ''
       let url =
         prefix && req.url.startsWith(prefix)
@@ -67,7 +87,7 @@ export default async function createServer(root = '', conf = {}) {
 
       if (prefix && req.url === '/') {
         res.setHeader('Location', DEPLOY_PATH)
-        res.writeHead(302, 'Redirect')
+        res.writeHead(302, USE_HTTPS ? void 0 : 'Redirect')
         return res.end('')
       }
 
@@ -92,7 +112,7 @@ export default async function createServer(root = '', conf = {}) {
             pageName = tmp.join('.')
             // 页面不存在时输出404, 避免进程崩溃退出
             if (!conf.pages[pageName]) {
-              res.writeHead(404, 'Not Found')
+              res.writeHead(404, USE_HTTPS ? void 0 : 'Not Found')
               return res.end(`Oops!!! 404 Not Found`)
             }
 
@@ -130,8 +150,13 @@ export default async function createServer(root = '', conf = {}) {
 
       if (isIndex) {
         res.setHeader('content-type', MIME_TYPES.html)
-        res.writeHead(200, 'OK')
+        res.writeHead(200, USE_HTTPS ? void 0 : 'OK')
-        res.end('<ul>' + indexPage + '</ul>')
+        res.end(
+          '<style>body{font:14px/1.5 Arial}ol{display:flex;flex-wrap:wrap;}li{width:30%;}a{color:teal}a:visited{color:orange;}</style>' +
+            '<div>注意: 你看到这个页面, 仅在开发时可见。<br>仅为了方便开发多页应用时访问自己想要修改的页面, 而不需要手动输入地址。</div><ol>' +
+            indexPage +
+            '</ol>'
+        )
       } else {
         res.setHeader('accept-ranges', 'bytes')
 
@@ -146,21 +171,39 @@ export default async function createServer(root = '', conf = {}) {
               let entry = fs.cat(page.entry)?.toString()
               let html = fs.cat(join(process.cwd(), 'index.html')).toString()
 
-              entry = parseJs(entry, conf.imports, {
+              entry = parseJs(
+                entry,
+                conf.imports,
+                {
                   IS_MPA,
                   currentPage,
                   IS_ENTRY: true,
-                DEPLOY_PATH
+                  DEPLOY_PATH,
-              })
+                  LEGACY_MODE,
+                  isCustomElement,
+                  plugin,
+                  define
+                },
+                page.entry
+              )
 
-              code = parseHtml(html, { page, imports: conf.imports, entry })
+              for (let fn of plugin) {
+                entry = await fn('js', entry)
+              }
+
+              code = parseHtml(html, {
+                page,
+                imports: conf.imports,
+                entry,
+                LEGACY_MODE
+              })
             }
 
             break
 
           case 'vue':
             {
-              let rpath = pathname.replace(/@\//, '')
+              let rpath = pathname.replace('@/', '')
               let file
 
               if (IS_MPA) {
@@ -175,19 +218,28 @@ export default async function createServer(root = '', conf = {}) {
               }
               if (!fs.isfile(file)) {
                 friendlyErrors(pathname, ext)
-                res.writeHead(404, 'Not Found')
+                res.writeHead(404, USE_HTTPS ? void 0 : 'Not Found')
                 res.end('')
                 return
               }
 
-              code = compileVue(file, conf.imports, {
+              code = await compileVue(file, conf.imports, {
                 IS_MPA,
                 currentPage,
                 SOURCE_DIR,
                 CACHE,
-                DEPLOY_PATH
+                DEPLOY_PATH,
+                INJECT_SCSS,
+                LEGACY_MODE,
+                isCustomElement,
+                plugin,
+                define
               })
 
+              for (let fn of plugin) {
+                code = await fn('js', code)
+              }
+
               res.setHeader('content-type', MIME_TYPES.js)
             }
             break
@@ -195,27 +247,43 @@ export default async function createServer(root = '', conf = {}) {
           case 'scss':
           case 'css':
             {
-              let file = join(SOURCE_DIR, pathname.replace(/@\//, ''))
+              let file = join(SOURCE_DIR, pathname.replace('@/', ''))
               if (!fs.isfile(file)) {
-                file = join(PUBLIC_DIR, pathname.replace(/@\//, ''))
+                file = join(PUBLIC_DIR, pathname.replace('@/', ''))
 
                 if (!fs.isfile(file)) {
                   friendlyErrors(pathname, ext)
-                  res.setHeader('content-type', MIME_TYPES.html)
+                  res.writeHead(404, USE_HTTPS ? void 0 : 'Not Found')
-                  res.writeHead(404, 'Not Found')
                   res.end('')
                   return
                 }
               }
+              if (file === conf.inject?.scss) {
+                console.log(red('设置为注入的样式文件不可被vue/js文件引用\n'))
+                res.writeHead(404, USE_HTTPS ? void 0 : 'Not Found')
+                res.end('')
+                return
+              }
               code = compileScss(file)
+              for (let fn of plugin) {
+                code = await fn('css', code)
+              }
               res.setHeader('content-type', MIME_TYPES.css)
             }
             break
 
           case 'js':
+          case 'wasm':
             {
-              let rpath = pathname.replace(/@\//, '')
+              let rpath = pathname.replace('@/', '')
               let file
+              let isJson = false
+              let isWasm = rpath.endsWith('.wasm')
+
+              if (rpath.endsWith('json.js')) {
+                isJson = true
+                rpath = rpath.slice(0, -3)
+              }
 
               if (IS_MPA) {
                 // 判断前后2个值相等, 避免出现目录名和页面名字相同时走错逻辑
@@ -231,19 +299,43 @@ export default async function createServer(root = '', conf = {}) {
               if (fs.isfile(file)) {
                 code = fs.cat(file)
               } else if (fs.isfile(join(PUBLIC_DIR, rpath))) {
-                code = fs.cat(join(PUBLIC_DIR, rpath))
+                file = join(PUBLIC_DIR, rpath)
+                code = fs.cat(file)
               } else {
                 friendlyErrors(rpath, ext)
-                res.writeHead(404, 'Not Found')
+                res.writeHead(404, USE_HTTPS ? void 0 : 'Not Found')
                 res.end('')
                 return
               }
-              code = parseJs(code + '', conf.imports, {
+              if (isJson) {
+                try {
+                  code =
+                    'export default ' + JSON.stringify(JSON.parse(code + ''))
+                } catch (err) {
+                  console.log('%s 语法错误: %s', rpath, red(err.message))
+                }
+              } else if (isWasm) {
+                //
+              } else {
+                code = parseJs(
+                  code + '',
+                  conf.imports,
+                  {
                     IS_MPA,
                     currentPage,
-                DEPLOY_PATH
+                    DEPLOY_PATH,
-              })
+                    LEGACY_MODE,
-              res.setHeader('content-type', MIME_TYPES.js)
+                    isCustomElement,
+                    plugin,
+                    define
+                  },
+                  file
+                )
+                for (let fn of plugin) {
+                  code = await fn('js', code)
+                }
+              }
+              res.setHeader('content-type', MIME_TYPES[ext])
             }
 
             break
@@ -276,16 +368,23 @@ export default async function createServer(root = '', conf = {}) {
             }
             if (code === null) {
               friendlyErrors(pathname, ext)
-              res.writeHead(404, 'Not Found')
+              res.writeHead(404, USE_HTTPS ? void 0 : 'Not Found')
               res.end('')
               return
             }
             break
         }
 
-        res.setHeader('content-length', Buffer.byteLength(code || noc))
+        if (ENABLE_GZIP) {
-        res.writeHead(200, 'OK')
+          code = gzip(code || noc)
-        res.end(code || noc)
+          res.setHeader('Content-Encoding', 'gzip')
+        } else {
+          code = code || noc
+        }
+
+        res.setHeader('Content-Length', Buffer.byteLength(code))
+        res.writeHead(200, USE_HTTPS ? void 0 : 'OK')
+        res.end(code)
       }
     })
 
@@ -302,7 +401,7 @@ export default async function createServer(root = '', conf = {}) {
       console.log(
         '  本地: %s://%s:%d%s',
         USE_HTTPS ? 'https' : 'http',
-        '127.0.0.1',
+        USE_HTTPS ? 'localhost' : '127.0.0.1',
         PORT,
         DEPLOY_PATH
       )
@@ -315,7 +414,7 @@ export default async function createServer(root = '', conf = {}) {
       )
       chokidar
         .watch([SOURCE_DIR, PUBLIC_DIR, join(root, './index.html')])
-        .on('all', (act, filePath) => {
+        .on('all', async (act, filePath) => {
           if (ready) {
             let file = filePath.slice(SOURCE_DIR.length)
 
@@ -326,7 +425,18 @@ export default async function createServer(root = '', conf = {}) {
                 case 'css':
                 case 'scss':
                   {
-                    let content = fs.cat(filePath).toString()
+                    let content = ''
+                    if (filePath === conf.inject?.scss) {
+                      return
+                    }
+                    if (ext === 'scss') {
+                      content = compileScss(filePath)
+                    } else {
+                      content = fs.cat(filePath).toString()
+                    }
+                    for (let fn of plugin) {
+                      content = await fn('css', content)
+                    }
                     ws.send({
                       action: 'render',
                       data: { path: file.replace(/\\/g, '/'), content }
@@ -336,12 +446,17 @@ export default async function createServer(root = '', conf = {}) {
 
                 case 'vue':
                   {
-                    let content = compileVue(filePath, conf.imports, {
+                    let content = await compileVue(filePath, conf.imports, {
                       IS_MPA,
                       currentPage,
                       SOURCE_DIR,
                       CACHE,
-                      DEPLOY_PATH
+                      DEPLOY_PATH,
+                      INJECT_SCSS,
+                      LEGACY_MODE,
+                      isCustomElement,
+                      plugin,
+                      define
                     })
                     let tmp = CACHE[filePath]
                     if (tmp.changed) {

lib/prod.js

@@ -1,88 +1,124 @@
+import { join, dirname, parse, normalize } from 'node:path'
+import { Worker, parentPort } from 'node:worker_threads'
+import os from 'node:os'
 import fs from 'iofs'
-import { join, dirname, parse } from 'path'
-import Es from 'esbuild'
-import { compileScss, parseJs, compileVue, parseHtml } from './compile-vue.js'
 
-export default function compile(root = '', dist = '', conf = {}) {
+import { compileFiles } from './compile.js'
+import { defaultCustomElement } from './utils.js'
+
+const IS_WIN = process.platform === 'win32'
+const PREFIX = IS_WIN ? 'pages\\' : 'pages/'
+// 4核(或4线程)以上的CPU, 才开启多线程编译。且线程开销太高, 开太多线程效率反而不高。
+const CPU_CORES = os.cpus().length > 5 ? 6 : os.cpus().length
+const THREADS_NUM = CPU_CORES > 3 ? CPU_CORES - 1 : 0
+const __filename = normalize(import.meta.url.slice(IS_WIN ? 8 : 7))
+const __dirname = dirname(__filename)
+const WORKER_POOL = new Set() // 线程池
+const JOBS_QUEUE = [] // 任务队列
+
+function readFile(file) {
+  return (file && fs.cat(file)?.toString()) || ''
+}
+
+function doJob() {
+  while (JOBS_QUEUE.length && WORKER_POOL.size) {
+    let job = JOBS_QUEUE.shift()
+    let worker = WORKER_POOL.values().next().value
+
+    WORKER_POOL.delete(worker)
+
+    worker.once('message', _ => {
+      if (JOBS_QUEUE.length) {
+        WORKER_POOL.add(worker)
+        doJob()
+      } else {
+        worker.terminate()
+      }
+    })
+
+    worker.postMessage(job)
+  }
+}
+
+export default function compile(root = '', dist = '', conf = {}, verbose) {
   //
   const SOURCE_DIR = join(root, 'src')
   const PUBLIC_DIR = join(root, 'public')
   const DEPLOY_PATH = conf.base || '' // 部署目录, 默认是根目录部署
-  const IS_MPA = Object.keys(conf.pages).length > 1
+  const PAGES_KEYS = Object.keys(conf.pages)
+  const IS_MPA = PAGES_KEYS.length > 1
+  const PAGES_PREFIX = PAGES_KEYS.map(it =>
+    IS_WIN ? `${PREFIX + it}\\` : `${PREFIX + it}/`
+  )
+  const INJECT_SCSS = readFile(conf.inject?.scss)
+  const LEGACY_MODE = !!conf.legacy
+  const {
+    ABS_CONFIG_FILEPATH,
+    compileOptions = {},
+    define = {},
+    plugin = []
+  } = conf
+  const { isCustomElement = defaultCustomElement } = compileOptions
+
+  conf.inject = conf.inject || { scss: '' }
 
   let timeStart = Date.now()
-  let template = fs.cat(join(process.env.PWD, 'index.html')).toString()
+  let template = fs.cat(join(process.cwd(), 'index.html')).toString()
-
+  let list = new Map()
-  let list = fs
+  let options = {
-    .ls(SOURCE_DIR, true)
+    IS_MPA,
-    .map(it => ({
+    SOURCE_DIR,
-      name: it.slice(SOURCE_DIR.length + 1),
+    DEPLOY_PATH,
-      path: it,
+    INJECT_SCSS,
-      ext: parse(it).ext
+    LEGACY_MODE,
-    }))
+    ABS_CONFIG_FILEPATH,
-    .filter(it => fs.isfile(it.path) && it.ext !== '')
+    define
-
-  let compileFiles = function (currentPage, page, files) {
-    let options = { IS_MPA, currentPage, SOURCE_DIR, DEPLOY_PATH }
-
-    for (let it of files) {
-      // 入口文件, 特殊处理
-      if (page && it.path === page.entry) {
-        let entry = fs.cat(page.entry).toString()
-
-        entry = parseJs(entry, conf.imports, { ...options, IS_ENTRY: true })
-
-        let code = parseHtml(template, { page, imports: conf.imports, entry })
-
-        fs.echo(code, join(dist, `${currentPage}.html`))
-        continue
   }
 
-      console.log('  解析 %s ...', it.name)
+  fs.ls(SOURCE_DIR, true).forEach(path => {
+    if (fs.isdir(path)) {
+      return
+    }
 
-      let pageDir = IS_MPA && currentPage ? `pages/${currentPage}` : ''
+    let name = path.slice(SOURCE_DIR.length + 1)
+    let it = {
+      name,
+      ext: parse(name).ext
+    }
+    if (it.ext) {
+      if (IS_MPA && it.name.startsWith(PREFIX)) {
+        if (PAGES_PREFIX.some(it => it.startsWith(it.name))) {
+          list.set(path, it)
+        }
+        return
+      }
 
-      switch (it.ext) {
+      if (path === conf.inject.scss) {
-        case '.vue':
+        return
-          {
+      }
-            let code = compileVue(it.path, conf.imports, options)
+      list.set(path, it)
+    }
+  })
 
-            Es.transform(code, { minify: true }).then(r => {
+  // 创建线程池
-              fs.echo(
+  if (THREADS_NUM > 0 && (IS_MPA || list.size > THREADS_NUM * 10)) {
-                r.code,
+    // 页面数过少时, 线程数量不能比页面数多
-                join(dist, 'assets/', pageDir, it.name.replace(/\.vue$/, '.js'))
+    let max = Math.min(THREADS_NUM, PAGES_KEYS.length)
+
+    for (let i = 0; i < max; i++) {
+      WORKER_POOL.add(
+        new Worker(join(__dirname, './thread.js'), {
+          workerData: {
+            options,
+            verbose,
+            dist,
+            imports: conf.imports
+          }
+        })
       )
-            })
-          }
-          break
-
-        case '.js':
-          {
-            let code = fs.cat(it.path)
-
-            code = parseJs(code + '', conf.imports, options)
-            Es.transform(code, { minify: true }).then(r => {
-              fs.echo(r.code, join(dist, 'assets/', pageDir, it.name))
-            })
-          }
-          break
-
-        case '.scss':
-        case '.css':
-          {
-            let code = compileScss(it.path)
-            if (!it.name.startsWith('assets')) {
-              it.name = 'assets/' + it.name
-            }
-            fs.echo(code, join(dist, it.name.replace(/\.scss$/, '.css')))
-          }
-          break
-
-        default:
-          fs.cp(it.path, join(dist, it.name))
-          break
-      }
     }
+  } else {
+    options.isCustomElement = isCustomElement
   }
 
   // 优先处理静态目录, 之后的源码目录中, 以便如果有产生相同的文件名, 则覆盖静态目录中的文件
@@ -91,53 +127,98 @@ export default function compile(root = '', dist = '', conf = {}) {
     fs.ls(PUBLIC_DIR, true).forEach(it => {
       let ext = parse(it).ext
 
-      if (ext === '') {
+      if (ext && fs.isfile(it)) {
-        return
-      }
-
-      if (fs.isfile(it)) {
         let name = it.slice(PUBLIC_DIR.length + 1)
-        console.log('  复制 %s ...', name)
+        verbose && console.log('  复制 %s ...', name)
         fs.cp(it, join(dist, name))
       }
     })
   }
 
-  for (let currentPage in conf.pages) {
+  if (IS_MPA) {
+    for (let currentPage of PAGES_KEYS) {
       let page = conf.pages[currentPage]
       let dir = dirname(page.entry)
-    let files = list
+      let files = new Map()
-    if (IS_MPA) {
+      let chunk = new Map()
-      files = []
+
-      fs.ls(dir, true).forEach(it => {
+      fs.ls(dir, true).forEach(path => {
-        if (fs.isdir(it)) {
+        if (fs.isdir(path)) {
           return
         }
-        let idx = list.findIndex(_ => _.path === it)
+
-        let name = it.slice(dir.length + 1)
+        let name = path.slice(dir.length + 1)
         let ext = parse(name).ext
 
         if (ext === '') {
           return
         }
 
-        list.splice(idx, 1)
+        list.delete(path)
-
+        files.set(path, { name, ext })
-        files.push({
-          name,
-          path: it,
-          ext
       })
+
+      if (THREADS_NUM > 0) {
+        chunk.set(currentPage, { page, files })
+        JOBS_QUEUE.push(chunk)
+        doJob()
+      } else {
+        console.log(`正在生成 ${currentPage}.html ...`)
+        compileFiles(currentPage, page, files, options, {
+          verbose,
+          dist,
+          imports: conf.imports
         })
       }
-    console.log('正在生成 %s ...', `${currentPage}.html`)
-    compileFiles(currentPage, page, files)
     }
 
-  if (IS_MPA) {
+    // 公共依赖
+    if (THREADS_NUM > 0) {
+      let chunk = new Map()
+      chunk.set('', { page: null, files: list })
+
+      JOBS_QUEUE.push(chunk)
+      doJob()
+    } else {
       console.log('\n正在解析公共依赖 ...')
-    compileFiles('', null, list)
+      compileFiles('', null, list, options, {
+        verbose,
+        dist,
+        imports: conf.imports
+      })
+    }
+  } else {
+    // 每个线程处理的文件数
+    let chunkSize = Math.ceil(list.size / THREADS_NUM)
+    let currentPage = PAGES_KEYS[0]
+    let page = conf.pages[currentPage]
+
+    console.log(`正在生成 ${currentPage}.html ...`)
+
+    if (THREADS_NUM > 0 && list.size > THREADS_NUM * 10) {
+      list = [...list]
+      for (let i = 0; i < THREADS_NUM; i++) {
+        let start = i * chunkSize
+        let end = start + chunkSize
+        let chunk = new Map()
+
+        chunk.set(currentPage, { page, files: list.slice(start, end) })
+
+        JOBS_QUEUE.push(chunk)
+        doJob()
+      }
+    } else {
+      options.plugin = plugin
+      options.isCustomElement = isCustomElement
+      compileFiles(currentPage, page, list, options, {
+        verbose,
+        dist,
+        imports: conf.imports
+      })
+    }
   }
 
+  process.on('exit', _ => {
     console.log('\n页面处理完成, 耗时 %ss\n', (Date.now() - timeStart) / 1000)
+  })
 }

lib/thread.js

@@ -0,0 +1,40 @@
+/**
+ * 子线程
+ * @author yutent<yutent.io@gmail.com>
+ * @date 2023/06/14 16:15:39
+ */
+import { parentPort, workerData } from 'node:worker_threads'
+import { compileFiles } from './compile.js'
+import { defaultCustomElement } from './utils.js'
+
+const { options, verbose, dist, imports } = workerData
+const { ABS_CONFIG_FILEPATH } = options
+
+const { compileOptions = {}, plugin = [] } = await import(
+  ABS_CONFIG_FILEPATH
+).then(r => r.default)
+
+const { isCustomElement = defaultCustomElement } = compileOptions
+options.isCustomElement = isCustomElement
+options.plugin = plugin
+
+//
+async function doJob(job) {
+  let [currentPage, { page, files }] = job.entries().next().value
+
+  options.IS_MPA &&
+    console.log(
+      currentPage
+        ? `正在生成 ${currentPage}.html ...`
+        : '\n正在解析公共依赖 ...'
+    )
+
+  await compileFiles(currentPage, page, files, options, {
+    verbose,
+    dist,
+    imports
+  })
+  parentPort.postMessage(true)
+}
+
+parentPort.on('message', doJob)

lib/utils.js

@@ -1,4 +1,33 @@
+/**
+ * {一些工具类函数}
+ * @author yutent<yutent.io@gmail.com>
+ * @date 2023/05/22 14:52:00
+ */
+
+import { createHash, randomUUID } from 'node:crypto'
+import { join } from 'node:path'
+import { gzipSync } from 'node:zlib'
 import { red, cyan, blue } from 'kolorist'
+import { LEGACY_POLYFILL } from './constants.js'
+
+// 修正路径合并 避免在windows下被转义
+export function urlJoin(...args) {
+  return join(...args).replace(/\\/g, '/')
+}
+
+export function uuid() {
+  return randomUUID().slice(-8)
+}
+
+export function md5(str = '') {
+  let sum = createHash('md5')
+  sum.update(str, 'utf8')
+  return sum.digest('hex').slice(0, 8)
+}
+
+export function gzip(val) {
+  return gzipSync(val)
+}
 
 export function friendlyErrors(pathname, ext = '') {
   console.log(cyan(pathname), red(`not found!!!`))
@@ -7,3 +36,70 @@ export function friendlyErrors(pathname, ext = '') {
     blue(`/index.${ext}`)
   )
 }
+
+export function createHmrScript(legacy, session = '') {
+  return `
+  !(function vue_live_hmr(){
+    let ws = new WebSocket(\`ws\${location.protocol === 'https:' ? 's' : ''}://\${location.host}/ws-fite-hmr?session=\${btoa(location.pathname).replace(/[=\+\/]/g, '')}&lock=\${localStorage.getItem(location.pathname) || 0}\`)
+  
+    ws.addEventListener('open', function (r) {
+      if(vue_live_hmr.closed){
+        delete vue_live_hmr.closed
+        location.reload()
+      }
+      console.log('fite hmr ready...')
+    })
+  
+    ws.addEventListener('close', function(){
+      vue_live_hmr.closed = true
+      if (localStorage.getItem(location.pathname) === '1') {
+        return
+      }
+      setTimeout(vue_live_hmr, 2000)
+    })
+    
+    ws.addEventListener('message', function (ev) {
+      var { action, data } = JSON.parse(ev.data)
+    
+      switch (action) {
+        case 'reload':
+          location.reload()
+          break
+    
+        case 'render':
+          {
+            ${
+              legacy
+                ? `
+            let stylesheet = document.head.children.namedItem(data.path)
+            if (stylesheet) {
+              stylesheet.textContent = data.content
+            }
+            `
+                : `
+            let tmp = [...document.adoptedStyleSheets]
+            for (let i = -1, it; (it = tmp[++i]); ) {
+              if (it.path === data.path) {
+                let stylesheet = new CSSStyleSheet()
+                stylesheet.path = data.path
+                stylesheet.replaceSync(data.content)
+                tmp[i] = stylesheet
+                document.adoptedStyleSheets = tmp
+                break
+              }
+            }
+            `
+            }
+          }
+          break
+      }
+    })
+    ${LEGACY_POLYFILL}
+  })()
+  `
+}
+
+// 默认的 web components 判断
+export function defaultCustomElement(tag) {
+  return tag.startsWith('wc-')
+}

lib/ws.js

@@ -6,28 +6,40 @@
 import { WebSocketServer } from 'ws'
 
 class WebSocket {
-  #ws = null // ws实例
+  #clients = new Map()
   #queue = [] // 消息队列
 
   constructor(server) {
     if (server.listening) {
-      let conn = new WebSocketServer({ server, path: '/ws-vue-live' })
+      let conn = new WebSocketServer({ server, path: '/ws-fite-hmr' })
-      conn.on('connection', ws => {
+      conn.on('connection', (client, req) => {
-        this.#ws = ws
+        let params = new URLSearchParams(req.url.slice(req.url.indexOf('?')))
-        // ws.on('message', data => {
+        let session = params.get('session')
-        //   console.log(data + '');
+        let lock = +params.get('lock')
-        // })
+
+        if (lock === 1) {
+          client.close()
+        } else {
+          this.#clients.set(session, client)
+
+          client.once('close', _ => {
+            this.#clients.delete(session)
+          })
+
           while (this.#queue.length) {
             let msg = this.#queue.shift()
             this.send(msg)
           }
+        }
       })
     }
   }
 
   send(msg = {}) {
-    if (this.#ws) {
+    if (this.#clients.size) {
-      this.#ws.send(JSON.stringify(msg))
+      for (let [key, client] of this.#clients) {
+        client.send(JSON.stringify(msg))
+      }
     } else {
       this.#queue.push(msg)
     }

package.json

@@ -1,7 +1,7 @@
 {
   "name": "fite",
   "type": "module",
-  "version": "0.9.4",
+  "version": "1.4.4",
   "bin": {
     "fite": "index.js"
   },
@@ -19,5 +19,6 @@
   },
   "engines": {
     "node": ">=16.6.0"
-  }
+  },
+  "license": "MIT"
 }