typescript-to-lua 1.3.2 → 1.4.0

package/dist/lualib-build/plugin.js

@@ -0,0 +1,138 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.lualibDiagnostic = void 0;
+const ts = require("typescript");
+const tstl = require("..");
+const path = require("path");
+const lualib_1 = require("../transformation/utils/lualib");
+const LuaLib_1 = require("../LuaLib");
+const util_1 = require("./util");
+const utils_1 = require("../utils");
+exports.lualibDiagnostic = (0, utils_1.createDiagnosticFactoryWithCode)(200000, (message, file) => ({
+    messageText: message,
+    file,
+    start: file && 0,
+    length: file && 0,
+}));
+class LuaLibPlugin {
+    constructor() {
+        // Plugin members
+        this.visitors = {
+            [ts.SyntaxKind.SourceFile]: this.lualibFileVisitor.bind(this),
+        };
+        this.printer = (program, emitHost, fileName, file) => new LuaLibPrinter(emitHost, program, fileName).print(file);
+        // Internals
+        this.featureExports = new Map();
+        this.featureDependencies = new Map();
+    }
+    afterPrint(program, options, emitHost, result) {
+        var _a;
+        void options;
+        // Write lualib dependency json
+        const { result: luaLibModuleInfo, diagnostics } = this.createLuaLibModulesInfo();
+        const emitBOM = (_a = options.emitBOM) !== null && _a !== void 0 ? _a : false;
+        emitHost.writeFile(path.join(tstl.getEmitOutDir(program), LuaLib_1.luaLibModulesInfoFileName), JSON.stringify(luaLibModuleInfo, null, 2), emitBOM);
+        // Create map of result files keyed by their 'lualib name'
+        const exportedLualibFeatures = new Map(result.map(f => [path.basename(f.fileName).split(".")[0], f.code]));
+        // Figure out the order required in the bundle by recursively resolving all dependency features
+        const allFeatures = Object.values(LuaLib_1.LuaLibFeature);
+        const orderedFeatures = (0, LuaLib_1.resolveRecursiveLualibFeatures)(allFeatures, emitHost, luaLibModuleInfo);
+        // Concatenate lualib files into bundle with exports table and add lualib_bundle.lua to results
+        let lualibBundle = orderedFeatures.map(f => exportedLualibFeatures.get(LuaLib_1.LuaLibFeature[f])).join("\n");
+        const exports = allFeatures.flatMap(feature => luaLibModuleInfo[feature].exports);
+        lualibBundle += `\nreturn {\n${exports.map(exportName => `  ${exportName} = ${exportName}`).join(",\n")}\n}\n`;
+        result.push({ fileName: "lualib_bundle.lua", code: lualibBundle });
+        return diagnostics;
+    }
+    lualibFileVisitor(file, context) {
+        const featureName = path.basename(file.fileName, ".ts");
+        if (!(featureName in tstl.LuaLibFeature)) {
+            context.diagnostics.push((0, exports.lualibDiagnostic)(`File is not a lualib feature: ${featureName}`, file));
+        }
+        // Transpile file as normal with tstl
+        const fileResult = context.superTransformNode(file)[0];
+        const usedFeatures = new Set((0, lualib_1.getUsedLuaLibFeatures)(context));
+        // Get all imports in file
+        const importNames = new Set();
+        const imports = file.statements.filter(ts.isImportDeclaration);
+        for (const { importClause, moduleSpecifier } of imports) {
+            if ((importClause === null || importClause === void 0 ? void 0 : importClause.namedBindings) && ts.isNamedImports(importClause.namedBindings)) {
+                for (const { name } of importClause.namedBindings.elements) {
+                    importNames.add(name.text);
+                }
+            }
+            // track lualib imports
+            if (ts.isStringLiteral(moduleSpecifier)) {
+                const featureName = path.basename(moduleSpecifier.text, ".ts");
+                if (featureName in tstl.LuaLibFeature) {
+                    usedFeatures.add(featureName);
+                }
+            }
+        }
+        const filteredStatements = fileResult.statements
+            .filter(s => !(0, util_1.isExportTableDeclaration)(s) && !(0, util_1.isRequire)(s) && !(0, util_1.isImport)(s, importNames) && !(0, util_1.isExportsReturn)(s))
+            .map(statement => {
+            if ((0, util_1.isExportAlias)(statement)) {
+                const name = statement.left[0];
+                const exportName = statement.right[0].index.value;
+                if (name.text === exportName)
+                    return undefined; // Remove "x = x" statements
+                return tstl.createAssignmentStatement(name, tstl.createIdentifier(exportName));
+            }
+            return statement;
+        })
+            .filter(statement => statement !== undefined);
+        const exportNames = filteredStatements.filter(util_1.isExportAssignment).map(s => s.left[0].index.value);
+        if (!filteredStatements.every(util_1.isExportAssignment)) {
+            // If there are local statements, wrap them in a do ... end with exports outside
+            const exports = tstl.createVariableDeclarationStatement(exportNames.map(k => tstl.createIdentifier(k)));
+            // transform export assignments to local assignments
+            const bodyStatements = filteredStatements.map(s => (0, util_1.isExportAssignment)(s)
+                ? tstl.createAssignmentStatement(tstl.createIdentifier(s.left[0].index.value), s.right[0])
+                : s);
+            fileResult.statements = [exports, tstl.createDoStatement(bodyStatements)];
+        }
+        else {
+            // transform export assignments to local variable declarations
+            fileResult.statements = filteredStatements.map(s => tstl.createVariableDeclarationStatement(tstl.createIdentifier(s.left[0].index.value), s.right[0]));
+        }
+        // Save dependency information
+        this.featureExports.set(featureName, new Set(exportNames));
+        if (usedFeatures.size > 0) {
+            this.featureDependencies.set(featureName, usedFeatures);
+        }
+        return fileResult;
+    }
+    createLuaLibModulesInfo() {
+        const result = {};
+        const diagnostics = [];
+        for (const feature of Object.values(tstl.LuaLibFeature)) {
+            const exports = this.featureExports.get(feature);
+            if (!exports) {
+                diagnostics.push((0, exports.lualibDiagnostic)(`Missing file for lualib feature: ${feature}`));
+                continue;
+            }
+            const dependencies = this.featureDependencies.get(feature);
+            result[feature] = {
+                exports: Array.from(exports),
+                dependencies: dependencies ? Array.from(dependencies) : undefined,
+            };
+        }
+        return { result: result, diagnostics };
+    }
+}
+class LuaLibPrinter extends tstl.LuaPrinter {
+    // Strip all exports during print
+    printTableIndexExpression(expression) {
+        if (tstl.isIdentifier(expression.table) &&
+            expression.table.text === "____exports" &&
+            tstl.isStringLiteral(expression.index)) {
+            return super.printExpression(tstl.createIdentifier(expression.index.value));
+        }
+        return super.printTableIndexExpression(expression);
+    }
+}
+const pluginInstance = new LuaLibPlugin();
+// eslint-disable-next-line import/no-default-export
+exports.default = pluginInstance;
+//# sourceMappingURL=plugin.js.map

