mindcraft/src/agent/commands.js

import { actionsList } from './commands/actions.js';
import { queryList } from './commands/queries.js';

const commandList = queryList.concat(actionsList);
const commandMap = {};
for (let command of commandList) {
    commandMap[command.name] = command;
}

export function getCommand(name) {
    return commandMap[name];
}

const commandRegex = /!(\w+)(?:\(((?:[^)(]+|'[^']*'|"[^"]*")*)\))?/
const argRegex = /(?:"[^"]*"|'[^']*'|[^,])+/g;

export function containsCommand(message) {
    const commandMatch = message.match(commandRegex);
    if (commandMatch) {
        const commandName = "!"+commandMatch[1];
        if (commandList.some((command) => command.name === commandName))
            return commandName;
    }
    return null;
}

function parseCommandMessage(message) {
    const commandMatch = message.match(commandRegex);
    if (commandMatch) {
        const commandName = "!"+commandMatch[1];
        if (!commandMatch[2])
            return { commandName, args: [] };
        let args = commandMatch[2].match(argRegex);
        if (args) {
            for (let i = 0; i < args.length; i++) {
                args[i] = args[i].trim();
            }

            for (let i = 0; i < args.length; i++) {
                let arg = args[i];
                if ((arg.startsWith('"') && arg.endsWith('"')) || (arg.startsWith("'") && arg.endsWith("'"))) {
                    args[i] = arg.substring(1, arg.length-1);
                } else if (!isNaN(arg)) {
                    args[i] = Number(arg);
                }
            }
        }
        else
            args = [];

        return { commandName, args };
    }
    return null;
}

function numParams(command) {
    if (!command.params)
        return 0;
    return Object.keys(command.params).length;
}

export async function executeCommand(agent, message) {
    let parsed = parseCommandMessage(message);
    if (parsed) {
        const command = getCommand(parsed.commandName);
        let numArgs = 0;
        if (parsed.args) {
            numArgs = parsed.args.length;
        }
        console.log('parsed command:', parsed);
        if (numArgs !== numParams(command))
            return `Command ${command.name} was given incorrect number of arguments`;
        else
            return await command.perform(agent, ...parsed.args);
    }
    else
        return `Command is incorrectly formatted`;
}

export function getCommandDocs() {
    let docs = `\n*COMMAND DOCS\n You can use the following commands to perform actions and get information about the world. 
    Use the commands with the syntax: !commandName or !commandName("arg1", 1.2, ...) if the command takes arguments.\n
    Do not use codeblocks. Only use one command in each response, trailing commands and comments will be ignored. Use these commands frequently in your responses!\n`;
    for (let command of commandList) {
        docs += command.name + ': ' + command.description + '\n';
        if (command.params) {
            docs += 'Params:\n';
            for (let param in command.params) {
                docs += param + ': ' + command.params[param] + '\n';
            }
        }
    }
    return docs + '*\n';
}