oscar-surf/misskey
oscar-surf/misskey
0
0
Fork 0
Code Issues Activity

spec(aiscript): Mk:apiの呼び出しにエンドポイントごとのレートリミットを設定 (MisskeyIO#522)

Browse source
This commit is contained in:
まっちゃとーにゅ 2024-03-15 14:16:48 +09:00 • committed by GitHub
parent 9d1cdadccc
commit cd7ab5d0f9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 624 additions and 540 deletions
Download patch file Download diff file Expand all files Collapse all files

4
packages/frontend/src/scripts/aiscript/api.ts
View file

@ -11,6 +11,7 @@ import { miLocalStorage } from '@/local-storage.js';
import { customEmojis } from '@/custom-emojis.js';
import { url, lang } from '@/config.js';
import { nyaize } from '@/scripts/nyaize.js';
import { RateLimiter } from '@/scripts/rate-limiter.js';
export function aiScriptReadline(q: string): Promise<string> {
return new Promise(ok => {
@ -23,6 +24,8 @@ export function aiScriptReadline(q: string): Promise<string> {
}
export function createAiScriptEnv(opts) {
const rateLimiter = new RateLimiter<string>({ duration: 1000 * 15, max: 30 });
return {
USER_ID: $i ? values.STR($i.id) : values.NULL,
USER_NAME: $i ? values.STR($i.name) : values.NULL,
@ -55,6 +58,7 @@ export function createAiScriptEnv(opts) {
if (typeof token.value !== 'string') throw new Error('invalid token');
}
const actualToken: string|null = token?.value ?? opts.token ?? null;
if (!rateLimiter.hit(ep.value)) return values.ERROR('rate_limited', values.NULL);
return misskeyApi(ep.value, utils.valToJs(param), actualToken).then(res => {
return utils.jsToVal(res);
}, err => {