crankscript 0.9.3 → 0.9.5

package/assets/plugin.ts

@@ -0,0 +1,297 @@
+import * as ts from 'typescript';
+import * as tstl from 'typescript-to-lua';
+import { FunctionVisitor, TransformationContext } from 'typescript-to-lua';
+import * as lua from 'typescript-to-lua/dist/LuaAST';
+import { ScopeType } from 'typescript-to-lua/dist/transformation/utils/scope';
+import { transformCallAndArguments } from 'typescript-to-lua/dist/transformation/visitors/call';
+import { transformClassInstanceFields } from 'typescript-to-lua/dist/transformation/visitors/class/members/fields';
+import {
+    getExtendedNode,
+    isStaticNode,
+} from 'typescript-to-lua/dist/transformation/visitors/class/utils';
+import {
+    transformFunctionBodyContent,
+    transformFunctionToExpression,
+    transformParameters,
+} from 'typescript-to-lua/dist/transformation/visitors/function';
+
+function createClassCall(
+    context: tstl.TransformationContext,
+    className: tstl.Identifier,
+    extendsNode?: ts.ExpressionWithTypeArguments
+): tstl.Statement {
+    // class('X')
+    const classCall = tstl.createCallExpression(
+        tstl.createIdentifier('class'),
+        [tstl.createStringLiteral(className.text)]
+    );
+    let classCreationExpression: tstl.Expression;
+    if (extendsNode) {
+        // class('X').extends(Blah)
+        classCreationExpression = tstl.createCallExpression(
+            tstl.createTableIndexExpression(
+                classCall,
+                tstl.createStringLiteral('extends')
+            ),
+            [context.transformExpression(extendsNode.expression)]
+        );
+    } else {
+        classCreationExpression = tstl.createCallExpression(
+            tstl.createTableIndexExpression(
+                classCall,
+                tstl.createStringLiteral('extends')
+            ),
+            [tstl.createIdentifier('Object')]
+        );
+    }
+    return tstl.createExpressionStatement(classCreationExpression);
+}
+
+export function transformPropertyName(
+    context: TransformationContext,
+    node: ts.PropertyName
+): tstl.Expression {
+    if (ts.isComputedPropertyName(node)) {
+        return context.transformExpression(node.expression);
+    } else if (ts.isIdentifier(node)) {
+        return tstl.createStringLiteral(node.text);
+    } else if (ts.isPrivateIdentifier(node)) {
+        throw new Error('PrivateIdentifier is not supported');
+    } else {
+        return context.transformExpression(node);
+    }
+}
+
+function transformConstructor(
+    context: TransformationContext,
+    className: tstl.Identifier,
+    instanceFields: ts.PropertyDeclaration[],
+    constructor?: ts.ConstructorDeclaration
+): tstl.Statement | undefined {
+    const methodName = 'init';
+    context.pushScope(ScopeType.Function);
+    const bodyStatements: tstl.Statement[] = [];
+    let params: tstl.Identifier[];
+    if (constructor) {
+        [params] = transformParameters(
+            context,
+            constructor?.parameters,
+            tstl.createIdentifier('self')
+        );
+    } else {
+        params = [tstl.createIdentifier('self')];
+    }
+    bodyStatements.push(
+        tstl.createExpressionStatement(
+            tstl.createCallExpression(
+                tstl.createTableIndexExpression(
+                    tstl.createTableIndexExpression(
+                        className,
+                        tstl.createStringLiteral('super')
+                    ),
+                    tstl.createStringLiteral('init')
+                ),
+                params
+            )
+        )
+    );
+    const classInstanceFields = transformClassInstanceFields(
+        context,
+        instanceFields
+    );
+    // initializers have to come before any body of the constructor
+    bodyStatements.push(...classInstanceFields);
+    if (constructor?.body) {
+        const body = transformFunctionBodyContent(context, constructor.body);
+        // if the first expression in the body is a super call, ignore it, because we have
+        // constructed our own super call.
+        // if it's not, make sure to include the entire body.
+        const firstStatement = constructor.body.statements[0];
+        if (
+            firstStatement &&
+            ts.isExpressionStatement(firstStatement) &&
+            ts.isCallExpression(firstStatement.expression) &&
+            firstStatement.expression.expression.kind ===
+                ts.SyntaxKind.SuperKeyword
+        ) {
+            bodyStatements.push(...body.slice(1));
+        } else {
+            bodyStatements.push(...body);
+        }
+    }
+    context.popScope();
+    return tstl.createAssignmentStatement(
+        tstl.createTableIndexExpression(
+            className,
+            tstl.createStringLiteral(methodName)
+        ),
+        tstl.createFunctionExpression(tstl.createBlock(bodyStatements), params)
+    );
+}
+
+function transformMethodDeclaration(
+    context: TransformationContext,
+    node: ts.MethodDeclaration,
+    className: tstl.Identifier
+): tstl.Statement | undefined {
+    const [functionExpression] = transformFunctionToExpression(context, node);
+    return tstl.createAssignmentStatement(
+        tstl.createTableIndexExpression(
+            className,
+            transformPropertyName(context, node.name)
+        ),
+        functionExpression
+    );
+}
+interface ClassSuperInfo {
+    className: lua.Identifier;
+    extendedTypeNode?: ts.ExpressionWithTypeArguments;
+}
+
+export const transformClassDeclaration: FunctionVisitor<
+    ts.ClassLikeDeclaration
+> = (
+    declaration,
+    context: TransformationContext & { classSuperInfos?: [ClassSuperInfo] }
+) => {
+    let className: tstl.Identifier;
+    if (declaration.name) {
+        className = tstl.createIdentifier(declaration.name.text);
+    } else {
+        className = tstl.createIdentifier(
+            context.createTempName('class'),
+            declaration
+        );
+    }
+
+    const extension = getExtendedNode(declaration);
+    if (context.classSuperInfos) {
+        context.classSuperInfos.push({
+            className,
+            extendedTypeNode: extension,
+        });
+    } else {
+        context.classSuperInfos = [{ className, extendedTypeNode: extension }];
+    }
+
+    // Get all properties with value
+    const properties = declaration.members
+        .filter(ts.isPropertyDeclaration)
+        .filter((member) => member.initializer);
+
+    // Divide properties into static and non-static
+    const instanceFields = properties.filter((prop) => !isStaticNode(prop));
+
+    const statements: tstl.Statement[] = [];
+
+    // class('X')
+    statements.push(createClassCall(context, className, extension));
+
+    // function X:init()
+    //   X.super.init(self)
+    // end
+    const constructor = declaration.members.find(
+        (n): n is ts.ConstructorDeclaration =>
+            ts.isConstructorDeclaration(n) && n.body !== undefined
+    );
+    const transformedConstructor = transformConstructor(
+        context,
+        className,
+        instanceFields,
+        constructor
+    );
+    if (transformedConstructor) {
+        statements.push(transformedConstructor);
+    }
+
+    const methods = declaration.members
+        .filter(ts.isMethodDeclaration)
+        .map((method) => transformMethodDeclaration(context, method, className))
+        .filter((method): method is tstl.Statement => method !== undefined);
+    statements.push(...methods);
+
+    return statements;
+};
+
+const transformNewExpression: FunctionVisitor<ts.NewExpression> = (
+    node,
+    context
+) => {
+    const signature = context.checker.getResolvedSignature(node);
+    const [name, params] = transformCallAndArguments(
+        context,
+        node.expression,
+        node.arguments ?? [ts.factory.createTrue()],
+        signature
+    );
+    return tstl.createCallExpression(name, params);
+};
+
+export const transformSuperExpression: FunctionVisitor<ts.SuperExpression> = (
+    expression,
+    context: TransformationContext & { classSuperInfos?: ClassSuperInfo[] }
+) => {
+    const superInfos = context.classSuperInfos;
+    let superInfo: ClassSuperInfo | undefined = undefined;
+    if (superInfos) {
+        superInfo = superInfos[superInfos.length - 1];
+    }
+    if (!superInfo) return lua.createAnonymousIdentifier(expression);
+    const { className } = superInfo;
+
+    // Using `super` without extended type node is a TypeScript error
+    // const extendsExpression = extendedTypeNode?.expression;
+    // let baseClassName: lua.AssignmentLeftHandSideExpression | undefined;
+
+    // if (extendsExpression && ts.isIdentifier(extendsExpression)) {
+    //     const symbol = context.checker.getSymbolAtLocation(extendsExpression);
+    //     if (symbol && !isSymbolExported(context, symbol)) {
+    //         // Use "baseClassName" if base is a simple identifier
+    //         baseClassName = transformIdentifier(context, extendsExpression);
+    //     }
+    // }
+
+    // if (!baseClassName) {
+    //     // Use "className.____super" if the base is not a simple identifier
+    //     baseClassName = lua.createTableIndexExpression(
+    //         className,
+    //         lua.createStringLiteral('____super'),
+    //         expression
+    //     );
+    // }
+
+    return lua.createTableIndexExpression(
+        className,
+        lua.createStringLiteral('super')
+    );
+};
+
+const plugin = {
+    visitors: {
+        [ts.SyntaxKind.ClassDeclaration]: transformClassDeclaration,
+        [ts.SyntaxKind.SuperKeyword]: transformSuperExpression,
+        [ts.SyntaxKind.NewExpression]: transformNewExpression,
+        [ts.SyntaxKind.CallExpression]: (node, context) => {
+            if (
+                ts.isIdentifier(node.expression) &&
+                node.expression.escapedText === 'require'
+            ) {
+                console.log(1);
+                const normalNode = context.superTransformExpression(
+                    node
+                ) as unknown as {
+                    expression: { text: string; originalName: string };
+                };
+
+                normalNode.expression.text = 'import';
+                normalNode.expression.originalName = 'import';
+
+                return normalNode as unknown as lua.Expression;
+            } else {
+                return context.superTransformExpression(node);
+            }
+        },
+    },
+} satisfies tstl.Plugin;
+
+export default plugin;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "crankscript",
-    "version": "0.9.3",
+    "version": "0.9.5",
     "scripts": {
         "dev": "tsx src/index.ts",
         "post-build": "tsc-alias --project tsconfig.json"
@@ -17,9 +17,10 @@
         "ink": "^5.0.1",
         "react": "^18.3.1",
         "ts-morph": "^23.0.0",
+        "ts-node": "^10.9.2",
         "typanion": "^3.14.0",
-        "typescript-to-lua": "^1.27.0",
-        "typescript": "~5.6.2"
+        "typescript": "~5.6.2",
+        "typescript-to-lua": "^1.27.0"
     },
     "type": "module",
     "main": "./src/index.js",

