Class ApiConnectionHandler

framework/framework/api.ts:222–270

}

export class ApiConnectionHandler extends ConnectionHandler {
    dispose: () => Promise<void> | void;
    isRpc: boolean;

    @param('op', Types.String)
    async prepare({ }, op: string) {
        if (op === 'rpc') {
            this.isRpc = true;
            return;
        }
        if (!APIS[op]) throw new BadRequestError(`Invalid API operation: ${op}`);
        if (APIS[op].type !== 'Subscription') {
            throw new BadRequestError('Only subscription operations are supported');
        }
        handleArguments(this.args);
        // @ts-ignore
        await this.ctx.parallel('handler/api/before', this);
        // @ts-ignore
        await this.ctx.parallel(`handler/api/before/${op}`, this);
        this.dispose = await this.ctx.api.execute(
            this, op, { domainId: this.args.domainId, ...this.args, ...(this.args.args || {}) },
            (m, args) => (this.ctx.parallel as any)(m, args), this.args.projection, (p) => this.send(p),
        );
    }

    async message(message) {
        if (!this.isRpc) throw new BadRequestError('Only RPC operations are supported');
        if (typeof message === 'string') {
            try {
                message = JSON.parse(message);
            } catch (e) {
                throw new BadRequestError('Invalid message');
            }
        }
        if (!APIS[message.op]) throw new BadRequestError(`Invalid API operation: ${message.op}`);
        if (APIS[message.op].type !== 'Subscription') {
            throw new BadRequestError('Only subscription operations are supported');
        }
        handleArguments(message);
        const result = await this.ctx.api.execute(
            this, message.op, message.args, (m, args) => (this.ctx.parallel as any)(m, args), message.projection,
        );
        this.send(result);
    }

    async cleanup() {
        await this.dispose?.();
    }
}

export async function applyApiHandler(ctx: Context, name: string, path: string) {
    ctx.plugin(ApiService);