front-cpu 0.1.2 → 0.1.4

package/src/plugins/vite-call-source.ts

@@ -0,0 +1,388 @@
+import type { Plugin } from 'vite'
+
+export interface FrontCpuCallSourcePluginOptions {
+  /**
+   * 是否启用插件
+   * 默认 true
+   */
+  enabled?: boolean
+  /**
+   * 目标包名（默认 ['front-cpu', 'front-cpu/debug']）
+   */
+  packageNames?: string[]
+  /**
+   * 生效阶段
+   * - serve: 仅开发环境（默认）
+   * - build: 仅构建
+   * - all: 开发和构建都启用
+   */
+  apply?: 'serve' | 'build' | 'all'
+}
+
+interface InjectContext {
+  id: string
+  packageNames: string[]
+}
+
+interface Edit {
+  start: number
+  end: number
+  value: string
+}
+
+interface ParsedArg {
+  text: string
+}
+
+const SUPPORTED_EXT_RE = /\.(ts|tsx|js|jsx|mjs|cjs|vue)$/i
+const CALL_SOURCE_MARKER_RE = /\bcallSource\s*:/
+
+/**
+ * 编译期注入 front-cpu dispatch 调用来源
+ */
+export function frontCpuCallSourcePlugin(
+  options: FrontCpuCallSourcePluginOptions = {}
+): Plugin {
+  const {
+    enabled = true,
+    packageNames = ['front-cpu', 'front-cpu/debug'],
+    apply = 'serve',
+  } = options
+
+  return {
+    name: 'front-cpu-call-source',
+    enforce: 'post',
+    apply(_, env) {
+      if (!enabled) return false
+      if (apply === 'all') return true
+      if (apply === 'serve') return env.command === 'serve'
+      return env.command === 'build'
+    },
+    transform(code, id) {
+      if (!shouldTransform(id)) return null
+      const transformed = injectCallSourceToDispatch(code, {
+        id,
+        packageNames,
+      })
+      if (transformed === code) return null
+      return {
+        code: transformed,
+        map: null,
+      }
+    },
+  }
+}
+
+/**
+ * 纯函数：对单个模块代码进行注入改写（便于测试）
+ */
+export function injectCallSourceToDispatch(code: string, context: InjectContext): string {
+  const pipelineVars = collectPipelineVariables(code, context.packageNames)
+  if (pipelineVars.size === 0) return code
+
+  const edits: Edit[] = []
+  const normalizedId = normalizeInjectedFilePath(context.id)
+
+  for (const variable of pipelineVars) {
+    const pattern = new RegExp(
+      `\\b${escapeRegExp(variable)}\\s*(?:\\?\\.)?\\s*\\.\\s*dispatch\\s*\\(`,
+      'g'
+    )
+    let match: RegExpExecArray | null
+    while ((match = pattern.exec(code)) !== null) {
+      const matchedText = match[0] || ''
+      const openParenOffset = matchedText.lastIndexOf('(')
+      if (openParenOffset < 0) continue
+
+      const openParenIndex = match.index + openParenOffset
+      const closeParenIndex = findMatchingParen(code, openParenIndex)
+      if (closeParenIndex < 0) continue
+
+      const argsText = code.slice(openParenIndex + 1, closeParenIndex)
+      const args = splitTopLevelArgs(argsText)
+      if (args.length < 2) continue
+
+      // 用户已显式传 callSource，不覆盖
+      if (args.length >= 4 && CALL_SOURCE_MARKER_RE.test(args[3]?.text || '')) {
+        continue
+      }
+
+      const location = indexToLineColumn(code, match.index)
+      const callSourceLiteral = createCallSourceLiteral(normalizedId, location.line, location.column)
+      const nextArgsText = buildInjectedArgs(args, callSourceLiteral)
+      if (nextArgsText === argsText) continue
+
+      edits.push({
+        start: openParenIndex + 1,
+        end: closeParenIndex,
+        value: nextArgsText,
+      })
+    }
+  }
+
+  if (edits.length === 0) return code
+  return applyEdits(code, edits)
+}
+
+function shouldTransform(id: string): boolean {
+  const clean = normalizeFileId(id)
+  if (!SUPPORTED_EXT_RE.test(clean) && !id.includes('?vue&type=script')) {
+    return false
+  }
+  if (clean.includes('/node_modules/')) {
+    return false
+  }
+  return true
+}
+
+function collectPipelineVariables(code: string, packageNames: string[]): Set<string> {
+  const constructors = collectPipelineConstructors(code, packageNames)
+  if (constructors.size === 0) return new Set()
+
+  const variables = new Set<string>()
+  for (const ctor of constructors) {
+    const declarationRegex = new RegExp(
+      `\\b(?:const|let|var)\\s+([A-Za-z_$][\\w$]*)\\s*=\\s*new\\s+${escapeRegExp(ctor)}\\s*\\(`,
+      'g'
+    )
+    let m: RegExpExecArray | null
+    while ((m = declarationRegex.exec(code)) !== null) {
+      if (m[1]) variables.add(m[1])
+    }
+
+    const assignmentRegex = new RegExp(
+      `\\b([A-Za-z_$][\\w$]*)\\s*=\\s*new\\s+${escapeRegExp(ctor)}\\s*\\(`,
+      'g'
+    )
+    while ((m = assignmentRegex.exec(code)) !== null) {
+      if (m[1]) variables.add(m[1])
+    }
+  }
+
+  return variables
+}
+
+function collectPipelineConstructors(code: string, packageNames: string[]): Set<string> {
+  const constructors = new Set<string>()
+  const importRegex = /import\s+(?:type\s+)?{([\s\S]*?)}\s+from\s+['"]([^'"]+)['"]/g
+
+  let match: RegExpExecArray | null
+  while ((match = importRegex.exec(code)) !== null) {
+    const importedFrom = match[2]
+    if (!importedFrom || !packageNames.includes(importedFrom)) continue
+
+    const specifiers = (match[1] || '').split(',')
+    for (const specifier of specifiers) {
+      const trimmed = specifier.trim()
+      if (!trimmed) continue
+
+      // Pipeline
+      if (trimmed === 'Pipeline') {
+        constructors.add('Pipeline')
+        continue
+      }
+
+      // Pipeline as Xxx
+      const aliasMatch = trimmed.match(/^Pipeline\s+as\s+([A-Za-z_$][\w$]*)$/)
+      if (aliasMatch?.[1]) {
+        constructors.add(aliasMatch[1])
+      }
+    }
+  }
+
+  return constructors
+}
+
+function createCallSourceLiteral(file: string, line: number, column: number): string {
+  return `{ file: ${JSON.stringify(file)}, line: ${line}, column: ${column}, raw: 'compile-time' }`
+}
+
+function buildInjectedArgs(args: ParsedArg[], callSourceLiteral: string): string {
+  const normalized = args.map((arg) => arg.text.trim())
+
+  if (normalized.length >= 4) {
+    const existingOptions = normalized[3]
+    normalized[3] = `{ ...(${existingOptions || 'undefined'}), callSource: ${callSourceLiteral} }`
+    return normalized.join(', ')
+  }
+
+  if (normalized.length === 3) {
+    normalized.push(`{ callSource: ${callSourceLiteral} }`)
+    return normalized.join(', ')
+  }
+
+  if (normalized.length === 2) {
+    normalized.push('undefined')
+    normalized.push(`{ callSource: ${callSourceLiteral} }`)
+    return normalized.join(', ')
+  }
+
+  return args.map((arg) => arg.text).join(', ')
+}
+
+function splitTopLevelArgs(input: string): ParsedArg[] {
+  const args: ParsedArg[] = []
+  let start = 0
+  let parenDepth = 0
+  let bracketDepth = 0
+  let braceDepth = 0
+  let quote: '"' | "'" | '`' | null = null
+  let escaped = false
+
+  for (let i = 0; i < input.length; i++) {
+    const ch = input[i]
+    if (!ch) continue
+
+    if (quote) {
+      if (escaped) {
+        escaped = false
+        continue
+      }
+      if (ch === '\\') {
+        escaped = true
+        continue
+      }
+      if (ch === quote) {
+        quote = null
+      }
+      continue
+    }
+
+    if (ch === "'" || ch === '"' || ch === '`') {
+      quote = ch
+      continue
+    }
+
+    if (ch === '(') parenDepth++
+    else if (ch === ')') parenDepth--
+    else if (ch === '[') bracketDepth++
+    else if (ch === ']') bracketDepth--
+    else if (ch === '{') braceDepth++
+    else if (ch === '}') braceDepth--
+
+    if (ch === ',' && parenDepth === 0 && bracketDepth === 0 && braceDepth === 0) {
+      args.push({ text: input.slice(start, i) })
+      start = i + 1
+    }
+  }
+
+  const tail = input.slice(start)
+  if (tail.trim().length > 0 || input.trim().length > 0) {
+    args.push({ text: tail })
+  }
+
+  return args
+}
+
+function findMatchingParen(code: string, openIndex: number): number {
+  let depth = 0
+  let quote: '"' | "'" | '`' | null = null
+  let escaped = false
+  let inLineComment = false
+  let inBlockComment = false
+
+  for (let i = openIndex; i < code.length; i++) {
+    const ch = code[i]
+    const next = code[i + 1]
+    if (!ch) continue
+
+    if (inLineComment) {
+      if (ch === '\n') inLineComment = false
+      continue
+    }
+
+    if (inBlockComment) {
+      if (ch === '*' && next === '/') {
+        inBlockComment = false
+        i++
+      }
+      continue
+    }
+
+    if (quote) {
+      if (escaped) {
+        escaped = false
+        continue
+      }
+      if (ch === '\\') {
+        escaped = true
+        continue
+      }
+      if (ch === quote) {
+        quote = null
+      }
+      continue
+    }
+
+    if (ch === '/' && next === '/') {
+      inLineComment = true
+      i++
+      continue
+    }
+
+    if (ch === '/' && next === '*') {
+      inBlockComment = true
+      i++
+      continue
+    }
+
+    if (ch === "'" || ch === '"' || ch === '`') {
+      quote = ch
+      continue
+    }
+
+    if (ch === '(') {
+      depth++
+      continue
+    }
+
+    if (ch === ')') {
+      depth--
+      if (depth === 0) return i
+    }
+  }
+
+  return -1
+}
+
+function indexToLineColumn(code: string, index: number): { line: number; column: number } {
+  let line = 1
+  let lineStart = 0
+  for (let i = 0; i < index; i++) {
+    if (code[i] === '\n') {
+      line++
+      lineStart = i + 1
+    }
+  }
+  return {
+    line,
+    column: index - lineStart + 1,
+  }
+}
+
+function applyEdits(code: string, edits: Edit[]): string {
+  const sorted = [...edits].sort((a, b) => b.start - a.start)
+  let result = code
+  for (const edit of sorted) {
+    result = result.slice(0, edit.start) + edit.value + result.slice(edit.end)
+  }
+  return result
+}
+
+function normalizeFileId(id: string): string {
+  const withoutQuery = id.split('?')[0] || id
+  return withoutQuery.replace(/\\/g, '/')
+}
+
+function normalizeInjectedFilePath(id: string): string {
+  const normalized = normalizeFileId(id)
+  const srcIndex = normalized.lastIndexOf('/src/')
+  if (srcIndex >= 0) {
+    return normalized.slice(srcIndex + 1)
+  }
+  return normalized
+}
+
+function escapeRegExp(value: string): string {
+  return value.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')
+}

package/src/types.ts

@@ -50,6 +50,19 @@ export interface InstructionContext {
   callSource?: CallSource
 }
 
+/**
+ * dispatch 可选参数
+ */
+export interface DispatchOptions {
+  /** 指令标签（用于批量取消/筛选） */
+  tags?: string[]
+  /**
+   * 调用源信息（优先使用编译期注入）
+   * 未提供时会在运行时 fallback 到 Error.stack 解析
+   */
+  callSource?: CallSource
+}
+
 /**
  * WB阶段执行信息
  */

package/src/vite.ts

@@ -0,0 +1,5 @@
+export {
+  frontCpuCallSourcePlugin,
+  injectCallSourceToDispatch,
+  type FrontCpuCallSourcePluginOptions,
+} from './plugins/vite-call-source'