hippofish/packages/client/src/plugin.ts
2023-10-22 17:33:40 +09:00

182 lines
4.1 KiB
TypeScript

import { AiScript, utils, values } from "@syuilo/aiscript";
import { deserialize } from "@syuilo/aiscript/built/serializer";
import { jsToVal } from "@syuilo/aiscript/built/interpreter/util";
import { createAiScriptEnv } from "@/scripts/aiscript/api";
import { inputText } from "@/os";
import {
noteActions,
notePostInterruptors,
noteViewInterruptors,
postFormActions,
userActions,
} from "@/store";
const pluginContexts = new Map<string, AiScript>();
export function install(plugin) {
console.info("Plugin installed:", plugin.name, `v${plugin.version}`);
const aiscript = new AiScript(
createPluginEnv({
plugin,
storageKey: `plugins:${plugin.id}`,
}),
{
in: (q) => {
return new Promise((ok) => {
inputText({
title: q,
}).then(({ canceled, result: a }) => {
ok(a);
});
});
},
out: (value) => {
console.log(value);
},
log: (type, params) => {},
},
);
initPlugin({ plugin, aiscript });
aiscript.exec(deserialize(plugin.ast));
}
function createPluginEnv(opts) {
const config = new Map();
for (const [k, v] of Object.entries(opts.plugin.config || {})) {
config.set(
k,
jsToVal(
typeof opts.plugin.configData[k] !== "undefined"
? opts.plugin.configData[k]
: v.default,
),
);
}
return {
...createAiScriptEnv({ ...opts, token: opts.plugin.token }),
// #region Deprecated
"Mk:register_post_form_action": values.FN_NATIVE(([title, handler]) => {
registerPostFormAction({
pluginId: opts.plugin.id,
title: title.value,
handler,
});
}),
"Mk:register_user_action": values.FN_NATIVE(([title, handler]) => {
registerUserAction({
pluginId: opts.plugin.id,
title: title.value,
handler,
});
}),
"Mk:register_note_action": values.FN_NATIVE(([title, handler]) => {
registerNoteAction({
pluginId: opts.plugin.id,
title: title.value,
handler,
});
}),
// #endregion
"Plugin:register_post_form_action": values.FN_NATIVE(([title, handler]) => {
registerPostFormAction({
pluginId: opts.plugin.id,
title: title.value,
handler,
});
}),
"Plugin:register_user_action": values.FN_NATIVE(([title, handler]) => {
registerUserAction({
pluginId: opts.plugin.id,
title: title.value,
handler,
});
}),
"Plugin:register_note_action": values.FN_NATIVE(([title, handler]) => {
registerNoteAction({
pluginId: opts.plugin.id,
title: title.value,
handler,
});
}),
"Plugin:register_note_view_interruptor": values.FN_NATIVE(([handler]) => {
registerNoteViewInterruptor({
pluginId: opts.plugin.id,
handler,
});
}),
"Plugin:register_note_post_interruptor": values.FN_NATIVE(([handler]) => {
registerNotePostInterruptor({
pluginId: opts.plugin.id,
handler,
});
}),
"Plugin:open_url": values.FN_NATIVE(([url]) => {
window.open(url.value, "_blank");
}),
"Plugin:config": values.OBJ(config),
};
}
function initPlugin({ plugin, aiscript }) {
pluginContexts.set(plugin.id, aiscript);
}
function registerPostFormAction({ pluginId, title, handler }) {
postFormActions.push({
title,
handler: (form, update) => {
pluginContexts.get(pluginId).execFn(handler, [
utils.jsToVal(form),
values.FN_NATIVE(([key, value]) => {
update(key.value, value.value);
}),
]);
},
});
}
function registerUserAction({ pluginId, title, handler }) {
userActions.push({
title,
handler: (user) => {
pluginContexts.get(pluginId).execFn(handler, [utils.jsToVal(user)]);
},
});
}
function registerNoteAction({ pluginId, title, handler }) {
noteActions.push({
title,
handler: (note) => {
pluginContexts.get(pluginId).execFn(handler, [utils.jsToVal(note)]);
},
});
}
function registerNoteViewInterruptor({ pluginId, handler }) {
noteViewInterruptors.push({
handler: async (note) => {
return utils.valToJs(
await pluginContexts
.get(pluginId)
.execFn(handler, [utils.jsToVal(note)]),
);
},
});
}
function registerNotePostInterruptor({ pluginId, handler }) {
notePostInterruptors.push({
handler: async (note) => {
return utils.valToJs(
await pluginContexts
.get(pluginId)
.execFn(handler, [utils.jsToVal(note)]),
);
},
});
}