iod/page-assist
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support Edge browser

Browse Source
This commit is contained in:
n4ze3m 2025-02-08 23:06:09 +05:30
parent 87c2024771
commit ef51cf85ff
10 changed files with 114 additions and 192 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
package.json
View File

@ -7,8 +7,10 @@
"scripts": { "scripts": {
"dev": "cross-env TARGET=chrome wxt", "dev": "cross-env TARGET=chrome wxt",
"dev:firefox": "cross-env TARGET=firefox wxt -b firefox", "dev:firefox": "cross-env TARGET=firefox wxt -b firefox",
"dev:edge": "cross-env TARGET=chrome wxt -b edge",
"build": "cross-env TARGET=chrome wxt build", "build": "cross-env TARGET=chrome wxt build",
"build:firefox": "cross-env TARGET=firefox wxt build -b firefox", "build:firefox": "cross-env TARGET=firefox wxt build -b firefox",
"build:edge": "cross-env TARGET=chrome wxt build -b edge",
"zip": "cross-env TARGET=chrome wxt zip", "zip": "cross-env TARGET=chrome wxt zip",
"zip:firefox": "cross-env TARGET=firefox wxt zip -b firefox", "zip:firefox": "cross-env TARGET=firefox wxt zip -b firefox",
"compile": "tsc --noEmit", "compile": "tsc --noEmit",

256
src/entries/background.ts
View File

@ -35,15 +35,9 @@ export default defineBackground({
} }
}) })
if (import.meta.env.BROWSER === "chrome") { chrome.action.onClicked.addListener((tab) => {
chrome.action.onClicked.addListener((tab) => { chrome.tabs.create({ url: chrome.runtime.getURL("/options.html") })
chrome.tabs.create({ url: chrome.runtime.getURL("/options.html") }) })
})
} else {
browser.browserAction.onClicked.addListener((tab) => {
browser.tabs.create({ url: browser.runtime.getURL("/options.html") })
})
}
const contextMenuTitle = { const contextMenuTitle = {
webUi: browser.i18n.getMessage("openOptionToChat"), webUi: browser.i18n.getMessage("openOptionToChat"),
@ -91,176 +85,98 @@ export default defineBackground({
contexts: ["selection"] contexts: ["selection"]
}) })
if (import.meta.env.BROWSER === "chrome") { browser.contextMenus.onClicked.addListener(async (info, tab) => {
browser.contextMenus.onClicked.addListener(async (info, tab) => { if (info.menuItemId === "open-side-panel-pa") {
if (info.menuItemId === "open-side-panel-pa") { chrome.sidePanel.open({
chrome.sidePanel.open({ tabId: tab.id!
tabId: tab.id! })
} else if (info.menuItemId === "open-web-ui-pa") {
browser.tabs.create({
url: browser.runtime.getURL("/options.html")
})
} else if (info.menuItemId === "summarize-pa") {
chrome.sidePanel.open({
tabId: tab.id!
})
// this is a bad method hope somone can fix it :)
setTimeout(async () => {
await browser.runtime.sendMessage({
from: "background",
type: "summary",
text: info.selectionText
}) })
} else if (info.menuItemId === "open-web-ui-pa") { }, isCopilotRunning ? 0 : 5000)
browser.tabs.create({
url: browser.runtime.getURL("/options.html") } else if (info.menuItemId === "rephrase-pa") {
chrome.sidePanel.open({
tabId: tab.id!
})
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "rephrase",
from: "background",
text: info.selectionText
}) })
} else if (info.menuItemId === "summarize-pa") { }, isCopilotRunning ? 0 : 5000)
chrome.sidePanel.open({
tabId: tab.id! } else if (info.menuItemId === "translate-pg") {
chrome.sidePanel.open({
tabId: tab.id!
})
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "translate",
from: "background",
text: info.selectionText
}) })
// this is a bad method hope somone can fix it :) }, isCopilotRunning ? 0 : 5000)
setTimeout(async () => { } else if (info.menuItemId === "explain-pa") {
await browser.runtime.sendMessage({ chrome.sidePanel.open({
from: "background", tabId: tab.id!
type: "summary", })
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "rephrase-pa") { setTimeout(async () => {
chrome.sidePanel.open({ await browser.runtime.sendMessage({
tabId: tab.id! type: "explain",
from: "background",
text: info.selectionText
}) })
setTimeout(async () => { }, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "custom-pg") {
chrome.sidePanel.open({
tabId: tab.id!
})
await browser.runtime.sendMessage({ setTimeout(async () => {
type: "rephrase", await browser.runtime.sendMessage({
from: "background", type: "custom",
text: info.selectionText from: "background",
}) text: info.selectionText
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "translate-pg") {
chrome.sidePanel.open({
tabId: tab.id!
}) })
}, isCopilotRunning ? 0 : 5000)
}
})
setTimeout(async () => { browser.commands.onCommand.addListener((command) => {
await browser.runtime.sendMessage({ switch (command) {
type: "translate", case "execute_side_panel":
from: "background", chrome.tabs.query(
text: info.selectionText { active: true, currentWindow: true },
}) async (tabs) => {
}, isCopilotRunning ? 0 : 5000) const tab = tabs[0]
} else if (info.menuItemId === "explain-pa") { chrome.sidePanel.open({
chrome.sidePanel.open({ tabId: tab.id!
tabId: tab.id! })
}) }
)
break
default:
break
}
})
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "explain",
from: "background",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "custom-pg") {
chrome.sidePanel.open({
tabId: tab.id!
})
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "custom",
from: "background",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
}
})
browser.commands.onCommand.addListener((command) => {
switch (command) {
case "execute_side_panel":
chrome.tabs.query(
{ active: true, currentWindow: true },
async (tabs) => {
const tab = tabs[0]
chrome.sidePanel.open({
tabId: tab.id!
})
}
)
break
default:
break
}
})
}
if (import.meta.env.BROWSER === "firefox") {
browser.contextMenus.onClicked.addListener((info, tab) => {
if (info.menuItemId === "open-side-panel-pa") {
browser.sidebarAction.toggle()
} else if (info.menuItemId === "open-web-ui-pa") {
browser.tabs.create({
url: browser.runtime.getURL("/options.html")
})
} else if (info.menuItemId === "summarize-pa") {
if (!isCopilotRunning) {
browser.sidebarAction.toggle()
}
setTimeout(async () => {
await browser.runtime.sendMessage({
from: "background",
type: "summary",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "rephrase-pa") {
if (!isCopilotRunning) {
browser.sidebarAction.toggle()
}
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "rephrase",
from: "background",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "translate-pg") {
if (!isCopilotRunning) {
browser.sidebarAction.toggle()
}
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "translate",
from: "background",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "explain-pa") {
if (!isCopilotRunning) {
browser.sidebarAction.toggle()
}
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "explain",
from: "background",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
} else if (info.menuItemId === "custom-pg") {
if (!isCopilotRunning) {
browser.sidebarAction.toggle()
}
setTimeout(async () => {
await browser.runtime.sendMessage({
type: "custom",
from: "background",
text: info.selectionText
})
}, isCopilotRunning ? 0 : 5000)
}
})
browser.commands.onCommand.addListener((command) => {
switch (command) {
case "execute_side_panel":
browser.sidebarAction.toggle()
break
default:
break
}
})
}
}, },
persistent: true persistent: true
}) })

