Refactoring
This commit is contained in:
syuilo
parent
3444b9c9c8
commit
c3cd6ad2d2
@ -1,543 +0,0 @@
|
|||||||
/**
|
|
||||||
* AiScript
|
|
||||||
* evaluator & type checker
|
|
||||||
*/
|
|
||||||
|
|
||||||
import autobind from 'autobind-decorator';
|
|
||||||
import * as seedrandom from 'seedrandom';
|
|
||||||
|
|
||||||
import {
|
|
||||||
faSuperscript,
|
|
||||||
faAlignLeft,
|
|
||||||
faShareAlt,
|
|
||||||
faSquareRootAlt,
|
|
||||||
faPlus,
|
|
||||||
faMinus,
|
|
||||||
faTimes,
|
|
||||||
faDivide,
|
|
||||||
faList,
|
|
||||||
faQuoteRight,
|
|
||||||
faEquals,
|
|
||||||
faGreaterThan,
|
|
||||||
faLessThan,
|
|
||||||
faGreaterThanEqual,
|
|
||||||
faLessThanEqual,
|
|
||||||
faExclamation,
|
|
||||||
faNotEqual,
|
|
||||||
faDice,
|
|
||||||
faSortNumericUp,
|
|
||||||
faExchangeAlt,
|
|
||||||
faRecycle,
|
|
||||||
} from '@fortawesome/free-solid-svg-icons';
|
|
||||||
import { faFlag } from '@fortawesome/free-regular-svg-icons';
|
|
||||||
|
|
||||||
import { version } from '../../config';
|
|
||||||
|
|
||||||
export type Block<V = any> = {
|
|
||||||
id: string;
|
|
||||||
type: string;
|
|
||||||
args: Block[];
|
|
||||||
value: V;
|
|
||||||
};
|
|
||||||
|
|
||||||
type FnBlock = Block<{
|
|
||||||
slots: {
|
|
||||||
name: string;
|
|
||||||
type: Type;
|
|
||||||
}[];
|
|
||||||
expression: Block;
|
|
||||||
}>;
|
|
||||||
|
|
||||||
export type Variable = Block & {
|
|
||||||
name: string;
|
|
||||||
};
|
|
||||||
|
|
||||||
type Type = 'string' | 'number' | 'boolean' | 'stringArray';
|
|
||||||
|
|
||||||
type TypeError = {
|
|
||||||
arg: number;
|
|
||||||
expect: Type;
|
|
||||||
actual: Type;
|
|
||||||
};
|
|
||||||
|
|
||||||
const funcDefs = {
|
|
||||||
if: { in: ['boolean', 0, 0], out: 0, category: 'flow', icon: faShareAlt, },
|
|
||||||
for: { in: ['number', 'function'], out: null, category: 'flow', icon: faRecycle, },
|
|
||||||
not: { in: ['boolean'], out: 'boolean', category: 'logical', icon: faFlag, },
|
|
||||||
or: { in: ['boolean', 'boolean'], out: 'boolean', category: 'logical', icon: faFlag, },
|
|
||||||
and: { in: ['boolean', 'boolean'], out: 'boolean', category: 'logical', icon: faFlag, },
|
|
||||||
add: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faPlus, },
|
|
||||||
subtract: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faMinus, },
|
|
||||||
multiply: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faTimes, },
|
|
||||||
divide: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faDivide, },
|
|
||||||
eq: { in: [0, 0], out: 'boolean', category: 'comparison', icon: faEquals, },
|
|
||||||
notEq: { in: [0, 0], out: 'boolean', category: 'comparison', icon: faNotEqual, },
|
|
||||||
gt: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faGreaterThan, },
|
|
||||||
lt: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faLessThan, },
|
|
||||||
gtEq: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faGreaterThanEqual, },
|
|
||||||
ltEq: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faLessThanEqual, },
|
|
||||||
strLen: { in: ['string'], out: 'number', category: 'text', icon: faQuoteRight, },
|
|
||||||
strPick: { in: ['string', 'number'], out: 'string', category: 'text', icon: faQuoteRight, },
|
|
||||||
strReplace: { in: ['string', 'string', 'string'], out: 'string', category: 'text', icon: faQuoteRight, },
|
|
||||||
strReverse: { in: ['string'], out: 'string', category: 'text', icon: faQuoteRight, },
|
|
||||||
join: { in: ['stringArray', 'string'], out: 'string', category: 'text', icon: faQuoteRight, },
|
|
||||||
stringToNumber: { in: ['string'], out: 'number', category: 'convert', icon: faExchangeAlt, },
|
|
||||||
numberToString: { in: ['number'], out: 'string', category: 'convert', icon: faExchangeAlt, },
|
|
||||||
splitStrByLine: { in: ['string'], out: 'stringArray', category: 'convert', icon: faExchangeAlt, },
|
|
||||||
rannum: { in: ['number', 'number'], out: 'number', category: 'random', icon: faDice, },
|
|
||||||
dailyRannum: { in: ['number', 'number'], out: 'number', category: 'random', icon: faDice, },
|
|
||||||
seedRannum: { in: [null, 'number', 'number'], out: 'number', category: 'random', icon: faDice, },
|
|
||||||
random: { in: ['number'], out: 'boolean', category: 'random', icon: faDice, },
|
|
||||||
dailyRandom: { in: ['number'], out: 'boolean', category: 'random', icon: faDice, },
|
|
||||||
seedRandom: { in: [null, 'number'], out: 'boolean', category: 'random', icon: faDice, },
|
|
||||||
randomPick: { in: [0], out: 0, category: 'random', icon: faDice, },
|
|
||||||
dailyRandomPick: { in: [0], out: 0, category: 'random', icon: faDice, },
|
|
||||||
seedRandomPick: { in: [null, 0], out: 0, category: 'random', icon: faDice, },
|
|
||||||
};
|
|
||||||
|
|
||||||
const literalDefs = {
|
|
||||||
text: { out: 'string', category: 'value', icon: faQuoteRight, },
|
|
||||||
multiLineText: { out: 'string', category: 'value', icon: faAlignLeft, },
|
|
||||||
textList: { out: 'stringArray', category: 'value', icon: faList, },
|
|
||||||
number: { out: 'number', category: 'value', icon: faSortNumericUp, },
|
|
||||||
ref: { out: null, category: 'value', icon: faSuperscript, },
|
|
||||||
fn: { out: 'function', category: 'value', icon: faSuperscript, },
|
|
||||||
};
|
|
||||||
|
|
||||||
const blockDefs = [
|
|
||||||
...Object.entries(literalDefs).map(([k, v]) => ({
|
|
||||||
type: k, out: v.out, category: v.category, icon: v.icon
|
|
||||||
})),
|
|
||||||
...Object.entries(funcDefs).map(([k, v]) => ({
|
|
||||||
type: k, out: v.out, category: v.category, icon: v.icon
|
|
||||||
}))
|
|
||||||
];
|
|
||||||
|
|
||||||
function isFnBlock(block: Block): block is FnBlock {
|
|
||||||
return block.type === 'fn';
|
|
||||||
}
|
|
||||||
|
|
||||||
type PageVar = { name: string; value: any; type: Type; };
|
|
||||||
|
|
||||||
const envVarsDef = {
|
|
||||||
AI: 'string',
|
|
||||||
URL: 'string',
|
|
||||||
VERSION: 'string',
|
|
||||||
LOGIN: 'boolean',
|
|
||||||
NAME: 'string',
|
|
||||||
USERNAME: 'string',
|
|
||||||
USERID: 'string',
|
|
||||||
NOTES_COUNT: 'number',
|
|
||||||
FOLLOWERS_COUNT: 'number',
|
|
||||||
FOLLOWING_COUNT: 'number',
|
|
||||||
IS_CAT: 'boolean',
|
|
||||||
MY_NOTES_COUNT: 'number',
|
|
||||||
MY_FOLLOWERS_COUNT: 'number',
|
|
||||||
MY_FOLLOWING_COUNT: 'number',
|
|
||||||
SEED: null,
|
|
||||||
YMD: 'string',
|
|
||||||
};
|
|
||||||
|
|
||||||
class AiScriptError extends Error {
|
|
||||||
public info?: any;
|
|
||||||
|
|
||||||
constructor(message: string, info?: any) {
|
|
||||||
super(message);
|
|
||||||
|
|
||||||
this.info = info;
|
|
||||||
|
|
||||||
// Maintains proper stack trace for where our error was thrown (only available on V8)
|
|
||||||
if (Error.captureStackTrace) {
|
|
||||||
Error.captureStackTrace(this, AiScriptError);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
class Scope {
|
|
||||||
private layerdStates: Record<string, any>[];
|
|
||||||
public name: string;
|
|
||||||
|
|
||||||
constructor(layerdStates: Scope['layerdStates'], name?: Scope['name']) {
|
|
||||||
this.layerdStates = layerdStates;
|
|
||||||
this.name = name || 'anonymous';
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public createChildScope(states: Record<string, any>, name?: Scope['name']): Scope {
|
|
||||||
const layer = [states, ...this.layerdStates];
|
|
||||||
return new Scope(layer, name);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 指定した名前の変数の値を取得します
|
|
||||||
* @param name 変数名
|
|
||||||
*/
|
|
||||||
@autobind
|
|
||||||
public getState(name: string): any {
|
|
||||||
for (const later of this.layerdStates) {
|
|
||||||
const state = later[name];
|
|
||||||
if (state !== undefined) {
|
|
||||||
return state;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
throw new AiScriptError(
|
|
||||||
`No such variable '${name}' in scope '${this.name}'`, {
|
|
||||||
scope: this.layerdStates
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export class AiScript {
|
|
||||||
private variables: Variable[];
|
|
||||||
private pageVars: PageVar[];
|
|
||||||
private envVars: Record<keyof typeof envVarsDef, any>;
|
|
||||||
|
|
||||||
public static envVarsDef = envVarsDef;
|
|
||||||
public static blockDefs = blockDefs;
|
|
||||||
public static funcDefs = funcDefs;
|
|
||||||
private opts: {
|
|
||||||
randomSeed?: string; user?: any; visitor?: any; page?: any; url?: string;
|
|
||||||
};
|
|
||||||
|
|
||||||
constructor(variables: Variable[] = [], pageVars: PageVar[] = [], opts: AiScript['opts'] = {}) {
|
|
||||||
this.variables = variables;
|
|
||||||
this.pageVars = pageVars;
|
|
||||||
this.opts = opts;
|
|
||||||
|
|
||||||
const date = new Date();
|
|
||||||
|
|
||||||
this.envVars = {
|
|
||||||
AI: 'kawaii',
|
|
||||||
VERSION: version,
|
|
||||||
URL: opts.page ? `${opts.url}/@${opts.page.user.username}/pages/${opts.page.name}` : '',
|
|
||||||
LOGIN: opts.visitor != null,
|
|
||||||
NAME: opts.visitor ? opts.visitor.name : '',
|
|
||||||
USERNAME: opts.visitor ? opts.visitor.username : '',
|
|
||||||
USERID: opts.visitor ? opts.visitor.id : '',
|
|
||||||
NOTES_COUNT: opts.visitor ? opts.visitor.notesCount : 0,
|
|
||||||
FOLLOWERS_COUNT: opts.visitor ? opts.visitor.followersCount : 0,
|
|
||||||
FOLLOWING_COUNT: opts.visitor ? opts.visitor.followingCount : 0,
|
|
||||||
IS_CAT: opts.visitor ? opts.visitor.isCat : false,
|
|
||||||
MY_NOTES_COUNT: opts.user ? opts.user.notesCount : 0,
|
|
||||||
MY_FOLLOWERS_COUNT: opts.user ? opts.user.followersCount : 0,
|
|
||||||
MY_FOLLOWING_COUNT: opts.user ? opts.user.followingCount : 0,
|
|
||||||
SEED: opts.randomSeed ? opts.randomSeed : '',
|
|
||||||
YMD: `${date.getFullYear()}/${date.getMonth() + 1}/${date.getDate()}`
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public injectVars(vars: Variable[]) {
|
|
||||||
this.variables = vars;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public injectPageVars(pageVars: PageVar[]) {
|
|
||||||
this.pageVars = pageVars;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public updatePageVar(name: string, value: any) {
|
|
||||||
this.pageVars.find(v => v.name === name).value = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public updateRandomSeed(seed: string) {
|
|
||||||
this.opts.randomSeed = seed;
|
|
||||||
this.envVars.SEED = seed;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public static isLiteralBlock(v: Block) {
|
|
||||||
if (v.type === null) return true;
|
|
||||||
if (literalDefs[v.type]) return true;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public typeCheck(v: Block): TypeError | null {
|
|
||||||
if (AiScript.isLiteralBlock(v)) return null;
|
|
||||||
|
|
||||||
const def = AiScript.funcDefs[v.type];
|
|
||||||
if (def == null) {
|
|
||||||
throw new Error('Unknown type: ' + v.type);
|
|
||||||
}
|
|
||||||
|
|
||||||
const generic: Type[] = [];
|
|
||||||
|
|
||||||
for (let i = 0; i < def.in.length; i++) {
|
|
||||||
const arg = def.in[i];
|
|
||||||
const type = this.typeInference(v.args[i]);
|
|
||||||
if (type === null) continue;
|
|
||||||
|
|
||||||
if (typeof arg === 'number') {
|
|
||||||
if (generic[arg] === undefined) {
|
|
||||||
generic[arg] = type;
|
|
||||||
} else if (type !== generic[arg]) {
|
|
||||||
return {
|
|
||||||
arg: i,
|
|
||||||
expect: generic[arg],
|
|
||||||
actual: type
|
|
||||||
};
|
|
||||||
}
|
|
||||||
} else if (type !== arg) {
|
|
||||||
return {
|
|
||||||
arg: i,
|
|
||||||
expect: arg,
|
|
||||||
actual: type
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public getExpectedType(v: Block, slot: number): Type | null {
|
|
||||||
const def = AiScript.funcDefs[v.type];
|
|
||||||
if (def == null) {
|
|
||||||
throw new Error('Unknown type: ' + v.type);
|
|
||||||
}
|
|
||||||
|
|
||||||
const generic: Type[] = [];
|
|
||||||
|
|
||||||
for (let i = 0; i < def.in.length; i++) {
|
|
||||||
const arg = def.in[i];
|
|
||||||
const type = this.typeInference(v.args[i]);
|
|
||||||
if (type === null) continue;
|
|
||||||
|
|
||||||
if (typeof arg === 'number') {
|
|
||||||
if (generic[arg] === undefined) {
|
|
||||||
generic[arg] = type;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (typeof def.in[slot] === 'number') {
|
|
||||||
return generic[def.in[slot]] || null;
|
|
||||||
} else {
|
|
||||||
return def.in[slot];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public typeInference(v: Block): Type | null {
|
|
||||||
if (v.type === null) return null;
|
|
||||||
if (v.type === 'text') return 'string';
|
|
||||||
if (v.type === 'multiLineText') return 'string';
|
|
||||||
if (v.type === 'textList') return 'stringArray';
|
|
||||||
if (v.type === 'number') return 'number';
|
|
||||||
if (v.type === 'ref') {
|
|
||||||
const variable = this.variables.find(va => va.name === v.value);
|
|
||||||
if (variable) {
|
|
||||||
return this.typeInference(variable);
|
|
||||||
}
|
|
||||||
|
|
||||||
const pageVar = this.pageVars.find(va => va.name === v.value);
|
|
||||||
if (pageVar) {
|
|
||||||
return pageVar.type;
|
|
||||||
}
|
|
||||||
|
|
||||||
const envVar = AiScript.envVarsDef[v.value];
|
|
||||||
if (envVar !== undefined) {
|
|
||||||
return envVar;
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
if (v.type === 'fn') return null; // todo
|
|
||||||
if (v.type.startsWith('fn:')) return null; // todo
|
|
||||||
|
|
||||||
const generic: Type[] = [];
|
|
||||||
|
|
||||||
const def = AiScript.funcDefs[v.type];
|
|
||||||
|
|
||||||
for (let i = 0; i < def.in.length; i++) {
|
|
||||||
const arg = def.in[i];
|
|
||||||
if (typeof arg === 'number') {
|
|
||||||
const type = this.typeInference(v.args[i]);
|
|
||||||
|
|
||||||
if (generic[arg] === undefined) {
|
|
||||||
generic[arg] = type;
|
|
||||||
} else {
|
|
||||||
if (type !== generic[arg]) {
|
|
||||||
generic[arg] = null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (typeof def.out === 'number') {
|
|
||||||
return generic[def.out];
|
|
||||||
} else {
|
|
||||||
return def.out;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public getVarsByType(type: Type | null): Variable[] {
|
|
||||||
if (type == null) return this.variables;
|
|
||||||
return this.variables.filter(x => (this.typeInference(x) === null) || (this.typeInference(x) === type));
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public getVarByName(name: string): Variable {
|
|
||||||
return this.variables.find(x => x.name === name);
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public getEnvVarsByType(type: Type | null): string[] {
|
|
||||||
if (type == null) return Object.keys(AiScript.envVarsDef);
|
|
||||||
return Object.entries(AiScript.envVarsDef).filter(([k, v]) => v === null || type === v).map(([k, v]) => k);
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public getPageVarsByType(type: Type | null): string[] {
|
|
||||||
if (type == null) return this.pageVars.map(v => v.name);
|
|
||||||
return this.pageVars.filter(v => type === v.type).map(v => v.name);
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
private interpolate(str: string, scope: Scope) {
|
|
||||||
return str.replace(/\{(.+?)\}/g, match => {
|
|
||||||
const v = scope.getState(match.slice(1, -1).trim());
|
|
||||||
return v == null ? 'NULL' : v.toString();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public evaluateVars(): Record<string, any> {
|
|
||||||
const values: Record<string, any> = {};
|
|
||||||
|
|
||||||
for (const [k, v] of Object.entries(this.envVars)) {
|
|
||||||
values[k] = v;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const v of this.pageVars) {
|
|
||||||
values[v.name] = v.value;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const v of this.variables) {
|
|
||||||
values[v.name] = this.evaluate(v, new Scope([values]));
|
|
||||||
}
|
|
||||||
|
|
||||||
return values;
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
private evaluate(block: Block, scope: Scope): any {
|
|
||||||
if (block.type === null) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (block.type === 'number') {
|
|
||||||
return parseInt(block.value, 10);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (block.type === 'text' || block.type === 'multiLineText') {
|
|
||||||
return this.interpolate(block.value || '', scope);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (block.type === 'textList') {
|
|
||||||
return block.value.trim().split('\n');
|
|
||||||
}
|
|
||||||
|
|
||||||
if (block.type === 'ref') {
|
|
||||||
return scope.getState(block.value);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (isFnBlock(block)) { // ユーザー関数定義
|
|
||||||
return {
|
|
||||||
slots: block.value.slots.map(x => x.name),
|
|
||||||
exec: slotArg => {
|
|
||||||
return this.evaluate(block.value.expression, scope.createChildScope(slotArg, block.id));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
if (block.type.startsWith('fn:')) { // ユーザー関数呼び出し
|
|
||||||
const fnName = block.type.split(':')[1];
|
|
||||||
const fn = scope.getState(fnName);
|
|
||||||
const args = {};
|
|
||||||
for (let i = 0; i < fn.slots.length; i++) {
|
|
||||||
const name = fn.slots[i];
|
|
||||||
args[name] = this.evaluate(block.args[i], scope);
|
|
||||||
}
|
|
||||||
return fn.exec(args);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (block.args === undefined) return null;
|
|
||||||
|
|
||||||
const date = new Date();
|
|
||||||
const day = `${this.opts.visitor ? this.opts.visitor.id : ''} ${date.getFullYear()}/${date.getMonth() + 1}/${date.getDate()}`;
|
|
||||||
|
|
||||||
const funcs: { [p in keyof typeof funcDefs]: any } = {
|
|
||||||
not: (a) => !a,
|
|
||||||
eq: (a, b) => a === b,
|
|
||||||
notEq: (a, b) => a !== b,
|
|
||||||
gt: (a, b) => a > b,
|
|
||||||
lt: (a, b) => a < b,
|
|
||||||
gtEq: (a, b) => a >= b,
|
|
||||||
ltEq: (a, b) => a <= b,
|
|
||||||
or: (a, b) => a || b,
|
|
||||||
and: (a, b) => a && b,
|
|
||||||
if: (bool, a, b) => bool ? a : b,
|
|
||||||
for: (times, fn) => {
|
|
||||||
const result = [];
|
|
||||||
for (let i = 0; i < times; i++) {
|
|
||||||
result.push(fn.exec({
|
|
||||||
[fn.slots[0]]: i + 1
|
|
||||||
}));
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
},
|
|
||||||
add: (a, b) => a + b,
|
|
||||||
subtract: (a, b) => a - b,
|
|
||||||
multiply: (a, b) => a * b,
|
|
||||||
divide: (a, b) => a / b,
|
|
||||||
strLen: (a) => a.length,
|
|
||||||
strPick: (a, b) => a[b - 1],
|
|
||||||
strReplace: (a, b, c) => a.split(b).join(c),
|
|
||||||
strReverse: (a) => a.split('').reverse().join(''),
|
|
||||||
join: (texts, separator) => texts.join(separator || ''),
|
|
||||||
stringToNumber: (a) => parseInt(a),
|
|
||||||
numberToString: (a) => a.toString(),
|
|
||||||
splitStrByLine: (a) => a.split('\n'),
|
|
||||||
random: (probability) => Math.floor(seedrandom(`${this.opts.randomSeed}:${block.id}`)() * 100) < probability,
|
|
||||||
rannum: (min, max) => min + Math.floor(seedrandom(`${this.opts.randomSeed}:${block.id}`)() * (max - min + 1)),
|
|
||||||
randomPick: (list) => list[Math.floor(seedrandom(`${this.opts.randomSeed}:${block.id}`)() * list.length)],
|
|
||||||
dailyRandom: (probability) => Math.floor(seedrandom(`${day}:${block.id}`)() * 100) < probability,
|
|
||||||
dailyRannum: (min, max) => min + Math.floor(seedrandom(`${day}:${block.id}`)() * (max - min + 1)),
|
|
||||||
dailyRandomPick: (list) => list[Math.floor(seedrandom(`${day}:${block.id}`)() * list.length)],
|
|
||||||
seedRandom: (seed, probability) => Math.floor(seedrandom(seed)() * 100) < probability,
|
|
||||||
seedRannum: (seed, min, max) => min + Math.floor(seedrandom(seed)() * (max - min + 1)),
|
|
||||||
seedRandomPick: (seed, list) => list[Math.floor(seedrandom(seed)() * list.length)],
|
|
||||||
};
|
|
||||||
|
|
||||||
const fnName = block.type;
|
|
||||||
const fn = funcs[fnName];
|
|
||||||
if (fn == null) {
|
|
||||||
throw new AiScriptError(`No such function '${fnName}'`);
|
|
||||||
} else {
|
|
||||||
return fn(...block.args.map(x => this.evaluate(x, scope)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@autobind
|
|
||||||
public isUsedName(name: string) {
|
|
||||||
if (this.variables.some(v => v.name === name)) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (this.pageVars.some(v => v.name === name)) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (AiScript.envVarsDef[name]) {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -57,7 +57,7 @@ import Vue from 'vue';
|
|||||||
import i18n from '../../../../i18n';
|
import i18n from '../../../../i18n';
|
||||||
import XContainer from './page-editor.container.vue';
|
import XContainer from './page-editor.container.vue';
|
||||||
import { faSuperscript, faPencilAlt, faSquareRootAlt } from '@fortawesome/free-solid-svg-icons';
|
import { faSuperscript, faPencilAlt, faSquareRootAlt } from '@fortawesome/free-solid-svg-icons';
|
||||||
import { AiScript } from '../../../scripts/aiscript';
|
import { isLiteralBlock, funcDefs, blockDefs } from '../../../../../../misc/aiscript/index';
|
||||||
import * as uuid from 'uuid';
|
import * as uuid from 'uuid';
|
||||||
|
|
||||||
export default Vue.extend({
|
export default Vue.extend({
|
||||||
@ -97,7 +97,6 @@ export default Vue.extend({
|
|||||||
|
|
||||||
data() {
|
data() {
|
||||||
return {
|
return {
|
||||||
AiScript,
|
|
||||||
error: null,
|
error: null,
|
||||||
warn: null,
|
warn: null,
|
||||||
slots: '',
|
slots: '',
|
||||||
@ -109,7 +108,7 @@ export default Vue.extend({
|
|||||||
icon(): any {
|
icon(): any {
|
||||||
if (this.value.type === null) return null;
|
if (this.value.type === null) return null;
|
||||||
if (this.value.type.startsWith('fn:')) return null;
|
if (this.value.type.startsWith('fn:')) return null;
|
||||||
return AiScript.blockDefs.find(x => x.type === this.value.type).icon;
|
return blockDefs.find(x => x.type === this.value.type).icon;
|
||||||
},
|
},
|
||||||
typeText(): any {
|
typeText(): any {
|
||||||
if (this.value.type === null) return null;
|
if (this.value.type === null) return null;
|
||||||
@ -160,17 +159,17 @@ export default Vue.extend({
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (AiScript.isLiteralBlock(this.value)) return;
|
if (isLiteralBlock(this.value)) return;
|
||||||
|
|
||||||
const empties = [];
|
const empties = [];
|
||||||
for (let i = 0; i < AiScript.funcDefs[this.value.type].in.length; i++) {
|
for (let i = 0; i < funcDefs[this.value.type].in.length; i++) {
|
||||||
const id = uuid.v4();
|
const id = uuid.v4();
|
||||||
empties.push({ id, type: null });
|
empties.push({ id, type: null });
|
||||||
}
|
}
|
||||||
Vue.set(this.value, 'args', empties);
|
Vue.set(this.value, 'args', empties);
|
||||||
|
|
||||||
for (let i = 0; i < AiScript.funcDefs[this.value.type].in.length; i++) {
|
for (let i = 0; i < funcDefs[this.value.type].in.length; i++) {
|
||||||
const inType = AiScript.funcDefs[this.value.type].in[i];
|
const inType = funcDefs[this.value.type].in[i];
|
||||||
if (typeof inType !== 'number') {
|
if (typeof inType !== 'number') {
|
||||||
if (inType === 'number') this.value.args[i].type = 'number';
|
if (inType === 'number') this.value.args[i].type = 'number';
|
||||||
if (inType === 'string') this.value.args[i].type = 'text';
|
if (inType === 'string') this.value.args[i].type = 'text';
|
||||||
|
|||||||
@ -92,7 +92,8 @@ import { faSave, faStickyNote, faTrashAlt } from '@fortawesome/free-regular-svg-
|
|||||||
import XVariable from './page-editor.script-block.vue';
|
import XVariable from './page-editor.script-block.vue';
|
||||||
import XBlock from './page-editor.block.vue';
|
import XBlock from './page-editor.block.vue';
|
||||||
import * as uuid from 'uuid';
|
import * as uuid from 'uuid';
|
||||||
import { AiScript } from '../../../scripts/aiscript';
|
import { blockDefs } from '../../../../../../misc/aiscript/index';
|
||||||
|
import { ASTypeChecker } from '../../../../../../misc/aiscript/type-checker';
|
||||||
import { url } from '../../../../config';
|
import { url } from '../../../../config';
|
||||||
import { collectPageVars } from '../../../scripts/collect-page-vars';
|
import { collectPageVars } from '../../../scripts/collect-page-vars';
|
||||||
|
|
||||||
@ -150,14 +151,14 @@ export default Vue.extend({
|
|||||||
},
|
},
|
||||||
|
|
||||||
created() {
|
created() {
|
||||||
this.aiScript = new AiScript();
|
this.aiScript = new ASTypeChecker();
|
||||||
|
|
||||||
this.$watch('variables', () => {
|
this.$watch('variables', () => {
|
||||||
this.aiScript.injectVars(this.variables);
|
this.aiScript.variables = this.variables;
|
||||||
}, { deep: true });
|
}, { deep: true });
|
||||||
|
|
||||||
this.$watch('content', () => {
|
this.$watch('content', () => {
|
||||||
this.aiScript.injectPageVars(collectPageVars(this.content));
|
this.aiScript.pageVars = collectPageVars(this.content);
|
||||||
}, { deep: true });
|
}, { deep: true });
|
||||||
|
|
||||||
if (this.page) {
|
if (this.page) {
|
||||||
@ -352,7 +353,7 @@ export default Vue.extend({
|
|||||||
getScriptBlockList(type: string = null) {
|
getScriptBlockList(type: string = null) {
|
||||||
const list = [];
|
const list = [];
|
||||||
|
|
||||||
const blocks = AiScript.blockDefs.filter(block => type === null || block.out === null || block.out === type);
|
const blocks = blockDefs.filter(block => type === null || block.out === null || block.out === type);
|
||||||
|
|
||||||
for (const block of blocks) {
|
for (const block of blocks) {
|
||||||
const category = list.find(x => x.category === block.category);
|
const category = list.find(x => x.category === block.category);
|
||||||
|
|||||||
@ -21,12 +21,12 @@ import i18n from '../../../../i18n';
|
|||||||
import { faICursor, faPlus, faSquareRootAlt } from '@fortawesome/free-solid-svg-icons';
|
import { faICursor, faPlus, faSquareRootAlt } from '@fortawesome/free-solid-svg-icons';
|
||||||
import { faSave, faStickyNote } from '@fortawesome/free-regular-svg-icons';
|
import { faSave, faStickyNote } from '@fortawesome/free-regular-svg-icons';
|
||||||
import XBlock from './page.block.vue';
|
import XBlock from './page.block.vue';
|
||||||
import { AiScript } from '../../../scripts/aiscript';
|
import { ASEvaluator } from '../../../../../../misc/aiscript/evaluator';
|
||||||
import { collectPageVars } from '../../../scripts/collect-page-vars';
|
import { collectPageVars } from '../../../scripts/collect-page-vars';
|
||||||
import { url } from '../../../../config';
|
import { url } from '../../../../config';
|
||||||
|
|
||||||
class Script {
|
class Script {
|
||||||
public aiScript: AiScript;
|
public aiScript: ASEvaluator;
|
||||||
private onError: any;
|
private onError: any;
|
||||||
public vars: Record<string, any>;
|
public vars: Record<string, any>;
|
||||||
|
|
||||||
@ -86,7 +86,7 @@ export default Vue.extend({
|
|||||||
}).then(page => {
|
}).then(page => {
|
||||||
this.page = page;
|
this.page = page;
|
||||||
const pageVars = this.getPageVars();
|
const pageVars = this.getPageVars();
|
||||||
this.script = new Script(new AiScript(this.page.variables, pageVars, {
|
this.script = new Script(new ASEvaluator(this.page.variables, pageVars, {
|
||||||
randomSeed: Math.random(),
|
randomSeed: Math.random(),
|
||||||
user: page.user,
|
user: page.user,
|
||||||
visitor: this.$store.state.i,
|
visitor: this.$store.state.i,
|
||||||
|
|||||||
250
src/misc/aiscript/evaluator.ts
Normal file
250
src/misc/aiscript/evaluator.ts
Normal file
@ -0,0 +1,250 @@
|
|||||||
|
import autobind from 'autobind-decorator';
|
||||||
|
import * as seedrandom from 'seedrandom';
|
||||||
|
import { Variable, PageVar, envVarsDef, funcDefs, Block, isFnBlock } from '.';
|
||||||
|
|
||||||
|
type Fn = {
|
||||||
|
slots: string[];
|
||||||
|
exec: (args: Record<string, any>) => ReturnType<ASEvaluator['evaluate']>;
|
||||||
|
};
|
||||||
|
|
||||||
|
class AiScriptError extends Error {
|
||||||
|
public info?: any;
|
||||||
|
|
||||||
|
constructor(message: string, info?: any) {
|
||||||
|
super(message);
|
||||||
|
|
||||||
|
this.info = info;
|
||||||
|
|
||||||
|
// Maintains proper stack trace for where our error was thrown (only available on V8)
|
||||||
|
if (Error.captureStackTrace) {
|
||||||
|
Error.captureStackTrace(this, AiScriptError);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class Scope {
|
||||||
|
private layerdStates: Record<string, any>[];
|
||||||
|
public name: string;
|
||||||
|
|
||||||
|
constructor(layerdStates: Scope['layerdStates'], name?: Scope['name']) {
|
||||||
|
this.layerdStates = layerdStates;
|
||||||
|
this.name = name || 'anonymous';
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public createChildScope(states: Record<string, any>, name?: Scope['name']): Scope {
|
||||||
|
const layer = [states, ...this.layerdStates];
|
||||||
|
return new Scope(layer, name);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 指定した名前の変数の値を取得します
|
||||||
|
* @param name 変数名
|
||||||
|
*/
|
||||||
|
@autobind
|
||||||
|
public getState(name: string): any {
|
||||||
|
for (const later of this.layerdStates) {
|
||||||
|
const state = later[name];
|
||||||
|
if (state !== undefined) {
|
||||||
|
return state;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new AiScriptError(
|
||||||
|
`No such variable '${name}' in scope '${this.name}'`, {
|
||||||
|
scope: this.layerdStates
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* AiScript evaluator
|
||||||
|
*/
|
||||||
|
export class ASEvaluator {
|
||||||
|
private variables: Variable[];
|
||||||
|
private pageVars: PageVar[];
|
||||||
|
private envVars: Record<keyof typeof envVarsDef, any>;
|
||||||
|
|
||||||
|
private opts: {
|
||||||
|
randomSeed: string; user?: any; visitor?: any; page?: any; url?: string; version: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
constructor(variables: Variable[], pageVars: PageVar[], opts: ASEvaluator['opts']) {
|
||||||
|
this.variables = variables;
|
||||||
|
this.pageVars = pageVars;
|
||||||
|
this.opts = opts;
|
||||||
|
|
||||||
|
const date = new Date();
|
||||||
|
|
||||||
|
this.envVars = {
|
||||||
|
AI: 'kawaii',
|
||||||
|
VERSION: opts.version,
|
||||||
|
URL: opts.page ? `${opts.url}/@${opts.page.user.username}/pages/${opts.page.name}` : '',
|
||||||
|
LOGIN: opts.visitor != null,
|
||||||
|
NAME: opts.visitor ? opts.visitor.name : '',
|
||||||
|
USERNAME: opts.visitor ? opts.visitor.username : '',
|
||||||
|
USERID: opts.visitor ? opts.visitor.id : '',
|
||||||
|
NOTES_COUNT: opts.visitor ? opts.visitor.notesCount : 0,
|
||||||
|
FOLLOWERS_COUNT: opts.visitor ? opts.visitor.followersCount : 0,
|
||||||
|
FOLLOWING_COUNT: opts.visitor ? opts.visitor.followingCount : 0,
|
||||||
|
IS_CAT: opts.visitor ? opts.visitor.isCat : false,
|
||||||
|
MY_NOTES_COUNT: opts.user ? opts.user.notesCount : 0,
|
||||||
|
MY_FOLLOWERS_COUNT: opts.user ? opts.user.followersCount : 0,
|
||||||
|
MY_FOLLOWING_COUNT: opts.user ? opts.user.followingCount : 0,
|
||||||
|
SEED: opts.randomSeed ? opts.randomSeed : '',
|
||||||
|
YMD: `${date.getFullYear()}/${date.getMonth() + 1}/${date.getDate()}`
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public updatePageVar(name: string, value: any) {
|
||||||
|
const pageVar = this.pageVars.find(v => v.name === name);
|
||||||
|
if (pageVar !== undefined) {
|
||||||
|
pageVar.value = value;
|
||||||
|
} else {
|
||||||
|
throw new AiScriptError(`No such page var '${name}'`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public updateRandomSeed(seed: string) {
|
||||||
|
this.opts.randomSeed = seed;
|
||||||
|
this.envVars.SEED = seed;
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public getVarByName(name: string): Variable {
|
||||||
|
const v = this.variables.find(x => x.name === name);
|
||||||
|
if (v !== undefined) {
|
||||||
|
return v;
|
||||||
|
} else {
|
||||||
|
throw new AiScriptError(`No such variable '${name}'`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
private interpolate(str: string, scope: Scope) {
|
||||||
|
return str.replace(/\{(.+?)\}/g, match => {
|
||||||
|
const v = scope.getState(match.slice(1, -1).trim());
|
||||||
|
return v == null ? 'NULL' : v.toString();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public evaluateVars(): Record<string, any> {
|
||||||
|
const values: Record<string, any> = {};
|
||||||
|
|
||||||
|
for (const [k, v] of Object.entries(this.envVars)) {
|
||||||
|
values[k] = v;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const v of this.pageVars) {
|
||||||
|
values[v.name] = v.value;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const v of this.variables) {
|
||||||
|
values[v.name] = this.evaluate(v, new Scope([values]));
|
||||||
|
}
|
||||||
|
|
||||||
|
return values;
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
private evaluate(block: Block, scope: Scope): any {
|
||||||
|
if (block.type === null) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (block.type === 'number') {
|
||||||
|
return parseInt(block.value, 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (block.type === 'text' || block.type === 'multiLineText') {
|
||||||
|
return this.interpolate(block.value || '', scope);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (block.type === 'textList') {
|
||||||
|
return block.value.trim().split('\n');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (block.type === 'ref') {
|
||||||
|
return scope.getState(block.value);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isFnBlock(block)) { // ユーザー関数定義
|
||||||
|
return {
|
||||||
|
slots: block.value.slots.map(x => x.name),
|
||||||
|
exec: (slotArg: Record<string, any>) => {
|
||||||
|
return this.evaluate(block.value.expression, scope.createChildScope(slotArg, block.id));
|
||||||
|
}
|
||||||
|
} as Fn;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (block.type.startsWith('fn:')) { // ユーザー関数呼び出し
|
||||||
|
const fnName = block.type.split(':')[1];
|
||||||
|
const fn = scope.getState(fnName);
|
||||||
|
const args = {} as Record<string, any>;
|
||||||
|
for (let i = 0; i < fn.slots.length; i++) {
|
||||||
|
const name = fn.slots[i];
|
||||||
|
args[name] = this.evaluate(block.args[i], scope);
|
||||||
|
}
|
||||||
|
return fn.exec(args);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (block.args === undefined) return null;
|
||||||
|
|
||||||
|
const date = new Date();
|
||||||
|
const day = `${this.opts.visitor ? this.opts.visitor.id : ''} ${date.getFullYear()}/${date.getMonth() + 1}/${date.getDate()}`;
|
||||||
|
|
||||||
|
const funcs: { [p in keyof typeof funcDefs]: Function } = {
|
||||||
|
not: (a: boolean) => !a,
|
||||||
|
or: (a: boolean, b: boolean) => a || b,
|
||||||
|
and: (a: boolean, b: boolean) => a && b,
|
||||||
|
eq: (a: any, b: any) => a === b,
|
||||||
|
notEq: (a: any, b: any) => a !== b,
|
||||||
|
gt: (a: number, b: number) => a > b,
|
||||||
|
lt: (a: number, b: number) => a < b,
|
||||||
|
gtEq: (a: number, b: number) => a >= b,
|
||||||
|
ltEq: (a: number, b: number) => a <= b,
|
||||||
|
if: (bool: boolean, a: any, b: any) => bool ? a : b,
|
||||||
|
for: (times: number, fn: Fn) => {
|
||||||
|
const result = [];
|
||||||
|
for (let i = 0; i < times; i++) {
|
||||||
|
result.push(fn.exec({
|
||||||
|
[fn.slots[0]]: i + 1
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
},
|
||||||
|
add: (a: number, b: number) => a + b,
|
||||||
|
subtract: (a: number, b: number) => a - b,
|
||||||
|
multiply: (a: number, b: number) => a * b,
|
||||||
|
divide: (a: number, b: number) => a / b,
|
||||||
|
strLen: (a: string) => a.length,
|
||||||
|
strPick: (a: string, b: number) => a[b - 1],
|
||||||
|
strReplace: (a: string, b: string, c: string) => a.split(b).join(c),
|
||||||
|
strReverse: (a: string) => a.split('').reverse().join(''),
|
||||||
|
join: (texts: string[], separator: string) => texts.join(separator || ''),
|
||||||
|
stringToNumber: (a: string) => parseInt(a),
|
||||||
|
numberToString: (a: number) => a.toString(),
|
||||||
|
splitStrByLine: (a: string) => a.split('\n'),
|
||||||
|
random: (probability: number) => Math.floor(seedrandom(`${this.opts.randomSeed}:${block.id}`)() * 100) < probability,
|
||||||
|
rannum: (min: number, max: number) => min + Math.floor(seedrandom(`${this.opts.randomSeed}:${block.id}`)() * (max - min + 1)),
|
||||||
|
randomPick: (list: any[]) => list[Math.floor(seedrandom(`${this.opts.randomSeed}:${block.id}`)() * list.length)],
|
||||||
|
dailyRandom: (probability: number) => Math.floor(seedrandom(`${day}:${block.id}`)() * 100) < probability,
|
||||||
|
dailyRannum: (min: number, max: number) => min + Math.floor(seedrandom(`${day}:${block.id}`)() * (max - min + 1)),
|
||||||
|
dailyRandomPick: (list: any[]) => list[Math.floor(seedrandom(`${day}:${block.id}`)() * list.length)],
|
||||||
|
seedRandom: (seed: any, probability: number) => Math.floor(seedrandom(seed)() * 100) < probability,
|
||||||
|
seedRannum: (seed: any, min: number, max: number) => min + Math.floor(seedrandom(seed)() * (max - min + 1)),
|
||||||
|
seedRandomPick: (seed: any, list: any[]) => list[Math.floor(seedrandom(seed)() * list.length)],
|
||||||
|
};
|
||||||
|
|
||||||
|
const fnName = block.type;
|
||||||
|
const fn = (funcs as any)[fnName];
|
||||||
|
if (fn == null) {
|
||||||
|
throw new AiScriptError(`No such function '${fnName}'`);
|
||||||
|
} else {
|
||||||
|
return fn(...block.args.map(x => this.evaluate(x, scope)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
131
src/misc/aiscript/index.ts
Normal file
131
src/misc/aiscript/index.ts
Normal file
@ -0,0 +1,131 @@
|
|||||||
|
/**
|
||||||
|
* AiScript
|
||||||
|
*/
|
||||||
|
|
||||||
|
import {
|
||||||
|
faSuperscript,
|
||||||
|
faAlignLeft,
|
||||||
|
faShareAlt,
|
||||||
|
faPlus,
|
||||||
|
faMinus,
|
||||||
|
faTimes,
|
||||||
|
faDivide,
|
||||||
|
faList,
|
||||||
|
faQuoteRight,
|
||||||
|
faEquals,
|
||||||
|
faGreaterThan,
|
||||||
|
faLessThan,
|
||||||
|
faGreaterThanEqual,
|
||||||
|
faLessThanEqual,
|
||||||
|
faNotEqual,
|
||||||
|
faDice,
|
||||||
|
faSortNumericUp,
|
||||||
|
faExchangeAlt,
|
||||||
|
faRecycle,
|
||||||
|
} from '@fortawesome/free-solid-svg-icons';
|
||||||
|
import { faFlag } from '@fortawesome/free-regular-svg-icons';
|
||||||
|
|
||||||
|
export type Block<V = any> = {
|
||||||
|
id: string;
|
||||||
|
type: string;
|
||||||
|
args: Block[];
|
||||||
|
value: V;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type FnBlock = Block<{
|
||||||
|
slots: {
|
||||||
|
name: string;
|
||||||
|
type: Type;
|
||||||
|
}[];
|
||||||
|
expression: Block;
|
||||||
|
}>;
|
||||||
|
|
||||||
|
export type Variable = Block & {
|
||||||
|
name: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type Type = 'string' | 'number' | 'boolean' | 'stringArray';
|
||||||
|
|
||||||
|
export const funcDefs = {
|
||||||
|
if: { in: ['boolean', 0, 0], out: 0, category: 'flow', icon: faShareAlt, },
|
||||||
|
for: { in: ['number', 'function'], out: null, category: 'flow', icon: faRecycle, },
|
||||||
|
not: { in: ['boolean'], out: 'boolean', category: 'logical', icon: faFlag, },
|
||||||
|
or: { in: ['boolean', 'boolean'], out: 'boolean', category: 'logical', icon: faFlag, },
|
||||||
|
and: { in: ['boolean', 'boolean'], out: 'boolean', category: 'logical', icon: faFlag, },
|
||||||
|
add: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faPlus, },
|
||||||
|
subtract: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faMinus, },
|
||||||
|
multiply: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faTimes, },
|
||||||
|
divide: { in: ['number', 'number'], out: 'number', category: 'operation', icon: faDivide, },
|
||||||
|
eq: { in: [0, 0], out: 'boolean', category: 'comparison', icon: faEquals, },
|
||||||
|
notEq: { in: [0, 0], out: 'boolean', category: 'comparison', icon: faNotEqual, },
|
||||||
|
gt: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faGreaterThan, },
|
||||||
|
lt: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faLessThan, },
|
||||||
|
gtEq: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faGreaterThanEqual, },
|
||||||
|
ltEq: { in: ['number', 'number'], out: 'boolean', category: 'comparison', icon: faLessThanEqual, },
|
||||||
|
strLen: { in: ['string'], out: 'number', category: 'text', icon: faQuoteRight, },
|
||||||
|
strPick: { in: ['string', 'number'], out: 'string', category: 'text', icon: faQuoteRight, },
|
||||||
|
strReplace: { in: ['string', 'string', 'string'], out: 'string', category: 'text', icon: faQuoteRight, },
|
||||||
|
strReverse: { in: ['string'], out: 'string', category: 'text', icon: faQuoteRight, },
|
||||||
|
join: { in: ['stringArray', 'string'], out: 'string', category: 'text', icon: faQuoteRight, },
|
||||||
|
stringToNumber: { in: ['string'], out: 'number', category: 'convert', icon: faExchangeAlt, },
|
||||||
|
numberToString: { in: ['number'], out: 'string', category: 'convert', icon: faExchangeAlt, },
|
||||||
|
splitStrByLine: { in: ['string'], out: 'stringArray', category: 'convert', icon: faExchangeAlt, },
|
||||||
|
rannum: { in: ['number', 'number'], out: 'number', category: 'random', icon: faDice, },
|
||||||
|
dailyRannum: { in: ['number', 'number'], out: 'number', category: 'random', icon: faDice, },
|
||||||
|
seedRannum: { in: [null, 'number', 'number'], out: 'number', category: 'random', icon: faDice, },
|
||||||
|
random: { in: ['number'], out: 'boolean', category: 'random', icon: faDice, },
|
||||||
|
dailyRandom: { in: ['number'], out: 'boolean', category: 'random', icon: faDice, },
|
||||||
|
seedRandom: { in: [null, 'number'], out: 'boolean', category: 'random', icon: faDice, },
|
||||||
|
randomPick: { in: [0], out: 0, category: 'random', icon: faDice, },
|
||||||
|
dailyRandomPick: { in: [0], out: 0, category: 'random', icon: faDice, },
|
||||||
|
seedRandomPick: { in: [null, 0], out: 0, category: 'random', icon: faDice, },
|
||||||
|
};
|
||||||
|
|
||||||
|
export const literalDefs = {
|
||||||
|
text: { out: 'string', category: 'value', icon: faQuoteRight, },
|
||||||
|
multiLineText: { out: 'string', category: 'value', icon: faAlignLeft, },
|
||||||
|
textList: { out: 'stringArray', category: 'value', icon: faList, },
|
||||||
|
number: { out: 'number', category: 'value', icon: faSortNumericUp, },
|
||||||
|
ref: { out: null, category: 'value', icon: faSuperscript, },
|
||||||
|
fn: { out: 'function', category: 'value', icon: faSuperscript, },
|
||||||
|
};
|
||||||
|
|
||||||
|
export const blockDefs = [
|
||||||
|
...Object.entries(literalDefs).map(([k, v]) => ({
|
||||||
|
type: k, out: v.out, category: v.category, icon: v.icon
|
||||||
|
})),
|
||||||
|
...Object.entries(funcDefs).map(([k, v]) => ({
|
||||||
|
type: k, out: v.out, category: v.category, icon: v.icon
|
||||||
|
}))
|
||||||
|
];
|
||||||
|
|
||||||
|
export function isFnBlock(block: Block): block is FnBlock {
|
||||||
|
return block.type === 'fn';
|
||||||
|
}
|
||||||
|
|
||||||
|
export type PageVar = { name: string; value: any; type: Type; };
|
||||||
|
|
||||||
|
export const envVarsDef = {
|
||||||
|
AI: 'string',
|
||||||
|
URL: 'string',
|
||||||
|
VERSION: 'string',
|
||||||
|
LOGIN: 'boolean',
|
||||||
|
NAME: 'string',
|
||||||
|
USERNAME: 'string',
|
||||||
|
USERID: 'string',
|
||||||
|
NOTES_COUNT: 'number',
|
||||||
|
FOLLOWERS_COUNT: 'number',
|
||||||
|
FOLLOWING_COUNT: 'number',
|
||||||
|
IS_CAT: 'boolean',
|
||||||
|
MY_NOTES_COUNT: 'number',
|
||||||
|
MY_FOLLOWERS_COUNT: 'number',
|
||||||
|
MY_FOLLOWING_COUNT: 'number',
|
||||||
|
SEED: null,
|
||||||
|
YMD: 'string',
|
||||||
|
};
|
||||||
|
|
||||||
|
export function isLiteralBlock(v: Block) {
|
||||||
|
if (v.type === null) return true;
|
||||||
|
if (literalDefs[v.type]) return true;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
176
src/misc/aiscript/type-checker.ts
Normal file
176
src/misc/aiscript/type-checker.ts
Normal file
@ -0,0 +1,176 @@
|
|||||||
|
import autobind from 'autobind-decorator';
|
||||||
|
import { Type, Block, funcDefs, envVarsDef, Variable, PageVar, isLiteralBlock } from '.';
|
||||||
|
|
||||||
|
type TypeError = {
|
||||||
|
arg: number;
|
||||||
|
expect: Type;
|
||||||
|
actual: Type;
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* AiScript type checker
|
||||||
|
*/
|
||||||
|
export class ASTypeChecker {
|
||||||
|
public variables: Variable[];
|
||||||
|
public pageVars: PageVar[];
|
||||||
|
|
||||||
|
constructor(variables: ASTypeChecker['variables'] = [], pageVars: ASTypeChecker['pageVars'] = []) {
|
||||||
|
this.variables = variables;
|
||||||
|
this.pageVars = pageVars;
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public typeCheck(v: Block): TypeError | null {
|
||||||
|
if (isLiteralBlock(v)) return null;
|
||||||
|
|
||||||
|
const def = funcDefs[v.type];
|
||||||
|
if (def == null) {
|
||||||
|
throw new Error('Unknown type: ' + v.type);
|
||||||
|
}
|
||||||
|
|
||||||
|
const generic: Type[] = [];
|
||||||
|
|
||||||
|
for (let i = 0; i < def.in.length; i++) {
|
||||||
|
const arg = def.in[i];
|
||||||
|
const type = this.typeInference(v.args[i]);
|
||||||
|
if (type === null) continue;
|
||||||
|
|
||||||
|
if (typeof arg === 'number') {
|
||||||
|
if (generic[arg] === undefined) {
|
||||||
|
generic[arg] = type;
|
||||||
|
} else if (type !== generic[arg]) {
|
||||||
|
return {
|
||||||
|
arg: i,
|
||||||
|
expect: generic[arg],
|
||||||
|
actual: type
|
||||||
|
};
|
||||||
|
}
|
||||||
|
} else if (type !== arg) {
|
||||||
|
return {
|
||||||
|
arg: i,
|
||||||
|
expect: arg,
|
||||||
|
actual: type
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public getExpectedType(v: Block, slot: number): Type | null {
|
||||||
|
const def = funcDefs[v.type];
|
||||||
|
if (def == null) {
|
||||||
|
throw new Error('Unknown type: ' + v.type);
|
||||||
|
}
|
||||||
|
|
||||||
|
const generic: Type[] = [];
|
||||||
|
|
||||||
|
for (let i = 0; i < def.in.length; i++) {
|
||||||
|
const arg = def.in[i];
|
||||||
|
const type = this.typeInference(v.args[i]);
|
||||||
|
if (type === null) continue;
|
||||||
|
|
||||||
|
if (typeof arg === 'number') {
|
||||||
|
if (generic[arg] === undefined) {
|
||||||
|
generic[arg] = type;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (typeof def.in[slot] === 'number') {
|
||||||
|
return generic[def.in[slot]] || null;
|
||||||
|
} else {
|
||||||
|
return def.in[slot];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public typeInference(v: Block): Type | null {
|
||||||
|
if (v.type === null) return null;
|
||||||
|
if (v.type === 'text') return 'string';
|
||||||
|
if (v.type === 'multiLineText') return 'string';
|
||||||
|
if (v.type === 'textList') return 'stringArray';
|
||||||
|
if (v.type === 'number') return 'number';
|
||||||
|
if (v.type === 'ref') {
|
||||||
|
const variable = this.variables.find(va => va.name === v.value);
|
||||||
|
if (variable) {
|
||||||
|
return this.typeInference(variable);
|
||||||
|
}
|
||||||
|
|
||||||
|
const pageVar = this.pageVars.find(va => va.name === v.value);
|
||||||
|
if (pageVar) {
|
||||||
|
return pageVar.type;
|
||||||
|
}
|
||||||
|
|
||||||
|
const envVar = envVarsDef[v.value];
|
||||||
|
if (envVar !== undefined) {
|
||||||
|
return envVar;
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
if (v.type === 'fn') return null; // todo
|
||||||
|
if (v.type.startsWith('fn:')) return null; // todo
|
||||||
|
|
||||||
|
const generic: Type[] = [];
|
||||||
|
|
||||||
|
const def = funcDefs[v.type];
|
||||||
|
|
||||||
|
for (let i = 0; i < def.in.length; i++) {
|
||||||
|
const arg = def.in[i];
|
||||||
|
if (typeof arg === 'number') {
|
||||||
|
const type = this.typeInference(v.args[i]);
|
||||||
|
|
||||||
|
if (generic[arg] === undefined) {
|
||||||
|
generic[arg] = type;
|
||||||
|
} else {
|
||||||
|
if (type !== generic[arg]) {
|
||||||
|
generic[arg] = null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (typeof def.out === 'number') {
|
||||||
|
return generic[def.out];
|
||||||
|
} else {
|
||||||
|
return def.out;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public getVarsByType(type: Type | null): Variable[] {
|
||||||
|
if (type == null) return this.variables;
|
||||||
|
return this.variables.filter(x => (this.typeInference(x) === null) || (this.typeInference(x) === type));
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public getEnvVarsByType(type: Type | null): string[] {
|
||||||
|
if (type == null) return Object.keys(envVarsDef);
|
||||||
|
return Object.entries(envVarsDef).filter(([k, v]) => v === null || type === v).map(([k, v]) => k);
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public getPageVarsByType(type: Type | null): string[] {
|
||||||
|
if (type == null) return this.pageVars.map(v => v.name);
|
||||||
|
return this.pageVars.filter(v => type === v.type).map(v => v.name);
|
||||||
|
}
|
||||||
|
|
||||||
|
@autobind
|
||||||
|
public isUsedName(name: string) {
|
||||||
|
if (this.variables.some(v => v.name === name)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this.pageVars.some(v => v.name === name)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (envVarsDef[name]) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user