@shelken/pi-command-history 0.1.3

package/LICENSE

@@ -0,0 +1,22 @@
+MIT License
+
+Copyright (c) 2026 Ross Z
+Modifications Copyright (c) 2026 shelken
+
+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.

package/README.md

@@ -0,0 +1,30 @@
+# pi-command-history
+
+按工作目录持久化输入历史。同一目录下跨 session 可用 `shift+up` / `shift+down` 回填。
+
+初始 fork 自 [ross-jill-ws/pi-command-history](https://github.com/ross-jill-ws/pi-command-history)。
+
+## 安装
+
+```bash
+pi install npm:@shelken/pi-command-history
+```
+
+## 快捷键
+
+| 快捷键 | 作用 |
+|---|---|
+| `shift+up` | 更早的历史 |
+| `shift+down` | 更新的历史 |
+
+- 用户输入都会保存（含 `/` 命令）
+- 去重：重复输入会移到最新
+- 每个目录最多 500 条（`session_start`/`load` 时压缩）
+- 文件：`~/.pi/folder-history/*.jsonl`（每行 `{ cwd, text }`，append 写入）
+- 文件名把路径 `/` 换成 `-` 可能碰撞，靠行内 `cwd` 隔离
+
+## 验证
+
+```bash
+bun --filter @shelken/pi-command-history test
+```

package/index.ts

@@ -0,0 +1,188 @@
+/**
+ * 按工作目录持久化输入历史，shift+up/down 跨 session 回填。
+ * 存储：~/.pi/folder-history/<path-with-dashes>.jsonl
+ *
+ * 文件名把 `/` 换成 `-`，`/a-b` 与 `/a/b` 会撞同一文件；
+ * 因此每行必须带 cwd，读写都按 cwd 过滤。
+ *
+ * 写入用 append（并发更安全）；超过上限时在 load 时压缩回写。
+ */
+
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import {
+  appendFileSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  renameSync,
+  writeFileSync,
+} from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+
+export const HISTORY_DIR = join(homedir(), ".pi", "folder-history");
+export const MAX_HISTORY = 500;
+
+interface HistoryRow {
+  cwd: string;
+  text: string;
+}
+
+function historyFile(cwd: string, historyDir: string): string {
+  return join(historyDir, `${cwd.replace(/\//g, "-")}.jsonl`);
+}
+
+function readRows(file: string): HistoryRow[] {
+  if (!existsSync(file)) return [];
+  try {
+    const rows: HistoryRow[] = [];
+    for (const line of readFileSync(file, "utf-8").split("\n")) {
+      if (!line.trim()) continue;
+      try {
+        const entry = JSON.parse(line) as { text?: string; cwd?: string };
+        if (!entry.text || !entry.cwd) continue;
+        rows.push({ cwd: entry.cwd, text: entry.text });
+      } catch {
+        // skip malformed lines
+      }
+    }
+    return rows;
+  } catch {
+    return [];
+  }
+}
+
+function uniqueTexts(texts: string[]): string[] {
+  const seen = new Map<string, number>();
+  texts.forEach((text, i) => seen.set(text, i));
+  return [...seen.entries()]
+    .sort((a, b) => a[1] - b[1])
+    .map(([text]) => text);
+}
+
+function writeRows(file: string, rows: HistoryRow[]): void {
+  const body =
+    rows.map((row) => JSON.stringify(row)).join("\n") + (rows.length > 0 ? "\n" : "");
+  const tmp = `${file}.${process.pid}.tmp`;
+  writeFileSync(tmp, body, "utf-8");
+  renameSync(tmp, file);
+}
+
+/** 压缩：各 cwd 只保留最近 MAX_HISTORY 条唯一记录。 */
+function compactFile(file: string): void {
+  const rows = readRows(file);
+  const byCwd = new Map<string, string[]>();
+  for (const row of rows) {
+    const list = byCwd.get(row.cwd) ?? [];
+    list.push(row.text);
+    byCwd.set(row.cwd, list);
+  }
+
+  const compacted: HistoryRow[] = [];
+  for (const [cwd, texts] of byCwd) {
+    for (const text of uniqueTexts(texts).slice(-MAX_HISTORY)) {
+      compacted.push({ cwd, text });
+    }
+  }
+
+  // 只有确实变短才回写，避免无意义 IO
+  if (compacted.length < rows.length) {
+    writeRows(file, compacted);
+  }
+}
+
+export function loadHistory(cwd: string, historyDir = HISTORY_DIR): string[] {
+  const file = historyFile(cwd, historyDir);
+  const rows = readRows(file);
+  const oursRaw = rows.filter((r) => r.cwd === cwd).map((r) => r.text);
+  const unique = uniqueTexts(oursRaw);
+
+  // 文件膨胀时在 load 路径压缩（session_start 触发一次即可）
+  if (rows.length > MAX_HISTORY || unique.length > MAX_HISTORY) {
+    compactFile(file);
+  }
+
+  return unique.slice(-MAX_HISTORY);
+}
+
+/** append 新输入；截断在下次 load/compact 时完成。 */
+export function appendHistory(
+  cwd: string,
+  text: string,
+  historyDir = HISTORY_DIR,
+): void {
+  mkdirSync(historyDir, { recursive: true });
+  const file = historyFile(cwd, historyDir);
+  appendFileSync(file, `${JSON.stringify({ cwd, text })}\n`, "utf-8");
+}
+
+export function pushUniqueHistory(history: string[], text: string): string[] {
+  const next = history.filter((item) => item !== text);
+  next.push(text);
+  if (next.length > MAX_HISTORY) {
+    return next.slice(-MAX_HISTORY);
+  }
+  return next;
+}
+
+export default function commandHistoryExtension(pi: ExtensionAPI): void {
+  let history: string[] = [];
+  let historyIndex = -1;
+  let savedEditorText = "";
+  let currentCwd = "";
+
+  const showPreviousCommand = (ctx: ExtensionContext) => {
+    if (history.length === 0) return;
+
+    if (historyIndex === -1) {
+      savedEditorText = ctx.ui.getEditorText();
+    }
+
+    const nextIndex = historyIndex + 1;
+    if (nextIndex >= history.length) return;
+
+    historyIndex = nextIndex;
+    ctx.ui.setEditorText(history[history.length - 1 - historyIndex]);
+  };
+
+  const showNextCommand = (ctx: ExtensionContext) => {
+    if (historyIndex <= -1) return;
+
+    historyIndex--;
+
+    if (historyIndex === -1) {
+      ctx.ui.setEditorText(savedEditorText);
+    } else {
+      ctx.ui.setEditorText(history[history.length - 1 - historyIndex]);
+    }
+  };
+
+  pi.on("session_start", (_event, ctx) => {
+    currentCwd = ctx.cwd;
+    history = loadHistory(currentCwd);
+    historyIndex = -1;
+    savedEditorText = "";
+  });
+
+  pi.on("input", (event, _ctx) => {
+    const text = event.text?.trim();
+    if (!text || !currentCwd) return;
+
+    appendHistory(currentCwd, text);
+    history = pushUniqueHistory(history, text);
+    historyIndex = -1;
+    savedEditorText = "";
+
+    return { action: "continue" as const };
+  });
+
+  pi.registerShortcut("shift+up", {
+    description: "Previous command from folder history",
+    handler: showPreviousCommand,
+  });
+
+  pi.registerShortcut("shift+down", {
+    description: "Next command from folder history",
+    handler: showNextCommand,
+  });
+}

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@shelken/pi-command-history",
+  "version": "0.1.3",
+  "description": "按工作目录持久化输入历史，支持 shift+up/down 回填",
+  "type": "module",
+  "license": "MIT",
+  "files": [
+    "index.ts"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/shelken/pi-extensions.git",
+    "directory": "extensions/pi-command-history"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "test": "vitest run"
+  },
+  "keywords": [
+    "pi-package",
+    "command-history"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "peerDependenciesMeta": {
+    "@earendil-works/pi-coding-agent": {
+      "optional": true
+    }
+  }
+}