dsh-plugin-prompt-tool 2.0.1 → 3.0.0

package/engine/interpolate.mjs

@@ -13,12 +13,14 @@
  *   - 键字符集：字母数字、下划线、点、中文、连字符（与 ST setvar/getvar 一致）。
  */
 
-import { sessionEvents } from './shared.mjs'
+import { currentEvents } from './history.mjs'
+import { machinePaths } from './machine-paths.mjs'
 import { createHash } from 'node:crypto'
 
 /** 会话事件中最后一条指定类型消息的文本（事件倒序扫描；无则空串）。 */
 function lastMessageOf(session, type) {
-  const events = sessionEvents(session)
+  // 这两个宏是官方按**模型可见消息**求值的运行时事实，压缩遮蔽掉的旧正文不再是「最后一条」。
+  const events = currentEvents(session)
   for (let index = events.length - 1; index >= 0; index--) {
     const event = events[index]
     if (event?.type !== type) continue
@@ -95,29 +97,45 @@ const DYNAMIC_MACROS = {
   pipe: () => '|',
 }
 
-/** 内置变量：路径类事实（无会话上下文时回退进程 cwd）；两条插值通道同源。 */
+/** 内置变量：路径类事实取机器路径事实的唯一来源（无会话上下文时回退进程 cwd）；两条插值通道同源。 */
 function builtinVariables(session) {
-  return {
-    DSH_HOME: process.env.DSH_HOME ?? (process.env.USERPROFILE ? `${process.env.USERPROFILE}\\.dsh` : ''),
-    WORKSPACE: process.env.DSH_WORKSPACE ?? session?.header?.cwd ?? process.cwd(),
-    CWD: session?.header?.cwd ?? process.cwd(),
-  }
+  return machinePaths(session)
+}
+
+/** 内置变量名的单一来源：从 builtinVariables 的键派生（惰性，不在 import 期取 cwd）。 */
+let builtinNames
+function builtinNameSet() {
+  return (builtinNames ??= new Set(Object.keys(builtinVariables())))
+}
+
+/**
+ * 保留名判定（保留名的唯一判据，写入与读取共用）：内建名大小写敏感（`dsh_home`
+ * 不是内建），动态宏名大小写不敏感（`TIME` 即 `time`）。会话变量不得占用这些名字，
+ * 否则字面值会遮蔽内建事实与动态宏。
+ */
+export function isReservedInterpolationName(key) {
+  const name = String(key ?? '')
+  return builtinNameSet().has(name) || Object.hasOwn(DYNAMIC_MACROS, name.toLowerCase())
 }
 
 /** 模板变量插值：配置 variables 优先，ST 运行时宏次之，内置 {{DSH_HOME}} / {{WORKSPACE}} / {{CWD}} 兜底。 */
 export function interpolateVariables(text, variables, session, keep, sourceId = '') {
   const builtins = builtinVariables(session)
   const active = new Set()
+  // pick 只为「同会话、同模板、同一次出现」稳定：整段正文 input 不进 seed，
+  // 否则无关正文长度一变取值就漂移；出现序号让同一正文里的多处 pick 各自取值。
+  // sourceId 必须只依赖标识（模板名、条目 id/键位）：省略时所有调用位置共用同一个 seed。
+  let pickCount = 0
   // 限制展开增加的字符数、递归深度与工作量；原始正文不截断，失败引用留给出口清洗。
   let remainingChars = 1024 * 1024
   let remainingExpansions = 4096
   function render(input, depth = 0) {
-    return normalizeMacroSyntax(input).replace(REFERENCE_RE, (whole, key, arg, offset) => {
+    return normalizeMacroSyntax(input).replace(REFERENCE_RE, (whole, key, arg) => {
       if (arg === undefined && keep?.has(key)) return `{{${key}}}`
       if (active.has(key)) return whole
       let value
       if (Object.hasOwn(variables, key)) value = String(variables[key] ?? '')
-      else if (key.toLowerCase() === 'pick') value = pickRandom(arg, JSON.stringify([session?.id ?? '', sourceId, input, offset]))
+      else if (key.toLowerCase() === 'pick') value = pickRandom(arg, JSON.stringify([session?.id ?? '', sourceId, pickCount++]))
       else if (Object.hasOwn(DYNAMIC_MACROS, key.toLowerCase())) value = DYNAMIC_MACROS[key.toLowerCase()](arg, session)
       else if (Object.hasOwn(builtins, key)) value = builtins[key]
       else return whole
@@ -142,9 +160,11 @@ export function interpolateVariables(text, variables, session, keep, sourceId =
  * 静态层插值：配置 variables 优先，ST 运行时宏取空串，内置路径变量按无会话语义解析。
  * @param keep 需要交给官方变量通道按 assembly 求值的名字（运行时事实与已注册变量）——
  *   命中即原样保留引用，不做静态替换。
+ * @param sourceId 调用点的稳定身份（模板名等只依赖标识的串），只影响 {{pick}} 的 seed；
+ *   缺省时与传 `''` 等价（所有位置共用 seed）。
  */
-export function interpolateStatic(text, variables, keep) {
-  return interpolateVariables(text, variables, undefined, keep)
+export function interpolateStatic(text, variables, keep, sourceId) {
+  return interpolateVariables(text, variables, undefined, keep, sourceId)
 }
 
 /**
@@ -188,7 +208,6 @@ const MACRO_SINGLE_COLON_RE = new RegExp(`\\{\\{\\s*(${MACRO_ALIASES})\\s*:(?!:)
 
 /**
  * ST 宏写法归一：{{roll 1d6}} / {{roll:1d6}} → {{roll::1d6}}。
- * 与 `src/host/sillytavern.ts` 的导入期归一同源（两处物理隔离，改动需同步）。
  */
 export function normalizeMacroSyntax(text) {
   return text
@@ -196,6 +215,18 @@ export function normalizeMacroSyntax(text) {
     .replace(MACRO_SINGLE_COLON_RE, (_whole, macro, arg) => `{{${String(macro).toLowerCase()}::${arg}}}`)
 }
 
+/**
+ * 按已声明键集合剥离引用（`variablesEnabled=false` 的编译期清洗）：命中声明键的
+ * `{{键}}` / `{{ 键 }}` / `{{键::参数}}` 整段移除，未声明键与内置引用原样保留。
+ * 纯函数（内部用 `replace` 调用 `REFERENCE_RE`）：不导出带 `/g` 的正则，否则
+ * 调用方的 `lastIndex` 会互相污染。
+ */
+export function stripDeclaredRefs(text, keys) {
+  return typeof text === 'string'
+    ? text.replace(REFERENCE_RE, (whole, key) => keys.has(key) ? '' : whole)
+    : text
+}
+
 /**
  * 官方插值通道（system-section / runtime-context）专用清洗。
  *