@apmantza/greedysearch-pi 1.8.0 → 1.8.2

package/extractors/perplexity.mjs

@@ -1,141 +1,145 @@
-#!/usr/bin/env node
-
-// extractors/perplexity.mjs
-// Navigate Perplexity, wait for streaming to complete, return clean answer + sources.
-//
-// Usage:
-//   node extractors/perplexity.mjs "<query>" [--tab <prefix>]
-//
-// Output (stdout): JSON { answer, sources, query, url }
-// Errors go to stderr only — stdout is always clean JSON for piping.
-//
-// TODO: Refactor - this file has 42 lines duplicated with google-ai.mjs (line 28)
-
-import {
-	cdp,
-	formatAnswer,
-	getOrOpenTab,
-	handleError,
-	injectClipboardInterceptor,
-	outputJson,
-	parseArgs,
-	parseSourcesFromMarkdown,
-	validateQuery,
-	waitForStreamComplete,
-} from "./common.mjs";
-import { dismissConsent } from "./consent.mjs";
-import { SELECTORS } from "./selectors.mjs";
-
-const S = SELECTORS.perplexity;
-const GLOBAL_VAR = "__pplxClipboard";
-
-// ============================================================================
-// Extraction
-// ============================================================================
-
-async function extractAnswer(tab) {
-	await cdp([
-		"eval",
-		tab,
-		`document.querySelector('${S.copyButton}')?.click()`,
-	]);
-	await new Promise((r) => setTimeout(r, 400));
-
-	let answer = await cdp(["eval", tab, `window.${GLOBAL_VAR} || ''`]);
-
-	// Retry once if clipboard is empty (Perplexity might be slow to write)
-	if (!answer) {
-		console.error("[perplexity] Clipboard empty, retrying in 2s...");
-		await cdp([
-			"eval",
-			tab,
-			`document.querySelector('${S.copyButton}')?.click()`,
-		]);
-		await new Promise((r) => setTimeout(r, 2000));
-		answer = await cdp(["eval", tab, `window.${GLOBAL_VAR} || ''`]);
-	}
-
-	if (!answer) throw new Error("Clipboard interceptor returned empty text");
-
-	const sources = parseSourcesFromMarkdown(answer);
-	return { answer: answer.trim(), sources };
-}
-
-// ============================================================================
-// Main
-// ============================================================================
-
-const USAGE =
-	'Usage: node extractors/perplexity.mjs "<query>" [--tab <prefix>]\n';
-
-async function main() {
-	const args = process.argv.slice(2);
-	validateQuery(args, USAGE);
-
-	const { query, tabPrefix, short } = parseArgs(args);
-
-	try {
-		// Refresh page list so cache is current
-		await cdp(["list"]);
-
-		const tab = await getOrOpenTab(tabPrefix);
-
-		// Navigate to homepage and use the search box (direct ?q= URLs trigger bot redirect)
-		await cdp(["nav", tab, "https://www.perplexity.ai/"], 35000);
-		await dismissConsent(tab, cdp);
-
-		// Wait for React app to mount input (up to 8s)
-		const deadline = Date.now() + 8000;
-		while (Date.now() < deadline) {
-			const found = await cdp([
-				"eval",
-				tab,
-				`!!document.querySelector('${S.input}')`,
-			]).catch(() => "false");
-			if (found === "true") break;
-			await new Promise((r) => setTimeout(r, 400));
-		}
-		await new Promise((r) => setTimeout(r, 300));
-
-		await injectClipboardInterceptor(tab, GLOBAL_VAR);
-		await cdp(["click", tab, S.input]);
-		await new Promise((r) => setTimeout(r, 400));
-		await cdp(["type", tab, query]);
-		await new Promise((r) => setTimeout(r, 400));
-
-		// Submit with Enter (most reliable across Chrome instances)
-		await cdp([
-			"eval",
-			tab,
-			`document.querySelector('${S.input}')?.dispatchEvent(new KeyboardEvent('keydown',{key:'Enter',bubbles:true,keyCode:13})), 'ok'`,
-		]);
-
-		await waitForStreamComplete(tab, {
-			timeout: 30000,
-			interval: 600,
-			stableRounds: 3,
-			selector: "document.body",
-		});
-
-		const { answer, sources } = await extractAnswer(tab);
-
-		if (!answer)
-			throw new Error(
-				"No answer extracted — Perplexity may not have responded",
-			);
-
-		const finalUrl = await cdp(["eval", tab, "document.location.href"]).catch(
-			() => "",
-		);
-		outputJson({
-			query,
-			url: finalUrl,
-			answer: formatAnswer(answer, short),
-			sources,
-		});
-	} catch (e) {
-		handleError(e);
-	}
-}
-
-main();
+#!/usr/bin/env node
+
+// extractors/perplexity.mjs
+// Navigate Perplexity, wait for streaming to complete, return clean answer + sources.
+//
+// Usage:
+//   node extractors/perplexity.mjs "<query>" [--tab <prefix>]
+//
+// Output (stdout): JSON { answer, sources, query, url }
+// Errors go to stderr only — stdout is always clean JSON for piping.
+//
+// TODO: Refactor - this file has 42 lines duplicated with google-ai.mjs (line 28)
+
+import {
+	cdp,
+	formatAnswer,
+	getOrOpenTab,
+	handleError,
+	injectClipboardInterceptor,
+	outputJson,
+	parseArgs,
+	parseSourcesFromMarkdown,
+	validateQuery,
+	waitForStreamComplete,
+} from "./common.mjs";
+import { dismissConsent } from "./consent.mjs";
+import { SELECTORS } from "./selectors.mjs";
+
+const S = SELECTORS.perplexity;
+const GLOBAL_VAR = "__pplxClipboard";
+
+// ============================================================================
+// Language-agnostic copy button finder
+// ============================================================================
+
+function findCopyButtonJsExpression() {
+	// Perplexity uses SVG icons via <use xlink:href="#pplx-icon-copy">
+	// This works across all locales since it doesn't depend on aria-label text
+	return `Array.from(document.querySelectorAll('button')).find(b => b.innerHTML.includes('#pplx-icon-copy'))`;
+}
+
+// ============================================================================
+// Extraction
+// ============================================================================
+
+async function extractAnswer(tab) {
+	const copyBtnExpr = findCopyButtonJsExpression();
+
+	await cdp(["eval", tab, `${copyBtnExpr}?.click()`]);
+	await new Promise((r) => setTimeout(r, 400));
+
+	let answer = await cdp(["eval", tab, `window.${GLOBAL_VAR} || ''`]);
+
+	// Retry once if clipboard is empty (Perplexity might be slow to write)
+	if (!answer) {
+		console.error("[perplexity] Clipboard empty, retrying in 2s...");
+		await cdp(["eval", tab, `${copyBtnExpr}?.click()`]);
+		await new Promise((r) => setTimeout(r, 2000));
+		answer = await cdp(["eval", tab, `window.${GLOBAL_VAR} || ''`]);
+	}
+
+	if (!answer) throw new Error("Clipboard interceptor returned empty text");
+
+	const sources = parseSourcesFromMarkdown(answer);
+	return { answer: answer.trim(), sources };
+}
+
+// ============================================================================
+// Main
+// ============================================================================
+
+const USAGE =
+	'Usage: node extractors/perplexity.mjs "<query>" [--tab <prefix>]\n';
+
+async function main() {
+	const args = process.argv.slice(2);
+	validateQuery(args, USAGE);
+
+	const { query, tabPrefix, short } = parseArgs(args);
+
+	try {
+		// Refresh page list so cache is current
+		await cdp(["list"]);
+
+		const tab = await getOrOpenTab(tabPrefix);
+
+		// Navigate to homepage and use the search box (direct ?q= URLs trigger bot redirect)
+		await cdp(["nav", tab, "https://www.perplexity.ai/"], 35000);
+		await dismissConsent(tab, cdp);
+
+		// Wait for React app to mount input (up to 8s)
+		const deadline = Date.now() + 8000;
+		while (Date.now() < deadline) {
+			const found = await cdp([
+				"eval",
+				tab,
+				`!!document.querySelector('${S.input}')`,
+			]).catch(() => "false");
+			if (found === "true") break;
+			await new Promise((r) => setTimeout(r, 400));
+		}
+		await new Promise((r) => setTimeout(r, 300));
+
+		await injectClipboardInterceptor(tab, GLOBAL_VAR);
+		await cdp(["click", tab, S.input]);
+		await new Promise((r) => setTimeout(r, 400));
+		await cdp(["type", tab, query]);
+		await new Promise((r) => setTimeout(r, 400));
+
+		// Submit with Enter (most reliable across Chrome instances)
+		await cdp([
+			"eval",
+			tab,
+			`document.querySelector('${S.input}')?.dispatchEvent(new KeyboardEvent('keydown',{key:'Enter',bubbles:true,keyCode:13})), 'ok'`,
+		]);
+
+		await waitForStreamComplete(tab, {
+			timeout: 30000,
+			interval: 600,
+			stableRounds: 3,
+			selector: "document.body",
+		});
+
+		const { answer, sources } = await extractAnswer(tab);
+
+		if (!answer)
+			throw new Error(
+				"No answer extracted — Perplexity may not have responded",
+			);
+
+		const finalUrl = await cdp(["eval", tab, "document.location.href"]).catch(
+			() => "",
+		);
+		outputJson({
+			query,
+			url: finalUrl,
+			answer: formatAnswer(answer, short),
+			sources,
+		});
+	} catch (e) {
+		handleError(e);
+	}
+}
+
+main();

