pi-atuin 0.1.1 → 0.1.3

package/history-store.ts

@@ -25,6 +25,26 @@ export interface HistoryEntry {
 
 let writeQueue: Promise<void> = Promise.resolve();
 
+// ponytail: in-memory dedup guard — input event may fire twice per entry.
+// prevents duplicate writes to JSONL within 5s window.
+const recentWrites = new Map<string, number>();
+const WRITE_DEDUP_MS = 5_000;
+
+export function isRecentWrite(text: string): boolean {
+	const last = recentWrites.get(text);
+	if (last && Date.now() - last < WRITE_DEDUP_MS) return true;
+	return false;
+}
+
+export function markRecentWrite(text: string): void {
+	recentWrites.set(text, Date.now());
+	// Cleanup entries older than WRITE_DEDUP_MS
+	const cutoff = Date.now() - WRITE_DEDUP_MS;
+	for (const [k, ts] of recentWrites) {
+		if (ts < cutoff) recentWrites.delete(k);
+	}
+}
+
 // ---------------------------------------------------------------------------
 // JSONL persistence
 // ---------------------------------------------------------------------------
@@ -143,7 +163,7 @@ async function loadAllEntries(): Promise<HistoryEntry[]> {
 		const key = entry.text.trim();
 		const existing = seen.get(key);
 		if (!existing || Math.abs(entry.timestamp - existing.timestamp) > 5000) {
-			seen.set(`${key}-${entry.timestamp}`, entry);
+			seen.set(key, entry);
 		}
 	}
 

package/index.ts

@@ -9,7 +9,7 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { CustomEditor } from "@earendil-works/pi-coding-agent";
 import { matchesKey, Key } from "@earendil-works/pi-tui";
-import { addEntry, invalidateCache, listRecent, writeToAtuin } from "./history-store.js";
+import { addEntry, invalidateCache, listRecent, writeToAtuin, isRecentWrite, markRecentWrite } from "./history-store.js";
 import { HistorySearchComponent } from "./search-ui.js";
 
 export default function piAtuin(pi: ExtensionAPI) {
@@ -17,6 +17,9 @@ export default function piAtuin(pi: ExtensionAPI) {
 	pi.on("input", async (event, ctx) => {
 		if (event.source === "interactive" && event.text.trim()) {
 			const text = event.text.trim();
+			// Guard against duplicate input events (pi may fire input twice)
+			if (isRecentWrite(text)) return { action: "continue" as const };
+			markRecentWrite(text);
 			await addEntry(text, ctx.cwd);
 			// Also write to atuin DB so shell atuin can search pi prompts
 			writeToAtuin(text, ctx.cwd);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-atuin",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Atuin-style fuzzy history search for pi. Press ↑ for interactive history.",
   "license": "MIT",
   "author": "RealAlexandreAI",
@@ -18,7 +18,8 @@
   "pi": {
     "extensions": [
       "./index.ts"
-    ]
+    ],
+    "image": "https://raw.githubusercontent.com/RealAlexandreAI/pi-atuin/master/screenshot.png"
   },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*"