porffor 0.42.6 → 0.42.8

package/compiler/builtins/math.ts

@@ -279,7 +279,7 @@ export const __Math_cbrt = (y: number): number => {
   if (!Number.isFinite(y)) return y;
 
   // Babylonian method
-  let x = Math.abs(y);
+  let x:number = Math.abs(y);
 
   let prev: number;
 

package/compiler/builtins_precompiled.js

@@ -1780,9 +1780,9 @@ params:[124,127],typedParams:1,returns:[124,127],typedReturns:1,
 locals:[124,124],localNames:["y","y#type","x","prev"],
 }
 this.__Math_cbrt = {
-wasm:(_,{builtin})=>[[32,0],[68,0],[97],[4,64],[68,0],[65,1],[15],[11],[32,0],[16,builtin('__Number_isFinite')],[68,0],[97],[4,64],[32,0],[65,1],[15],[11],[32,0],[16,builtin('__Math_abs')],[33,2],[65,1],[33,3],[3,64],[2,64],[32,2],[33,4],[68,2],[32,2],[162],[32,0],[32,2],[32,2],[162],[163],[160],[68,3],[163],[34,2],[65,1],[33,3],[26],[32,4],[32,2],[161],[16,builtin('__Math_abs')],[68,1e-15],[100],[13,1],[11],[11],[32,0],[68,0],[99],[4,124],[32,2],[154],[65,1],[33,5],[5],[32,2],[32,3],[33,5],[11],[32,5],[15]],
+wasm:(_,{builtin})=>[[32,0],[68,0],[97],[4,64],[68,0],[65,1],[15],[11],[32,0],[16,builtin('__Number_isFinite')],[68,0],[97],[4,64],[32,0],[65,1],[15],[11],[32,0],[16,builtin('__Math_abs')],[33,2],[3,64],[2,64],[32,2],[33,3],[68,2],[32,2],[162],[32,0],[32,2],[32,2],[162],[163],[160],[68,3],[163],[33,2],[32,3],[32,2],[161],[16,builtin('__Math_abs')],[68,1e-15],[100],[13,1],[11],[11],[32,0],[68,0],[99],[4,124],[32,2],[154],[65,1],[33,4],[5],[32,2],[65,1],[33,4],[11],[32,4],[15]],
 params:[124,127],typedParams:1,returns:[124,127],typedReturns:1,
-locals:[124,127,124,127],localNames:["y","y#type","x","x#type","prev","#last_type"],
+locals:[124,124,127],localNames:["y","y#type","x","prev","#last_type"],
 }
 this.__Math_hypot = {
 wasm:(_,{builtin})=>[[32,0],[32,0],[162],[32,2],[32,2],[162],[160],[65,1],[16,builtin('__Math_sqrt')],[34,4],[15]],

package/compiler/codegen.js

@@ -399,9 +399,19 @@ const generateIdent = (scope, decl) => {
 
     if (local?.idx === undefined) {
       if (name === 'arguments' && scope.name !== 'main' && !scope.arrow) {
-        // todo: stub arguments as [] for now :)
+        // todo: not compliant
+        let len = countLength(scope);
+        const names = new Array(len);
+        const off = scope.constr ? 4 : 0;
+        for (const x in scope.locals) {
+          const i = scope.locals[x].idx - off;
+          if (i >= 0 && i % 2 === 0 && i < len * 2) {
+            names[i / 2] = x;
+          }
+        }
+
         return generateArray(scope, {
-          elements: []
+          elements: names.map(x => ({ type: 'Identifier', name: x }))
         }, false, '#arguments');
       }
 
@@ -1340,8 +1350,12 @@ const setType = (scope, name, type) => {
 };
 
 const getLastType = scope => {
+  if (!scope.locals['#last_type']) return number(TYPES.number, Valtype.i32);
+
   scope.gotLastType = true;
-  return [ [ Opcodes.local_get, localTmp(scope, '#last_type', Valtype.i32) ] ];
+  return [
+    [ Opcodes.local_get, localTmp(scope, '#last_type', Valtype.i32) ]
+  ];
 };
 
 const setLastType = (scope, type = []) => {
@@ -1388,11 +1402,7 @@ const getNodeType = (scope, node) => {
 
       if (name == null) {
         // iife
-        if (scope.locals['#last_type']) return getLastType(scope);
-
-        // presume
-        if (Prefs.warnAssumedType) console.warn(`Indirect call assumed to be number`);
-        return TYPES.number;
+        return getLastType(scope);
       }
 
       const func = funcByName(name);
@@ -1426,38 +1436,7 @@ const getNodeType = (scope, node) => {
         return TYPES.number;
       }
 
-      if (scope.locals['#last_type']) return getLastType(scope);
-
-      // presume
-      if (Prefs.warnAssumedType) console.warn(`Call to ${name} assumed to be number`);
-      return TYPES.number;
-
-      // let protoFunc;
-      // // ident.func()
-      // if (name && name.startsWith('__')) {
-      //   const spl = name.slice(2).split('_');
-
-      //   const baseName = spl.slice(0, -1).join('_');
-      //   const baseType = getType(scope, baseName);
-
-      //   const func = spl[spl.length - 1];
-      //   protoFunc = prototypeFuncs[baseType]?.[func];
-      // }
-
-      // // literal.func()
-      // if (!name && node.callee.type === 'MemberExpression') {
-      //   if (node.callee.object.regex) {
-      //     const funcName = node.callee.property.name;
-      //     return Rhemyn[funcName] ? TYPES.boolean : TYPES.undefined;
-      //   }
-
-      //   const baseType = getNodeType(scope, node.callee.object);
-
-      //   const func = node.callee.property.name;
-      //   protoFunc = prototypeFuncs[baseType]?.[func];
-      // }
-
-      // if (protoFunc) return protoFunc.returnType;
+      return getLastType(scope);
     }
 
     if (node.type === 'ExpressionStatement') {
@@ -1505,10 +1484,7 @@ const getNodeType = (scope, node) => {
       if (knownLeft === TYPES.bytestring || knownRight === TYPES.bytestring)
         guess = TYPES.bytestring;
 
-      if (scope.locals['#last_type']) return getLastType(scope);
-
-      // presume
-      return TYPES.number;
+      return getLastType(scope);
     }
 
     if (node.type === 'UnaryExpression') {
@@ -1542,11 +1518,7 @@ const getNodeType = (scope, node) => {
         }
       }
 
-      if (scope.locals['#last_type']) return getLastType(scope);
-
-      // presume
-      if (Prefs.warnAssumedType) console.warn(`Member access to field .${name} assumed to be number`);
-      return TYPES.number;
+      return getLastType(scope);
     }
 
     if (node.type === 'TemplateLiteral') {
@@ -1579,11 +1551,7 @@ const getNodeType = (scope, node) => {
       }
     }
 
-    if (scope.locals['#last_type']) return getLastType(scope);
-
-    // presume
-    if (Prefs.warnAssumedType) console.warn(`AST node ${node.type} assumed to be number`);
-    return TYPES.number;
+    return getLastType(scope);
   })();
 
   const out = typeof ret === 'number' ? number(ret, Valtype.i32) : ret;
@@ -2447,6 +2415,31 @@ const generateCall = (scope, decl, _global, _name, unusedValue = false) => {
     paramOffset += 4;
   }
 
+  if (args.at(-1)?.type === 'SpreadElement') {
+    // hack: support spread element if last by doing essentially:
+    // const foo = (a, b, c, d) => ...
+    // foo(a, b, ...c) -> _ = c; foo(a, b, _[0], _[1])
+    const arg = args.at(-1).argument;
+    out.push(
+      ...generate(scope, arg),
+      [ Opcodes.local_set, localTmp(scope, '#spread') ],
+      ...getNodeType(scope, arg),
+      [ Opcodes.local_set, localTmp(scope, '#spread#type', Valtype.i32) ]
+    );
+
+    args.pop();
+    const leftover = paramCount - args.length;
+    for (let i = 0; i < leftover; i++) {
+      args.push({
+        type: 'MemberExpression',
+        object: { type: 'Identifier', name: '#spread' },
+        property: { type: 'Literal', value: i },
+        computed: true,
+        optional: false
+      });
+    }
+  }
+
   if (func && args.length < paramCount) {
     // too little args, push undefineds
     args = args.concat(new Array(paramCount - (func.hasRestArgument ? 1 : 0) - args.length).fill(DEFAULT_VALUE()));
@@ -3689,7 +3682,10 @@ const ifIdentifierErrors = (scope, decl) => {
 const generateUnary = (scope, decl) => {
   switch (decl.operator) {
     case '+':
-      // todo/opt: skip ToNumber if already known number type
+      // opt: skip ToNumber if already known as number type
+      generate(scope, decl.argument); // hack: fix last type not being defined for getNodeType before generation
+      const known = knownType(scope, getNodeType(scope, decl.argument));
+      if (known === TYPES.number) return generate(scope, decl.argument);
 
       // 13.5.4 Unary + Operator, 13.5.4.1 Runtime Semantics: Evaluation
       // https://tc39.es/ecma262/#sec-unary-plus-operator-runtime-semantics-evaluation
@@ -3707,7 +3703,7 @@ const generateUnary = (scope, decl) => {
       });
 
     case '-':
-      // * -1
+      // +x * -1
 
       if (decl.prefix && decl.argument.type === 'Literal' && typeof decl.argument.value === 'number') {
         // if -n, just return that as a const
@@ -3715,7 +3711,12 @@ const generateUnary = (scope, decl) => {
       }
 
       return [
-        ...generate(scope, decl.argument),
+        ...generate(scope, {
+          type: 'UnaryExpression',
+          operator: '+',
+          prefix: true,
+          argument: decl.argument
+        }),
         ...(valtype === 'f64' ? [ [ Opcodes.f64_neg ] ] : [ ...number(-1), [ Opcodes.mul ] ])
       ];
 
@@ -3731,7 +3732,12 @@ const generateUnary = (scope, decl) => {
 
     case '~':
       return [
-        ...generate(scope, decl.argument),
+        ...generate(scope, {
+          type: 'UnaryExpression',
+          operator: '+',
+          prefix: true,
+          argument: decl.argument
+        }),
         Opcodes.i32_to,
         [ Opcodes.i32_const, ...signedLEB128(-1) ],
         [ Opcodes.i32_xor ],

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "porffor",
   "description": "a basic experimental wip aot optimizing js -> wasm engine/compiler/runtime in js",
-  "version": "0.42.6+d8f9d3f11",
+  "version": "0.42.8+06fc0f2c7",
   "author": "CanadaHonk",
   "license": "MIT",
   "scripts": {},

package/runner/index.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import fs from 'node:fs';
-globalThis.version = '0.42.6+d8f9d3f11';
+globalThis.version = '0.42.8+06fc0f2c7';
 
 // deno compat
 if (typeof process === 'undefined' && typeof Deno !== 'undefined') {

package/runner/repl.js

@@ -1,5 +1,6 @@
 import { TYPE_NAMES } from '../compiler/types.js';
 import compile from '../compiler/wrap.js';
+import parse from '../compiler/parse.js';
 
 import util from 'node:util';
 
@@ -83,7 +84,13 @@ const run = (source, _context, _filename, callback, run = true) => {
   // hack: print "secret" before latest code ran to only enable printing for new code
 
   source = source.trim();
-  if (source.startsWith('{') && source.endsWith('}')) source = '(' + source + ')';
+  if (source.startsWith('{') && source.endsWith('}')) {
+    const wrapped = '(' + source + ')';
+    try {
+      parse(wrapped);
+      source = wrapped;
+    } catch {}
+  }
 
   let toRun = (prev ? (prev + `;\nprint(-0x1337);\n`) : '') + source;