package/extractors/selectors.mjs

@@ -1,52 +1,54 @@
-// extractors/selectors.mjs
-// Centralized CSS selectors for all engines.
-// Update selectors here when a site changes its UI.
-
-export const SELECTORS = {
-	// ──────────────────────────────────────────────
-	// Perplexity (perplexity.ai)
-	// ──────────────────────────────────────────────
-	perplexity: {
-		input: "#ask-input",
-		copyButton: 'button[aria-label="Copy"]',
-		sourceItem: "[data-pplx-citation-url]",
-		sourceLink: "a",
-		consent: "#onetrust-accept-btn-handler",
-	},
-
-	// ──────────────────────────────────────────────
-	// Bing Copilot (copilot.microsoft.com)
-	// ──────────────────────────────────────────────
-	bing: {
-		input: "#userInput",
-		copyButton: 'button[data-testid="copy-ai-message-button"]',
-		sourceLink: 'a[href^="http"][target="_blank"]',
-		sourceExclude: "copilot.microsoft.com",
-		consent: "#onetrust-accept-btn-handler",
-	},
-
-	// ──────────────────────────────────────────────
-	// Google AI Mode (google.com/search?udm=50)
-	// ──────────────────────────────────────────────
-	google: {
-		answerContainer: ".pWvJNd",
-		sourceLink: 'a[href^="http"]',
-		sourceExclude: ["google.", "gstatic", "googleapis"],
-		sourceHeadingParent: "[data-snhf]",
-		consent: '#L2AGLb, button[jsname="b3VHJd"], .tHlp8d',
-	},
-
-	// ──────────────────────────────────────────────
-	// Gemini (gemini.google.com/app)
-	// ──────────────────────────────────────────────
-	gemini: {
-		input: "rich-textarea .ql-editor",
-		copyButton: 'button[aria-label="Copy"]',
-		sendButton: 'button[aria-label*="Send"]',
-		sourcesSidebarButton: "button.legacy-sources-sidebar-button",
-		sourcesExclude: ["gemini.google", "gstatic", "google.com/search"],
-		citationButtonPattern: 'button[aria-label*="citation from"]',
-		// For parsing citation aria-labels: "View source details for citation from {name}. Opens side panel."
-		citationNameRegex: /from\s+(.+?)\.\s/,
-	},
-};
+// extractors/selectors.mjs
+// Centralized CSS selectors for all engines.
+// Update selectors here when a site changes its UI.
+
+export const SELECTORS = {
+	// ──────────────────────────────────────────────
+	// Perplexity (perplexity.ai)
+	// ──────────────────────────────────────────────
+	perplexity: {
+		input: "#ask-input",
+		// Note: copy button found via JS in extractor (language-agnostic)
+		copyButton: null,
+		sourceItem: "[data-pplx-citation-url]",
+		sourceLink: "a",
+		consent: "#onetrust-accept-btn-handler",
+	},
+
+	// ──────────────────────────────────────────────
+	// Bing Copilot (copilot.microsoft.com)
+	// ──────────────────────────────────────────────
+	bing: {
+		input: "#userInput",
+		copyButton: 'button[data-testid="copy-ai-message-button"]',
+		sourceLink: 'a[href^="http"][target="_blank"]',
+		sourceExclude: "copilot.microsoft.com",
+		consent: "#onetrust-accept-btn-handler",
+	},
+
+	// ──────────────────────────────────────────────
+	// Google AI Mode (google.com/search?udm=50)
+	// ──────────────────────────────────────────────
+	google: {
+		answerContainer: ".pWvJNd",
+		sourceLink: 'a[href^="http"]',
+		sourceExclude: ["google.", "gstatic", "googleapis"],
+		sourceHeadingParent: "[data-snhf]",
+		consent: '#L2AGLb, button[jsname="b3VHJd"], .tHlp8d',
+	},
+
+	// ──────────────────────────────────────────────
+	// Gemini (gemini.google.com/app)
+	// ──────────────────────────────────────────────
+	gemini: {
+		input: "rich-textarea .ql-editor",
+		// Language-agnostic: use Material icon data attributes (work across locales)
+		copyButton: 'button:has(mat-icon[data-mat-icon-name="content_copy"])',
+		sendButton: 'button:has(mat-icon[data-mat-icon-name="send"]), .send-button',
+		sourcesSidebarButton: "button.legacy-sources-sidebar-button",
+		sourcesExclude: ["gemini.google", "gstatic", "google.com/search"],
+		citationButtonPattern: 'button[aria-label*="citation from"]',
+		// For parsing citation aria-labels: "View source details for citation from {name}. Opens side panel."
+		citationNameRegex: /from\s+(.+?)\.\s/,
+	},
+};

