mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-23 18:10:26 +08:00
f83d9fc03c
feat: add filter mode and list functionality to selection assistant - Introduced new filter mode options (default, whitelist, blacklist) for the selection assistant. - Added methods to set and get filter mode and filter list in ConfigManager. - Enhanced SelectionService to manage filter mode and list, affecting text selection processing. - Updated UI components to allow users to configure filter settings. - Localized new filter settings in multiple languages.
253 lines
15 KiB
TypeScript
253 lines
15 KiB
TypeScript
import type { ExtractChunkData } from '@cherrystudio/embedjs-interfaces'
|
|
import { electronAPI } from '@electron-toolkit/preload'
|
|
import { IpcChannel } from '@shared/IpcChannel'
|
|
import { FileType, KnowledgeBaseParams, KnowledgeItem, MCPServer, Shortcut, WebDavConfig } from '@types'
|
|
import { contextBridge, ipcRenderer, OpenDialogOptions, shell, webUtils } from 'electron'
|
|
import { Notification } from 'src/renderer/src/types/notification'
|
|
import { CreateDirectoryOptions } from 'webdav'
|
|
|
|
import type { ActionItem } from '../renderer/src/types/selectionTypes'
|
|
|
|
// Custom APIs for renderer
|
|
const api = {
|
|
getAppInfo: () => ipcRenderer.invoke(IpcChannel.App_Info),
|
|
reload: () => ipcRenderer.invoke(IpcChannel.App_Reload),
|
|
setProxy: (proxy: string | undefined) => ipcRenderer.invoke(IpcChannel.App_Proxy, proxy),
|
|
checkForUpdate: () => ipcRenderer.invoke(IpcChannel.App_CheckForUpdate),
|
|
showUpdateDialog: () => ipcRenderer.invoke(IpcChannel.App_ShowUpdateDialog),
|
|
setLanguage: (lang: string) => ipcRenderer.invoke(IpcChannel.App_SetLanguage, lang),
|
|
setLaunchOnBoot: (isActive: boolean) => ipcRenderer.invoke(IpcChannel.App_SetLaunchOnBoot, isActive),
|
|
setLaunchToTray: (isActive: boolean) => ipcRenderer.invoke(IpcChannel.App_SetLaunchToTray, isActive),
|
|
setTray: (isActive: boolean) => ipcRenderer.invoke(IpcChannel.App_SetTray, isActive),
|
|
setTrayOnClose: (isActive: boolean) => ipcRenderer.invoke(IpcChannel.App_SetTrayOnClose, isActive),
|
|
setTheme: (theme: 'light' | 'dark' | 'auto') => ipcRenderer.invoke(IpcChannel.App_SetTheme, theme),
|
|
handleZoomFactor: (delta: number, reset: boolean = false) =>
|
|
ipcRenderer.invoke(IpcChannel.App_HandleZoomFactor, delta, reset),
|
|
setAutoUpdate: (isActive: boolean) => ipcRenderer.invoke(IpcChannel.App_SetAutoUpdate, isActive),
|
|
openWebsite: (url: string) => ipcRenderer.invoke(IpcChannel.Open_Website, url),
|
|
getCacheSize: () => ipcRenderer.invoke(IpcChannel.App_GetCacheSize),
|
|
clearCache: () => ipcRenderer.invoke(IpcChannel.App_ClearCache),
|
|
notification: {
|
|
send: (notification: Notification) => ipcRenderer.invoke(IpcChannel.Notification_Send, notification)
|
|
},
|
|
system: {
|
|
getDeviceType: () => ipcRenderer.invoke(IpcChannel.System_GetDeviceType),
|
|
getHostname: () => ipcRenderer.invoke(IpcChannel.System_GetHostname)
|
|
},
|
|
devTools: {
|
|
toggle: () => ipcRenderer.invoke(IpcChannel.System_ToggleDevTools)
|
|
},
|
|
zip: {
|
|
compress: (text: string) => ipcRenderer.invoke(IpcChannel.Zip_Compress, text),
|
|
decompress: (text: Buffer) => ipcRenderer.invoke(IpcChannel.Zip_Decompress, text)
|
|
},
|
|
backup: {
|
|
backup: (fileName: string, data: string, destinationPath?: string, skipBackupFile?: boolean) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_Backup, fileName, data, destinationPath, skipBackupFile),
|
|
restore: (backupPath: string) => ipcRenderer.invoke(IpcChannel.Backup_Restore, backupPath),
|
|
backupToWebdav: (data: string, webdavConfig: WebDavConfig) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_BackupToWebdav, data, webdavConfig),
|
|
restoreFromWebdav: (webdavConfig: WebDavConfig) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_RestoreFromWebdav, webdavConfig),
|
|
listWebdavFiles: (webdavConfig: WebDavConfig) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_ListWebdavFiles, webdavConfig),
|
|
checkConnection: (webdavConfig: WebDavConfig) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_CheckConnection, webdavConfig),
|
|
createDirectory: (webdavConfig: WebDavConfig, path: string, options?: CreateDirectoryOptions) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_CreateDirectory, webdavConfig, path, options),
|
|
deleteWebdavFile: (fileName: string, webdavConfig: WebDavConfig) =>
|
|
ipcRenderer.invoke(IpcChannel.Backup_DeleteWebdavFile, fileName, webdavConfig)
|
|
},
|
|
file: {
|
|
select: (options?: OpenDialogOptions) => ipcRenderer.invoke(IpcChannel.File_Select, options),
|
|
upload: (file: FileType) => ipcRenderer.invoke(IpcChannel.File_Upload, file),
|
|
delete: (fileId: string) => ipcRenderer.invoke(IpcChannel.File_Delete, fileId),
|
|
read: (fileId: string) => ipcRenderer.invoke(IpcChannel.File_Read, fileId),
|
|
clear: () => ipcRenderer.invoke(IpcChannel.File_Clear),
|
|
get: (filePath: string) => ipcRenderer.invoke(IpcChannel.File_Get, filePath),
|
|
create: (fileName: string) => ipcRenderer.invoke(IpcChannel.File_Create, fileName),
|
|
write: (filePath: string, data: Uint8Array | string) => ipcRenderer.invoke(IpcChannel.File_Write, filePath, data),
|
|
writeWithId: (id: string, content: string) => ipcRenderer.invoke(IpcChannel.File_WriteWithId, id, content),
|
|
open: (options?: OpenDialogOptions) => ipcRenderer.invoke(IpcChannel.File_Open, options),
|
|
openPath: (path: string) => ipcRenderer.invoke(IpcChannel.File_OpenPath, path),
|
|
save: (path: string, content: string | NodeJS.ArrayBufferView, options?: any) =>
|
|
ipcRenderer.invoke(IpcChannel.File_Save, path, content, options),
|
|
selectFolder: () => ipcRenderer.invoke(IpcChannel.File_SelectFolder),
|
|
saveImage: (name: string, data: string) => ipcRenderer.invoke(IpcChannel.File_SaveImage, name, data),
|
|
base64Image: (fileId: string) => ipcRenderer.invoke(IpcChannel.File_Base64Image, fileId),
|
|
saveBase64Image: (data: string) => ipcRenderer.invoke(IpcChannel.File_SaveBase64Image, data),
|
|
download: (url: string, isUseContentType?: boolean) =>
|
|
ipcRenderer.invoke(IpcChannel.File_Download, url, isUseContentType),
|
|
copy: (fileId: string, destPath: string) => ipcRenderer.invoke(IpcChannel.File_Copy, fileId, destPath),
|
|
binaryImage: (fileId: string) => ipcRenderer.invoke(IpcChannel.File_BinaryImage, fileId),
|
|
base64File: (fileId: string) => ipcRenderer.invoke(IpcChannel.File_Base64File, fileId),
|
|
getPathForFile: (file: File) => webUtils.getPathForFile(file)
|
|
},
|
|
fs: {
|
|
read: (path: string) => ipcRenderer.invoke(IpcChannel.Fs_Read, path)
|
|
},
|
|
export: {
|
|
toWord: (markdown: string, fileName: string) => ipcRenderer.invoke(IpcChannel.Export_Word, markdown, fileName)
|
|
},
|
|
openPath: (path: string) => ipcRenderer.invoke(IpcChannel.Open_Path, path),
|
|
shortcuts: {
|
|
update: (shortcuts: Shortcut[]) => ipcRenderer.invoke(IpcChannel.Shortcuts_Update, shortcuts)
|
|
},
|
|
knowledgeBase: {
|
|
create: (base: KnowledgeBaseParams) => ipcRenderer.invoke(IpcChannel.KnowledgeBase_Create, base),
|
|
reset: (base: KnowledgeBaseParams) => ipcRenderer.invoke(IpcChannel.KnowledgeBase_Reset, base),
|
|
delete: (id: string) => ipcRenderer.invoke(IpcChannel.KnowledgeBase_Delete, id),
|
|
add: ({
|
|
base,
|
|
item,
|
|
forceReload = false
|
|
}: {
|
|
base: KnowledgeBaseParams
|
|
item: KnowledgeItem
|
|
forceReload?: boolean
|
|
}) => ipcRenderer.invoke(IpcChannel.KnowledgeBase_Add, { base, item, forceReload }),
|
|
remove: ({ uniqueId, uniqueIds, base }: { uniqueId: string; uniqueIds: string[]; base: KnowledgeBaseParams }) =>
|
|
ipcRenderer.invoke(IpcChannel.KnowledgeBase_Remove, { uniqueId, uniqueIds, base }),
|
|
search: ({ search, base }: { search: string; base: KnowledgeBaseParams }) =>
|
|
ipcRenderer.invoke(IpcChannel.KnowledgeBase_Search, { search, base }),
|
|
rerank: ({ search, base, results }: { search: string; base: KnowledgeBaseParams; results: ExtractChunkData[] }) =>
|
|
ipcRenderer.invoke(IpcChannel.KnowledgeBase_Rerank, { search, base, results })
|
|
},
|
|
window: {
|
|
setMinimumSize: (width: number, height: number) =>
|
|
ipcRenderer.invoke(IpcChannel.Windows_SetMinimumSize, width, height),
|
|
resetMinimumSize: () => ipcRenderer.invoke(IpcChannel.Windows_ResetMinimumSize)
|
|
},
|
|
gemini: {
|
|
uploadFile: (file: FileType, { apiKey, baseURL }: { apiKey: string; baseURL: string }) =>
|
|
ipcRenderer.invoke(IpcChannel.Gemini_UploadFile, file, { apiKey, baseURL }),
|
|
base64File: (file: FileType) => ipcRenderer.invoke(IpcChannel.Gemini_Base64File, file),
|
|
retrieveFile: (file: FileType, apiKey: string) => ipcRenderer.invoke(IpcChannel.Gemini_RetrieveFile, file, apiKey),
|
|
listFiles: (apiKey: string) => ipcRenderer.invoke(IpcChannel.Gemini_ListFiles, apiKey),
|
|
deleteFile: (fileId: string, apiKey: string) => ipcRenderer.invoke(IpcChannel.Gemini_DeleteFile, fileId, apiKey)
|
|
},
|
|
config: {
|
|
set: (key: string, value: any, isNotify: boolean = false) =>
|
|
ipcRenderer.invoke(IpcChannel.Config_Set, key, value, isNotify),
|
|
get: (key: string) => ipcRenderer.invoke(IpcChannel.Config_Get, key)
|
|
},
|
|
miniWindow: {
|
|
show: () => ipcRenderer.invoke(IpcChannel.MiniWindow_Show),
|
|
hide: () => ipcRenderer.invoke(IpcChannel.MiniWindow_Hide),
|
|
close: () => ipcRenderer.invoke(IpcChannel.MiniWindow_Close),
|
|
toggle: () => ipcRenderer.invoke(IpcChannel.MiniWindow_Toggle),
|
|
setPin: (isPinned: boolean) => ipcRenderer.invoke(IpcChannel.MiniWindow_SetPin, isPinned)
|
|
},
|
|
aes: {
|
|
encrypt: (text: string, secretKey: string, iv: string) =>
|
|
ipcRenderer.invoke(IpcChannel.Aes_Encrypt, text, secretKey, iv),
|
|
decrypt: (encryptedData: string, iv: string, secretKey: string) =>
|
|
ipcRenderer.invoke(IpcChannel.Aes_Decrypt, encryptedData, iv, secretKey)
|
|
},
|
|
mcp: {
|
|
removeServer: (server: MCPServer) => ipcRenderer.invoke(IpcChannel.Mcp_RemoveServer, server),
|
|
restartServer: (server: MCPServer) => ipcRenderer.invoke(IpcChannel.Mcp_RestartServer, server),
|
|
stopServer: (server: MCPServer) => ipcRenderer.invoke(IpcChannel.Mcp_StopServer, server),
|
|
listTools: (server: MCPServer) => ipcRenderer.invoke(IpcChannel.Mcp_ListTools, server),
|
|
callTool: ({ server, name, args }: { server: MCPServer; name: string; args: any }) =>
|
|
ipcRenderer.invoke(IpcChannel.Mcp_CallTool, { server, name, args }),
|
|
listPrompts: (server: MCPServer) => ipcRenderer.invoke(IpcChannel.Mcp_ListPrompts, server),
|
|
getPrompt: ({ server, name, args }: { server: MCPServer; name: string; args?: Record<string, any> }) =>
|
|
ipcRenderer.invoke(IpcChannel.Mcp_GetPrompt, { server, name, args }),
|
|
listResources: (server: MCPServer) => ipcRenderer.invoke(IpcChannel.Mcp_ListResources, server),
|
|
getResource: ({ server, uri }: { server: MCPServer; uri: string }) =>
|
|
ipcRenderer.invoke(IpcChannel.Mcp_GetResource, { server, uri }),
|
|
getInstallInfo: () => ipcRenderer.invoke(IpcChannel.Mcp_GetInstallInfo),
|
|
checkMcpConnectivity: (server: any) => ipcRenderer.invoke(IpcChannel.Mcp_CheckConnectivity, server)
|
|
},
|
|
shell: {
|
|
openExternal: (url: string, options?: Electron.OpenExternalOptions) => shell.openExternal(url, options)
|
|
},
|
|
copilot: {
|
|
getAuthMessage: (headers?: Record<string, string>) =>
|
|
ipcRenderer.invoke(IpcChannel.Copilot_GetAuthMessage, headers),
|
|
getCopilotToken: (device_code: string, headers?: Record<string, string>) =>
|
|
ipcRenderer.invoke(IpcChannel.Copilot_GetCopilotToken, device_code, headers),
|
|
saveCopilotToken: (access_token: string) => ipcRenderer.invoke(IpcChannel.Copilot_SaveCopilotToken, access_token),
|
|
getToken: (headers?: Record<string, string>) => ipcRenderer.invoke(IpcChannel.Copilot_GetToken, headers),
|
|
logout: () => ipcRenderer.invoke(IpcChannel.Copilot_Logout),
|
|
getUser: (token: string) => ipcRenderer.invoke(IpcChannel.Copilot_GetUser, token)
|
|
},
|
|
// Binary related APIs
|
|
isBinaryExist: (name: string) => ipcRenderer.invoke(IpcChannel.App_IsBinaryExist, name),
|
|
getBinaryPath: (name: string) => ipcRenderer.invoke(IpcChannel.App_GetBinaryPath, name),
|
|
installUVBinary: () => ipcRenderer.invoke(IpcChannel.App_InstallUvBinary),
|
|
installBunBinary: () => ipcRenderer.invoke(IpcChannel.App_InstallBunBinary),
|
|
protocol: {
|
|
onReceiveData: (callback: (data: { url: string; params: any }) => void) => {
|
|
const listener = (_event: Electron.IpcRendererEvent, data: { url: string; params: any }) => {
|
|
callback(data)
|
|
}
|
|
ipcRenderer.on('protocol-data', listener)
|
|
return () => {
|
|
ipcRenderer.off('protocol-data', listener)
|
|
}
|
|
}
|
|
},
|
|
nutstore: {
|
|
getSSOUrl: () => ipcRenderer.invoke(IpcChannel.Nutstore_GetSsoUrl),
|
|
decryptToken: (token: string) => ipcRenderer.invoke(IpcChannel.Nutstore_DecryptToken, token),
|
|
getDirectoryContents: (token: string, path: string) =>
|
|
ipcRenderer.invoke(IpcChannel.Nutstore_GetDirectoryContents, token, path)
|
|
},
|
|
searchService: {
|
|
openSearchWindow: (uid: string) => ipcRenderer.invoke(IpcChannel.SearchWindow_Open, uid),
|
|
closeSearchWindow: (uid: string) => ipcRenderer.invoke(IpcChannel.SearchWindow_Close, uid),
|
|
openUrlInSearchWindow: (uid: string, url: string) => ipcRenderer.invoke(IpcChannel.SearchWindow_OpenUrl, uid, url)
|
|
},
|
|
webview: {
|
|
setOpenLinkExternal: (webviewId: number, isExternal: boolean) =>
|
|
ipcRenderer.invoke(IpcChannel.Webview_SetOpenLinkExternal, webviewId, isExternal)
|
|
},
|
|
storeSync: {
|
|
subscribe: () => ipcRenderer.invoke(IpcChannel.StoreSync_Subscribe),
|
|
unsubscribe: () => ipcRenderer.invoke(IpcChannel.StoreSync_Unsubscribe),
|
|
onUpdate: (action: any) => ipcRenderer.invoke(IpcChannel.StoreSync_OnUpdate, action)
|
|
},
|
|
selection: {
|
|
hideToolbar: () => ipcRenderer.invoke(IpcChannel.Selection_ToolbarHide),
|
|
writeToClipboard: (text: string) => ipcRenderer.invoke(IpcChannel.Selection_WriteToClipboard, text),
|
|
determineToolbarSize: (width: number, height: number) =>
|
|
ipcRenderer.invoke(IpcChannel.Selection_ToolbarDetermineSize, width, height),
|
|
setEnabled: (enabled: boolean) => ipcRenderer.invoke(IpcChannel.Selection_SetEnabled, enabled),
|
|
setTriggerMode: (triggerMode: string) => ipcRenderer.invoke(IpcChannel.Selection_SetTriggerMode, triggerMode),
|
|
setFollowToolbar: (isFollowToolbar: boolean) =>
|
|
ipcRenderer.invoke(IpcChannel.Selection_SetFollowToolbar, isFollowToolbar),
|
|
setFilterMode: (filterMode: string) => ipcRenderer.invoke(IpcChannel.Selection_SetFilterMode, filterMode),
|
|
setFilterList: (filterList: string[]) => ipcRenderer.invoke(IpcChannel.Selection_SetFilterList, filterList),
|
|
processAction: (actionItem: ActionItem) => ipcRenderer.invoke(IpcChannel.Selection_ProcessAction, actionItem),
|
|
closeActionWindow: () => ipcRenderer.invoke(IpcChannel.Selection_ActionWindowClose),
|
|
minimizeActionWindow: () => ipcRenderer.invoke(IpcChannel.Selection_ActionWindowMinimize),
|
|
pinActionWindow: (isPinned: boolean) => ipcRenderer.invoke(IpcChannel.Selection_ActionWindowPin, isPinned)
|
|
}
|
|
}
|
|
|
|
// Use `contextBridge` APIs to expose Electron APIs to
|
|
// renderer only if context isolation is enabled, otherwise
|
|
// just add to the DOM global.
|
|
if (process.contextIsolated) {
|
|
try {
|
|
contextBridge.exposeInMainWorld('electron', electronAPI)
|
|
contextBridge.exposeInMainWorld('api', api)
|
|
contextBridge.exposeInMainWorld('obsidian', {
|
|
getVaults: () => ipcRenderer.invoke(IpcChannel.Obsidian_GetVaults),
|
|
getFolders: (vaultName: string) => ipcRenderer.invoke(IpcChannel.Obsidian_GetFiles, vaultName),
|
|
getFiles: (vaultName: string) => ipcRenderer.invoke(IpcChannel.Obsidian_GetFiles, vaultName)
|
|
})
|
|
} catch (error) {
|
|
console.error(error)
|
|
}
|
|
} else {
|
|
// @ts-ignore (define in dts)
|
|
window.electron = electronAPI
|
|
// @ts-ignore (define in dts)
|
|
window.api = api
|
|
}
|
|
|
|
export type WindowApiType = typeof api
|