package/dist/lualib-build/util.d.ts

@@ -0,0 +1,19 @@
+import * as tstl from "..";
+export declare function isExportTableDeclaration(node: tstl.Node): node is tstl.VariableDeclarationStatement & {
+    left: [];
+};
+export declare function isExportTable(node: tstl.Node): node is tstl.Identifier;
+export declare type ExportTableIndex = tstl.TableIndexExpression & {
+    index: tstl.StringLiteral;
+};
+export declare function isExportTableIndex(node: tstl.Node): node is ExportTableIndex;
+export declare function isExportAlias(node: tstl.Node): node is tstl.VariableDeclarationStatement & {
+    right: [ExportTableIndex];
+};
+export declare type ExportAssignment = tstl.AssignmentStatement & {
+    left: [ExportTableIndex];
+};
+export declare function isExportAssignment(node: tstl.Node): node is ExportAssignment;
+export declare function isRequire(node: tstl.Node): boolean | undefined;
+export declare function isImport(node: tstl.Node, importNames: Set<string>): boolean;
+export declare function isExportsReturn(node: tstl.Node): boolean;

package/dist/lualib-build/util.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isExportsReturn = exports.isImport = exports.isRequire = exports.isExportAssignment = exports.isExportAlias = exports.isExportTableIndex = exports.isExportTable = exports.isExportTableDeclaration = void 0;
+const tstl = require("..");
+function isExportTableDeclaration(node) {
+    return tstl.isVariableDeclarationStatement(node) && isExportTable(node.left[0]);
+}
+exports.isExportTableDeclaration = isExportTableDeclaration;
+function isExportTable(node) {
+    return tstl.isIdentifier(node) && node.text === "____exports";
+}
+exports.isExportTable = isExportTable;
+function isExportTableIndex(node) {
+    return tstl.isTableIndexExpression(node) && isExportTable(node.table) && tstl.isStringLiteral(node.index);
+}
+exports.isExportTableIndex = isExportTableIndex;
+function isExportAlias(node) {
+    return tstl.isVariableDeclarationStatement(node) && node.right !== undefined && isExportTableIndex(node.right[0]);
+}
+exports.isExportAlias = isExportAlias;
+function isExportAssignment(node) {
+    return tstl.isAssignmentStatement(node) && isExportTableIndex(node.left[0]);
+}
+exports.isExportAssignment = isExportAssignment;
+function isRequire(node) {
+    return (tstl.isVariableDeclarationStatement(node) &&
+        node.right &&
+        tstl.isCallExpression(node.right[0]) &&
+        tstl.isIdentifier(node.right[0].expression) &&
+        node.right[0].expression.text === "require");
+}
+exports.isRequire = isRequire;
+function isImport(node, importNames) {
+    return tstl.isVariableDeclarationStatement(node) && importNames.has(node.left[0].text);
+}
+exports.isImport = isImport;
+function isExportsReturn(node) {
+    return (tstl.isReturnStatement(node) &&
+        tstl.isIdentifier(node.expressions[0]) &&
+        node.expressions[0].text === "____exports");
+}
+exports.isExportsReturn = isExportsReturn;
+//# sourceMappingURL=util.js.map

