boardgame-core/src/utils/command/command-registry.ts

import type { Command, CommandSchema } from './types.js';
import type {CommandResult, CommandRunner, CommandRunnerContext, PromptEvent} from './command-runner.js';
import { parseCommand } from './command-parse.js';
import { applyCommandSchema } from './command-validate.js';
import { parseCommandSchema } from './schema-parse.js';
import {AsyncQueue} from "../async-queue";

export type CommandRegistry<TContext> = Map<string, CommandRunner<TContext, unknown>>;

export function createCommandRegistry<TContext>(): CommandRegistry<TContext> {
    return new Map();
}

export function registerCommand<TContext, TResult>(
    registry: CommandRegistry<TContext>,
    runner: CommandRunner<TContext, TResult>
): void {
    registry.set(runner.schema.name, runner as CommandRunner<TContext, unknown>);
}

export function unregisterCommand<TContext>(
    registry: CommandRegistry<TContext>,
    name: string
): void {
    registry.delete(name);
}

export function hasCommand<TContext>(
    registry: CommandRegistry<TContext>,
    name: string
): boolean {
    return registry.has(name);
}

export function getCommand<TContext>(
    registry: CommandRegistry<TContext>,
    name: string
): CommandRunner<TContext, unknown> | undefined {
    return registry.get(name);
}

type Listener = (e: PromptEvent) => void;

export type CommandRunnerContextExport<TContext> = CommandRunnerContext<TContext> & {
    registry: CommandRegistry<TContext>;
    promptQueue: AsyncQueue<PromptEvent>;
    _activePrompt: PromptEvent | null;
    _resolvePrompt: (command: Command) => void;
    _rejectPrompt: (error: Error) => void;
    _pendingInput: string | null;
};

export function createCommandRunnerContext<TContext>(
    registry: CommandRegistry<TContext>,
    context: TContext
): CommandRunnerContextExport<TContext> {
    const listeners = new Set<Listener>();

    const on = (_event: 'prompt', listener: Listener) => {
        listeners.add(listener);
    };

    const off = (_event: 'prompt', listener: Listener) => {
        listeners.delete(listener);
    };

    let activePrompt: PromptEvent | null = null;

    const resolvePrompt = (command: Command) => {
        if (activePrompt) {
            activePrompt.resolve(command);
            activePrompt = null;
        }
    };

    const rejectPrompt = (error: Error) => {
        if (activePrompt) {
            activePrompt.reject(error);
            activePrompt = null;
        }
    };

    const prompt = (schema: Parameters<CommandRunnerContext<TContext>['prompt']>[0]): Promise<Command> => {
        const resolvedSchema = typeof schema === 'string' ? parseCommandSchema(schema) : schema;
        return new Promise((resolve, reject) => {
            activePrompt = { schema: resolvedSchema, resolve, reject };
            const event: PromptEvent = { schema: resolvedSchema, resolve, reject };
            for (const listener of listeners) {
                listener(event);
            }
        });
    };

    const runnerCtx: CommandRunnerContextExport<TContext> = {
        registry,
        context,
        run: <T=unknown>(input: string) => runCommandWithContext(runnerCtx, input) as Promise<CommandResult<T>>,
        runParsed: (command: Command) => runCommandParsedWithContext(runnerCtx, command),
        prompt,
        on,
        off,
        _activePrompt: null,
        _resolvePrompt: resolvePrompt,
        _rejectPrompt: rejectPrompt,
        _pendingInput: null,
        promptQueue: null!
    };

    Object.defineProperty(runnerCtx, '_activePrompt', {
        get: () => activePrompt,
    });
    
    let promptQueue: AsyncQueue<PromptEvent>;
    Object.defineProperty(runnerCtx, 'promptQueue', {
        get(){
            if (!promptQueue) {
                promptQueue = new AsyncQueue();
                listeners.add(async (event) => {
                    promptQueue.push(event);
                });
            }
            return promptQueue;
        }
    });

    return runnerCtx;
}

async function executeWithRunnerContext<TContext>(
    runnerCtx: CommandRunnerContextExport<TContext>,
    runner: CommandRunner<TContext, unknown>,
    command: Command
): Promise<CommandResult> {
    try {
        const result = await runner.run.call(runnerCtx, command);
        return { success: true, result };
    } catch (e) {
        const error = e as Error;
        return { success: false, error: error.message };
    }
}

export async function runCommand<TContext>(
    registry: CommandRegistry<TContext>,
    context: TContext,
    input: string
): Promise<CommandResult> {
    const runnerCtx = createCommandRunnerContext(registry, context);
    return await runCommandWithContext(runnerCtx, input);
}

async function runCommandWithContext<TContext>(
    runnerCtx: CommandRunnerContextExport<TContext>,
    input: string
): Promise<{ success: true; result: unknown } | { success: false; error: string }> {
    const command = parseCommand(input);
    return await runCommandParsedWithContext(runnerCtx, command);
}

export async function runCommandParsed<TContext>(
    registry: CommandRegistry<TContext>,
    context: TContext,
    command: Command
): Promise<CommandResult> {
    const runnerCtx = createCommandRunnerContext(registry, context);
    return await runCommandParsedWithContext(runnerCtx, command);
}

async function runCommandParsedWithContext<TContext>(
    runnerCtx: CommandRunnerContextExport<TContext>,
    command: Command
): Promise<CommandResult> {
    const runner = runnerCtx.registry.get(command.name);
    if (!runner) {
        return { success: false, error: `Unknown command: ${command.name}` };
    }

    const validationResult = applyCommandSchema(command, runner.schema);
    if (!validationResult.valid) {
        return { success: false, error: validationResult.errors.join('; ') };
    }

    return await executeWithRunnerContext(runnerCtx, runner, validationResult.command);
}