mirror of
https://github.com/th-ch/youtube-music.git
synced 2026-01-11 10:31:47 +00:00
82 lines
3.1 KiB
TypeScript
82 lines
3.1 KiB
TypeScript
// Used for caching
|
|
import path from 'node:path';
|
|
import fs, { promises } from 'node:fs';
|
|
|
|
import { ElectronBlocker } from '@ghostery/adblocker-electron';
|
|
import { app, net } from 'electron';
|
|
|
|
const SOURCES = [
|
|
'https://raw.githubusercontent.com/kbinani/adblock-youtube-ads/master/signed.txt',
|
|
// UBlock Origin
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/filters.txt',
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/quick-fixes.txt',
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/unbreak.txt',
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/filters-2020.txt',
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/filters-2021.txt',
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/filters-2022.txt',
|
|
'https://raw.githubusercontent.com/ghostery/adblocker/master/packages/adblocker/assets/ublock-origin/filters-2023.txt',
|
|
// Fanboy Annoyances
|
|
'https://secure.fanboy.co.nz/fanboy-annoyance_ubo.txt',
|
|
// AdGuard
|
|
'https://filters.adtidy.org/extension/ublock/filters/122_optimized.txt',
|
|
];
|
|
|
|
let blocker: ElectronBlocker | undefined;
|
|
|
|
export const loadAdBlockerEngine = async (
|
|
session: Electron.Session | undefined = undefined,
|
|
cache: boolean = true,
|
|
additionalBlockLists: string[] = [],
|
|
disableDefaultLists: boolean | unknown[] = false,
|
|
) => {
|
|
// Only use cache if no additional blocklists are passed
|
|
const cacheDirectory = path.join(app.getPath('userData'), 'adblock_cache');
|
|
if (!fs.existsSync(cacheDirectory)) {
|
|
fs.mkdirSync(cacheDirectory);
|
|
}
|
|
const cachingOptions =
|
|
cache && additionalBlockLists.length === 0
|
|
? {
|
|
path: path.join(cacheDirectory, 'adblocker-engine.bin'),
|
|
read: promises.readFile,
|
|
write: promises.writeFile,
|
|
}
|
|
: undefined;
|
|
const lists = [
|
|
...((disableDefaultLists && !Array.isArray(disableDefaultLists)) ||
|
|
(Array.isArray(disableDefaultLists) && disableDefaultLists.length > 0)
|
|
? []
|
|
: SOURCES),
|
|
...additionalBlockLists,
|
|
];
|
|
|
|
try {
|
|
blocker = await ElectronBlocker.fromLists(
|
|
(url: string) => net.fetch(url),
|
|
lists,
|
|
{
|
|
enableCompression: true,
|
|
// When generating the engine for caching, do not load network filters
|
|
// So that enhancing the session works as expected
|
|
// Allowing to define multiple webRequest listeners
|
|
loadNetworkFilters: session !== undefined,
|
|
},
|
|
cachingOptions,
|
|
);
|
|
if (session) {
|
|
blocker.enableBlockingInSession(session);
|
|
}
|
|
} catch (error) {
|
|
console.error('Error loading adBlocker engine', error);
|
|
}
|
|
};
|
|
|
|
export const unloadAdBlockerEngine = (session: Electron.Session) => {
|
|
if (blocker) {
|
|
blocker.disableBlockingInSession(session);
|
|
}
|
|
};
|
|
|
|
export const isBlockerEnabled = (session: Electron.Session) =>
|
|
blocker !== undefined && blocker.isBlockingEnabled(session);
|