enabled support for lua string instance methods (#449)

tomblind · web-flow · commit 2a8846afb4b1 · 2019-02-24T16:01:36.000-07:00
diff --git a/src/LuaTransformer.ts b/src/LuaTransformer.ts
@@ -3513,6 +3513,35 @@ export class LuaTransformer {
                 const firstParamPlusOne = this.expressionPlusOne(params[0]);
                 return this.createStringCall("byte", node, caller, firstParamPlusOne);
             }
+            case "byte":
+            case "char":
+            case "dump":
+            case "find":
+            case "format":
+            case "gmatch":
+            case "gsub":
+            case "len":
+            case "lower":
+            case "match":
+            case "pack":
+            case "packsize":
+            case "rep":
+            case "reverse":
+            case "sub":
+            case "unpack":
+            case "upper":
+                // Allow lua's string instance methods
+                let stringVariable = this.transformExpression(expression.expression);
+                if (ts.isStringLiteral(expression.expression)) {
+                    // "foo":method() needs to be ("foo"):method()
+                    stringVariable = tstl.createParenthesizedExpression(stringVariable);
+                }
+                return tstl.createMethodCallExpression(
+                    stringVariable,
+                    this.transformIdentifier(expression.name),
+                    params,
+                    node
+                );
             default:
                 throw TSTLErrors.UnsupportedProperty("string", expressionName, node);
         }
diff --git a/test/unit/string.spec.ts b/test/unit/string.spec.ts
@@ -14,6 +14,17 @@ export class StringTests
         }).toThrowError(TranspileError, "Unsupported property on string: testThisIsNoMember");
     }
 
+    @Test("Suported lua string function")
+    public stringSuportedLuaFunction(): void {
+        Expect(util.transpileAndExecute(
+                `return "test".upper()`,
+                undefined,
+                undefined,
+                `interface String { upper(): string; }`
+            )
+        ).toBe("TEST");
+    }
+
     @TestCase([])
     @TestCase([65])
     @TestCase([65, 66])