package/dist/transformation/builtins/function.js

@@ -19,7 +19,8 @@ function transformFunctionPrototypeCall(context, node) {
     const expressionName = expression.name.text;
     switch (expressionName) {
         case "apply":
-            return lua.createCallExpression(caller, [params[0], (0, lua_ast_1.createUnpackCall)(context, params[1], node.arguments[1])], node);
+            const nonContextArgs = params.length > 1 ? [(0, lua_ast_1.createUnpackCall)(context, params[1], node.arguments[1])] : [];
+            return lua.createCallExpression(caller, [params[0], ...nonContextArgs], node);
         case "bind":
             return (0, lualib_1.transformLuaLibFunction)(context, lualib_1.LuaLibFeature.FunctionBind, node, caller, ...params);
         case "call":

package/dist/transformation/builtins/index.d.ts

@@ -4,3 +4,4 @@ import { TransformationContext } from "../context";
 export declare function transformBuiltinPropertyAccessExpression(context: TransformationContext, node: ts.PropertyAccessExpression): lua.Expression | undefined;
 export declare function transformBuiltinCallExpression(context: TransformationContext, node: ts.CallExpression, isOptionalCall: boolean): lua.Expression | undefined;
 export declare function transformBuiltinIdentifierExpression(context: TransformationContext, node: ts.Identifier): lua.Expression | undefined;
+export declare function checkForLuaLibType(context: TransformationContext, type: ts.Type): void;

package/dist/transformation/builtins/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.transformBuiltinIdentifierExpression = exports.transformBuiltinCallExpression = exports.transformBuiltinPropertyAccessExpression = void 0;
+exports.checkForLuaLibType = exports.transformBuiltinIdentifierExpression = exports.transformBuiltinCallExpression = exports.transformBuiltinPropertyAccessExpression = void 0;
 const ts = require("typescript");
 const lua = require("../../LuaAST");
 const utils_1 = require("../../utils");
@@ -8,7 +8,6 @@ const lua_ast_1 = require("../utils/lua-ast");
 const lualib_1 = require("../utils/lualib");
 const symbols_1 = require("../utils/symbols");
 const typescript_1 = require("../utils/typescript");
-const new_1 = require("../visitors/class/new");
 const array_1 = require("./array");
 const console_1 = require("./console");
 const function_1 = require("./function");
@@ -22,6 +21,14 @@ const symbol_1 = require("./symbol");
 const diagnostics_1 = require("../utils/diagnostics");
 function transformBuiltinPropertyAccessExpression(context, node) {
     const ownerType = context.checker.getTypeAtLocation(node.expression);
+    if (ts.isIdentifier(node.expression) && (0, typescript_1.isStandardLibraryType)(context, ownerType, undefined)) {
+        switch (node.expression.text) {
+            case "Math":
+                return (0, math_1.transformMathProperty)(context, node);
+            case "Symbol":
+                (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Symbol);
+        }
+    }
     if ((0, typescript_1.isStringType)(context, ownerType)) {
         return (0, string_1.transformStringProperty)(context, node);
     }
@@ -31,14 +38,6 @@ function transformBuiltinPropertyAccessExpression(context, node) {
     if ((0, typescript_1.isFunctionType)(ownerType)) {
         return (0, function_1.transformFunctionProperty)(context, node);
     }
-    if (ts.isIdentifier(node.expression) && (0, typescript_1.isStandardLibraryType)(context, ownerType, undefined)) {
-        switch (node.expression.text) {
-            case "Math":
-                return (0, math_1.transformMathProperty)(context, node);
-            case "Symbol":
-                (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Symbol);
-        }
-    }
 }
 exports.transformBuiltinPropertyAccessExpression = transformBuiltinPropertyAccessExpression;
 function transformBuiltinCallExpression(context, node, isOptionalCall) {
@@ -48,8 +47,7 @@ function transformBuiltinCallExpression(context, node, isOptionalCall) {
     };
     const expressionType = context.checker.getTypeAtLocation(node.expression);
     if (ts.isIdentifier(node.expression) && (0, typescript_1.isStandardLibraryType)(context, expressionType, undefined)) {
-        // TODO:
-        (0, new_1.checkForLuaLibType)(context, expressionType);
+        checkForLuaLibType(context, expressionType);
         const result = (0, global_1.transformGlobalCall)(context, node);
         if (result) {
             if (isOptionalCall)
@@ -102,22 +100,30 @@ function transformBuiltinCallExpression(context, node, isOptionalCall) {
                 return (0, promise_1.transformPromiseConstructorCall)(context, node);
         }
     }
-    if ((0, typescript_1.isStringType)(context, ownerType) && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
+    const isStringFunction = (0, typescript_1.isStringType)(context, ownerType) ||
+        (expression.questionDotToken && (0, typescript_1.isNullableType)(context, ownerType, typescript_1.isStringType));
+    if (isStringFunction && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
         if (isOptionalCall)
             return unsupportedOptionalCall();
         return (0, string_1.transformStringPrototypeCall)(context, node);
     }
-    if ((0, typescript_1.isNumberType)(context, ownerType) && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
+    const isNumberFunction = (0, typescript_1.isNumberType)(context, ownerType) ||
+        (expression.questionDotToken && (0, typescript_1.isNullableType)(context, ownerType, typescript_1.isNumberType));
+    if (isNumberFunction && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
         if (isOptionalCall)
             return unsupportedOptionalCall();
         return (0, number_1.transformNumberPrototypeCall)(context, node);
     }
-    if ((0, typescript_1.isArrayType)(context, ownerType) && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
+    const isArrayFunction = (0, typescript_1.isArrayType)(context, ownerType) ||
+        (expression.questionDotToken && (0, typescript_1.isNullableType)(context, ownerType, typescript_1.isArrayType));
+    if (isArrayFunction && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
         if (isOptionalCall)
             return unsupportedOptionalCall();
         return (0, array_1.transformArrayPrototypeCall)(context, node);
     }
-    if ((0, typescript_1.isFunctionType)(ownerType) && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
+    const isFunctionFunction = (0, typescript_1.isFunctionType)(ownerType) ||
+        (expression.questionDotToken && (0, typescript_1.isNullableType)(context, ownerType, (_, t) => (0, typescript_1.isFunctionType)(t)));
+    if (isFunctionFunction && (0, typescript_1.hasStandardLibrarySignature)(context, node)) {
         if (isOptionalCall)
             return unsupportedOptionalCall();
         return (0, function_1.transformFunctionPrototypeCall)(context, node);
@@ -143,4 +149,45 @@ function transformBuiltinIdentifierExpression(context, node) {
     }
 }
 exports.transformBuiltinIdentifierExpression = transformBuiltinIdentifierExpression;
+const builtinErrorTypeNames = new Set([
+    "Error",
+    "ErrorConstructor",
+    "RangeError",
+    "RangeErrorConstructor",
+    "ReferenceError",
+    "ReferenceErrorConstructor",
+    "SyntaxError",
+    "SyntaxErrorConstructor",
+    "TypeError",
+    "TypeErrorConstructor",
+    "URIError",
+    "URIErrorConstructor",
+]);
+function checkForLuaLibType(context, type) {
+    if (!type.symbol)
+        return;
+    const name = context.checker.getFullyQualifiedName(type.symbol);
+    switch (name) {
+        case "Map":
+        case "MapConstructor":
+            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Map);
+            return;
+        case "Set":
+        case "SetConstructor":
+            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Set);
+            return;
+        case "WeakMap":
+        case "WeakMapConstructor":
+            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.WeakMap);
+            return;
+        case "WeakSet":
+        case "WeakSetConstructor":
+            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.WeakSet);
+            return;
+    }
+    if (builtinErrorTypeNames.has(name)) {
+        (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Error);
+    }
+}
+exports.checkForLuaLibType = checkForLuaLibType;
 //# sourceMappingURL=index.js.map

package/dist/transformation/context/index.js

@@ -1,7 +1,11 @@
 "use strict";
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
-    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
 }) : (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
     o[k2] = m[k];

package/dist/transformation/utils/diagnostics.d.ts

@@ -94,3 +94,6 @@ export declare const unsupportedBuiltinOptionalCall: ((node: ts.Node, ...args: a
 export declare const unsupportedOptionalCompileMembersOnly: ((node: ts.Node, ...args: any[]) => ts.Diagnostic) & {
     code: number;
 };
+export declare const undefinedInArrayLiteral: ((node: ts.Node, ...args: any[]) => ts.Diagnostic) & {
+    code: number;
+};

package/dist/transformation/utils/diagnostics.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.unsupportedOptionalCompileMembersOnly = exports.unsupportedBuiltinOptionalCall = exports.awaitMustBeInAsyncFunction = exports.notAllowedOptionalAssignment = exports.annotationDeprecated = exports.annotationRemoved = exports.invalidTableExtensionUse = exports.invalidOperatorMappingUse = exports.invalidMultiReturnAccess = exports.invalidMultiTypeToEmptyPatternOrArrayLiteral = exports.invalidMultiTypeToNonArrayLiteral = exports.invalidMultiFunctionReturnType = exports.invalidMultiFunctionUse = exports.unsupportedVarDeclaration = exports.invalidAmbientIdentifierName = exports.unsupportedProperty = exports.unsupportedForTarget = exports.unsupportedRightShiftOperator = exports.unsupportedAccessorInObjectLiteral = exports.invalidPairsIterableWithoutDestructuring = exports.invalidMultiIterableWithoutDestructuring = exports.invalidRangeControlVariable = exports.invalidVarargUse = exports.invalidRangeUse = exports.annotationInvalidArgumentCount = exports.decoratorInvalidContext = exports.unsupportedOverloadAssignment = exports.unsupportedSelfFunctionConversion = exports.unsupportedNoSelfFunctionConversion = exports.forbiddenForIn = exports.unsupportedNodeKind = void 0;
+exports.undefinedInArrayLiteral = exports.unsupportedOptionalCompileMembersOnly = exports.unsupportedBuiltinOptionalCall = exports.awaitMustBeInAsyncFunction = exports.notAllowedOptionalAssignment = exports.annotationDeprecated = exports.annotationRemoved = exports.invalidTableExtensionUse = exports.invalidOperatorMappingUse = exports.invalidMultiReturnAccess = exports.invalidMultiTypeToEmptyPatternOrArrayLiteral = exports.invalidMultiTypeToNonArrayLiteral = exports.invalidMultiFunctionReturnType = exports.invalidMultiFunctionUse = exports.unsupportedVarDeclaration = exports.invalidAmbientIdentifierName = exports.unsupportedProperty = exports.unsupportedForTarget = exports.unsupportedRightShiftOperator = exports.unsupportedAccessorInObjectLiteral = exports.invalidPairsIterableWithoutDestructuring = exports.invalidMultiIterableWithoutDestructuring = exports.invalidRangeControlVariable = exports.invalidVarargUse = exports.invalidRangeUse = exports.annotationInvalidArgumentCount = exports.decoratorInvalidContext = exports.unsupportedOverloadAssignment = exports.unsupportedSelfFunctionConversion = exports.unsupportedNoSelfFunctionConversion = exports.forbiddenForIn = exports.unsupportedNodeKind = void 0;
 const ts = require("typescript");
 const CompilerOptions_1 = require("../../CompilerOptions");
 const utils_1 = require("../../utils");
@@ -59,4 +59,5 @@ exports.notAllowedOptionalAssignment = createErrorDiagnosticFactory("The left-ha
 exports.awaitMustBeInAsyncFunction = createErrorDiagnosticFactory("Await can only be used inside async functions.");
 exports.unsupportedBuiltinOptionalCall = createErrorDiagnosticFactory("Optional calls are not supported for builtin or language extension functions.");
 exports.unsupportedOptionalCompileMembersOnly = createErrorDiagnosticFactory("Optional calls are not supported on enums marked with @compileMembersOnly.");
+exports.undefinedInArrayLiteral = createErrorDiagnosticFactory("Array literals may not contain undefined or null.");
 //# sourceMappingURL=diagnostics.js.map

package/dist/transformation/utils/typescript/index.js

@@ -1,7 +1,11 @@
 "use strict";
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
-    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
 }) : (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
     o[k2] = m[k];

package/dist/transformation/utils/typescript/types.d.ts

@@ -3,8 +3,10 @@ import { TransformationContext } from "../../context";
 export declare function isTypeWithFlags(context: TransformationContext, type: ts.Type, flags: ts.TypeFlags): boolean;
 export declare function typeAlwaysSatisfies(context: TransformationContext, type: ts.Type, predicate: (type: ts.Type) => boolean): boolean;
 export declare function typeCanSatisfy(context: TransformationContext, type: ts.Type, predicate: (type: ts.Type) => boolean): boolean;
+export declare function isNullishType(context: TransformationContext, type: ts.Type): boolean;
 export declare function isStringType(context: TransformationContext, type: ts.Type): boolean;
 export declare function isNumberType(context: TransformationContext, type: ts.Type): boolean;
+export declare function isNullableType(context: TransformationContext, type: ts.Type, isType: (c: TransformationContext, t: ts.Type) => boolean): boolean;
 /**
  * Iterate over a type and its bases until the callback returns true.
  */

package/dist/transformation/utils/typescript/types.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.canBeFalsyWhenNotNull = exports.canBeFalsy = exports.isFunctionType = exports.isArrayType = exports.forTypeOrAnySupertype = exports.isNumberType = exports.isStringType = exports.typeCanSatisfy = exports.typeAlwaysSatisfies = exports.isTypeWithFlags = void 0;
+exports.canBeFalsyWhenNotNull = exports.canBeFalsy = exports.isFunctionType = exports.isArrayType = exports.forTypeOrAnySupertype = exports.isNullableType = exports.isNumberType = exports.isStringType = exports.isNullishType = exports.typeCanSatisfy = exports.typeAlwaysSatisfies = exports.isTypeWithFlags = void 0;
 const ts = require("typescript");
 function isTypeWithFlags(context, type, flags) {
     const predicate = (type) => (type.flags & flags) !== 0;
@@ -46,6 +46,10 @@ function typeCanSatisfy(context, type, predicate) {
     return false;
 }
 exports.typeCanSatisfy = typeCanSatisfy;
+function isNullishType(context, type) {
+    return isTypeWithFlags(context, type, ts.TypeFlags.Undefined | ts.TypeFlags.Null | ts.TypeFlags.VoidLike);
+}
+exports.isNullishType = isNullishType;
 function isStringType(context, type) {
     return isTypeWithFlags(context, type, ts.TypeFlags.String | ts.TypeFlags.StringLike | ts.TypeFlags.StringLiteral);
 }
@@ -54,6 +58,11 @@ function isNumberType(context, type) {
     return isTypeWithFlags(context, type, ts.TypeFlags.Number | ts.TypeFlags.NumberLike | ts.TypeFlags.NumberLiteral);
 }
 exports.isNumberType = isNumberType;
+function isNullableType(context, type, isType) {
+    return (typeCanSatisfy(context, type, t => isType(context, t)) &&
+        typeAlwaysSatisfies(context, type, t => isType(context, t) || isNullishType(context, t)));
+}
+exports.isNullableType = isNullableType;
 function isExplicitArrayType(context, type) {
     if (type.symbol) {
         const baseConstraint = context.checker.getBaseConstraintOfType(type);

package/dist/transformation/visitors/class/index.js

@@ -15,9 +15,8 @@ const accessors_1 = require("./members/accessors");
 const constructor_1 = require("./members/constructor");
 const fields_1 = require("./members/fields");
 const method_1 = require("./members/method");
-const new_1 = require("./new");
-const setup_1 = require("./setup");
 const utils_2 = require("./utils");
+const setup_1 = require("./setup");
 const transformClassDeclaration = (declaration, context) => {
     // If declaration is a default export, transform to export variable assignment instead
     if ((0, export_1.hasDefaultExportModifier)(declaration)) {
@@ -61,9 +60,6 @@ function transformClassLikeDeclaration(classDeclaration, context, nameOverride)
     const extendedType = (0, utils_2.getExtendedType)(context, classDeclaration);
     const superInfo = (0, utils_1.getOrUpdate)(classSuperInfos, context, () => []);
     superInfo.push({ className, extendedTypeNode });
-    if (extendedType) {
-        (0, new_1.checkForLuaLibType)(context, extendedType);
-    }
     // Get all properties with value
     const properties = classDeclaration.members.filter(ts.isPropertyDeclaration).filter(member => member.initializer);
     // Divide properties into static and non-static

package/dist/transformation/visitors/class/members/constructor.js

@@ -4,7 +4,6 @@ exports.transformConstructorDeclaration = exports.createConstructorName = export
 const ts = require("typescript");
 const lua = require("../../../../LuaAST");
 const lua_ast_1 = require("../../../utils/lua-ast");
-const preceding_statements_1 = require("../../../utils/preceding-statements");
 const scope_1 = require("../../../utils/scope");
 const function_1 = require("../../function");
 const identifier_1 = require("../../identifier");
@@ -30,7 +29,7 @@ function transformConstructorDeclaration(context, statement, className, instance
     const bodyWithFieldInitializers = (0, function_1.transformFunctionBodyHeader)(context, scope, statement.parameters, restParamName);
     // Check for field declarations in constructor
     const constructorFieldsDeclarations = statement.parameters.filter(p => p.modifiers !== undefined);
-    const [fieldsPrecedingStatements, classInstanceFields] = (0, preceding_statements_1.transformInPrecedingStatementScope)(context, () => (0, fields_1.transformClassInstanceFields)(context, instanceFields));
+    const classInstanceFields = (0, fields_1.transformClassInstanceFields)(context, instanceFields);
     // If there are field initializers and the first statement is a super call,
     // move super call between default assignments and initializers
     if ((constructorFieldsDeclarations.length > 0 || classInstanceFields.length > 0) &&
@@ -55,7 +54,6 @@ function transformConstructorDeclaration(context, statement, className, instance
         }
         // else { TypeScript error: A parameter property may not be declared using a binding pattern }
     }
-    bodyWithFieldInitializers.push(...fieldsPrecedingStatements);
     bodyWithFieldInitializers.push(...classInstanceFields);
     bodyWithFieldInitializers.push(...body);
     const block = lua.createBlock(bodyWithFieldInitializers);

package/dist/transformation/visitors/class/members/fields.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.transformStaticPropertyDeclaration = exports.transformClassInstanceFields = exports.createPropertyDecoratingExpression = void 0;
 const lua = require("../../../../LuaAST");
 const lua_ast_1 = require("../../../utils/lua-ast");
+const preceding_statements_1 = require("../../../utils/preceding-statements");
 const literal_1 = require("../../literal");
 const decorators_1 = require("../decorators");
 const method_1 = require("./method");
@@ -17,14 +18,17 @@ exports.createPropertyDecoratingExpression = createPropertyDecoratingExpression;
 function transformClassInstanceFields(context, instanceFields) {
     const statements = [];
     for (const f of instanceFields) {
-        // Get identifier
-        const fieldName = (0, literal_1.transformPropertyName)(context, f.name);
-        const value = f.initializer ? context.transformExpression(f.initializer) : undefined;
-        // self[fieldName]
-        const selfIndex = lua.createTableIndexExpression((0, lua_ast_1.createSelfIdentifier)(), fieldName);
-        // self[fieldName] = value
-        const assignClassField = lua.createAssignmentStatement(selfIndex, value, f);
-        statements.push(assignClassField);
+        const [precedingStatements, statement] = (0, preceding_statements_1.transformInPrecedingStatementScope)(context, () => {
+            // Get identifier
+            const fieldName = (0, literal_1.transformPropertyName)(context, f.name);
+            const value = f.initializer ? context.transformExpression(f.initializer) : undefined;
+            // self[fieldName]
+            const selfIndex = lua.createTableIndexExpression((0, lua_ast_1.createSelfIdentifier)(), fieldName);
+            // self[fieldName] = value
+            const assignClassField = lua.createAssignmentStatement(selfIndex, value, f);
+            return assignClassField;
+        });
+        statements.push(...precedingStatements, statement);
     }
     return statements;
 }

package/dist/transformation/visitors/class/new.d.ts

@@ -1,4 +1,3 @@
 import * as ts from "typescript";
-import { FunctionVisitor, TransformationContext } from "../../context";
-export declare function checkForLuaLibType(context: TransformationContext, type: ts.Type): void;
+import { FunctionVisitor } from "../../context";
 export declare const transformNewExpression: FunctionVisitor<ts.NewExpression>;

package/dist/transformation/visitors/class/new.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.transformNewExpression = exports.checkForLuaLibType = void 0;
+exports.transformNewExpression = void 0;
 const ts = require("typescript");
 const lua = require("../../../LuaAST");
 const annotations_1 = require("../../utils/annotations");
@@ -8,44 +8,6 @@ const diagnostics_1 = require("../../utils/diagnostics");
 const lualib_1 = require("../../utils/lualib");
 const call_1 = require("../call");
 const table_1 = require("../language-extensions/table");
-const builtinErrorTypeNames = new Set([
-    "Error",
-    "ErrorConstructor",
-    "RangeError",
-    "RangeErrorConstructor",
-    "ReferenceError",
-    "ReferenceErrorConstructor",
-    "SyntaxError",
-    "SyntaxErrorConstructor",
-    "TypeError",
-    "TypeErrorConstructor",
-    "URIError",
-    "URIErrorConstructor",
-]);
-// TODO: Do it in identifier?
-function checkForLuaLibType(context, type) {
-    if (!type.symbol)
-        return;
-    const name = context.checker.getFullyQualifiedName(type.symbol);
-    switch (name) {
-        case "Map":
-            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Map);
-            return;
-        case "Set":
-            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Set);
-            return;
-        case "WeakMap":
-            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.WeakMap);
-            return;
-        case "WeakSet":
-            (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.WeakSet);
-            return;
-    }
-    if (builtinErrorTypeNames.has(name)) {
-        (0, lualib_1.importLuaLibFeature)(context, lualib_1.LuaLibFeature.Error);
-    }
-}
-exports.checkForLuaLibType = checkForLuaLibType;
 const transformNewExpression = (node, context) => {
     var _a, _b;
     const type = context.checker.getTypeAtLocation(node);
@@ -58,7 +20,6 @@ const transformNewExpression = (node, context) => {
     }
     const signature = context.checker.getResolvedSignature(node);
     const [name, params] = (0, call_1.transformCallAndArguments)(context, node.expression, (_a = node.arguments) !== null && _a !== void 0 ? _a : [ts.factory.createTrue()], signature);
-    checkForLuaLibType(context, type);
     const customConstructorAnnotation = annotations.get(annotations_1.AnnotationKind.CustomConstructor);
     if (customConstructorAnnotation) {
         if (customConstructorAnnotation.args.length === 1) {