typescript-to-lua 1.12.0 → 1.13.0

package/dist/CompilerOptions.d.ts

@@ -39,7 +39,8 @@ export type CompilerOptions = OmitIndexSignature<ts.CompilerOptions> & TypeScrip
 export declare enum LuaLibImportKind {
     None = "none",
     Inline = "inline",
-    Require = "require"
+    Require = "require",
+    RequireMinimal = "require-minimal"
 }
 export declare enum LuaTarget {
     Universal = "universal",

package/dist/CompilerOptions.js

@@ -8,6 +8,7 @@ var LuaLibImportKind;
     LuaLibImportKind["None"] = "none";
     LuaLibImportKind["Inline"] = "inline";
     LuaLibImportKind["Require"] = "require";
+    LuaLibImportKind["RequireMinimal"] = "require-minimal";
 })(LuaLibImportKind = exports.LuaLibImportKind || (exports.LuaLibImportKind = {}));
 var LuaTarget;
 (function (LuaTarget) {

package/dist/LuaLib.d.ts

@@ -110,8 +110,12 @@ export type LuaLibModulesInfo = Record<LuaLibFeature, LuaLibFeatureInfo>;
 export declare function resolveLuaLibDir(luaTarget: LuaTarget): string;
 export declare const luaLibModulesInfoFileName = "lualib_module_info.json";
 export declare function getLuaLibModulesInfo(luaTarget: LuaTarget, emitHost: EmitHost): LuaLibModulesInfo;
+export declare function getLuaLibExportToFeatureMap(luaTarget: LuaTarget, emitHost: EmitHost): ReadonlyMap<string, LuaLibFeature>;
 export declare function readLuaLibFeature(feature: LuaLibFeature, luaTarget: LuaTarget, emitHost: EmitHost): string;
 export declare function resolveRecursiveLualibFeatures(features: Iterable<LuaLibFeature>, luaTarget: LuaTarget, emitHost: EmitHost, luaLibModulesInfo?: LuaLibModulesInfo): LuaLibFeature[];
 export declare function loadInlineLualibFeatures(features: Iterable<LuaLibFeature>, luaTarget: LuaTarget, emitHost: EmitHost): string;
 export declare function loadImportedLualibFeatures(features: Iterable<LuaLibFeature>, luaTarget: LuaTarget, emitHost: EmitHost): lua.Statement[];
 export declare function getLuaLibBundle(luaTarget: LuaTarget, emitHost: EmitHost): string;
+export declare function getLualibBundleReturn(exportedValues: string[]): string;
+export declare function buildMinimalLualibBundle(features: Iterable<LuaLibFeature>, luaTarget: LuaTarget, emitHost: EmitHost): string;
+export declare function findUsedLualibFeatures(luaTarget: LuaTarget, emitHost: EmitHost, luaContents: string[]): Set<LuaLibFeature>;

package/dist/LuaLib.js

@@ -1,9 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getLuaLibBundle = exports.loadImportedLualibFeatures = exports.loadInlineLualibFeatures = exports.resolveRecursiveLualibFeatures = exports.readLuaLibFeature = exports.getLuaLibModulesInfo = exports.luaLibModulesInfoFileName = exports.resolveLuaLibDir = exports.LuaLibFeature = void 0;
+exports.findUsedLualibFeatures = exports.buildMinimalLualibBundle = exports.getLualibBundleReturn = exports.getLuaLibBundle = exports.loadImportedLualibFeatures = exports.loadInlineLualibFeatures = exports.resolveRecursiveLualibFeatures = exports.readLuaLibFeature = exports.getLuaLibExportToFeatureMap = exports.getLuaLibModulesInfo = exports.luaLibModulesInfoFileName = exports.resolveLuaLibDir = exports.LuaLibFeature = void 0;
 const path = require("path");
 const lua = require("./LuaAST");
 const CompilerOptions_1 = require("./CompilerOptions");
+const utils_1 = require("./utils");
 var LuaLibFeature;
 (function (LuaLibFeature) {
     LuaLibFeature["ArrayConcat"] = "ArrayConcat";
@@ -114,26 +115,47 @@ exports.resolveLuaLibDir = resolveLuaLibDir;
 exports.luaLibModulesInfoFileName = "lualib_module_info.json";
 const luaLibModulesInfo = new Map();
 function getLuaLibModulesInfo(luaTarget, emitHost) {
-    const lualibPath = path.join(resolveLuaLibDir(luaTarget), exports.luaLibModulesInfoFileName);
-    if (!luaLibModulesInfo.has(lualibPath)) {
+    if (!luaLibModulesInfo.has(luaTarget)) {
+        const lualibPath = path.join(resolveLuaLibDir(luaTarget), exports.luaLibModulesInfoFileName);
         const result = emitHost.readFile(lualibPath);
         if (result !== undefined) {
-            luaLibModulesInfo.set(lualibPath, JSON.parse(result));
+            luaLibModulesInfo.set(luaTarget, JSON.parse(result));
         }
         else {
             throw new Error(`Could not load lualib dependencies from '${lualibPath}'`);
         }
     }
-    return luaLibModulesInfo.get(lualibPath);
+    return luaLibModulesInfo.get(luaTarget);
 }
 exports.getLuaLibModulesInfo = getLuaLibModulesInfo;
+// This caches the names of lualib exports to their LuaLibFeature, avoiding a linear search for every lookup
+const lualibExportToFeature = new Map();
+function getLuaLibExportToFeatureMap(luaTarget, emitHost) {
+    if (!lualibExportToFeature.has(luaTarget)) {
+        const luaLibModulesInfo = getLuaLibModulesInfo(luaTarget, emitHost);
+        const map = new Map();
+        for (const [feature, info] of Object.entries(luaLibModulesInfo)) {
+            for (const exportName of info.exports) {
+                map.set(exportName, feature);
+            }
+        }
+        lualibExportToFeature.set(luaTarget, map);
+    }
+    return lualibExportToFeature.get(luaTarget);
+}
+exports.getLuaLibExportToFeatureMap = getLuaLibExportToFeatureMap;
+const lualibFeatureCache = new Map();
 function readLuaLibFeature(feature, luaTarget, emitHost) {
-    const featurePath = path.join(resolveLuaLibDir(luaTarget), `${feature}.lua`);
-    const luaLibFeature = emitHost.readFile(featurePath);
-    if (luaLibFeature === undefined) {
-        throw new Error(`Could not load lualib feature from '${featurePath}'`);
+    const featureMap = (0, utils_1.getOrUpdate)(lualibFeatureCache, luaTarget, () => new Map());
+    if (!featureMap.has(feature)) {
+        const featurePath = path.join(resolveLuaLibDir(luaTarget), `${feature}.lua`);
+        const luaLibFeature = emitHost.readFile(featurePath);
+        if (luaLibFeature === undefined) {
+            throw new Error(`Could not load lualib feature from '${featurePath}'`);
+        }
+        featureMap.set(feature, luaLibFeature);
     }
-    return luaLibFeature;
+    return featureMap.get(feature);
 }
 exports.readLuaLibFeature = readLuaLibFeature;
 function resolveRecursiveLualibFeatures(features, luaTarget, emitHost, luaLibModulesInfo = getLuaLibModulesInfo(luaTarget, emitHost)) {
@@ -157,23 +179,20 @@ function resolveRecursiveLualibFeatures(features, luaTarget, emitHost, luaLibMod
 }
 exports.resolveRecursiveLualibFeatures = resolveRecursiveLualibFeatures;
 function loadInlineLualibFeatures(features, luaTarget, emitHost) {
-    let result = "";
-    for (const feature of resolveRecursiveLualibFeatures(features, luaTarget, emitHost)) {
-        const luaLibFeature = readLuaLibFeature(feature, luaTarget, emitHost);
-        result += luaLibFeature + "\n";
-    }
-    return result;
+    return resolveRecursiveLualibFeatures(features, luaTarget, emitHost)
+        .map(feature => readLuaLibFeature(feature, luaTarget, emitHost))
+        .join("\n");
 }
 exports.loadInlineLualibFeatures = loadInlineLualibFeatures;
 function loadImportedLualibFeatures(features, luaTarget, emitHost) {
     const luaLibModuleInfo = getLuaLibModulesInfo(luaTarget, emitHost);
     const imports = Array.from(features).flatMap(feature => luaLibModuleInfo[feature].exports);
-    const requireCall = lua.createCallExpression(lua.createIdentifier("require"), [
-        lua.createStringLiteral("lualib_bundle"),
-    ]);
     if (imports.length === 0) {
         return [];
     }
+    const requireCall = lua.createCallExpression(lua.createIdentifier("require"), [
+        lua.createStringLiteral("lualib_bundle"),
+    ]);
     const luaLibId = lua.createIdentifier("____lualib");
     const importStatement = lua.createVariableDeclarationStatement(luaLibId, requireCall);
     const statements = [importStatement];
@@ -199,4 +218,35 @@ function getLuaLibBundle(luaTarget, emitHost) {
     return luaLibBundleContent.get(lualibPath);
 }
 exports.getLuaLibBundle = getLuaLibBundle;
+function getLualibBundleReturn(exportedValues) {
+    return `\nreturn {\n${exportedValues.map(exportName => `  ${exportName} = ${exportName}`).join(",\n")}\n}\n`;
+}
+exports.getLualibBundleReturn = getLualibBundleReturn;
+function buildMinimalLualibBundle(features, luaTarget, emitHost) {
+    const code = loadInlineLualibFeatures(features, luaTarget, emitHost);
+    const moduleInfo = getLuaLibModulesInfo(luaTarget, emitHost);
+    const exports = Array.from(features).flatMap(feature => moduleInfo[feature].exports);
+    return code + getLualibBundleReturn(exports);
+}
+exports.buildMinimalLualibBundle = buildMinimalLualibBundle;
+function findUsedLualibFeatures(luaTarget, emitHost, luaContents) {
+    const features = new Set();
+    const exportToFeatureMap = getLuaLibExportToFeatureMap(luaTarget, emitHost);
+    for (const lua of luaContents) {
+        const regex = /^local (\w+) = ____lualib\.(\w+)$/gm;
+        while (true) {
+            const match = regex.exec(lua);
+            if (!match)
+                break;
+            const [, localName, exportName] = match;
+            if (localName !== exportName)
+                continue;
+            const feature = exportToFeatureMap.get(exportName);
+            if (feature)
+                features.add(feature);
+        }
+    }
+    return features;
+}
+exports.findUsedLualibFeatures = findUsedLualibFeatures;
 //# sourceMappingURL=LuaLib.js.map

package/dist/LuaPrinter.js

@@ -135,7 +135,8 @@ class LuaPrinter {
         }
         const luaTarget = (_a = this.options.luaTarget) !== null && _a !== void 0 ? _a : CompilerOptions_1.LuaTarget.Universal;
         const luaLibImport = (_b = this.options.luaLibImport) !== null && _b !== void 0 ? _b : CompilerOptions_1.LuaLibImportKind.Require;
-        if (luaLibImport === CompilerOptions_1.LuaLibImportKind.Require && file.luaLibFeatures.size > 0) {
+        if ((luaLibImport === CompilerOptions_1.LuaLibImportKind.Require || luaLibImport === CompilerOptions_1.LuaLibImportKind.RequireMinimal) &&
+            file.luaLibFeatures.size > 0) {
             // Import lualib features
             sourceChunks = this.printStatementArray((0, LuaLib_1.loadImportedLualibFeatures)(file.luaLibFeatures, luaTarget, this.emitHost));
         }

package/dist/lualib/5.0/ArraySlice.lua

@@ -1,10 +1,6 @@
 local function __TS__ArraySlice(self, first, last)
     local len = table.getn(self)
-    local ____first_0 = first
-    if ____first_0 == nil then
-        ____first_0 = 0
-    end
-    first = ____first_0
+    first = first or 0
     if first < 0 then
         first = len + first
         if first < 0 then
@@ -15,11 +11,7 @@ local function __TS__ArraySlice(self, first, last)
             first = len
         end
     end
-    local ____last_1 = last
-    if ____last_1 == nil then
-        ____last_1 = len
-    end
-    last = ____last_1
+    last = last or len
     if last < 0 then
         last = len + last
         if last < 0 then

package/dist/lualib/5.0/ArraySplice.lua

@@ -22,11 +22,7 @@ local function __TS__ArraySplice(self, ...)
     elseif actualArgumentCount == 1 then
         actualDeleteCount = len - start
     else
-        local ____deleteCount_0 = deleteCount
-        if ____deleteCount_0 == nil then
-            ____deleteCount_0 = 0
-        end
-        actualDeleteCount = ____deleteCount_0
+        actualDeleteCount = deleteCount or 0
         if actualDeleteCount < 0 then
             actualDeleteCount = 0
         end

package/dist/lualib/5.0/ParseFloat.lua

@@ -10,9 +10,5 @@ local function __TS__ParseFloat(numberString)
         return ____temp_0
     end
     local number = tonumber(__TS__Match(numberString, "^%s*(-?%d+%.?%d*)"))
-    local ____number_1 = number
-    if ____number_1 == nil then
-        ____number_1 = 0 / 0
-    end
-    return ____number_1
+    return number or 0 / 0
 end

package/dist/lualib/5.0/StringCharCodeAt.lua

@@ -5,9 +5,5 @@ local function __TS__StringCharCodeAt(self, index)
     if index < 0 then
         return 0 / 0
     end
-    local ____string_byte_result_0 = string.byte(self, index + 1)
-    if ____string_byte_result_0 == nil then
-        ____string_byte_result_0 = 0 / 0
-    end
-    return ____string_byte_result_0
+    return string.byte(self, index + 1) or 0 / 0
 end

package/dist/lualib/5.0/lualib_bundle.lua

@@ -350,11 +350,7 @@ end
 
 local function __TS__ArraySlice(self, first, last)
     local len = table.getn(self)
-    local ____first_0 = first
-    if ____first_0 == nil then
-        ____first_0 = 0
-    end
-    first = ____first_0
+    first = first or 0
     if first < 0 then
         first = len + first
         if first < 0 then
@@ -365,11 +361,7 @@ local function __TS__ArraySlice(self, first, last)
             first = len
         end
     end
-    local ____last_1 = last
-    if ____last_1 == nil then
-        ____last_1 = len
-    end
-    last = ____last_1
+    last = last or len
     if last < 0 then
         last = len + last
         if last < 0 then
@@ -425,11 +417,7 @@ local function __TS__ArraySplice(self, ...)
     elseif actualArgumentCount == 1 then
         actualDeleteCount = len - start
     else
-        local ____deleteCount_0 = deleteCount
-        if ____deleteCount_0 == nil then
-            ____deleteCount_0 = 0
-        end
-        actualDeleteCount = ____deleteCount_0
+        actualDeleteCount = deleteCount or 0
         if actualDeleteCount < 0 then
             actualDeleteCount = 0
         end
@@ -1661,11 +1649,7 @@ local function __TS__ParseFloat(numberString)
         return ____temp_0
     end
     local number = tonumber(__TS__Match(numberString, "^%s*(-?%d+%.?%d*)"))
-    local ____number_1 = number
-    if ____number_1 == nil then
-        ____number_1 = 0 / 0
-    end
-    return ____number_1
+    return number or 0 / 0
 end
 
 local function __TS__StringSubstring(self, start, ____end)
@@ -2230,11 +2214,7 @@ local function __TS__StringCharCodeAt(self, index)
     if index < 0 then
         return 0 / 0
     end
-    local ____string_byte_result_0 = string.byte(self, index + 1)
-    if ____string_byte_result_0 == nil then
-        ____string_byte_result_0 = 0 / 0
-    end
-    return ____string_byte_result_0
+    return string.byte(self, index + 1) or 0 / 0
 end
 
 local function __TS__StringEndsWith(self, searchString, endPosition)

package/dist/lualib/universal/ArraySlice.lua

@@ -1,10 +1,6 @@
 local function __TS__ArraySlice(self, first, last)
     local len = #self
-    local ____first_0 = first
-    if ____first_0 == nil then
-        ____first_0 = 0
-    end
-    first = ____first_0
+    first = first or 0
     if first < 0 then
         first = len + first
         if first < 0 then
@@ -15,11 +11,7 @@ local function __TS__ArraySlice(self, first, last)
             first = len
         end
     end
-    local ____last_1 = last
-    if ____last_1 == nil then
-        ____last_1 = len
-    end
-    last = ____last_1
+    last = last or len
     if last < 0 then
         last = len + last
         if last < 0 then

package/dist/lualib/universal/ArraySplice.lua

@@ -22,11 +22,7 @@ local function __TS__ArraySplice(self, ...)
     elseif actualArgumentCount == 1 then
         actualDeleteCount = len - start
     else
-        local ____deleteCount_0 = deleteCount
-        if ____deleteCount_0 == nil then
-            ____deleteCount_0 = 0
-        end
-        actualDeleteCount = ____deleteCount_0
+        actualDeleteCount = deleteCount or 0
         if actualDeleteCount < 0 then
             actualDeleteCount = 0
         end

package/dist/lualib/universal/ParseFloat.lua

@@ -10,9 +10,5 @@ local function __TS__ParseFloat(numberString)
         return ____temp_0
     end
     local number = tonumber(__TS__Match(numberString, "^%s*(-?%d+%.?%d*)"))
-    local ____number_1 = number
-    if ____number_1 == nil then
-        ____number_1 = 0 / 0
-    end
-    return ____number_1
+    return number or 0 / 0
 end

package/dist/lualib/universal/SparseArraySpread.lua

@@ -1,8 +1,4 @@
 local function __TS__SparseArraySpread(sparseArray)
-    local ____unpack_0 = unpack
-    if ____unpack_0 == nil then
-        ____unpack_0 = table.unpack
-    end
-    local _unpack = ____unpack_0
+    local _unpack = unpack or table.unpack
     return _unpack(sparseArray, 1, sparseArray.sparseLength)
 end

package/dist/lualib/universal/StringCharCodeAt.lua

@@ -5,9 +5,5 @@ local function __TS__StringCharCodeAt(self, index)
     if index < 0 then
         return 0 / 0
     end
-    local ____string_byte_result_0 = string.byte(self, index + 1)
-    if ____string_byte_result_0 == nil then
-        ____string_byte_result_0 = 0 / 0
-    end
-    return ____string_byte_result_0
+    return string.byte(self, index + 1) or 0 / 0
 end

package/dist/lualib/universal/lualib_bundle.lua

@@ -349,11 +349,7 @@ end
 
 local function __TS__ArraySlice(self, first, last)
     local len = #self
-    local ____first_0 = first
-    if ____first_0 == nil then
-        ____first_0 = 0
-    end
-    first = ____first_0
+    first = first or 0
     if first < 0 then
         first = len + first
         if first < 0 then
@@ -364,11 +360,7 @@ local function __TS__ArraySlice(self, first, last)
             first = len
         end
     end
-    local ____last_1 = last
-    if ____last_1 == nil then
-        ____last_1 = len
-    end
-    last = ____last_1
+    last = last or len
     if last < 0 then
         last = len + last
         if last < 0 then
@@ -424,11 +416,7 @@ local function __TS__ArraySplice(self, ...)
     elseif actualArgumentCount == 1 then
         actualDeleteCount = len - start
     else
-        local ____deleteCount_0 = deleteCount
-        if ____deleteCount_0 == nil then
-            ____deleteCount_0 = 0
-        end
-        actualDeleteCount = ____deleteCount_0
+        actualDeleteCount = deleteCount or 0
         if actualDeleteCount < 0 then
             actualDeleteCount = 0
         end
@@ -1597,11 +1585,7 @@ local function __TS__ParseFloat(numberString)
         return ____temp_0
     end
     local number = tonumber(__TS__Match(numberString, "^%s*(-?%d+%.?%d*)"))
-    local ____number_1 = number
-    if ____number_1 == nil then
-        ____number_1 = 0 / 0
-    end
-    return ____number_1
+    return number or 0 / 0
 end
 
 local function __TS__StringSubstring(self, start, ____end)
@@ -1979,11 +1963,7 @@ local function __TS__SparseArrayPush(sparseArray, ...)
 end
 
 local function __TS__SparseArraySpread(sparseArray)
-    local ____unpack_0 = unpack
-    if ____unpack_0 == nil then
-        ____unpack_0 = table.unpack
-    end
-    local _unpack = ____unpack_0
+    local _unpack = unpack or table.unpack
     return _unpack(sparseArray, 1, sparseArray.sparseLength)
 end
 
@@ -2165,11 +2145,7 @@ local function __TS__StringCharCodeAt(self, index)
     if index < 0 then
         return 0 / 0
     end
-    local ____string_byte_result_0 = string.byte(self, index + 1)
-    if ____string_byte_result_0 == nil then
-        ____string_byte_result_0 = 0 / 0
-    end
-    return ____string_byte_result_0
+    return string.byte(self, index + 1) or 0 / 0
 end
 
 local function __TS__StringEndsWith(self, searchString, endPosition)

package/dist/lualib-build/plugin.js

@@ -49,7 +49,7 @@ class LuaLibPlugin {
         // 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`;
+        lualibBundle += (0, LuaLib_1.getLualibBundleReturn)(exports);
         result.push({ fileName: "lualib_bundle.lua", code: lualibBundle });
         return diagnostics;
     }

package/dist/transformation/builtins/array.js

@@ -42,10 +42,10 @@ function createTableLengthExpression(context, expression, node) {
  * return (#array + 1)
  */
 function transformSingleElementArrayPush(context, node, caller, param) {
-    const expressionIsUsed = !ts.isExpressionStatement((0, typescript_1.findFirstNonOuterParent)(node));
     const arrayIdentifier = lua.isIdentifier(caller) ? caller : (0, expression_list_1.moveToPrecedingTemp)(context, caller);
     // #array + 1
     let lengthExpression = lua.createBinaryExpression(createTableLengthExpression(context, arrayIdentifier), lua.createNumericLiteral(1), lua.SyntaxKind.AdditionOperator);
+    const expressionIsUsed = (0, typescript_1.expressionResultIsUsed)(node);
     if (expressionIsUsed) {
         // store length in a temp
         lengthExpression = (0, expression_list_1.moveToPrecedingTemp)(context, lengthExpression);

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

@@ -91,3 +91,9 @@ export declare const unsupportedOptionalCompileMembersOnly: ((node: ts.Node, ...
 export declare const undefinedInArrayLiteral: ((node: ts.Node, ...args: any[]) => ts.Diagnostic) & {
     code: number;
 };
+export declare const invalidMethodCallExtensionUse: ((node: ts.Node, ...args: any[]) => ts.Diagnostic) & {
+    code: number;
+};
+export declare const invalidSpreadInCallExtension: ((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.undefinedInArrayLiteral = exports.unsupportedOptionalCompileMembersOnly = exports.unsupportedBuiltinOptionalCall = exports.awaitMustBeInAsyncFunction = exports.notAllowedOptionalAssignment = exports.truthyOnlyConditionalValue = exports.annotationDeprecated = exports.invalidCallExtensionUse = exports.invalidMultiReturnAccess = exports.invalidMultiFunctionReturnType = exports.invalidMultiFunctionUse = exports.unsupportedVarDeclaration = exports.invalidAmbientIdentifierName = exports.unsupportedProperty = exports.unsupportedForTargetButOverrideAvailable = 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.invalidSpreadInCallExtension = exports.invalidMethodCallExtensionUse = exports.undefinedInArrayLiteral = exports.unsupportedOptionalCompileMembersOnly = exports.unsupportedBuiltinOptionalCall = exports.awaitMustBeInAsyncFunction = exports.notAllowedOptionalAssignment = exports.truthyOnlyConditionalValue = exports.annotationDeprecated = exports.invalidCallExtensionUse = exports.invalidMultiReturnAccess = exports.invalidMultiFunctionReturnType = exports.invalidMultiFunctionUse = exports.unsupportedVarDeclaration = exports.invalidAmbientIdentifierName = exports.unsupportedProperty = exports.unsupportedForTargetButOverrideAvailable = 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,6 @@ exports.awaitMustBeInAsyncFunction = createErrorDiagnosticFactory("Await can onl
 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.");
+exports.invalidMethodCallExtensionUse = createErrorDiagnosticFactory("This language extension must be called as a method.");
+exports.invalidSpreadInCallExtension = createErrorDiagnosticFactory("Spread elements are not supported in call extensions.");
 //# sourceMappingURL=diagnostics.js.map

package/dist/transformation/utils/language-extensions.d.ts

@@ -63,3 +63,7 @@ export declare enum IterableExtensionKind {
 export declare function isLuaIterable(context: TransformationContext, type: ts.Type): boolean;
 export declare function getIterableExtensionTypeForType(context: TransformationContext, type: ts.Type): IterableExtensionKind | undefined;
 export declare function getIterableExtensionKindForNode(context: TransformationContext, node: ts.Node): IterableExtensionKind | undefined;
+export declare const methodExtensionKinds: ReadonlySet<ExtensionKind>;
+export declare function getNaryCallExtensionArgs(context: TransformationContext, node: ts.CallExpression, kind: ExtensionKind, numArgs: number): readonly ts.Expression[] | undefined;
+export declare function getUnaryCallExtensionArg(context: TransformationContext, node: ts.CallExpression, kind: ExtensionKind): ts.Expression | undefined;
+export declare function getBinaryCallExtensionArgs(context: TransformationContext, node: ts.CallExpression, kind: ExtensionKind): readonly [ts.Expression, ts.Expression] | undefined;

package/dist/transformation/utils/language-extensions.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getIterableExtensionKindForNode = exports.getIterableExtensionTypeForType = exports.isLuaIterable = exports.IterableExtensionKind = exports.getExtensionKindForSymbol = exports.getExtensionKindForNode = exports.getExtensionKindForType = exports.ExtensionKind = void 0;
+exports.getBinaryCallExtensionArgs = exports.getUnaryCallExtensionArg = exports.getNaryCallExtensionArgs = exports.methodExtensionKinds = exports.getIterableExtensionKindForNode = exports.getIterableExtensionTypeForType = exports.isLuaIterable = exports.IterableExtensionKind = exports.getExtensionKindForSymbol = exports.getExtensionKindForNode = exports.getExtensionKindForType = exports.ExtensionKind = void 0;
 const ts = require("typescript");
+const diagnostics_1 = require("./diagnostics");
 var ExtensionKind;
 (function (ExtensionKind) {
     ExtensionKind["MultiFunction"] = "MultiFunction";
@@ -112,4 +113,68 @@ function getIterableExtensionKindForNode(context, node) {
     return getIterableExtensionTypeForType(context, type);
 }
 exports.getIterableExtensionKindForNode = getIterableExtensionKindForNode;
+exports.methodExtensionKinds = new Set([
+    ExtensionKind.AdditionOperatorMethodType,
+    ExtensionKind.SubtractionOperatorMethodType,
+    ExtensionKind.MultiplicationOperatorMethodType,
+    ExtensionKind.DivisionOperatorMethodType,
+    ExtensionKind.ModuloOperatorMethodType,
+    ExtensionKind.PowerOperatorMethodType,
+    ExtensionKind.FloorDivisionOperatorMethodType,
+    ExtensionKind.BitwiseAndOperatorMethodType,
+    ExtensionKind.BitwiseOrOperatorMethodType,
+    ExtensionKind.BitwiseExclusiveOrOperatorMethodType,
+    ExtensionKind.BitwiseLeftShiftOperatorMethodType,
+    ExtensionKind.BitwiseRightShiftOperatorMethodType,
+    ExtensionKind.ConcatOperatorMethodType,
+    ExtensionKind.LessThanOperatorMethodType,
+    ExtensionKind.GreaterThanOperatorMethodType,
+    ExtensionKind.NegationOperatorMethodType,
+    ExtensionKind.BitwiseNotOperatorMethodType,
+    ExtensionKind.LengthOperatorMethodType,
+    ExtensionKind.TableDeleteMethodType,
+    ExtensionKind.TableGetMethodType,
+    ExtensionKind.TableHasMethodType,
+    ExtensionKind.TableSetMethodType,
+    ExtensionKind.TableAddKeyMethodType,
+]);
+function getNaryCallExtensionArgs(context, node, kind, numArgs) {
+    let expressions;
+    if (node.arguments.some(ts.isSpreadElement)) {
+        context.diagnostics.push((0, diagnostics_1.invalidSpreadInCallExtension)(node));
+        return undefined;
+    }
+    if (exports.methodExtensionKinds.has(kind)) {
+        if (!(ts.isPropertyAccessExpression(node.expression) || ts.isElementAccessExpression(node.expression))) {
+            context.diagnostics.push((0, diagnostics_1.invalidMethodCallExtensionUse)(node));
+            return undefined;
+        }
+        if (node.arguments.length < numArgs - 1) {
+            // assumed to be TS error
+            return undefined;
+        }
+        expressions = [node.expression.expression, ...node.arguments];
+    }
+    else {
+        if (node.arguments.length < numArgs) {
+            // assumed to be TS error
+            return undefined;
+        }
+        expressions = node.arguments;
+    }
+    return expressions;
+}
+exports.getNaryCallExtensionArgs = getNaryCallExtensionArgs;
+function getUnaryCallExtensionArg(context, node, kind) {
+    var _a;
+    return (_a = getNaryCallExtensionArgs(context, node, kind, 1)) === null || _a === void 0 ? void 0 : _a[0];
+}
+exports.getUnaryCallExtensionArg = getUnaryCallExtensionArg;
+function getBinaryCallExtensionArgs(context, node, kind) {
+    const expressions = getNaryCallExtensionArgs(context, node, kind, 2);
+    if (expressions === undefined)
+        return undefined;
+    return [expressions[0], expressions[1]];
+}
+exports.getBinaryCallExtensionArgs = getBinaryCallExtensionArgs;
 //# sourceMappingURL=language-extensions.js.map

package/dist/transformation/utils/preceding-statements.d.ts

@@ -1,3 +1,7 @@
 import * as lua from "../../LuaAST";
 import { TransformationContext } from "../context";
-export declare function transformInPrecedingStatementScope<TReturn extends lua.Statement | lua.Statement[] | lua.Expression | lua.Expression[]>(context: TransformationContext, transformer: () => TReturn): [lua.Statement[], TReturn];
+export interface WithPrecedingStatements<T extends lua.Statement | lua.Statement[] | lua.Expression | lua.Expression[]> {
+    precedingStatements: lua.Statement[];
+    result: T;
+}
+export declare function transformInPrecedingStatementScope<TReturn extends lua.Statement | lua.Statement[] | lua.Expression | lua.Expression[]>(context: TransformationContext, transformer: () => TReturn): WithPrecedingStatements<TReturn>;

package/dist/transformation/utils/preceding-statements.js

@@ -5,7 +5,7 @@ function transformInPrecedingStatementScope(context, transformer) {
     context.pushPrecedingStatements();
     const statementOrStatements = transformer();
     const precedingStatements = context.popPrecedingStatements();
-    return [precedingStatements, statementOrStatements];
+    return { precedingStatements, result: statementOrStatements };
 }
 exports.transformInPrecedingStatementScope = transformInPrecedingStatementScope;
 //# sourceMappingURL=preceding-statements.js.map

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

@@ -8,6 +8,7 @@ export declare function hasExportEquals(sourceFile: ts.SourceFile): boolean;
  */
 export declare function findFirstNodeAbove<T extends ts.Node>(node: ts.Node, callback: (n: ts.Node) => n is T): T | undefined;
 export declare function findFirstNonOuterParent(node: ts.Node): ts.Node;
+export declare function expressionResultIsUsed(node: ts.Expression): boolean;
 export declare function getFirstDeclarationInFile(symbol: ts.Symbol, sourceFile: ts.SourceFile): ts.Declaration | undefined;
 export declare function isStandardLibraryDeclaration(context: TransformationContext, declaration: ts.Declaration): boolean;
 export declare function isStandardLibraryType(context: TransformationContext, type: ts.Type, name: string | undefined): boolean;