156 lines
4.0 KiB
TypeScript
156 lines
4.0 KiB
TypeScript
import { getOllamaURL, isOllamaRunning } from "../services/ollama"
|
|
import { Storage } from "@plasmohq/storage"
|
|
|
|
const progressHuman = (completed: number, total: number) => {
|
|
return ((completed / total) * 100).toFixed(0) + "%"
|
|
}
|
|
|
|
const clearBadge = () => {
|
|
chrome.action.setBadgeText({ text: "" })
|
|
chrome.action.setTitle({ title: "" })
|
|
}
|
|
const streamDownload = async (url: string, model: string) => {
|
|
url += "/api/pull"
|
|
const response = await fetch(url, {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json"
|
|
},
|
|
body: JSON.stringify({ model, stream: true })
|
|
})
|
|
|
|
const reader = response.body?.getReader()
|
|
|
|
const decoder = new TextDecoder()
|
|
|
|
let isSuccess = true
|
|
while (true) {
|
|
if (!reader) {
|
|
break
|
|
}
|
|
const { done, value } = await reader.read()
|
|
|
|
if (done) {
|
|
break
|
|
}
|
|
|
|
const text = decoder.decode(value)
|
|
try {
|
|
const json = JSON.parse(text.trim()) as {
|
|
status: string
|
|
total?: number
|
|
completed?: number
|
|
}
|
|
if (json.total && json.completed) {
|
|
chrome.action.setBadgeText({
|
|
text: progressHuman(json.completed, json.total)
|
|
})
|
|
chrome.action.setBadgeBackgroundColor({ color: "#0000FF" })
|
|
} else {
|
|
chrome.action.setBadgeText({ text: "🏋️♂️" })
|
|
chrome.action.setBadgeBackgroundColor({ color: "#FFFFFF" })
|
|
}
|
|
|
|
chrome.action.setTitle({ title: json.status })
|
|
|
|
if (json.status === "success") {
|
|
isSuccess = true
|
|
}
|
|
} catch (e) {
|
|
console.error(e)
|
|
}
|
|
}
|
|
|
|
if (isSuccess) {
|
|
chrome.action.setBadgeText({ text: "✅" })
|
|
chrome.action.setBadgeBackgroundColor({ color: "#00FF00" })
|
|
chrome.action.setTitle({ title: "Model pulled successfully" })
|
|
} else {
|
|
chrome.action.setBadgeText({ text: "❌" })
|
|
chrome.action.setBadgeBackgroundColor({ color: "#FF0000" })
|
|
chrome.action.setTitle({ title: "Model pull failed" })
|
|
}
|
|
|
|
setTimeout(() => {
|
|
clearBadge()
|
|
}, 5000)
|
|
}
|
|
export default defineBackground({
|
|
main() {
|
|
const storage = new Storage()
|
|
|
|
chrome.runtime.onMessage.addListener(async (message) => {
|
|
if (message.type === "sidepanel") {
|
|
chrome.tabs.query(
|
|
{ active: true, currentWindow: true },
|
|
async (tabs) => {
|
|
const tab = tabs[0]
|
|
chrome.sidePanel.open({
|
|
// tabId: tab.id!,
|
|
windowId: tab.windowId!
|
|
})
|
|
}
|
|
)
|
|
} else if (message.type === "pull_model") {
|
|
const ollamaURL = await getOllamaURL()
|
|
|
|
const isRunning = await isOllamaRunning()
|
|
|
|
if (!isRunning) {
|
|
chrome.action.setBadgeText({ text: "E" })
|
|
chrome.action.setBadgeBackgroundColor({ color: "#FF0000" })
|
|
chrome.action.setTitle({ title: "Ollama is not running" })
|
|
setTimeout(() => {
|
|
clearBadge()
|
|
}, 5000)
|
|
}
|
|
|
|
await streamDownload(ollamaURL, message.modelName)
|
|
}
|
|
})
|
|
|
|
chrome.action.onClicked.addListener((tab) => {
|
|
chrome.tabs.create({ url: chrome.runtime.getURL("options.html") })
|
|
})
|
|
|
|
chrome.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({
|
|
windowId: tab.windowId!
|
|
})
|
|
}
|
|
)
|
|
break
|
|
default:
|
|
break
|
|
}
|
|
})
|
|
|
|
chrome.contextMenus.create({
|
|
id: "open-side-panel-pa",
|
|
title: browser.i18n.getMessage("openSidePanelToChat"),
|
|
contexts: ["all"]
|
|
})
|
|
|
|
chrome.contextMenus.onClicked.addListener((info, tab) => {
|
|
if (info.menuItemId === "open-side-panel-pa") {
|
|
chrome.tabs.query(
|
|
{ active: true, currentWindow: true },
|
|
async (tabs) => {
|
|
const tab = tabs[0]
|
|
chrome.sidePanel.open({
|
|
tabId: tab.id!
|
|
})
|
|
}
|
|
)
|
|
}
|
|
})
|
|
},
|
|
persistent: true
|
|
})
|