package/index.ts

@@ -15,10 +15,10 @@ import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { Type } from "@sinclair/typebox";
 
 import { formatCodingTask } from "./src/formatters/coding.js";
-import { registerGreedySearchTool } from "./src/tools/greedy-search-handler.js";
+import { DEFAULTS } from "./src/search/defaults.mjs";
 import { registerDeepResearchTool } from "./src/tools/deep-research-handler.js";
+import { registerGreedySearchTool } from "./src/tools/greedy-search-handler.js";
 import { cdpAvailable, type ProgressUpdate } from "./src/tools/shared.js";
-import { DEFAULTS } from "./src/search/defaults.js";
 
 const __dir = dirname(fileURLToPath(import.meta.url));
 
@@ -62,22 +62,43 @@ export default function greedySearchExtension(pi: ExtensionAPI) {
 				},
 			),
 			mode: Type.Union(
-				[Type.Literal("code"), Type.Literal("review"), Type.Literal("plan"), Type.Literal("test"), Type.Literal("debug")],
+				[
+					Type.Literal("code"),
+					Type.Literal("review"),
+					Type.Literal("plan"),
+					Type.Literal("test"),
+					Type.Literal("debug"),
+				],
 				{
-					description: "Task mode: code (default), review (code review), plan (architect review), test (write tests), debug (root cause analysis)",
+					description:
+						"Task mode: code (default), review (code review), plan (architect review), test (write tests), debug (root cause analysis)",
 					default: "code",
 				},
 			),
