initial commit

kagi.js

@@ -0,0 +1,65 @@
+import { createDriver } from "./driver.js";
+
+export class SearchProviderError extends Error {
+  constructor(code, message) {
+    super(message);
+    this.code = code;
+  }
+}
+
+// Kagi Search Extraction
+export const kagiSearchScript = `return Array.from(document.querySelectorAll("div > .__sri-title"))
+  .map((i) => i.parentElement)
+  .map((i) => ({
+    title: i.children[0].querySelector("a").innerText,
+    url: i.children[0].querySelector("a").getAttribute("href"),
+    description: i.querySelector(".__sri-body").innerText,
+  }));`;
+
+// Build Kagi Search Url
+export function buildKagiSearchUrl(query, token) {
+  return `https://kagi.com/search?token=${encodeURIComponent(token)}&q=${encodeURIComponent(query)}`;
+}
+
+// Search Kagi
+export async function searchKagi({
+  query,
+  token = process.env.KAGI_TOKEN,
+  headless = true,
+  existingUrl,
+  timeoutMs = 5000,
+  intervalMs = 200,
+} = {}) {
+  if (!query) {
+    throw new SearchProviderError("QUERY_REQUIRED", "query is required");
+  }
+
+  // Validate Kagi Token
+  if (!token) {
+    throw new SearchProviderError(
+      "KAGI_TOKEN_REQUIRED",
+      "Kagi search requires --token or the KAGI_TOKEN environment variable.",
+    );
+  }
+
+  const driver = await createDriver({ headless, existingUrl });
+
+  try {
+    // Navigate To Kagi
+    await driver.get(buildKagiSearchUrl(query, token));
+
+    // Poll For Results
+    let result = [];
+    const start = Date.now();
+
+    while (Date.now() - start < timeoutMs) {
+      result = await driver.executeScript(kagiSearchScript);
+      if (Array.isArray(result) && result.length > 0) break;
+      await new Promise((resolve) => setTimeout(resolve, intervalMs));
+    }
+
+    return result;
+  } finally {
+    await driver.quit();
+  }
+}