@mammothb/pi-websearch 0.1.0 → 0.3.0

package/bin/searxng

@@ -0,0 +1,48 @@
+#!/usr/bin/env bash
+# searxng — Start/stop the local SearXNG instance for pi-websearch.
+#
+# Usage:
+#   searxng up      Start SearXNG (docker compose up -d)
+#   searxng down    Stop SearXNG (docker compose down)
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+COMPOSE_DIR="${SCRIPT_DIR}/../searxng"
+COMPOSE_FILE="${COMPOSE_DIR}/docker-compose.yml"
+
+up() {
+  # Ensure the core-config directory exists (needed for volume mount).
+  mkdir -p "${COMPOSE_DIR}/core-config"
+
+  # Check if the SearXNG core container is already running.
+  if docker compose -f "${COMPOSE_FILE}" ps --status running 2>/dev/null | grep -q "searxng-core"; then
+    echo "SearXNG is already running" >&2
+    return 0
+  fi
+
+  echo "Starting SearXNG..." >&2
+  docker compose -f "${COMPOSE_FILE}" up -d
+
+  # Read the port from .env or default to 8080.
+  local port
+  port=$(grep -E '^SEARXNG_PORT=' "${COMPOSE_DIR}/.env" 2>/dev/null | cut -d= -f2 || echo "8080")
+  port="${port:-8080}"
+
+  echo "SearXNG is starting on http://localhost:${port}" >&2
+  echo "It may take a few seconds to become ready." >&2
+}
+
+down() {
+  echo "Stopping SearXNG..." >&2
+  docker compose -f "${COMPOSE_FILE}" down
+  echo "SearXNG stopped." >&2
+}
+
+case "${1:-}" in
+  up)   up ;;
+  down) down ;;
+  *)
+    echo "Usage: $0 {up|down}" >&2
+    exit 1
+    ;;
+esac

package/index.ts

@@ -1,6 +1,40 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { createWebsearchTool } from "./src/websearch.js";
+import { loadConfig } from "./src/config";
+import {
+  registerInstance,
+  unregisterInstance,
+} from "./src/lib/searxng-manager";
+import { createWebsearchTool } from "./src/websearch";
 
 export default function (pi: ExtensionAPI) {
-  pi.registerTool(createWebsearchTool());
+  let registered = false;
+  let searxngActive = false;
+  let searxngScript: string | undefined;
+
+  pi.on("session_start", async (_, ctx) => {
+    const config = loadConfig(ctx.cwd);
+    pi.registerTool(createWebsearchTool(config));
+
+    searxngActive = config.provider === "searxng";
+    if (searxngActive) {
+      searxngScript = config.searxng.script;
+      await registerInstance(searxngScript);
+    }
+
+    if (!registered) {
+      registered = true;
+      const providerDetail =
+        config.provider === "searxng" ? config.searxng.url : config.exaMcp.url;
+      ctx.ui.notify(
+        `websearch: provider ${config.provider} (${providerDetail})`,
+        "info",
+      );
+    }
+  });
+
+  pi.on("session_shutdown", async (event, _ctx) => {
+    if (event.reason === "quit" && searxngActive) {
+      await unregisterInstance(searxngScript);
+    }
+  });
 }

package/package.json

@@ -1,14 +1,19 @@
 {
   "name": "@mammothb/pi-websearch",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "description": "A pi extension that adds a websearch tool for searching the web",
   "keywords": [
     "pi-package"
   ],
   "license": "MIT",
+  "bin": {
+    "searxng": "./bin/searxng"
+  },
   "files": [
     "index.ts",
-    "src"
+    "src",
+    "searxng",
+    "bin"
   ],
   "pi": {
     "extensions": [

package/searxng/.env

@@ -0,0 +1,16 @@
+# Read the documentation before using the `docker-compose.yml` file:
+# https://docs.searxng.org/admin/installation-docker.html
+#
+# Additional ENVs:
+# https://docs.searxng.org/admin/settings/settings_general.html#settings-general
+# https://docs.searxng.org/admin/settings/settings_server.html#settings-server
+
+# Use a specific version tag. E.g. "latest" or "2026.3.25-541c6c3cb".
+#SEARXNG_VERSION=latest
+
+# Listen to a specific address.
+#SEARXNG_HOST=[::]
+
+# Listen to a specific port.
+SEARXNG_PORT=8080
+FORCE_OWNERSHIP=false