7
src/hooks/useTTS.tsx
View File

@ -32,7 +32,10 @@ export const useTTS = () => {
if (isSSML) { if (isSSML) {
utterance = markdownToSSML(utterance) utterance = markdownToSSML(utterance)
} }
if (import.meta.env.BROWSER === "chrome") { if (
import.meta.env.BROWSER === "chrome" ||
import.meta.env.BROWSER === "edge"
) {
chrome.tts.speak(utterance, { chrome.tts.speak(utterance, {
voiceName: voice, voiceName: voice,
onEvent(event) { onEvent(event) {
@ -112,7 +115,7 @@ export const useTTS = () => {
return return
} }
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.tts.stop() chrome.tts.stop()
} else { } else {
window.speechSynthesis.cancel() window.speechSynthesis.cancel()

2
src/libs/get-html.ts
View File

@ -24,7 +24,7 @@ const _getHtml = () => {
export const getDataFromCurrentTab = async () => { export const getDataFromCurrentTab = async () => {
const result = new Promise((resolve) => { const result = new Promise((resolve) => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => { chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => {
const tab = tabs[0] const tab = tabs[0]

2
src/libs/get-screenshot.ts
View File

@ -1,6 +1,6 @@
const captureVisibleTab = () => { const captureVisibleTab = () => {
const result = new Promise<string>((resolve) => { const result = new Promise<string>((resolve) => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => { chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => {
const tab = tabs[0] const tab = tabs[0]
chrome.tabs.captureVisibleTab(null, { format: "png" }, (dataUrl) => { chrome.tabs.captureVisibleTab(null, { format: "png" }, (dataUrl) => {

2
src/libs/runtime.ts
View File

@ -6,7 +6,7 @@ export const urlRewriteRuntime = async function (
) { ) {
if (browser.runtime && browser.runtime.id) { if (browser.runtime && browser.runtime.id) {
const { isEnableRewriteUrl, rewriteUrl } = await getAdvancedOllamaSettings() const { isEnableRewriteUrl, rewriteUrl } = await getAdvancedOllamaSettings()
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
const url = new URL(domain) const url = new URL(domain)
const domains = [url.hostname] const domains = [url.hostname]
let origin = `${url.protocol}//${url.hostname}` let origin = `${url.protocol}//${url.hostname}`

2
src/parser/google-docs.ts
View File

@ -95,7 +95,7 @@ const getGoogleDocs = () => {
export const parseGoogleDocs = async () => { export const parseGoogleDocs = async () => {
const result = new Promise((resolve) => { const result = new Promise((resolve) => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => { chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => {
const tab = tabs[0] const tab = tabs[0]

2
src/services/tts.ts
View File

@ -21,7 +21,7 @@ export const setTTSProvider = async (ttsProvider: string) => {
} }
export const getBrowserTTSVoices = async () => { export const getBrowserTTSVoices = async () => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
const tts = await chrome.tts.getVoices() const tts = await chrome.tts.getVoices()
return tts return tts
} else { } else {

6
src/utils/action.ts
View File

@ -1,7 +1,7 @@
import { browser } from "wxt/browser" import { browser } from "wxt/browser"
export const setTitle = ({ title }: { title: string }) => { export const setTitle = ({ title }: { title: string }) => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.action.setTitle({ title }) chrome.action.setTitle({ title })
} else { } else {
browser.browserAction.setTitle({ title }) browser.browserAction.setTitle({ title })
@ -9,7 +9,7 @@ export const setTitle = ({ title }: { title: string }) => {
} }
export const setBadgeBackgroundColor = ({ color }: { color: string }) => { export const setBadgeBackgroundColor = ({ color }: { color: string }) => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.action.setBadgeBackgroundColor({ color }) chrome.action.setBadgeBackgroundColor({ color })
} else { } else {
browser.browserAction.setBadgeBackgroundColor({ color }) browser.browserAction.setBadgeBackgroundColor({ color })
@ -17,7 +17,7 @@ export const setBadgeBackgroundColor = ({ color }: { color: string }) => {
} }
export const setBadgeText = ({ text }: { text: string }) => { export const setBadgeText = ({ text }: { text: string }) => {
if (import.meta.env.BROWSER === "chrome") { if (import.meta.env.BROWSER === "chrome" || import.meta.env.BROWSER === "edge") {
chrome.action.setBadgeText({ text }) chrome.action.setBadgeText({ text })
} else { } else {
browser.browserAction.setBadgeText({ text }) browser.browserAction.setBadgeText({ text })

23
src/utils/ocr.ts
View File

@ -1,17 +1,18 @@
import { createWorker } from 'tesseract.js'; import { createWorker } from "tesseract.js"
export async function processImageForOCR(imageData: string): Promise<string> { export async function processImageForOCR(imageData: string): Promise<string> {
const worker = await createWorker('eng-fast', undefined, { const isOCROffline = import.meta.env.BROWSER === "edge"
workerPath: "/ocr/worker.min.js", const worker = await createWorker(!isOCROffline ? "eng-fast" : "eng", undefined, {
workerBlobURL: false, workerPath: "/ocr/worker.min.js",
corePath: "/ocr/tesseract-core-simd.js", workerBlobURL: false,
errorHandler: e => console.error(e), corePath: "/ocr/tesseract-core-simd.js",
langPath: "/ocr/lang" errorHandler: (e) => console.error(e),
}); langPath: !isOCROffline ? "/ocr/lang" : undefined
})
const result = await worker.recognize(imageData); const result = await worker.recognize(imageData)
await worker.terminate(); await worker.terminate()
return result.data.text; return result.data.text
} }