@@ -11,3 +11,5 @@ export const SidepanelSettings = () => {
)
}
+
+export default SidepanelSettings
diff --git a/src/services/app.ts b/src/services/app.ts
new file mode 100644
index 0000000..8f0cc66
--- /dev/null
+++ b/src/services/app.ts
@@ -0,0 +1,36 @@
+import { Storage } from "@plasmohq/storage"
+const storage = new Storage()
+
+const DEFAULT_URL_REWRITE_URL = "http://127.0.0.1:11434"
+
+export const isUrlRewriteEnabled = async () => {
+ const enabled = await storage.get
("urlRewriteEnabled")
+ return enabled
+}
+export const setUrlRewriteEnabled = async (enabled: boolean) => {
+ await storage.set("urlRewriteEnabled", enabled ? "true" : "false")
+}
+
+export const getRewriteUrl = async () => {
+ const rewriteUrl = await storage.get("rewriteUrl")
+ if (!rewriteUrl || rewriteUrl.trim() === "") {
+ return DEFAULT_URL_REWRITE_URL
+ }
+ return rewriteUrl
+}
+
+export const setRewriteUrl = async (url: string) => {
+ await storage.set("rewriteUrl", url)
+}
+
+export const getAdvancedOllamaSettings = async () => {
+ const [isEnableRewriteUrl, rewriteUrl] = await Promise.all([
+ isUrlRewriteEnabled(),
+ getRewriteUrl()
+ ])
+
+ return {
+ isEnableRewriteUrl,
+ rewriteUrl
+ }
+}
diff --git a/src/services/ollama.ts b/src/services/ollama.ts
index ccca697..e1ce53b 100644
--- a/src/services/ollama.ts
+++ b/src/services/ollama.ts
@@ -1,6 +1,6 @@
import { Storage } from "@plasmohq/storage"
import { cleanUrl } from "../libs/clean-url"
-import { chromeRunTime } from "../libs/runtime"
+import { urlRewriteRuntime } from "../libs/runtime"
const storage = new Storage()
@@ -22,10 +22,10 @@ Search results:
export const getOllamaURL = async () => {
const ollamaURL = await storage.get("ollamaURL")
if (!ollamaURL || ollamaURL.length === 0) {
- await chromeRunTime(DEFAULT_OLLAMA_URL)
+ await urlRewriteRuntime(DEFAULT_OLLAMA_URL)
return DEFAULT_OLLAMA_URL
}
- await chromeRunTime(cleanUrl(ollamaURL))
+ await urlRewriteRuntime(cleanUrl(ollamaURL))
return ollamaURL
}
@@ -163,7 +163,7 @@ export const setOllamaURL = async (ollamaURL: string) => {
"http://127.0.0.1:"
)
}
- await chromeRunTime(cleanUrl(formattedUrl))
+ await urlRewriteRuntime(cleanUrl(formattedUrl))
await storage.set("ollamaURL", cleanUrl(formattedUrl))
}
diff --git a/src/services/tts.ts b/src/services/tts.ts
index 2d38d3e..847efb4 100644
--- a/src/services/tts.ts
+++ b/src/services/tts.ts
@@ -21,8 +21,16 @@ export const setTTSProvider = async (ttsProvider: string) => {
}
export const getBrowserTTSVoices = async () => {
- const tts = await chrome.tts.getVoices()
- return tts
+ if (import.meta.env.BROWSER === "chrome") {
+ const tts = await chrome.tts.getVoices()
+ return tts
+ } else {
+ const tts = await speechSynthesis.getVoices()
+ return tts.map((voice) => ({
+ voiceName: voice.name,
+ lang: voice.lang
+ }))
+ }
}
export const getVoice = async () => {
diff --git a/src/utils/action.ts b/src/utils/action.ts
new file mode 100644
index 0000000..5881aa4
--- /dev/null
+++ b/src/utils/action.ts
@@ -0,0 +1,25 @@
+import { browser } from "wxt/browser"
+
+export const setTitle = ({ title }: { title: string }) => {
+ if (import.meta.env.BROWSER === "chrome") {
+ chrome.action.setTitle({ title })
+ } else {
+ browser.browserAction.setTitle({ title })
+ }
+}
+
+export const setBadgeBackgroundColor = ({ color }: { color: string }) => {
+ if (import.meta.env.BROWSER === "chrome") {
+ chrome.action.setBadgeBackgroundColor({ color })
+ } else {
+ browser.browserAction.setBadgeBackgroundColor({ color })
+ }
+}
+
+export const setBadgeText = ({ text }: { text: string }) => {
+ if (import.meta.env.BROWSER === "chrome") {
+ chrome.action.setBadgeText({ text })
+ } else {
+ browser.browserAction.setBadgeText({ text })
+ }
+}
diff --git a/src/web/search-engines/duckduckgo.ts b/src/web/search-engines/duckduckgo.ts
index 089e571..374ce60 100644
--- a/src/web/search-engines/duckduckgo.ts
+++ b/src/web/search-engines/duckduckgo.ts
@@ -1,5 +1,5 @@
import { cleanUrl } from "@/libs/clean-url"
-import { chromeRunTime } from "@/libs/runtime"
+import { urlRewriteRuntime } from "@/libs/runtime"
import { PageAssistHtmlLoader } from "@/loader/html"
import {
defaultEmbeddingChunkOverlap,
@@ -18,7 +18,7 @@ import { RecursiveCharacterTextSplitter } from "langchain/text_splitter"
import { MemoryVectorStore } from "langchain/vectorstores/memory"
export const localDuckDuckGoSearch = async (query: string) => {
- await chromeRunTime(cleanUrl("https://html.duckduckgo.com/html/?q=" + query))
+ await urlRewriteRuntime(cleanUrl("https://html.duckduckgo.com/html/?q=" + query), "duckduckgo")
const abortController = new AbortController()
setTimeout(() => abortController.abort(), 10000)
diff --git a/src/web/search-engines/google.ts b/src/web/search-engines/google.ts
index 98fa2a8..8f0a16a 100644
--- a/src/web/search-engines/google.ts
+++ b/src/web/search-engines/google.ts
@@ -7,7 +7,7 @@ import type { Document } from "@langchain/core/documents"
import { RecursiveCharacterTextSplitter } from "langchain/text_splitter"
import { MemoryVectorStore } from "langchain/vectorstores/memory"
import { cleanUrl } from "~/libs/clean-url"
-import { chromeRunTime } from "~/libs/runtime"
+import { urlRewriteRuntime } from "~/libs/runtime"
import { PageAssistHtmlLoader } from "~/loader/html"
import {
defaultEmbeddingChunkOverlap,
@@ -18,8 +18,9 @@ import {
export const localGoogleSearch = async (query: string) => {
- await chromeRunTime(
- cleanUrl("https://www.google.com/search?hl=en&q=" + query)
+ await urlRewriteRuntime(
+ cleanUrl("https://www.google.com/search?hl=en&q=" + query),
+ "google"
)
const abortController = new AbortController()
setTimeout(() => abortController.abort(), 10000)
diff --git a/src/web/search-engines/sogou.ts b/src/web/search-engines/sogou.ts
index ffa7e6f..274d03f 100644
--- a/src/web/search-engines/sogou.ts
+++ b/src/web/search-engines/sogou.ts
@@ -1,5 +1,5 @@
import { cleanUrl } from "@/libs/clean-url"
-import { chromeRunTime } from "@/libs/runtime"
+import { urlRewriteRuntime } from "@/libs/runtime"
import { PageAssistHtmlLoader } from "@/loader/html"
import {
defaultEmbeddingChunkOverlap,
@@ -25,7 +25,10 @@ const getCorrectTargeUrl = async (url: string) => {
return matches?.[1] || ""
}
export const localSogouSearch = async (query: string) => {
- await chromeRunTime(cleanUrl("https://www.sogou.com/web?query=" + query))
+ await urlRewriteRuntime(
+ cleanUrl("https://www.sogou.com/web?query=" + query),
+ "sogou"
+ )
const abortController = new AbortController()
diff --git a/tailwind.config.js b/tailwind.config.js
index a30235c..f4b105c 100644
--- a/tailwind.config.js
+++ b/tailwind.config.js
@@ -3,5 +3,5 @@ module.exports = {
mode: "jit",
darkMode: "class",
content: ["./src/**/*.tsx"],
- plugins: [require("@tailwindcss/forms"), require("@tailwindcss/typography")]
+ plugins: [require("@tailwindcss/forms"), require("@tailwindcss/typography"),]
}
diff --git a/wxt.config.ts b/wxt.config.ts
index d80478d..0f5aeba 100644
--- a/wxt.config.ts
+++ b/wxt.config.ts
@@ -2,35 +2,73 @@ import { defineConfig } from "wxt"
import react from "@vitejs/plugin-react"
import topLevelAwait from "vite-plugin-top-level-await"
+const chromeMV3Permissions = [
+ "storage",
+ "sidePanel",
+ "activeTab",
+ "scripting",
+ "declarativeNetRequest",
+ "action",
+ "unlimitedStorage",
+ "contextMenus",
+ "tts"
+]
+
+const firefoxMV2Permissions = [
+ "storage",
+ "activeTab",
+ "scripting",
+ "unlimitedStorage",
+ "contextMenus",
+ "webRequest",
+ "webRequestBlocking",
+ "http://*/*",
+ "https://*/*",
+ "file://*/*"
+]
+
// See https://wxt.dev/api/config.html
export default defineConfig({
vite: () => ({
- plugins: [react(),
+ plugins: [
+ react(),
topLevelAwait({
- promiseExportName: '__tla',
- promiseImportName: i => `__tla_${i}`,
- }),
+ promiseExportName: "__tla",
+ promiseImportName: (i) => `__tla_${i}`
+ })
],
build: {
rollupOptions: {
- external: [
- "langchain",
- "@langchain/community",
- ]
+ external: ["langchain", "@langchain/community"]
}
}
}),
entrypointsDir: "entries",
srcDir: "src",
outDir: "build",
+
manifest: {
- version: "1.1.6",
- name: '__MSG_extName__',
- description: '__MSG_extDescription__',
- default_locale: 'en',
+ version: "1.1.7",
+ name:
+ process.env.TARGET === "firefox"
+ ? "Page Assist - A Web UI for Local AI Models"
+ : "__MSG_extName__",
+ description: "__MSG_extDescription__",
+ default_locale: "en",
action: {},
author: "n4ze3m",
- host_permissions: ["http://*/*", "https://*/*", "file://*/*"],
+ browser_specific_settings:
+ process.env.TARGET === "firefox"
+ ? {
+ gecko: {
+ id: "page-assist@nazeem"
+ }
+ }
+ : undefined,
+ host_permissions:
+ process.env.TARGET !== "firefox"
+ ? ["http://*/*", "https://*/*", "file://*/*"]
+ : undefined,
commands: {
_execute_action: {
suggested_key: {
@@ -44,16 +82,9 @@ export default defineConfig({
}
}
},
- permissions: [
- "storage",
- "sidePanel",
- "activeTab",
- "scripting",
- "declarativeNetRequest",
- "action",
- "unlimitedStorage",
- "contextMenus",
- "tts"
- ]
+ permissions:
+ process.env.TARGET === "firefox"
+ ? firefoxMV2Permissions
+ : chromeMV3Permissions
}
})