initial commit

Author: hughobrien

README.md

@@ -0,0 +1,10 @@
+# Kagi Chrome Extension
+
+This enables basic Kagi features for your Chromium browser (Chrome, Brave, Edge, Opera, [etc.](https://en.wikipedia.org/wiki/Chromium_(web_browser)#Browsers_based_on_Chromium)).
+
+Install from the [Chrome Web Store](TODO ADD LINK) (not yet available)
+
+## Features
+- Sets Kagi as your default search engine
+- Preserves your login across private browsing.
+- Search-by-image by right-clicking on an image.

icons/icon_16px.png

@@ -0,0 +1,39 @@
+{
+  "manifest_version": 3,
+  "name": "Kagi Search",
+  "version": "1.0",
+  "description": "Set Kagi as your default search engine, while also preserving your session in private browsing. Remember to enable 'Allow Incognito'.",
+  "background": {
+    "service_worker": "service_worker.js",
+    "type": "module"
+  },
+  "icons": {
+    "16": "icons/icon_16px.png",
+    "32": "icons/icon_32px.png",
+    "48": "icons/icon_48px.png",
+    "180": "icons/icon_180px.png"
+  },
+  "action": {
+    "default_icon": "icons/icon_32px.png",
+    "default_title": "Kagi Search"
+  },
+  "permissions": [
+    "contextMenus",
+    "cookies",
+    "declarativeNetRequest",
+    "webRequest"
+  ],
+  "host_permissions": ["https://*.kagi.com/*"],
+  "chrome_settings_overrides": {
+    "search_provider": {
+      "name": "Kagi",
+      "search_url": "https://kagi.com/search?q={searchTerms}",
+      "favicon_url": "https://assets.kagi.com/v2/favicon-32x32.png",
+      "keyword": "@kagi",
+      "is_default": true,
+      "suggest_url": "https://kagi.com/api/autosuggest?q={searchTerms}",
+      "encoding": "UTF-8"
+    }
+  },
+  "incognito": "spanning"
+}

icons/icon_180px.png

@@ -0,0 +1,75 @@
+const kagiBaseUrl = "https://kagi.com/";
+let extensionToken = undefined; // use process memory to hold the token
+
+chrome.runtime.onInstalled.addListener((details) => {
+  if (details.reason === chrome.runtime.OnInstalledReason.INSTALL) {
+    chrome.tabs.create({ url: kagiBaseUrl });
+  }
+});
+
+async function loadTokenFromCookies() {
+  const cookie = await chrome.cookies.get({
+    url: kagiBaseUrl,
+    name: "kagi_session",
+  });
+
+  if (!cookie || !cookie.value || cookie.value.trim().length === 0) {
+    return;
+  }
+
+  let valuesDiffer = cookie.value !== extensionToken;
+  extensionToken = cookie.value;
+  if (valuesDiffer) await updateRules();
+}
+
+chrome.webRequest.onCompleted.addListener(loadTokenFromCookies, {
+  urls: [`${kagiBaseUrl}*`],
+});
+
+async function updateRules() {
+  // https://developer.chrome.com/docs/extensions/reference/api/declarativeNetRequest#dynamic-and-session-rules
+  // dynamic needed as the token can't be known ahead of time
+  await chrome.declarativeNetRequest.updateDynamicRules({
+    addRules: [
+      {
+        id: 1,
+        priority: 1,
+        action: {
+          type: "modifyHeaders",
+          requestHeaders: [
+            {
+              header: "Authorization",
+              value: extensionToken,
+              operation: "set",
+            },
+          ],
+        },
+        condition: {
+          urlFilter: `${kagiBaseUrl}*`,
+          resourceTypes: ["main_frame"],
+        },
+      },
+    ],
+    removeRuleIds: [1],
+  });
+}
+
+// Image Search
+function kagiImageSearch(info) {
+  const imageUrl = encodeURIComponent(info.srcUrl);
+  chrome.tabs.create({
+    url: `${kagiBaseUrl}images?q=${imageUrl}&reverse=reference`,
+  });
+}
+
+chrome.contextMenus.create({
+  id: "kagi-image-search",
+  title: "Kagi Image Search",
+  contexts: ["image"],
+});
+
+chrome.contextMenus.onClicked.addListener((info, tab) => {
+  if (info.menuItemId === "kagi-image-search") {
+    kagiImageSearch(info, tab);
+  }
+});