Add custom symbols

ark120202 · ark120202 · commit 5224ca06ad91 · 2019-02-16T17:17:54.000+05:00
diff --git a/src/LuaLib.ts b/src/LuaLib.ts
@@ -33,6 +33,7 @@ export enum LuaLibFeature {
     StringSplit = "StringSplit",
     StringConcat = "StringConcat",
     Symbol = "Symbol",
+    SymbolRegistry = "SymbolRegistry",
 }
 
 const luaLibDependencies: {[lib in LuaLibFeature]?: LuaLibFeature[]} = {
@@ -41,6 +42,7 @@ const luaLibDependencies: {[lib in LuaLibFeature]?: LuaLibFeature[]} = {
     Set: [LuaLibFeature.InstanceOf, LuaLibFeature.Iterator, LuaLibFeature.Symbol],
     WeakMap: [LuaLibFeature.InstanceOf, LuaLibFeature.Iterator, LuaLibFeature.Symbol],
     WeakSet: [LuaLibFeature.InstanceOf, LuaLibFeature.Iterator, LuaLibFeature.Symbol],
+    SymbolRegistry: [LuaLibFeature.Symbol],
 };
 
 export class LuaLib {
diff --git a/src/LuaTransformer.ts b/src/LuaTransformer.ts
@@ -2815,6 +2815,11 @@ export class LuaTransformer {
             parameters = this.transformArguments(node.arguments, signature);
         }
 
+        const expressionType = this.checker.getTypeAtLocation(node.expression);
+        if (expressionType.symbol && expressionType.symbol.escapedName === "SymbolConstructor") {
+            return this.transformLuaLibFunction(LuaLibFeature.Symbol, node, ...parameters);
+        }
+
         const callExpression = tstl.createCallExpression(callPath, parameters);
         return wrapResult ? this.wrapInTable(callExpression) : callExpression;
     }
@@ -2850,6 +2855,10 @@ export class LuaTransformer {
             return this.transformObjectCallExpression(node);
         }
 
+        if (ownerType.symbol && ownerType.symbol.escapedName === "SymbolConstructor") {
+            return this.transformSymbolCallExpression(node);
+        }
+
         switch (ownerType.flags) {
             case ts.TypeFlags.String:
             case ts.TypeFlags.StringLiteral:
@@ -3300,6 +3309,28 @@ export class LuaTransformer {
         }
     }
 
+    // Transpile a Symbol._ property
+    public transformSymbolCallExpression(expression: ts.CallExpression): tstl.CallExpression {
+        const method = expression.expression as ts.PropertyAccessExpression;
+        const parameters = this.transformArguments(expression.arguments);
+        const methodName = method.name.escapedText;
+
+        switch (methodName) {
+            case "for":
+            case "keyFor":
+                this.importLuaLibFeature(LuaLibFeature.SymbolRegistry);
+                const upperMethodName = methodName[0].toUpperCase() + methodName.slice(1);
+                const functionIdentifier = tstl.createIdentifier(`__TS__SymbolRegistry${upperMethodName}`);
+                return tstl.createCallExpression(functionIdentifier, parameters, expression);
+            default:
+                throw TSTLErrors.UnsupportedForTarget(
+                    `symbol property ${methodName}`,
+                    this.options.luaTarget,
+                    expression
+                );
+        }
+    }
+
     public transformArrayCallExpression(node: ts.CallExpression): tstl.CallExpression {
         const expression = node.expression as ts.PropertyAccessExpression;
         const params = this.transformArguments(node.arguments);
diff --git a/src/lualib/Symbol.ts b/src/lualib/Symbol.ts
@@ -1,3 +1,19 @@
+declare function setmetatable<T extends object>(obj: T, metatable: any): T;
+
+const symbolMetatable = {
+    __tostring(): string {
+        if (this.description === undefined) {
+            return 'Symbol()';
+        } else {
+            return 'Symbol(' + this.description + ')';
+        }
+    },
+};
+
+function __TS__Symbol(description?: string | number): symbol {
+    return setmetatable({ description }, symbolMetatable) as any;
+}
+
 Symbol = {
-    iterator: {},
+    iterator: __TS__Symbol('Symbol.iterator'),
 } as any;
diff --git a/src/lualib/SymbolRegistry.ts b/src/lualib/SymbolRegistry.ts
@@ -0,0 +1,15 @@
+const symbolRegistry: Record<string, symbol> = {};
+
+function __TS__SymbolRegistryFor(key: string): symbol {
+    if (!symbolRegistry[key])  {
+        symbolRegistry[key] = __TS__Symbol(key);
+    }
+
+    return symbolRegistry[key];
+}
+
+function __TS__SymbolRegistryKeyFor(sym: symbol): string {
+    for (const key in symbolRegistry) {
+        if (symbolRegistry[key] === sym) return key;
+    }
+}
diff --git a/test/unit/lualib/symbol.spec.ts b/test/unit/lualib/symbol.spec.ts
@@ -0,0 +1,79 @@
+import { Expect, Test, TestCase } from "alsatian";
+import * as util from "../../src/util";
+import { CompilerOptions, LuaLibImportKind } from '../../../src/CompilerOptions';
+
+export class SymbolTests {
+    private compilerOptions: CompilerOptions = {
+        lib: ["esnext"],
+        luaLibImport: LuaLibImportKind.Require,
+    };
+
+    @Test("symbol.toString()")
+    @TestCase()
+    @TestCase(1)
+    @TestCase("name")
+    public symbolToString(description?: string | number): void
+    {
+        const result = util.transpileAndExecute(`
+            return Symbol(${JSON.stringify(description)}).toString();
+        `, this.compilerOptions);
+
+        Expect(result).toBe(`Symbol(${description || ''})`);
+    }
+
+    @Test("symbol.description")
+    @TestCase()
+    @TestCase(1)
+    @TestCase("name")
+    public symbolDescription(description?: string | number): void
+    {
+        const result = util.transpileAndExecute(`
+            return Symbol(${JSON.stringify(description)}).description;
+        `, this.compilerOptions);
+
+        Expect(result).toBe(description);
+    }
+
+    @Test("Symbol.for")
+    public symbolFor(): void
+    {
+        const result = util.transpileAndExecute(`
+            return Symbol.for("name").description;
+        `, this.compilerOptions);
+
+        Expect(result).toBe("name");
+    }
+
+    @Test("Symbol.for reference")
+    public symbolForReference(): void
+    {
+        const result = util.transpileAndExecute(`
+            return Symbol.for("a") === Symbol.for("a");
+        `);
+
+        Expect(result).toBe(true);
+    }
+
+    @Test("Symbol.keyFor")
+    public symbolKeyFor(): void
+    {
+        const result = util.transpileAndExecute(`
+            const sym = Symbol.for("a");
+            Symbol.for("b");
+            return Symbol.keyFor(sym);
+        `);
+
+        Expect(result).toBe("a");
+    }
+
+    @Test("Symbol.keyFor empty")
+    public symbolKeyForEmpty(): void
+    {
+        const result = util.transpileAndExecute(`
+            Symbol.for("a");
+            return Symbol.keyFor(Symbol());
+        `);
+
+        Expect(result).toBe(undefined);
+    }
+}
