Added support for string.substring

Perryvw · Perryvw · commit bb1ae317ea37 · 2018-01-27T11:37:16.000+01:00
diff --git a/dist/Transpiler.js b/dist/Transpiler.js
@@ -485,6 +485,15 @@ var LuaTranspiler = /** @class */ (function () {
                 else {
                     return "(string.find(" + caller + "," + params + "+1,true) or 0)-1";
                 }
+            case "substring":
+                if (node.arguments.length == 1) {
+                    return "string.sub(" + caller + "," + params + "+1)";
+                }
+                else {
+                    var arg1 = this.transpileExpression(node.arguments[0]);
+                    var arg2 = this.transpileExpression(node.arguments[1]);
+                    return "string.sub(" + caller + "," + arg1 + "+1," + arg2 + "+1)";
+                }
             default:
                 throw new TranspileError("Unsupported string function: " + expression.name.escapedText, node);
         }
diff --git a/src/Transpiler.ts b/src/Transpiler.ts
@@ -529,6 +529,14 @@ export class LuaTranspiler {
                 } else {
                     return `(string.find(${caller},${params}+1,true) or 0)-1`;
                 }
+            case "substring":
+                if (node.arguments.length == 1) {
+                    return `string.sub(${caller},${params}+1)`;
+                } else {
+                    const arg1 = this.transpileExpression(node.arguments[0]);
+                    const arg2 = this.transpileExpression(node.arguments[1]);
+                    return `string.sub(${caller},${arg1}+1,${arg2}+1)`;
+                }
             default:
                 throw new TranspileError("Unsupported string function: " + expression.name.escapedText, node);
         }
diff --git a/test/src/test2.lua b/test/src/test2.lua
@@ -31,6 +31,8 @@ function TestClass.Test(self)
     print((string.find(str,"b",2+1,true) or 0)-1)
     print((string.find("abc","c",1,true) or 0)-1)
     print((string.find(str,"b",1,true) or 0)-1)
+    print(string.sub("abcde",2+1))
+    print(string.sub("abcde",1+1,3+1))
 end
 function TestClass.Test3(self,a,b)
     return ""
diff --git a/test/src/test2.ts b/test/src/test2.ts
@@ -47,7 +47,9 @@ export class TestClass {
         print(str[str.indexOf("b")]);
         print(str.indexOf("b", 2));
         print("abc".indexOf("c"));
-        print(str.indexOf("b"))
+        print(str.indexOf("b"));
+        print("abcde".substring(2))
+        print("abcde".substring(1, 3))
     }
 
     Test3(a: number, b: string): string {
