securityos/scripts/preloadIcons.js

92 lines
2.7 KiB
JavaScript

const { readdirSync, readFileSync, writeFileSync, existsSync } = require("fs");
const { extname, join } = require("path");
const { parse } = require("ini");
const HOME = "/Users/Public";
const DESKTOP_PATH = `${HOME}/Desktop`;
const START_MENU_PATH = `${HOME}/Start Menu`;
const ICON_PATH = "/System/Icons";
const SHORTCUT_ICON = `${ICON_PATH}/shortcut.webp`;
const NEW_FOLDER_ICON = `${ICON_PATH}/new_folder.webp`;
const USER_ICON_PATH = `${HOME}/Icons`;
const ICON_CACHE = `${USER_ICON_PATH}/Cache`;
const YT_ICON_CACHE = `${ICON_CACHE}/YouTube`;
const ICON_CACHE_EXTENSION = ".cache";
const VLC_SUBICON = "/System/Icons/16x16/vlc.webp";
const isYouTubeUrl = (url) =>
url.includes("youtube.com/") || url.includes("youtu.be/");
const getYouTubeUrlId = (url) => {
try {
const { pathname, searchParams } = new URL(url);
return searchParams.get("v") || pathname.split("/").pop() || "";
} catch {
// URL parsing failed
}
return "";
};
const getPublicDirectoryIcons = (directory) => {
const isDesktop = directory === DESKTOP_PATH;
const baseDirectory = join("./public", directory);
return readdirSync(baseDirectory).reduce((icons, file) => {
if (extname(file) === ".url") {
const {
InternetShortcut: {
BaseURL: pid = "",
IconFile: icon = "",
URL: url = "",
},
} = parse(readFileSync(join(baseDirectory, file)).toString());
if (icon) icons.push(encodeURI(icon));
if (isDesktop) {
if (pid === "VideoPlayer") {
if (!icons.includes(VLC_SUBICON)) icons.push(encodeURI(VLC_SUBICON));
if (isYouTubeUrl(url)) {
const iconFileName = `/${getYouTubeUrlId(
url
)}${ICON_CACHE_EXTENSION}`;
if (
existsSync(join("./public", YT_ICON_CACHE, `${iconFileName}`))
) {
icons.push(encodeURI(`${YT_ICON_CACHE}${iconFileName}`));
}
}
}
const iconPath = url || `${directory}/${file}`;
const iconCacheFileName = `${iconPath}${ICON_CACHE_EXTENSION}`;
if (
extname(iconPath) &&
existsSync(join("./public", ICON_CACHE, `${iconCacheFileName}`))
) {
icons.push(encodeURI(`${ICON_CACHE}${iconCacheFileName}`));
}
}
}
return icons;
}, []);
};
writeFileSync(
"./public/.index/desktopIcons.json",
JSON.stringify([SHORTCUT_ICON, ...getPublicDirectoryIcons(DESKTOP_PATH)])
);
writeFileSync(
"./public/.index/startMenuIcons.json",
JSON.stringify([NEW_FOLDER_ICON, ...getPublicDirectoryIcons(START_MENU_PATH)])
);