smueller/youtube-music
1
0
Fork 0
mirror of https://github.com/th-ch/youtube-music.git synced 2026-01-12 02:51:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(plugin): support dynamic plugin load / unload

Browse Source
This commit is contained in:
Su-Yong
2023-11-12 01:16:34 +09:00
parent 9c59f56aac
commit 2097f42efb
8 changed files with 87 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/plugins/utils/main/css.ts
View File

@ -2,9 +2,16 @@ import fs from 'node:fs';
type Unregister = () => void;
let isLoaded = false;
const cssToInject = new Map<string, ((unregister: Unregister) => void) | undefined>();
const cssToInjectFile = new Map<string, ((unregister: Unregister) => void) | undefined>();
export const injectCSS = (webContents: Electron.WebContents, css: string): Promise<Unregister> => {
export const injectCSS = async (webContents: Electron.WebContents, css: string): Promise<Unregister> => {
if (isLoaded) {
const key = await webContents.insertCSS(css);
return async () => await webContents.removeInsertedCSS(key);
}
return new Promise((resolve) => {
if (cssToInject.size === 0 && cssToInjectFile.size === 0) {
setupCssInjection(webContents);
@ -13,7 +20,12 @@ export const injectCSS = (webContents: Electron.WebContents, css: string): Promi
});
};
export const injectCSSAsFile = (webContents: Electron.WebContents, filepath: string): Promise<Unregister> => {
export const injectCSSAsFile = async (webContents: Electron.WebContents, filepath: string): Promise<Unregister> => {
if (isLoaded) {
const key = await webContents.insertCSS(fs.readFileSync(filepath, 'utf-8'));
return async () => await webContents.removeInsertedCSS(key);
}
return new Promise((resolve) => {
if (cssToInject.size === 0 && cssToInjectFile.size === 0) {
setupCssInjection(webContents);
@ -25,6 +37,8 @@ export const injectCSSAsFile = (webContents: Electron.WebContents, filepath: str
const setupCssInjection = (webContents: Electron.WebContents) => {
webContents.on('did-finish-load', () => {
isLoaded = true;
cssToInject.forEach(async (callback, css) => {
const key = await webContents.insertCSS(css);
const remove = async () => await webContents.removeInsertedCSS(key);