137 lines
4.2 KiB
TypeScript
137 lines
4.2 KiB
TypeScript
import { Indexable, PluginManifest, Plugin } from "@types";
|
|
import { navigation } from "@metro/common";
|
|
import { awaitSyncWrapper, createStorage, wrapSync } from "@lib/storage";
|
|
import logger from "@lib/logger";
|
|
import Subpage from "@ui/settings/components/Subpage";
|
|
|
|
// TODO: Properly implement hash-based updating
|
|
|
|
type EvaledPlugin = {
|
|
onLoad?(): void;
|
|
onUnload(): void;
|
|
settings: JSX.Element;
|
|
};
|
|
|
|
export const plugins = wrapSync(createStorage<Indexable<Plugin>>("VENDETTA_PLUGINS"));
|
|
const loadedPlugins: Indexable<EvaledPlugin> = {};
|
|
|
|
export async function fetchPlugin(id: string, enabled = true) {
|
|
if (!id.endsWith("/")) id += "/";
|
|
if (typeof id !== "string" || id in plugins) throw new Error("Plugin ID invalid or taken");
|
|
|
|
let pluginManifest: PluginManifest;
|
|
|
|
try {
|
|
pluginManifest = await (await fetch(id + "manifest.json", { cache: "no-store" })).json();
|
|
} catch {
|
|
throw new Error(`Failed to fetch manifest for ${id}`);
|
|
}
|
|
|
|
let pluginJs: string;
|
|
|
|
// TODO: Remove duplicate error if possible
|
|
try {
|
|
// by polymanifest spec, plugins should always specify their main file, but just in case
|
|
pluginJs = await (await fetch(id + (pluginManifest.main || "index.js"), { cache: "no-store" })).text();
|
|
} catch {
|
|
throw new Error(`Failed to fetch JS for ${id}`);
|
|
}
|
|
|
|
if (pluginJs.length === 0) throw new Error(`Failed to fetch JS for ${id}`);
|
|
|
|
plugins[id] = {
|
|
id: id,
|
|
manifest: pluginManifest,
|
|
enabled: false,
|
|
update: true,
|
|
js: pluginJs,
|
|
};
|
|
|
|
if (enabled) await startPlugin(id);
|
|
}
|
|
|
|
export async function evalPlugin(plugin: Plugin) {
|
|
const vendettaForPlugins = {
|
|
...window.vendetta,
|
|
plugin: {
|
|
manifest: plugin.manifest,
|
|
// Wrapping this with wrapSync is NOT an option.
|
|
storage: await createStorage<Indexable<any>>(plugin.id),
|
|
showSettings: () => showSettings(plugin),
|
|
}
|
|
};
|
|
const pluginString = `vendetta=>{return ${plugin.js}}\n//# sourceURL=${plugin.id}`;
|
|
|
|
const raw = (0, eval)(pluginString)(vendettaForPlugins);
|
|
const ret = typeof raw == "function" ? raw() : raw;
|
|
return ret.default || ret;
|
|
}
|
|
|
|
export async function startPlugin(id: string) {
|
|
const plugin = plugins[id];
|
|
if (!plugin) throw new Error("Attempted to start non-existent plugin");
|
|
|
|
try {
|
|
const pluginRet: EvaledPlugin = await evalPlugin(plugin);
|
|
loadedPlugins[id] = pluginRet;
|
|
pluginRet.onLoad?.();
|
|
plugin.enabled = true;
|
|
} catch(e) {
|
|
logger.error(`Plugin ${plugin.id} errored whilst loading, and will be unloaded`, e);
|
|
|
|
try {
|
|
loadedPlugins[plugin.id]?.onUnload?.();
|
|
} catch(e2) {
|
|
logger.error(`Plugin ${plugin.id} errored whilst unloading`, e2);
|
|
}
|
|
|
|
delete loadedPlugins[id];
|
|
plugin.enabled = false;
|
|
}
|
|
}
|
|
|
|
export function stopPlugin(id: string, disable = true) {
|
|
const plugin = plugins[id];
|
|
const pluginRet = loadedPlugins[id];
|
|
if (!plugin) throw new Error("Attempted to stop non-existent plugin");
|
|
if (!pluginRet) throw new Error("Attempted to stop a non-started plugin");
|
|
|
|
try {
|
|
loadedPlugins[plugin.id]?.onUnload?.();
|
|
} catch(e) {
|
|
logger.error(`Plugin ${plugin.id} errored whilst unloading`, e);
|
|
}
|
|
|
|
delete loadedPlugins[id];
|
|
disable && (plugin.enabled = false);
|
|
}
|
|
|
|
export function removePlugin(id: string) {
|
|
const plugin = plugins[id];
|
|
if (plugin.enabled) stopPlugin(id);
|
|
delete plugins[id];
|
|
}
|
|
|
|
export async function initPlugins() {
|
|
await awaitSyncWrapper(plugins);
|
|
|
|
const allIds = Object.keys(plugins);
|
|
await Promise.allSettled(allIds.map((pl) => fetchPlugin(pl, false)));
|
|
for (const pl of allIds.filter((pl) => plugins[pl].enabled)) startPlugin(pl);
|
|
|
|
return stopAllPlugins;
|
|
}
|
|
|
|
const stopAllPlugins = () => Object.keys(plugins).forEach(p => stopPlugin(p, false));
|
|
|
|
export const getSettings = (id: string) => loadedPlugins[id]?.settings;
|
|
|
|
export function showSettings(plugin: Plugin) {
|
|
const settings = getSettings(plugin.id);
|
|
if (!settings) return logger.error(`Plugin ${plugin.id} is not loaded or has no settings`);
|
|
|
|
navigation.push(Subpage, {
|
|
name: plugin.manifest.name,
|
|
children: settings,
|
|
});
|
|
}
|