123 lines
4.0 KiB
TypeScript
123 lines
4.0 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import { utils, values } from '@syuilo/aiscript';
|
|
import * as Misskey from 'misskey-js';
|
|
import { parse as parseMfm, toString as mfmAstToString } from 'mfm-js';
|
|
import type { MfmNode } from 'mfm-js';
|
|
import * as os from '@/os.js';
|
|
import { misskeyApi } from '@/scripts/misskey-api.js';
|
|
import { $i } from '@/account.js';
|
|
import { miLocalStorage } from '@/local-storage.js';
|
|
import { customEmojis } from '@/custom-emojis.js';
|
|
import { deepClone } from '@/scripts/clone.js';
|
|
import { url, lang } from '@@/js/config.js';
|
|
|
|
export function aiScriptReadline(q: string): Promise<string> {
|
|
return new Promise(ok => {
|
|
os.inputText({
|
|
title: q,
|
|
}).then(({ result: a }) => {
|
|
ok(a ?? '');
|
|
});
|
|
});
|
|
}
|
|
|
|
export function createAiScriptEnv(opts) {
|
|
return {
|
|
USER_ID: $i ? values.STR($i.id) : values.NULL,
|
|
USER_NAME: $i ? values.STR($i.name) : values.NULL,
|
|
USER_USERNAME: $i ? values.STR($i.username) : values.NULL,
|
|
CUSTOM_EMOJIS: utils.jsToVal(customEmojis.value),
|
|
LOCALE: values.STR(lang),
|
|
SERVER_URL: values.STR(url),
|
|
'Mk:dialog': values.FN_NATIVE(async ([title, text, type]) => {
|
|
await os.alert({
|
|
type: type ? type.value : 'info',
|
|
title: title.value,
|
|
text: text.value,
|
|
});
|
|
return values.NULL;
|
|
}),
|
|
'Mk:confirm': values.FN_NATIVE(async ([title, text, type]) => {
|
|
const confirm = await os.confirm({
|
|
type: type ? type.value : 'question',
|
|
title: title.value,
|
|
text: text.value,
|
|
});
|
|
return confirm.canceled ? values.FALSE : values.TRUE;
|
|
}),
|
|
'Mk:api': values.FN_NATIVE(async ([ep, param, token]) => {
|
|
utils.assertString(ep);
|
|
if (ep.value.includes('://')) throw new Error('invalid endpoint');
|
|
if (token) {
|
|
utils.assertString(token);
|
|
// バグがあればundefinedもあり得るため念のため
|
|
if (typeof token.value !== 'string') throw new Error('invalid token');
|
|
}
|
|
const actualToken: string|null = token?.value ?? opts.token ?? null;
|
|
return misskeyApi(ep.value, utils.valToJs(param), actualToken).then(res => {
|
|
return utils.jsToVal(res);
|
|
}, err => {
|
|
return values.ERROR('request_failed', utils.jsToVal(err));
|
|
});
|
|
}),
|
|
/* セキュリティ上の問題があるため無効化
|
|
'Mk:apiExternal': values.FN_NATIVE(async ([host, ep, param, token]) => {
|
|
utils.assertString(host);
|
|
utils.assertString(ep);
|
|
if (token) utils.assertString(token);
|
|
return os.apiExternal(host.value, ep.value, utils.valToJs(param), token?.value).then(res => {
|
|
return utils.jsToVal(res);
|
|
}, err => {
|
|
return values.ERROR('request_failed', utils.jsToVal(err));
|
|
});
|
|
}),
|
|
*/
|
|
'Mk:save': values.FN_NATIVE(([key, value]) => {
|
|
utils.assertString(key);
|
|
miLocalStorage.setItem(`aiscript:${opts.storageKey}:${key.value}`, JSON.stringify(utils.valToJs(value)));
|
|
return values.NULL;
|
|
}),
|
|
'Mk:load': values.FN_NATIVE(([key]) => {
|
|
utils.assertString(key);
|
|
return utils.jsToVal(JSON.parse(miLocalStorage.getItem(`aiscript:${opts.storageKey}:${key.value}`)));
|
|
}),
|
|
'Mk:url': values.FN_NATIVE(() => {
|
|
return values.STR(window.location.href);
|
|
}),
|
|
'Mk:nyaize': values.FN_NATIVE(([text]) => {
|
|
utils.assertString(text);
|
|
return values.STR(Misskey.nyaize(text.value));
|
|
}),
|
|
'Mk:replaceMfm': values.FN_NATIVE(async ([text, fn], opts) => {
|
|
utils.assertString(text);
|
|
utils.assertFunction(fn);
|
|
|
|
const ast = parseMfm(text.value);
|
|
|
|
async function replaceText(ast: MfmNode[]) {
|
|
utils.assertFunction(fn);
|
|
|
|
return await Promise.all(ast.map(async (node) => {
|
|
const out = deepClone(node);
|
|
if (out.type === 'text') {
|
|
const res = (await opts.topCall(fn, [values.STR(out.props.text)]));
|
|
utils.assertString(res);
|
|
out.props.text = res.value;
|
|
} else if (out.type !== 'plain' && 'children' in out && out.children != null && out.children.length > 0) {
|
|
out.children = await replaceText(out.children);
|
|
}
|
|
return out;
|
|
}));
|
|
}
|
|
|
|
const afterAst = await replaceText(ast);
|
|
|
|
return values.STR(mfmAstToString(afterAst));
|
|
}),
|
|
};
|
|
}
|