@cosmology/ast/module/registry/registry.js

Cosmos TypeScript AST generation

import * as t from '@babel/types';
import { identifier } from '../utils';
export const createTypeRegistryObject = (mutation) => {
    return t.objectProperty(t.stringLiteral(mutation.typeUrl), t.identifier(mutation.TypeName));
};
export const createTypeRegistry = (context, mutations) => {
    const useTelescopeGeneratedType = context.pluginValue('prototypes.typingsFormat.useTelescopeGeneratedType');
    let generatedTypeExpr;
    if (useTelescopeGeneratedType) {
        context.addUtil('TelescopeGeneratedType');
        generatedTypeExpr = t.tsTypeReference(t.identifier('TelescopeGeneratedType'), t.tsTypeParameterInstantiation([
            t.tsAnyKeyword(),
            t.tsAnyKeyword(),
            t.tsAnyKeyword()
        ]));
    }
    else {
        context.addUtil('GeneratedType');
        generatedTypeExpr = t.tsTypeReference(t.identifier('GeneratedType'));
    }
    return t.exportNamedDeclaration(t.variableDeclaration('const', [
        t.variableDeclarator(identifier('registry', t.tsTypeAnnotation(t.tsTypeReference(t.identifier('ReadonlyArray'), t.tsTypeParameterInstantiation([
            t.tsTupleType([
                t.tsStringKeyword(),
                generatedTypeExpr
            ])
        ])))), t.arrayExpression([
            ...mutations.map(mutation => t.arrayExpression([
                t.stringLiteral(mutation.typeUrl),
                t.identifier(mutation.TypeName)
            ]))
        ]))
    ]));
};
export const createRegistryLoader = (context) => {
    context.addUtil('Registry');
    return t.exportNamedDeclaration(t.variableDeclaration('const', [
        t.variableDeclarator(t.identifier('load'), t.arrowFunctionExpression([
            identifier('protoRegistry', t.tsTypeAnnotation(t.tsTypeReference(t.identifier('Registry'))))
        ], t.blockStatement([
            t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('registry'), t.identifier('forEach')), [
                t.arrowFunctionExpression([
                    t.arrayPattern([
                        t.identifier('typeUrl'),
                        t.identifier('mod')
                    ])
                ], t.blockStatement([
                    t.expressionStatement(t.callExpression(t.memberExpression(t.identifier('protoRegistry'), t.identifier('register')), [
                        t.identifier('typeUrl'),
                        t.identifier('mod')
                    ]))
                ]))
            ]))
        ])))
    ]));
};