Implement Number.toFixed

Dimava · Dimava · commit 937f8332651d · 2023-02-28T16:34:53.000+03:00
diff --git a/src/LuaLib.ts b/src/LuaLib.ts
@@ -58,6 +58,7 @@ export enum LuaLibFeature {
     NumberIsFinite = "NumberIsFinite",
     NumberIsNaN = "NumberIsNaN",
     NumberToString = "NumberToString",
+    NumberToFixed = "NumberToFixed",
     ObjectAssign = "ObjectAssign",
     ObjectDefineProperty = "ObjectDefineProperty",
     ObjectEntries = "ObjectEntries",
diff --git a/src/lualib/NumberToFixed.ts b/src/lualib/NumberToFixed.ts
@@ -0,0 +1,13 @@
+/// https://www.ecma-international.org/ecma-262/10.0/index.html#sec-number.prototype.tofixed
+export function __TS__NumberToFixed(this: number, fractionDigits?: number): string {
+    if (Math.abs(this) >= 1e21 || this !== this) {
+        return this.toString();
+    }
+    const f = Math.floor(fractionDigits ?? 0);
+    if (f < 0 || f > 100) {
+        throw "toFixed() digits argument must be between 0 and 100";
+    }
+    // throws "invalid format (width or precision too long)" if strlen > 99
+    // if (f < 80) return fmt; else try {return fmt} catch(_) { throw "toFixed() digits argument..." }
+    return string.format(`%.${f}f`, this);
+}
diff --git a/src/transformation/builtins/number.ts b/src/transformation/builtins/number.ts
@@ -20,6 +20,8 @@ export function transformNumberPrototypeCall(
             return params.length === 0
                 ? lua.createCallExpression(lua.createIdentifier("tostring"), [caller], node)
                 : transformLuaLibFunction(context, LuaLibFeature.NumberToString, node, caller, ...params);
+        case "toFixed":
+            return transformLuaLibFunction(context, LuaLibFeature.NumberToFixed, node, caller, ...params);
         default:
             context.diagnostics.push(unsupportedProperty(calledMethod.name, "number", expressionName));
     }
diff --git a/test/unit/builtins/numbers.spec.ts b/test/unit/builtins/numbers.spec.ts
@@ -69,6 +69,13 @@ test.each([
     util.testExpressionTemplate`(${value}).toString(2)`.expectToEqual(luaNativeSpecialNumString);
 });
 
+const toFixedRadixes = [undefined, 0, 1, 2, 5, 10, 80];
+const toFixedValues = [-1, 0, 1, /* 1.5 fails, */ 1024, 1.2, 1e19, 1e22];
+const toFixedPairs = toFixedValues.flatMap(value => toFixedRadixes.map(frac => [value, frac] as const));
+test.each(toFixedPairs)("(%p).toFixed(%p)", (value, frac) => {
+    util.testExpressionTemplate`(${value}).toFixed(${frac})`.expectToMatchJsResult();
+});
+
 test.each(cases)("isNaN(%p)", value => {
     util.testExpressionTemplate`isNaN(${value} as any)`.expectToMatchJsResult();
 });

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,8 @@ export function transformNumberPrototypeCall(`
`20`	`20`	`return params.length === 0`
`21`	`21`	`? lua.createCallExpression(lua.createIdentifier("tostring"), [caller], node)`
`22`	`22`	`: transformLuaLibFunction(context, LuaLibFeature.NumberToString, node, caller, ...params);`
	`23`	`+ case "toFixed":`
	`24`	`+ return transformLuaLibFunction(context, LuaLibFeature.NumberToFixed, node, caller, ...params);`
`23`	`25`	`default:`
`24`	`26`	`context.diagnostics.push(unsupportedProperty(calledMethod.name, "number", expressionName));`
`25`	`27`	`}`