package/src/commands/GenerateTypes/utils/createTypeProvider.js

@@ -79,10 +79,6 @@ export const createTypeProvider = (version)=>{
     const getStatements = ()=>{
         return provider.statements;
     };
-    const isPropertyStatic = (property)=>{
-        const { isStatic } = getPropertyDetails(property);
-        return isStatic;
-    };
     const getFunctionReturnType = (func)=>{
         const { returnType } = getFunctionDetails(func);
         return returnType;

package/src/commands/GenerateTypes/utils/createTypeProvider.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../libs/cli/src/commands/GenerateTypes/utils/createTypeProvider.ts"],"sourcesContent":["import { readFileSync } from 'fs';\nimport { existsSync, writeFileSync } from 'node:fs';\nimport { join } from 'node:path';\nimport {\n    FunctionDeclarationStructure,\n    ParameterDeclarationStructure,\n    StructureKind,\n} from 'ts-morph';\nimport { DataFolder } from '@/cli/constants.js';\nimport {\n    FunctionDescription,\n    FunctionDetails,\n    ParameterDetails,\n    PropertyDescription,\n    PropertyDetails,\n    TypeProviderData,\n} from '@/cli/types.js';\n\nfunction kebabToCamelCase(str: string): string {\n    return str.replace(/-([a-z])/g, (_, letter) => letter.toUpperCase());\n}\n\nexport const createTypeProvider = (version: string) => {\n    const path = join(DataFolder, `${version}.json`);\n    const fallbackProvider = existsSync(path)\n        ? (JSON.parse(readFileSync(path, 'utf-8')) as TypeProviderData)\n        : ({\n              globalStatements: [],\n              statements: [],\n              properties: {},\n              functions: {},\n          } satisfies TypeProviderData);\n    const provider = {\n        globalStatements: fallbackProvider.globalStatements,\n        statements: fallbackProvider.statements,\n        properties: {},\n        functions: {},\n    } as TypeProviderData;\n    const visitedProperties = new Map<string, PropertyDetails>();\n    const visitedFunctions = new Map<string, FunctionDetails>();\n\n    const getPropertyDetails = (property: PropertyDescription) => {\n        if (visitedProperties.has(property.signature)) {\n            return visitedProperties.get(property.signature) as PropertyDetails;\n        }\n\n        let result: PropertyDetails;\n        let prop = provider.properties[property.signature];\n\n        if (!prop) {\n            prop = fallbackProvider.properties[property.signature];\n        }\n\n        if (!prop) {\n            const details = {\n                signature: property.signature,\n                type: 'any',\n            } satisfies PropertyDetails;\n\n            provider.properties[property.signature] = details;\n\n            result = details;\n        } else {\n            provider.properties[property.signature] = prop;\n\n            result = prop;\n        }\n\n        visitedProperties.set(property.signature, result);\n\n        return result;\n    };\n\n    const getFunctionDetails = (func: FunctionDescription): FunctionDetails => {\n        if (visitedFunctions.has(func.signature)) {\n            return visitedFunctions.get(func.signature) as FunctionDetails;\n        }\n\n        let result: FunctionDetails;\n        let fn = provider.functions[func.signature];\n\n        if (!fn) {\n            fn = fallbackProvider.functions[func.signature];\n        }\n\n        if (!fn) {\n            const details = {\n                signature: func.signature,\n                parameters: func.parameters.map((p) => ({\n                    name: p.name,\n                    type: 'any',\n                })),\n                returnType: 'any',\n            } satisfies FunctionDetails;\n\n            provider.functions[func.signature] = details;\n\n            result = details;\n        } else {\n            provider.functions[func.signature] = fn;\n\n            result = fn;\n        }\n\n        visitedFunctions.set(func.signature, result);\n\n        return result;\n    };\n\n    const getGlobalStatements = () => {\n        return provider.globalStatements;\n    };\n\n    const getStatements = () => {\n        return provider.statements;\n    };\n\n    const isPropertyStatic = (property: PropertyDescription) => {\n        const { isStatic } = getPropertyDetails(property);\n\n        return isStatic;\n    };\n\n    const getFunctionReturnType = (func: FunctionDescription) => {\n        const { returnType } = getFunctionDetails(func);\n\n        return returnType;\n    };\n\n    const getParameterDetails = (\n        func: FunctionDescription,\n        parameter: string\n    ) => {\n        const { parameters } = getFunctionDetails(func);\n        const param = parameters.find((p) => p.name === parameter);\n\n        if (!param) {\n            return {\n                name: parameter,\n                type: 'any',\n            } satisfies ParameterDetails;\n        }\n\n        return param;\n    };\n\n    const getParameters = (\n        func: FunctionDescription\n    ): FunctionDeclarationStructure['parameters'] => {\n        const { overrideParameters = false, parameters } =\n            getFunctionDetails(func);\n        const getParameterFromDetails = (parameter: ParameterDetails) => {\n            return {\n                kind: StructureKind.Parameter,\n                name: kebabToCamelCase(parameter.name),\n                type: parameter.type,\n                ...(parameter.overrideOptions ?? {}),\n            } satisfies ParameterDeclarationStructure;\n        };\n\n        if (overrideParameters) {\n            return parameters.map((details) => {\n                return getParameterFromDetails(details);\n            });\n        }\n\n        return func.parameters.map((parameter) => {\n            const details = getParameterDetails(func, parameter.name);\n\n            return {\n                hasQuestionToken: !parameter.required,\n                ...getParameterFromDetails(details),\n            } satisfies ParameterDeclarationStructure;\n        });\n    };\n\n    const getFunctionOverrideOptions = (func: FunctionDescription) => {\n        return getFunctionDetails(func).overrideOptions ?? {};\n    };\n\n    const save = () => {\n        const contents = JSON.stringify(provider, null, 4) + '\\n';\n\n        writeFileSync(path, contents, 'utf-8');\n    };\n\n    return {\n        getGlobalStatements,\n        getStatements,\n        getPropertyDetails,\n        getFunctionReturnType,\n        getParameterDetails,\n        getParameters,\n        getFunctionOverrideOptions,\n        save,\n    };\n};\n"],"names":["readFileSync","existsSync","writeFileSync","join","StructureKind","DataFolder","kebabToCamelCase","str","replace","_","letter","toUpperCase","createTypeProvider","version","path","fallbackProvider","JSON","parse","globalStatements","statements","properties","functions","provider","visitedProperties","Map","visitedFunctions","getPropertyDetails","property","has","signature","get","result","prop","details","type","set","getFunctionDetails","func","fn","parameters","map","p","name","returnType","getGlobalStatements","getStatements","isPropertyStatic","isStatic","getFunctionReturnType","getParameterDetails","parameter","param","find","getParameters","overrideParameters","getParameterFromDetails","kind","Parameter","overrideOptions","hasQuestionToken","required","getFunctionOverrideOptions","save","contents","stringify"],"rangeMappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;","mappings":";AAAA,SAASA,YAAY,QAAQ,KAAK;AAClC,SAASC,UAAU,EAAEC,aAAa,QAAQ,UAAU;AACpD,SAASC,IAAI,QAAQ,YAAY;AACjC,SAGIC,aAAa,QACV,WAAW;AAClB,SAASC,UAAU,QAAQ,qBAAqB;AAUhD,SAASC,iBAAiBC,GAAW;IACjC,OAAOA,IAAIC,OAAO,CAAC,aAAa,CAACC,GAAGC,SAAWA,OAAOC,WAAW;AACrE;AAEA,OAAO,MAAMC,qBAAqB,CAACC;IAC/B,MAAMC,OAAOX,KAAKE,YAAY,CAAC,EAAEQ,QAAQ,KAAK,CAAC;IAC/C,MAAME,mBAAmBd,WAAWa,QAC7BE,KAAKC,KAAK,CAACjB,aAAac,MAAM,YAC9B;QACGI,kBAAkB,EAAE;QACpBC,YAAY,EAAE;QACdC,YAAY,CAAC;QACbC,WAAW,CAAC;IAChB;IACN,MAAMC,WAAW;QACbJ,kBAAkBH,iBAAiBG,gBAAgB;QACnDC,YAAYJ,iBAAiBI,UAAU;QACvCC,YAAY,CAAC;QACbC,WAAW,CAAC;IAChB;IACA,MAAME,oBAAoB,IAAIC;IAC9B,MAAMC,mBAAmB,IAAID;IAE7B,MAAME,qBAAqB,CAACC;QACxB,IAAIJ,kBAAkBK,GAAG,CAACD,SAASE,SAAS,GAAG;YAC3C,OAAON,kBAAkBO,GAAG,CAACH,SAASE,SAAS;QACnD;QAEA,IAAIE;QACJ,IAAIC,OAAOV,SAASF,UAAU,CAACO,SAASE,SAAS,CAAC;QAElD,IAAI,CAACG,MAAM;YACPA,OAAOjB,iBAAiBK,UAAU,CAACO,SAASE,SAAS,CAAC;QAC1D;QAEA,IAAI,CAACG,MAAM;YACP,MAAMC,UAAU;gBACZJ,WAAWF,SAASE,SAAS;gBAC7BK,MAAM;YACV;YAEAZ,SAASF,UAAU,CAACO,SAASE,SAAS,CAAC,GAAGI;YAE1CF,SAASE;QACb,OAAO;YACHX,SAASF,UAAU,CAACO,SAASE,SAAS,CAAC,GAAGG;YAE1CD,SAASC;QACb;QAEAT,kBAAkBY,GAAG,CAACR,SAASE,SAAS,EAAEE;QAE1C,OAAOA;IACX;IAEA,MAAMK,qBAAqB,CAACC;QACxB,IAAIZ,iBAAiBG,GAAG,CAACS,KAAKR,SAAS,GAAG;YACtC,OAAOJ,iBAAiBK,GAAG,CAACO,KAAKR,SAAS;QAC9C;QAEA,IAAIE;QACJ,IAAIO,KAAKhB,SAASD,SAAS,CAACgB,KAAKR,SAAS,CAAC;QAE3C,IAAI,CAACS,IAAI;YACLA,KAAKvB,iBAAiBM,SAAS,CAACgB,KAAKR,SAAS,CAAC;QACnD;QAEA,IAAI,CAACS,IAAI;YACL,MAAML,UAAU;gBACZJ,WAAWQ,KAAKR,SAAS;gBACzBU,YAAYF,KAAKE,UAAU,CAACC,GAAG,CAAC,CAACC,IAAO,CAAA;wBACpCC,MAAMD,EAAEC,IAAI;wBACZR,MAAM;oBACV,CAAA;gBACAS,YAAY;YAChB;YAEArB,SAASD,SAAS,CAACgB,KAAKR,SAAS,CAAC,GAAGI;YAErCF,SAASE;QACb,OAAO;YACHX,SAASD,SAAS,CAACgB,KAAKR,SAAS,CAAC,GAAGS;YAErCP,SAASO;QACb;QAEAb,iBAAiBU,GAAG,CAACE,KAAKR,SAAS,EAAEE;QAErC,OAAOA;IACX;IAEA,MAAMa,sBAAsB;QACxB,OAAOtB,SAASJ,gBAAgB;IACpC;IAEA,MAAM2B,gBAAgB;QAClB,OAAOvB,SAASH,UAAU;IAC9B;IAEA,MAAM2B,mBAAmB,CAACnB;QACtB,MAAM,EAAEoB,QAAQ,EAAE,GAAGrB,mBAAmBC;QAExC,OAAOoB;IACX;IAEA,MAAMC,wBAAwB,CAACX;QAC3B,MAAM,EAAEM,UAAU,EAAE,GAAGP,mBAAmBC;QAE1C,OAAOM;IACX;IAEA,MAAMM,sBAAsB,CACxBZ,MACAa;QAEA,MAAM,EAAEX,UAAU,EAAE,GAAGH,mBAAmBC;QAC1C,MAAMc,QAAQZ,WAAWa,IAAI,CAAC,CAACX,IAAMA,EAAEC,IAAI,KAAKQ;QAEhD,IAAI,CAACC,OAAO;YACR,OAAO;gBACHT,MAAMQ;gBACNhB,MAAM;YACV;QACJ;QAEA,OAAOiB;IACX;IAEA,MAAME,gBAAgB,CAClBhB;QAEA,MAAM,EAAEiB,qBAAqB,KAAK,EAAEf,UAAU,EAAE,GAC5CH,mBAAmBC;QACvB,MAAMkB,0BAA0B,CAACL;gBAKrBA;YAJR,OAAO;gBACHM,MAAMpD,cAAcqD,SAAS;gBAC7Bf,MAAMpC,iBAAiB4C,UAAUR,IAAI;gBACrCR,MAAMgB,UAAUhB,IAAI;eAChBgB,CAAAA,6BAAAA,UAAUQ,eAAe,YAAzBR,6BAA6B,CAAC;QAE1C;QAEA,IAAII,oBAAoB;YACpB,OAAOf,WAAWC,GAAG,CAAC,CAACP;gBACnB,OAAOsB,wBAAwBtB;YACnC;QACJ;QAEA,OAAOI,KAAKE,UAAU,CAACC,GAAG,CAAC,CAACU;YACxB,MAAMjB,UAAUgB,oBAAoBZ,MAAMa,UAAUR,IAAI;YAExD,OAAO;gBACHiB,kBAAkB,CAACT,UAAUU,QAAQ;eAClCL,wBAAwBtB;QAEnC;IACJ;IAEA,MAAM4B,6BAA6B,CAACxB;YACzBD;QAAP,OAAOA,CAAAA,sCAAAA,mBAAmBC,MAAMqB,eAAe,YAAxCtB,sCAA4C,CAAC;IACxD;IAEA,MAAM0B,OAAO;QACT,MAAMC,WAAW/C,KAAKgD,SAAS,CAAC1C,UAAU,MAAM,KAAK;QAErDpB,cAAcY,MAAMiD,UAAU;IAClC;IAEA,OAAO;QACHnB;QACAC;QACAnB;QACAsB;QACAC;QACAI;QACAQ;QACAC;IACJ;AACJ,EAAE"}
+{"version":3,"sources":["../../../../../../../libs/cli/src/commands/GenerateTypes/utils/createTypeProvider.ts"],"sourcesContent":["import { readFileSync } from 'fs';\nimport { existsSync, writeFileSync } from 'node:fs';\nimport { join } from 'node:path';\nimport {\n    FunctionDeclarationStructure,\n    ParameterDeclarationStructure,\n    StructureKind,\n} from 'ts-morph';\nimport { DataFolder } from '@/cli/constants.js';\nimport {\n    FunctionDescription,\n    FunctionDetails,\n    ParameterDetails,\n    PropertyDescription,\n    PropertyDetails,\n    TypeProviderData,\n} from '@/cli/types.js';\n\nfunction kebabToCamelCase(str: string): string {\n    return str.replace(/-([a-z])/g, (_, letter) => letter.toUpperCase());\n}\n\nexport const createTypeProvider = (version: string) => {\n    const path = join(DataFolder, `${version}.json`);\n    const fallbackProvider = existsSync(path)\n        ? (JSON.parse(readFileSync(path, 'utf-8')) as TypeProviderData)\n        : ({\n              globalStatements: [],\n              statements: [],\n              properties: {},\n              functions: {},\n          } satisfies TypeProviderData);\n    const provider = {\n        globalStatements: fallbackProvider.globalStatements,\n        statements: fallbackProvider.statements,\n        properties: {},\n        functions: {},\n    } as TypeProviderData;\n    const visitedProperties = new Map<string, PropertyDetails>();\n    const visitedFunctions = new Map<string, FunctionDetails>();\n\n    const getPropertyDetails = (property: PropertyDescription) => {\n        if (visitedProperties.has(property.signature)) {\n            return visitedProperties.get(property.signature) as PropertyDetails;\n        }\n\n        let result: PropertyDetails;\n        let prop = provider.properties[property.signature];\n\n        if (!prop) {\n            prop = fallbackProvider.properties[property.signature];\n        }\n\n        if (!prop) {\n            const details = {\n                signature: property.signature,\n                type: 'any',\n            } satisfies PropertyDetails;\n\n            provider.properties[property.signature] = details;\n\n            result = details;\n        } else {\n            provider.properties[property.signature] = prop;\n\n            result = prop;\n        }\n\n        visitedProperties.set(property.signature, result);\n\n        return result;\n    };\n\n    const getFunctionDetails = (func: FunctionDescription): FunctionDetails => {\n        if (visitedFunctions.has(func.signature)) {\n            return visitedFunctions.get(func.signature) as FunctionDetails;\n        }\n\n        let result: FunctionDetails;\n        let fn = provider.functions[func.signature];\n\n        if (!fn) {\n            fn = fallbackProvider.functions[func.signature];\n        }\n\n        if (!fn) {\n            const details = {\n                signature: func.signature,\n                parameters: func.parameters.map((p) => ({\n                    name: p.name,\n                    type: 'any',\n                })),\n                returnType: 'any',\n            } satisfies FunctionDetails;\n\n            provider.functions[func.signature] = details;\n\n            result = details;\n        } else {\n            provider.functions[func.signature] = fn;\n\n            result = fn;\n        }\n\n        visitedFunctions.set(func.signature, result);\n\n        return result;\n    };\n\n    const getGlobalStatements = () => {\n        return provider.globalStatements;\n    };\n\n    const getStatements = () => {\n        return provider.statements;\n    };\n\n    const getFunctionReturnType = (func: FunctionDescription) => {\n        const { returnType } = getFunctionDetails(func);\n\n        return returnType;\n    };\n\n    const getParameterDetails = (\n        func: FunctionDescription,\n        parameter: string\n    ) => {\n        const { parameters } = getFunctionDetails(func);\n        const param = parameters.find((p) => p.name === parameter);\n\n        if (!param) {\n            return {\n                name: parameter,\n                type: 'any',\n            } satisfies ParameterDetails;\n        }\n\n        return param;\n    };\n\n    const getParameters = (\n        func: FunctionDescription\n    ): FunctionDeclarationStructure['parameters'] => {\n        const { overrideParameters = false, parameters } =\n            getFunctionDetails(func);\n        const getParameterFromDetails = (parameter: ParameterDetails) => {\n            return {\n                kind: StructureKind.Parameter,\n                name: kebabToCamelCase(parameter.name),\n                type: parameter.type,\n                ...(parameter.overrideOptions ?? {}),\n            } satisfies ParameterDeclarationStructure;\n        };\n\n        if (overrideParameters) {\n            return parameters.map((details) => {\n                return getParameterFromDetails(details);\n            });\n        }\n\n        return func.parameters.map((parameter) => {\n            const details = getParameterDetails(func, parameter.name);\n\n            return {\n                hasQuestionToken: !parameter.required,\n                ...getParameterFromDetails(details),\n            } satisfies ParameterDeclarationStructure;\n        });\n    };\n\n    const getFunctionOverrideOptions = (func: FunctionDescription) => {\n        return getFunctionDetails(func).overrideOptions ?? {};\n    };\n\n    const save = () => {\n        const contents = JSON.stringify(provider, null, 4) + '\\n';\n\n        writeFileSync(path, contents, 'utf-8');\n    };\n\n    return {\n        getGlobalStatements,\n        getStatements,\n        getPropertyDetails,\n        getFunctionReturnType,\n        getParameterDetails,\n        getParameters,\n        getFunctionOverrideOptions,\n        save,\n    };\n};\n"],"names":["readFileSync","existsSync","writeFileSync","join","StructureKind","DataFolder","kebabToCamelCase","str","replace","_","letter","toUpperCase","createTypeProvider","version","path","fallbackProvider","JSON","parse","globalStatements","statements","properties","functions","provider","visitedProperties","Map","visitedFunctions","getPropertyDetails","property","has","signature","get","result","prop","details","type","set","getFunctionDetails","func","fn","parameters","map","p","name","returnType","getGlobalStatements","getStatements","getFunctionReturnType","getParameterDetails","parameter","param","find","getParameters","overrideParameters","getParameterFromDetails","kind","Parameter","overrideOptions","hasQuestionToken","required","getFunctionOverrideOptions","save","contents","stringify"],"rangeMappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;","mappings":";AAAA,SAASA,YAAY,QAAQ,KAAK;AAClC,SAASC,UAAU,EAAEC,aAAa,QAAQ,UAAU;AACpD,SAASC,IAAI,QAAQ,YAAY;AACjC,SAGIC,aAAa,QACV,WAAW;AAClB,SAASC,UAAU,QAAQ,qBAAqB;AAUhD,SAASC,iBAAiBC,GAAW;IACjC,OAAOA,IAAIC,OAAO,CAAC,aAAa,CAACC,GAAGC,SAAWA,OAAOC,WAAW;AACrE;AAEA,OAAO,MAAMC,qBAAqB,CAACC;IAC/B,MAAMC,OAAOX,KAAKE,YAAY,CAAC,EAAEQ,QAAQ,KAAK,CAAC;IAC/C,MAAME,mBAAmBd,WAAWa,QAC7BE,KAAKC,KAAK,CAACjB,aAAac,MAAM,YAC9B;QACGI,kBAAkB,EAAE;QACpBC,YAAY,EAAE;QACdC,YAAY,CAAC;QACbC,WAAW,CAAC;IAChB;IACN,MAAMC,WAAW;QACbJ,kBAAkBH,iBAAiBG,gBAAgB;QACnDC,YAAYJ,iBAAiBI,UAAU;QACvCC,YAAY,CAAC;QACbC,WAAW,CAAC;IAChB;IACA,MAAME,oBAAoB,IAAIC;IAC9B,MAAMC,mBAAmB,IAAID;IAE7B,MAAME,qBAAqB,CAACC;QACxB,IAAIJ,kBAAkBK,GAAG,CAACD,SAASE,SAAS,GAAG;YAC3C,OAAON,kBAAkBO,GAAG,CAACH,SAASE,SAAS;QACnD;QAEA,IAAIE;QACJ,IAAIC,OAAOV,SAASF,UAAU,CAACO,SAASE,SAAS,CAAC;QAElD,IAAI,CAACG,MAAM;YACPA,OAAOjB,iBAAiBK,UAAU,CAACO,SAASE,SAAS,CAAC;QAC1D;QAEA,IAAI,CAACG,MAAM;YACP,MAAMC,UAAU;gBACZJ,WAAWF,SAASE,SAAS;gBAC7BK,MAAM;YACV;YAEAZ,SAASF,UAAU,CAACO,SAASE,SAAS,CAAC,GAAGI;YAE1CF,SAASE;QACb,OAAO;YACHX,SAASF,UAAU,CAACO,SAASE,SAAS,CAAC,GAAGG;YAE1CD,SAASC;QACb;QAEAT,kBAAkBY,GAAG,CAACR,SAASE,SAAS,EAAEE;QAE1C,OAAOA;IACX;IAEA,MAAMK,qBAAqB,CAACC;QACxB,IAAIZ,iBAAiBG,GAAG,CAACS,KAAKR,SAAS,GAAG;YACtC,OAAOJ,iBAAiBK,GAAG,CAACO,KAAKR,SAAS;QAC9C;QAEA,IAAIE;QACJ,IAAIO,KAAKhB,SAASD,SAAS,CAACgB,KAAKR,SAAS,CAAC;QAE3C,IAAI,CAACS,IAAI;YACLA,KAAKvB,iBAAiBM,SAAS,CAACgB,KAAKR,SAAS,CAAC;QACnD;QAEA,IAAI,CAACS,IAAI;YACL,MAAML,UAAU;gBACZJ,WAAWQ,KAAKR,SAAS;gBACzBU,YAAYF,KAAKE,UAAU,CAACC,GAAG,CAAC,CAACC,IAAO,CAAA;wBACpCC,MAAMD,EAAEC,IAAI;wBACZR,MAAM;oBACV,CAAA;gBACAS,YAAY;YAChB;YAEArB,SAASD,SAAS,CAACgB,KAAKR,SAAS,CAAC,GAAGI;YAErCF,SAASE;QACb,OAAO;YACHX,SAASD,SAAS,CAACgB,KAAKR,SAAS,CAAC,GAAGS;YAErCP,SAASO;QACb;QAEAb,iBAAiBU,GAAG,CAACE,KAAKR,SAAS,EAAEE;QAErC,OAAOA;IACX;IAEA,MAAMa,sBAAsB;QACxB,OAAOtB,SAASJ,gBAAgB;IACpC;IAEA,MAAM2B,gBAAgB;QAClB,OAAOvB,SAASH,UAAU;IAC9B;IAEA,MAAM2B,wBAAwB,CAACT;QAC3B,MAAM,EAAEM,UAAU,EAAE,GAAGP,mBAAmBC;QAE1C,OAAOM;IACX;IAEA,MAAMI,sBAAsB,CACxBV,MACAW;QAEA,MAAM,EAAET,UAAU,EAAE,GAAGH,mBAAmBC;QAC1C,MAAMY,QAAQV,WAAWW,IAAI,CAAC,CAACT,IAAMA,EAAEC,IAAI,KAAKM;QAEhD,IAAI,CAACC,OAAO;YACR,OAAO;gBACHP,MAAMM;gBACNd,MAAM;YACV;QACJ;QAEA,OAAOe;IACX;IAEA,MAAME,gBAAgB,CAClBd;QAEA,MAAM,EAAEe,qBAAqB,KAAK,EAAEb,UAAU,EAAE,GAC5CH,mBAAmBC;QACvB,MAAMgB,0BAA0B,CAACL;gBAKrBA;YAJR,OAAO;gBACHM,MAAMlD,cAAcmD,SAAS;gBAC7Bb,MAAMpC,iBAAiB0C,UAAUN,IAAI;gBACrCR,MAAMc,UAAUd,IAAI;eAChBc,CAAAA,6BAAAA,UAAUQ,eAAe,YAAzBR,6BAA6B,CAAC;QAE1C;QAEA,IAAII,oBAAoB;YACpB,OAAOb,WAAWC,GAAG,CAAC,CAACP;gBACnB,OAAOoB,wBAAwBpB;YACnC;QACJ;QAEA,OAAOI,KAAKE,UAAU,CAACC,GAAG,CAAC,CAACQ;YACxB,MAAMf,UAAUc,oBAAoBV,MAAMW,UAAUN,IAAI;YAExD,OAAO;gBACHe,kBAAkB,CAACT,UAAUU,QAAQ;eAClCL,wBAAwBpB;QAEnC;IACJ;IAEA,MAAM0B,6BAA6B,CAACtB;YACzBD;QAAP,OAAOA,CAAAA,sCAAAA,mBAAmBC,MAAMmB,eAAe,YAAxCpB,sCAA4C,CAAC;IACxD;IAEA,MAAMwB,OAAO;QACT,MAAMC,WAAW7C,KAAK8C,SAAS,CAACxC,UAAU,MAAM,KAAK;QAErDpB,cAAcY,MAAM+C,UAAU;IAClC;IAEA,OAAO;QACHjB;QACAC;QACAnB;QACAoB;QACAC;QACAI;QACAQ;QACAC;IACJ;AACJ,EAAE"}