-			context: Type.Optional(Type.String({ description: "Optional code context/snippet to include with the task" })),
+			context: Type.Optional(
+				Type.String({
+					description: "Optional code context/snippet to include with the task",
+				}),
+			),
 		}),
 		execute: async (_toolCallId, params, signal, onUpdate) => {
-			const { task, engine = "gemini", mode = "code", context } = params as {
-				task: string; engine: string; mode: string; context?: string;
+			const {
+				task,
+				engine = "gemini",
+				mode = "code",
+				context,
+			} = params as {
+				task: string;
+				engine: string;
+				mode: string;
+				context?: string;
 			};
 
 			if (!cdpAvailable(__dir)) {
 				return {
-					content: [{ type: "text", text: "cdp.mjs missing — try reinstalling." }],
+					content: [
+						{ type: "text", text: "cdp.mjs missing — try reinstalling." },
+					],
 					details: {} as { raw?: Record<string, unknown> },
 				};
 			}
@@ -87,35 +108,67 @@ export default function greedySearchExtension(pi: ExtensionAPI) {
 
 			try {
 				onUpdate?.({
-					content: [{ type: "text", text: `**Coding task...** 🔄 ${engine === "all" ? "Gemini + Copilot" : engine} (${mode} mode)` }],
+					content: [
+						{
+							type: "text",
+							text: `**Coding task...** 🔄 ${engine === "all" ? "Gemini + Copilot" : engine} (${mode} mode)`,
+						},
+					],
 					details: { _progress: true },
 				} satisfies ProgressUpdate);
 
-				const data = await new Promise<Record<string, unknown>>((resolve, reject) => {
-					const proc = spawn("node", [join(__dir, "bin", "coding-task.mjs"), task, ...flags], {
-						stdio: ["ignore", "pipe", "pipe"],
-					});
-					let out = "";
-					let err = "";
-
-					const onAbort = () => { proc.kill("SIGTERM"); reject(new Error("Aborted")); };
-					signal?.addEventListener("abort", onAbort, { once: true });
-
-					proc.stdout.on("data", (d: Buffer) => (out += d));
-					proc.stderr.on("data", (d: Buffer) => (err += d));
-					proc.on("close", (code: number) => {
-						signal?.removeEventListener("abort", onAbort);
-						if (code !== 0) {
-							reject(new Error(err.trim() || `coding-task.mjs exited with code ${code}`));
-						} else {
-							try { resolve(JSON.parse(out.trim())); }
-							catch { reject(new Error(`Invalid JSON from coding-task.mjs: ${out.slice(0, 200)}`)); }
-						}
-					});
-
-					// Timeout after 3 minutes
-					setTimeout(() => { proc.kill("SIGTERM"); reject(new Error(`Coding task timed out after ${DEFAULTS.CODING_TASK_TIMEOUT / 1000}s`)); }, DEFAULTS.CODING_TASK_TIMEOUT);
-				});
+				const data = await new Promise<Record<string, unknown>>(
+					(resolve, reject) => {
+						const proc = spawn(
+							"node",
+							[join(__dir, "bin", "coding-task.mjs"), task, ...flags],
+							{
+								stdio: ["ignore", "pipe", "pipe"],
+							},
+						);
+						let out = "";
+						let err = "";
+
+						const onAbort = () => {
+							proc.kill("SIGTERM");
+							reject(new Error("Aborted"));
+						};
+						signal?.addEventListener("abort", onAbort, { once: true });
+
+						proc.stdout.on("data", (d: Buffer) => (out += d));
+						proc.stderr.on("data", (d: Buffer) => (err += d));
+						proc.on("close", (code: number) => {
+							signal?.removeEventListener("abort", onAbort);
+							if (code !== 0) {
+								reject(
+									new Error(
+										err.trim() || `coding-task.mjs exited with code ${code}`,
+									),
+								);
+							} else {
+								try {
+									resolve(JSON.parse(out.trim()));
+								} catch {
+									reject(
+										new Error(
+											`Invalid JSON from coding-task.mjs: ${out.slice(0, 200)}`,
+										),
+									);
+								}
+							}
+						});
+
+						// Timeout after 3 minutes
+						setTimeout(() => {
+							proc.kill("SIGTERM");
+							reject(
+								new Error(
+									`Coding task timed out after ${DEFAULTS.CODING_TASK_TIMEOUT / 1000}s`,
+								),
+							);
+						}, DEFAULTS.CODING_TASK_TIMEOUT);
+					},
+				);
 
 				const text = formatCodingTask(data);
 				return {
@@ -131,4 +184,95 @@ export default function greedySearchExtension(pi: ExtensionAPI) {
 			}
 		},
 	});
-}
+
+	// ─── /set-greedy-locale command ───────────────────────────────────────────
+	pi.registerCommand("set-greedy-locale", {
+		description:
+			"Set default locale for GreedySearch results (e.g., /set-greedy-locale de, /set-greedy-locale --clear, /set-greedy-locale --show)",
+		handler: async (args, ctx) => {
+			const arg = args.trim() || "--show";
+
+			if (arg === "--show") {
+				const config = loadUserConfig();
+				if (config.locale) {
+					ctx.ui.notify(`Default locale: ${config.locale}`, "info");
+				} else {
+					ctx.ui.notify("No default locale (uses: en)", "info");
+				}
+				return;
+			}
+
+			if (arg === "--clear") {
+				const config = loadUserConfig();
+				delete config.locale;
+				saveUserConfig(config);
+				ctx.ui.notify("Default locale cleared (now uses: en).", "info");
+				return;
+			}
+
+			// Set locale
+			const locale = arg.toLowerCase();
+			const VALID_LOCALES = [
+				"en",
+				"de",
+				"fr",
+				"es",
+				"it",
+				"pt",
+				"nl",
+				"pl",
+				"ru",
+				"ja",
+				"ko",
+				"zh",
+				"ar",
+				"hi",
+				"tr",
+				"sv",
+				"da",
+				"no",
+				"fi",
+				"cs",
+				"hu",
+				"ro",
+				"el",
+			];
+
+			if (!VALID_LOCALES.includes(locale)) {
+				ctx.ui.notify(
+					`Invalid locale "${locale}". Valid: ${VALID_LOCALES.join(", ")}`,
+					"error",
+				);
+				return;
+			}
+
+			const config = loadUserConfig();
+			config.locale = locale;
+			saveUserConfig(config);
+			ctx.ui.notify(`Default locale set to: ${locale}`, "info");
+		},
+	});
+}
+
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+// Config helpers for /set-greedy-locale command
+import { homedir } from "node:os";
+
+const USER_CONFIG_DIR = join(homedir(), ".config", "greedysearch");
+const USER_CONFIG_FILE = join(USER_CONFIG_DIR, "config.json");
+
+function loadUserConfig(): Record<string, string> {
+	try {
+		if (existsSync(USER_CONFIG_FILE)) {
+			return JSON.parse(readFileSync(USER_CONFIG_FILE, "utf8"));
+		}
+	} catch {
+		// Ignore parse errors
+	}
+	return {};
+}
+
+function saveUserConfig(config: Record<string, string>): void {
+	mkdirSync(USER_CONFIG_DIR, { recursive: true });
+	writeFileSync(USER_CONFIG_FILE, JSON.stringify(config, null, 2), "utf8");
+}