@foolsecret/pi-prompt 0.4.0

package/src/config.ts

@@ -0,0 +1,280 @@
+/**
+ * 配置管理器：~/.pi/pi-prompt.json 的读写 + 三轴档位解析。
+ *
+ * 配置存 `~/.pi/` 根目录（与 pi-usager.json 同层）；写入前 mkdirSync 防御
+ * （对齐 pi-usager 踩坑 7）。
+ *
+ * 档位解析优先级（会话显式设档 > 自动解析）：
+ *   show 自动 = perProvider[provider].show → env PI_PROMPT_SHOW(别名 PI_PROMPT_MODE)
+ *              → defaultShow → 'full'
+ *   write 自动 = perProvider[provider].write → env PI_PROMPT_WRITE → defaultWrite → 'normal'
+ *   do 自动   = perProvider[provider].do → env PI_PROMPT_DO → defaultDo → 'normal'
+ *
+ * v0.2 迁移：旧 v0.1 配置 defaultMode / perProvider 字符串 会被读成 show 轴值。
+ */
+
+import { readFileSync, writeFileSync, mkdirSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+import {
+	normalizeDoMode,
+	normalizeShowMode,
+	normalizeWriteMode,
+	DO_DEFAULT,
+	SHOW_DEFAULT,
+	WRITE_DEFAULT,
+	type DoMode,
+	type ShowMode,
+	type WriteMode,
+} from "./modes.ts";
+
+/** 配置文件路径（~/.pi/pi-prompt.json） */
+export const CONFIG_PATH: string = join(homedir(), ".pi", "pi-prompt.json");
+
+/** 单厂商的三轴覆盖 */
+export interface ProviderAxes {
+	show?: string;
+	write?: string;
+	do?: string;
+}
+
+/** 配置文件结构（defaultShow 可含 auto；perProvider 兼容旧字符串写法） */
+export interface PromptConfigFile {
+	defaultShow?: string;
+	defaultWrite?: string;
+	defaultDo?: string;
+	quietStartup?: boolean;
+	hideStatus?: boolean;
+	/** DeepSeek 高峰升档（默认开；false 关闭） */
+	peakUpgrade?: boolean;
+	/** max_tokens 上限（正整数；缺省/非正数 = 不开） */
+	maxTokensCap?: number;
+	/** auto 低频自动抽样（默认关；true 开启） */
+	autoSample?: boolean;
+	/** 抽样间隔（多少轮后抽一次；默认 50） */
+	autoSampleInterval?: number;
+	/** 当日抽样上限（默认 3） */
+	autoSampleMaxPerDay?: number;
+	perProvider?: Record<string, string | ProviderAxes>;
+}
+
+/** env 布尔开关解析：值存在且非空/0/false/no 视为真；未设置返回 undefined */
+function envTruthy(env: NodeJS.ProcessEnv | undefined, name: string): boolean | undefined {
+	const value = env?.[name];
+	if (value === undefined) return undefined;
+	const v = value.trim().toLowerCase();
+	return !(v === "" || v === "0" || v === "false" || v === "no");
+}
+
+/** 三轴默认值合计（status/重置用） */
+export interface AxesDefault {
+	show: ShowMode;
+	write: WriteMode;
+	do: DoMode;
+}
+
+/** 配置管理器类：惰性读档（含 v0.1 迁移）、三轴解析、持久化 */
+export class PromptConfigManager {
+	/** 当前配置（惰性加载后的缓存） */
+	private config: PromptConfigFile;
+	/** 是否已从磁盘读档 */
+	private loaded: boolean;
+
+	constructor(
+		private readonly env: NodeJS.ProcessEnv | undefined = process.env,
+		private readonly configPath: string = CONFIG_PATH,
+		config?: PromptConfigFile,
+	) {
+		// 注入测试用途的初始配置；未注入则首次 get 时惰性读档
+		this.config = config ?? {};
+		this.loaded = config !== undefined;
+	}
+
+	/** 惰性读档（strip BOM；损坏回退空配置；v0.1 defaultMode → defaultShow） */
+	private load(): PromptConfigFile {
+		if (this.loaded) return this.config;
+		this.loaded = true;
+		try {
+			const raw = readFileSync(this.configPath, "utf8").replace(/^\uFEFF/, "");
+			const parsed = JSON.parse(raw) as (Partial<PromptConfigFile> & { defaultMode?: unknown }) | null;
+			this.config = {
+				defaultShow:
+					typeof parsed?.defaultShow === "string"
+						? parsed.defaultShow
+						: typeof parsed?.defaultMode === "string"
+							? parsed.defaultMode
+							: undefined,
+				defaultWrite: typeof parsed?.defaultWrite === "string" ? parsed.defaultWrite : undefined,
+				defaultDo: typeof parsed?.defaultDo === "string" ? parsed.defaultDo : undefined,
+				quietStartup: parsed?.quietStartup === true,
+				hideStatus: parsed?.hideStatus === true,
+				perProvider: parsed?.perProvider ?? {},
+			};
+		} catch {
+			this.config = {};
+		}
+		return this.config;
+	}
+
+	/** show 轴默认档（含 auto；env → defaultShow → 内置） */
+	getShowDefault(): ShowMode {
+		const envValue = this.env?.PI_PROMPT_SHOW ?? this.env?.PI_PROMPT_MODE;
+		const envMode = normalizeShowMode(envValue);
+		if (envMode !== null) return envMode;
+		const configMode = normalizeShowMode(this.load().defaultShow);
+		if (configMode !== null) return configMode;
+		return SHOW_DEFAULT;
+	}
+
+	/** write 轴默认档（env → defaultWrite → 内置 normal） */
+	getWriteDefault(): WriteMode {
+		return normalizeWriteMode(this.env?.PI_PROMPT_WRITE) ?? normalizeWriteMode(this.load().defaultWrite) ?? WRITE_DEFAULT;
+	}
+
+	/** do 轴默认档（env → defaultDo → 内置 normal） */
+	getDoDefault(): DoMode {
+		return normalizeDoMode(this.env?.PI_PROMPT_DO) ?? normalizeDoMode(this.load().defaultDo) ?? DO_DEFAULT;
+	}
+
+	/** 三轴默认合计（status / /prompt 无参恢复用） */
+	getDefaults(): AxesDefault {
+		return { show: this.getShowDefault(), write: this.getWriteDefault(), do: this.getDoDefault() };
+	}
+
+	/** 解析某厂商 show 轴：perProvider.show → getShowDefault() */
+	showForProvider(provider: string | undefined): ShowMode {
+		if (provider !== undefined) {
+			const overridden = this.providerAxes(provider).show;
+			const normalized = normalizeShowMode(overridden);
+			if (normalized !== null) return normalized;
+		}
+		return this.getShowDefault();
+	}
+
+	/** 解析某厂商 write 轴 */
+	writeForProvider(provider: string | undefined): WriteMode {
+		if (provider !== undefined) {
+			const overridden = this.providerAxes(provider).write;
+			const normalized = normalizeWriteMode(overridden);
+			if (normalized !== null) return normalized;
+		}
+		return this.getWriteDefault();
+	}
+
+	/** 解析某厂商 do 轴 */
+	doForProvider(provider: string | undefined): DoMode {
+		if (provider !== undefined) {
+			const overridden = this.providerAxes(provider).do;
+			const normalized = normalizeDoMode(overridden);
+			if (normalized !== null) return normalized;
+		}
+		return this.getDoDefault();
+	}
+
+	/** 读取厂商覆盖并归一化为三轴对象（兼容旧字符串写法 → show） */
+	private providerAxes(provider: string): ProviderAxes {
+		const raw = this.load().perProvider?.[provider];
+		if (typeof raw === "string") return { show: raw };
+		return raw ?? {};
+	}
+
+	/** 懒查询厂商覆盖映射（已归一化，status 展示用） */
+	perProviderConfig(): Record<string, ProviderAxes> {
+		const result: Record<string, ProviderAxes> = {};
+		const config = this.load();
+		for (const [provider, raw] of Object.entries(config.perProvider ?? {})) {
+			result[provider] = typeof raw === "string" ? { show: raw } : raw;
+		}
+		return result;
+	}
+
+	/** 启动是否静默（env 优先，其次配置文件） */
+	isQuietStartup(): boolean {
+		return envTruthy(this.env, "PI_PROMPT_QUIET_STARTUP") ?? this.load().quietStartup === true;
+	}
+
+	/** 状态栏是否隐藏（env 优先，其次配置文件） */
+	isHideStatus(): boolean {
+		return envTruthy(this.env, "PI_PROMPT_HIDE_STATUS") ?? this.load().hideStatus === true;
+	}
+
+	/**
+	 * DeepSeek 高峰升档是否开启：默认开（纯成本优化）；env PI_PROMPT_PEAK=
+	 * 可显式关闭（0/false/no），其次配置文件 peakUpgrade:false。
+	 */
+	isPeakUpgrade(): boolean {
+		const env = envTruthy(this.env, "PI_PROMPT_PEAK");
+		if (env !== undefined) return env;
+		return this.load().peakUpgrade !== false;
+	}
+
+	/** max_tokens 上限：env PI_PROMPT_MAX_TOKENS 优先，其次配置文件 maxTokensCap；非正整数/缺省 → null 不开 */
+	maxTokensCap(): number | null {
+		const envRaw = this.env?.PI_PROMPT_MAX_TOKENS?.trim();
+		const envCap = envRaw !== undefined && envRaw !== "" && /^\d+$/.test(envRaw) ? Number(envRaw) : null;
+		const fileRaw = this.load().maxTokensCap;
+		const fileCap = typeof fileRaw === "number" ? fileRaw : null;
+		const cap = envCap ?? fileCap;
+		return cap !== null && cap > 0 ? cap : null;
+	}
+
+	/** auto 低频抽样是否开启（默认关；env PI_PROMPT_AUTO_SAMPLE=true/false 或配置 autoSample） */
+	isAutoSample(): boolean {
+		const env = envTruthy(this.env, "PI_PROMPT_AUTO_SAMPLE");
+		if (env !== undefined) return env;
+		return this.load().autoSample === true;
+	}
+
+	/** 抽样间隔（多少轮后触发一次；配置 autoSampleInterval，兜底 50） */
+	autoSampleInterval(): number {
+		const v = this.load().autoSampleInterval;
+		return typeof v === "number" && v >= 2 ? v : 50;
+	}
+
+	/** 当日抽样上限（配置 autoSampleMaxPerDay，兜底 3） */
+	autoSampleMaxPerDay(): number {
+		const v = this.load().autoSampleMaxPerDay;
+		return typeof v === "number" && v >= 1 ? v : 3;
+	}
+
+	/** 持久化某轴默认档（按轴校验；非法拒绝返回 false） */
+	writeDefaultAxis(axis: "show" | "write" | "do", value: string): boolean {
+		switch (axis) {
+			case "show": {
+				if (normalizeShowMode(value) === null) return false;
+				return this.persist({ defaultShow: value.toLowerCase() });
+			}
+			case "write": {
+				if (normalizeWriteMode(value) === null) return false;
+				return this.persist({ defaultWrite: value.toLowerCase() });
+			}
+			case "do": {
+				if (normalizeDoMode(value) === null) return false;
+				return this.persist({ defaultDo: value.toLowerCase() });
+			}
+			default:
+				// 运行时防御：类型约束外仍可能传入任意字符串
+				return false;
+		}
+	}
+
+	/** 写盘（mkdirSync 防御目录缺失）：合并当前磁盘内容（防多写互相覆盖）；失败返回 false */
+	private persist(patch: Partial<PromptConfigFile>): boolean {
+		try {
+			mkdirSync(dirname(this.configPath), { recursive: true });
+			let merged: PromptConfigFile = {};
+			try {
+				const raw = readFileSync(this.configPath, "utf8").replace(/^\uFEFF/, "");
+				const parsed = JSON.parse(raw) as Partial<PromptConfigFile> | null;
+				if (parsed && typeof parsed === "object") merged = parsed as PromptConfigFile;
+			} catch {
+				// 文件不存在或损坏：以空白为底继续合并
+			}
+			this.config = { ...merged, ...patch };
+			writeFileSync(this.configPath, JSON.stringify(this.config, null, 2), "utf8");
+			return true;
+		} catch {
+			return false;
+		}
+	}
+}

package/src/format.ts

@@ -0,0 +1,11 @@
+/**
+ * 展示格式化工具：把 token 计数转成易读的紧凑文本。
+ */
+
+/** 千分位/百万缩写（1_234_567 → "1.2M"），<1k 原样输出 */
+export function formatTokens(n: number): string {
+	const abs = Math.abs(n);
+	if (abs >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+	if (abs >= 1_000) return `${(n / 1_000).toFixed(1)}k`;
+	return String(n);
+}

package/src/modes.ts

@@ -0,0 +1,113 @@
+/**
+ * 三轴档位定义与规范化。
+ *
+ * pi-prompt v0.2 由三根正交轴组成（每轴独立注入、独立配置/会话覆盖）：
+ *   - show（输出风格）：normal / lite / full / ultra / auto
+ *        normal = 不注入（同 value full）；auto 是元档位：运行时由选档器
+ *        （auto.ts）按成本函数解析为四档之一，可作默认配置，但不是直接注入的档
+ *        （四档写死文案才保证字节稳定）。
+ *   - write（代码写量）：minimal / normal / complete
+ *   - do（行为力度）：direct / normal / deep
+ * review 不再是一个档位，改为独立审计技能 /prompt-review（会话级）。
+ *
+ * v0.2.2：show 轴旧档名 "off" 更名为 "normal"（与 write/do 的"不注入基线"统一）。
+ * 历史配置/env/台账里的 "off" 由下述 normalize 别名层归一为 normal，注入字节
+ * 零变化（off/normal 本就不注入），不会重置前缀 KV 缓存。
+ */
+
+/** show 轴全部取值（auto 为元档位） */
+export type ShowMode = "auto" | "normal" | "lite" | "full" | "ultra";
+
+/** show 轴可直接注入的运行时档位（auto 已解析后的结果） */
+export type RuntimeShowMode = Exclude<ShowMode, "auto">;
+
+/** show 轴全部取值数组（校验/补全用） */
+export const SHOW_MODES: readonly ShowMode[] = ["auto", "normal", "lite", "full", "ultra"];
+
+/** show 轴运行时档位数组（选档器输出域） */
+export const RUNTIME_SHOW_MODES: readonly RuntimeShowMode[] = ["normal", "lite", "full", "ultra"];
+
+/** show 轴内置默认档位（v0.2 起默认 auto：运行时按成本函数选档） */
+export const SHOW_DEFAULT: ShowMode = "auto";
+
+/** write 轴全部取值 */
+export type WriteMode = "minimal" | "normal" | "complete";
+
+/** write 轴默认档位（normal = 不注入） */
+export const WRITE_DEFAULT: WriteMode = "normal";
+
+/** write 轴全部取值数组 */
+export const WRITE_MODES: readonly WriteMode[] = ["minimal", "normal", "complete"];
+
+/** do 轴全部取值 */
+export type DoMode = "direct" | "normal" | "deep";
+
+/** do 轴默认档位（normal = 不注入） */
+export const DO_DEFAULT: DoMode = "normal";
+
+/** do 轴全部取值数组 */
+export const DO_MODES: readonly DoMode[] = ["direct", "normal", "deep"];
+
+/** show 轴档位 → 展示图标（保留供 /prompt status 文案用；footer 已改纯文字） */
+export const SHOW_ICONS: Record<ShowMode, string> = {
+	auto: "🔄",
+	normal: "",
+	lite: "🌿",
+	full: "⚡",
+	ultra: "🔥",
+};
+
+/**
+ * 严格归一化 show 轴取值（小写、去首尾空白）；非法返回 null。
+ * 接受 auto（元档位），调用方决定后续解析。
+ * v0.2.2 兼容别名：历史取值 "off" 归一为 "normal"。
+ */
+export function normalizeShowMode(value: unknown): ShowMode | null {
+	if (typeof value !== "string") return null;
+	const normalized = value.trim().toLowerCase();
+	if (normalized === "off") return "normal";
+	return (SHOW_MODES as readonly string[]).includes(normalized) ? (normalized as ShowMode) : null;
+}
+
+/** 严格归一化为 show 轴运行时档位（拒绝 auto）；非法返回 null；兼容别名 off→normal */
+export function normalizeRuntimeShow(value: unknown): RuntimeShowMode | null {
+	if (typeof value !== "string") return null;
+	const normalized = value.trim().toLowerCase();
+	if (normalized === "off") return "normal";
+	return (RUNTIME_SHOW_MODES as readonly string[]).includes(normalized)
+		? (normalized as RuntimeShowMode)
+		: null;
+}
+
+/** 严格归一化 write 轴取值；非法返回 null */
+export function normalizeWriteMode(value: unknown): WriteMode | null {
+	if (typeof value !== "string") return null;
+	const normalized = value.trim().toLowerCase();
+	return (WRITE_MODES as readonly string[]).includes(normalized) ? (normalized as WriteMode) : null;
+}
+
+/** 严格归一化 do 轴取值；非法返回 null */
+export function normalizeDoMode(value: unknown): DoMode | null {
+	if (typeof value !== "string") return null;
+	const normalized = value.trim().toLowerCase();
+	return (DO_MODES as readonly string[]).includes(normalized) ? (normalized as DoMode) : null;
+}
+
+/**
+ * 全词关停指令匹配："stop pi-prompt" / "normal mode"。
+ * 必须整条消息匹配（去尾标点、忽略大小写），防止任务中句子含词就误关
+ * （对齐 ponytail 踩坑：普通请求 "add a normal mode toggle" 也会被误伤）。
+ * 语义 = 全局回到三轴默认（show/默认档、write normal、do normal）。
+ */
+export function isDeactivationCommand(text: unknown): boolean {
+	const t = String(text ?? "")
+		.trim()
+		.toLowerCase()
+		.replace(/[.!?。！？\s]+$/, "");
+	return t === "stop pi-prompt" || t === "normal mode";
+}
+
+/** 三轴默认值汇总（重置到默认用） */
+export function defaultAxes(): { show: ShowMode; write: WriteMode; do: DoMode } {
+	return { show: SHOW_DEFAULT, write: WRITE_DEFAULT, do: DO_DEFAULT };
+}