TypeScriptToLua · lolleko · Dec 2, 2018 · Nov 27, 2018 · Nov 27, 2018 · Nov 27, 2018
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -50,7 +50,7 @@
         "@types/glob": "^5.0.35",
         "@types/node": "^9.6.23",
         "@types/yargs": "^11.1.1",
-        "alsatian": "^2.2.1",
+        "alsatian": "^2.3.0",
         "circular-json": "^0.5.5",
         "codecov": "3.0.2",
         "deep-equal": "^1.0.1",

diff --git a/src/LuaLibFeature.ts b/src/LuaLibFeature.ts
@@ -0,0 +1,22 @@
+export enum LuaLibFeature {
+    ArrayConcat = "ArrayConcat",
+    ArrayEvery = "ArrayEvery",
+    ArrayFilter = "ArrayFilter",
+    ArrayForEach = "ArrayForEach",
+    ArrayIndexOf = "ArrayIndexOf",
+    ArrayMap = "ArrayMap",
+    ArrayPush = "ArrayPush",
+    ArrayReverse = "ArrayReverse",
+    ArrayShift = "ArrayShift",
+    ArrayUnshift = "ArrayUnshift",
+    ArraySort = "ArraySort",
+    ArraySlice = "ArraySlice",
+    ArraySome = "ArraySome",
+    ArraySplice = "ArraySplice",
+    InstanceOf = "InstanceOf",
+    Map = "Map",
+    Set = "Set",
+    StringReplace = "StringReplace",
+    StringSplit = "StringSplit",
+    Ternary = "Ternary",
+}
diff --git a/src/TSHelper.ts b/src/TSHelper.ts
@@ -56,6 +56,16 @@ export class TSHelper {
         return false;
     }
 
+    public static isIdentifierExported(
+        identifier: ts.Identifier, scope: ts.ModuleDeclaration | ts.SourceFile, checker: ts.TypeChecker): boolean {
+        const identifierSymbol = checker.getTypeAtLocation(scope).getSymbol();
+        if (identifierSymbol.exports) {
+            return identifierSymbol.exports.has(identifier.escapedText);
+        }
+
+        return false;
+    }
+
     public static isInDestructingAssignment(node: ts.Node): boolean {
         return node.parent && ((ts.isVariableDeclaration(node.parent) && ts.isArrayBindingPattern(node.parent.name))
             || (ts.isBinaryExpression(node.parent) && ts.isArrayLiteralExpression(node.parent.left)));

diff --git a/src/TransformHelper.ts b/src/TransformHelper.ts
@@ -2,13 +2,12 @@ import * as ts from "typescript";
 
 export class TransformHelper {
     // Helper to create simple lua variable statement;
-    public static createLuaVariableStatement(
-        identifier: ts.Identifier,
-        expression: ts.Expression,
-        typeNode?: ts.TypeNode
-    ): ts.VariableStatement {
+    public static createLuaVariableStatement(identifier: ts.Identifier,
+                                             expression?: ts.Expression,
+                                             typeNode?: ts.TypeNode,
+                                             modifiers: ReadonlyArray<ts.Modifier> = []): ts.VariableStatement {
         const declaration = ts.createVariableDeclaration(identifier, typeNode, expression);
-        const statement = ts.createVariableStatement([], ts.createVariableDeclarationList([declaration]));
+        const statement = ts.createVariableStatement(modifiers, ts.createVariableDeclarationList([declaration]));
         return statement;
     }
 
@@ -18,4 +17,9 @@ export class TransformHelper {
             ts.createCall(requireIdentifier, [ts.createLiteralTypeNode(moduleSpecifier)], [moduleSpecifier]);
         return this.createLuaVariableStatement(identifier, requireCall);
     }
+
+    public static flatten<T>(arr: T[]): T[] {
+        const flat = [].concat(...arr);
+        return flat.some(Array.isArray) ? this.flatten(flat) : flat;
+    }
 }