@creationix/rex 0.3.0 → 0.3.1

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@creationix/rex",
-	"version": "0.3.0",
+	"version": "0.3.1",
 	"description": "Compiler and parser for the Rex language",
 	"keywords": [
 		"rex",
@@ -40,7 +40,8 @@
 		"compile": "bun run rex-cli.ts",
 		"verify:docs": "bun run verify-doc-examples.ts",
 		"pack:dry-run": "npm pack --dry-run",
-		"prepublishOnly": "bun run build:grammar && bun test && bun run build:js && npm run pack:dry-run"
+		"prepublishOnly": "bun run build:grammar && bun test && bun run build:js && npm run pack:dry-run",
+		"release": "npm publish"
 	},
 	"publishConfig": {
 		"access": "public"

package/rex-cli.js

@@ -100,12 +100,32 @@ function encodeBareOrLengthString(value) {
     return `${value}:`;
   return `${encodeUint(byteLength(value))},${value}`;
 }
+function splitDecimal(num) {
+  const match = num.toExponential().match(DEC_PARTS);
+  if (!match)
+    throw new Error(`Failed to split decimal for ${num}`);
+  const [, b1, b2 = "", e1] = match;
+  const base = Number.parseInt(b1 + b2, 10);
+  const exp = Number.parseInt(e1, 10) - b2.length;
+  return { base, exp };
+}
+function encodeDecimal(significand, power) {
+  return `${encodeZigzag(power)}*${encodeInt(significand)}`;
+}
 function encodeNumberNode(node) {
   const numberValue = node.value;
-  if (!Number.isFinite(numberValue))
-    throw new Error(`Cannot encode non-finite number: ${node.raw}`);
-  if (Number.isInteger(numberValue))
-    return encodeInt(numberValue);
+  if (Number.isNaN(numberValue))
+    return "5'";
+  if (numberValue === Infinity)
+    return "6'";
+  if (numberValue === -Infinity)
+    return "7'";
+  if (Number.isInteger(numberValue)) {
+    const { base, exp } = splitDecimal(numberValue);
+    if (exp >= 0 && exp <= 4)
+      return encodeInt(numberValue);
+    return encodeDecimal(base, exp);
+  }
   const raw = node.raw.toLowerCase();
   const sign = raw.startsWith("-") ? -1 : 1;
   const unsigned = sign < 0 ? raw.slice(1) : raw;
@@ -128,7 +148,7 @@ function encodeNumberNode(node) {
     significand /= 10;
     power += 1;
   }
-  return `${encodeZigzag(power)}*${encodeInt(significand)}`;
+  return encodeDecimal(significand, power);
 }
 function encodeOpcode(opcode) {
   return `${encodeUint(OPCODE_IDS[opcode])}%`;
@@ -429,6 +449,18 @@ function isPlainObject(value) {
 function isBareKeyName(key) {
   return /^[A-Za-z_][A-Za-z0-9_-]*$/.test(key);
 }
+function isNumericKey(key) {
+  if (key === "")
+    return false;
+  return String(Number(key)) === key && Number.isFinite(Number(key));
+}
+function stringifyKey(key) {
+  if (isBareKeyName(key))
+    return key;
+  if (isNumericKey(key))
+    return key;
+  return stringifyString(key);
+}
 function stringifyString(value) {
   return JSON.stringify(value);
 }
@@ -440,8 +472,12 @@ function stringifyInline(value) {
   if (typeof value === "boolean")
     return value ? "true" : "false";
   if (typeof value === "number") {
-    if (!Number.isFinite(value))
-      throw new Error("Rex stringify() cannot encode non-finite numbers");
+    if (Number.isNaN(value))
+      return "nan";
+    if (value === Infinity)
+      return "inf";
+    if (value === -Infinity)
+      return "-inf";
     return String(value);
   }
   if (typeof value === "string")
@@ -455,7 +491,7 @@ function stringifyInline(value) {
     const entries = Object.entries(value);
     if (entries.length === 0)
       return "{}";
-    const body = entries.map(([key, item]) => `${isBareKeyName(key) ? key : stringifyString(key)}: ${stringifyInline(item)}`).join(" ");
+    const body = entries.map(([key, item]) => `${stringifyKey(key)}: ${stringifyInline(item)}`).join(" ");
     return `{${body}}`;
   }
   throw new Error(`Rex stringify() cannot encode value of type ${typeof value}`);
@@ -492,7 +528,7 @@ ${indent}]`;
     if (entries.length === 0)
       return "{}";
     const lines = entries.map(([key, item]) => {
-      const keyText = isBareKeyName(key) ? key : stringifyString(key);
+      const keyText = stringifyKey(key);
       const rendered = stringifyPretty(item, depth + 1, indentSize, maxWidth);
       return `${childIndent}${keyText}: ${rendered}`;
     });
@@ -1150,7 +1186,16 @@ function inlineAdjacentPureAssignments(block) {
   return out;
 }
 function toNumberNode(value) {
-  return { type: "number", raw: String(value), value };
+  let raw;
+  if (Number.isNaN(value))
+    raw = "nan";
+  else if (value === Infinity)
+    raw = "inf";
+  else if (value === -Infinity)
+    raw = "-inf";
+  else
+    raw = String(value);
+  return { type: "number", raw, value };
 }
 function toStringNode(value) {
   return { type: "string", raw: JSON.stringify(value) };
@@ -1162,7 +1207,7 @@ function toLiteralNode(value) {
     return { type: "null" };
   if (typeof value === "boolean")
     return { type: "boolean", value };
-  if (typeof value === "number" && Number.isFinite(value))
+  if (typeof value === "number")
     return toNumberNode(value);
   if (typeof value === "string")
     return toStringNode(value);
@@ -1983,6 +2028,12 @@ function compile(source, options) {
   });
 }
 function parseNumber(raw) {
+  if (raw === "nan")
+    return NaN;
+  if (raw === "inf")
+    return Infinity;
+  if (raw === "-inf")
+    return -Infinity;
   if (/^-?0x/i.test(raw))
     return parseInt(raw, 16);
   if (/^-?0b/i.test(raw)) {
@@ -2054,7 +2105,7 @@ function buildPostfix(base, steps) {
   flushSegments();
   return current;
 }
-var require2, rexGrammarModule, rexGrammar, grammar, semantics, DIGITS = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-_", OPCODE_IDS, FIRST_NON_RESERVED_REF = 5, DOMAIN_DIGIT_INDEX, BINARY_TO_OPCODE, ASSIGN_COMPOUND_TO_OPCODE, activeEncodeOptions, DIGIT_SET, DIGIT_INDEX;
+var require2, rexGrammarModule, rexGrammar, grammar, semantics, DIGITS = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-_", OPCODE_IDS, FIRST_NON_RESERVED_REF = 8, DOMAIN_DIGIT_INDEX, BINARY_TO_OPCODE, ASSIGN_COMPOUND_TO_OPCODE, DEC_PARTS, activeEncodeOptions, DIGIT_SET, DIGIT_INDEX;
 var init_rex = __esm(() => {
   require2 = createRequire(import.meta.url);
   rexGrammarModule = require2("./rex.ohm-bundle.cjs");
@@ -2114,6 +2165,7 @@ var init_rex = __esm(() => {
     "|=": "or",
     "^=": "xor"
   };
+  DEC_PARTS = /^(-?\d)(?:\.(\d+))?e([+-]\d+)$/;
   DIGIT_SET = new Set(DIGITS.split(""));
   DIGIT_INDEX = new Map(Array.from(DIGITS).map((char, index) => [char, index]));
   semantics.addOperation("toIR", {
@@ -2462,7 +2514,10 @@ class CursorInterpreter {
         1: ctx.refs?.[1] ?? true,
         2: ctx.refs?.[2] ?? false,
         3: ctx.refs?.[3] ?? null,
-        4: ctx.refs?.[4] ?? undefined
+        4: ctx.refs?.[4] ?? undefined,
+        5: ctx.refs?.[5] ?? NaN,
+        6: ctx.refs?.[6] ?? Infinity,
+        7: ctx.refs?.[7] ?? -Infinity
       }
     };
     this.selfStack = ctx.selfStack && ctx.selfStack.length > 0 ? [...ctx.selfStack] : [initialSelf];
@@ -3019,25 +3074,39 @@ class CursorInterpreter {
       case OPCODES.do:
         return args.length ? args[args.length - 1] : undefined;
       case OPCODES.add:
+        if (args[0] === undefined || args[1] === undefined)
+          return;
         if (typeof args[0] === "string" || typeof args[1] === "string") {
-          return String(args[0] ?? "") + String(args[1] ?? "");
+          return String(args[0]) + String(args[1]);
         }
-        return Number(args[0] ?? 0) + Number(args[1] ?? 0);
+        return Number(args[0]) + Number(args[1]);
       case OPCODES.sub:
-        return Number(args[0] ?? 0) - Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) - Number(args[1]);
       case OPCODES.mul:
-        return Number(args[0] ?? 0) * Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) * Number(args[1]);
       case OPCODES.div:
-        return Number(args[0] ?? 0) / Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) / Number(args[1]);
       case OPCODES.mod:
-        return Number(args[0] ?? 0) % Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) % Number(args[1]);
       case OPCODES.neg:
-        return -Number(args[0] ?? 0);
+        if (args[0] === undefined)
+          return;
+        return -Number(args[0]);
       case OPCODES.not: {
         const value = args[0];
+        if (value === undefined)
+          return;
         if (typeof value === "boolean")
           return !value;
-        return ~Number(value ?? 0);
+        return ~Number(value);
       }
       case OPCODES.and: {
         const [a, b] = args;
@@ -3310,6 +3379,7 @@ __export(exports_rex_repl, {
   isIncomplete: () => isIncomplete,
   highlightRexc: () => highlightRexc,
   highlightLine: () => highlightLine,
+  highlightJSON: () => highlightJSON,
   formatVarState: () => formatVarState
 });
 import * as readline from "node:readline";
@@ -3460,6 +3530,32 @@ function highlightRexc(text) {
   }
   return out;
 }
+function highlightJSON(json) {
+  let result = "";
+  let lastIndex = 0;
+  JSON_TOKEN_RE.lastIndex = 0;
+  for (const m of json.matchAll(JSON_TOKEN_RE)) {
+    result += json.slice(lastIndex, m.index);
+    const text = m[0];
+    const g = m.groups;
+    if (g.key) {
+      result += C.cyan + g.key + C.reset + ":";
+    } else if (g.string) {
+      result += C.green + text + C.reset;
+    } else if (g.number) {
+      result += C.yellow + text + C.reset;
+    } else if (g.bool) {
+      result += C.yellow + text + C.reset;
+    } else if (g.null) {
+      result += C.dim + text + C.reset;
+    } else {
+      result += text;
+    }
+    lastIndex = m.index + text.length;
+  }
+  result += json.slice(lastIndex);
+  return result;
+}
 function stripStringsAndComments(source) {
   return source.replace(/\/\*[\s\S]*?\*\/|\/\/[^\n]*|"(?:[^"\\]|\\.)*"|'(?:[^'\\]|\\.)*'/g, (m) => " ".repeat(m.length));
 }
@@ -3707,7 +3803,7 @@ async function startRepl() {
       const ir = parseToIR(source);
       const lowered = state.optimize ? optimizeIR(ir) : ir;
       if (state.showIR) {
-        console.log(`${C.dim}  IR: ${JSON.stringify(lowered)}${C.reset}`);
+        console.log(`${C.dim}  IR:${C.reset} ${highlightJSON(JSON.stringify(lowered))}`);
       }
       const rexc = compile(source, { optimize: state.optimize });
       if (state.showRexc) {
@@ -3739,7 +3835,7 @@ async function startRepl() {
   });
   rl.prompt();
 }
-var req, version, C, TOKEN_RE, REXC_DIGITS, KEYWORDS, GAS_LIMIT = 1e7;
+var req, version, C, TOKEN_RE, REXC_DIGITS, JSON_TOKEN_RE, KEYWORDS, GAS_LIMIT = 1e7;
 var init_rex_repl = __esm(() => {
   init_rex();
   init_rexc_interpreter();
@@ -3757,8 +3853,9 @@ var init_rex_repl = __esm(() => {
     gray: "\x1B[90m",
     boldBlue: "\x1B[1;34m"
   };
-  TOKEN_RE = /(?<blockComment>\/\*[\s\S]*?(?:\*\/|$))|(?<lineComment>\/\/[^\n]*)|(?<dstring>"(?:[^"\\]|\\.)*"?)|(?<sstring>'(?:[^'\\]|\\.)*'?)|(?<keyword>\b(?:when|unless|while|for|do|end|in|of|and|or|else|break|continue|delete|self)(?![a-zA-Z0-9_-]))|(?<literal>\b(?:true|false|null|undefined)(?![a-zA-Z0-9_-]))|(?<typePred>\b(?:string|number|object|array|boolean)(?![a-zA-Z0-9_-]))|(?<num>\b(?:0x[0-9a-fA-F]+|0b[01]+|(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b)/g;
+  TOKEN_RE = /(?<blockComment>\/\*[\s\S]*?(?:\*\/|$))|(?<lineComment>\/\/[^\n]*)|(?<dstring>"(?:[^"\\]|\\.)*"?)|(?<sstring>'(?:[^'\\]|\\.)*'?)|(?<keyword>\b(?:when|unless|while|for|do|end|in|of|and|or|else|break|continue|delete|self)(?![a-zA-Z0-9_-]))|(?<literal>\b(?:true|false|null|undefined|nan)(?![a-zA-Z0-9_-])|-?\binf\b)|(?<typePred>\b(?:string|number|object|array|boolean)(?![a-zA-Z0-9_-]))|(?<num>\b(?:0x[0-9a-fA-F]+|0b[01]+|(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b)/g;
   REXC_DIGITS = new Set("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-_");
+  JSON_TOKEN_RE = /(?<key>"(?:[^"\\]|\\.)*")\s*:|(?<string>"(?:[^"\\]|\\.)*")|(?<number>-?(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b|(?<bool>true|false)|(?<null>null)|(?<brace>[{}[\]])|(?<punct>[:,])/g;
   KEYWORDS = [
     "when",
     "unless",
@@ -3779,6 +3876,8 @@ var init_rex_repl = __esm(() => {
     "false",
     "null",
     "undefined",
+    "nan",
+    "inf",
     "string",
     "number",
     "object",
@@ -3962,7 +4061,7 @@ async function main() {
     });
   } else {
     const { value } = evaluateSource(source);
-    output = JSON.stringify(value, null, 2);
+    output = stringify(value);
   }
   if (options.out) {
     await writeFile(options.out, `${output}

package/rex-cli.ts

@@ -1,4 +1,4 @@
-import { compile, parse, parseToIR } from "./rex.ts";
+import { compile, parse, parseToIR, stringify } from "./rex.ts";
 import { evaluateSource } from "./rexc-interpreter.ts";
 import { dirname, resolve } from "node:path";
 import { readFile, writeFile } from "node:fs/promises";
@@ -183,7 +183,7 @@ async function main() {
 		});
 	} else {
 		const { value } = evaluateSource(source);
-		output = JSON.stringify(value, null, 2);
+		output = stringify(value);
 	}
 
 	if (options.out) {

package/rex-repl.js

@@ -37,7 +37,7 @@ var OPCODE_IDS = {
   mod: 20,
   neg: 21
 };
-var FIRST_NON_RESERVED_REF = 5;
+var FIRST_NON_RESERVED_REF = 8;
 var DOMAIN_DIGIT_INDEX = new Map(Array.from(DIGITS).map((char, index) => [char, index]));
 var BINARY_TO_OPCODE = {
   add: "add",
@@ -151,12 +151,33 @@ function encodeBareOrLengthString(value) {
     return `${value}:`;
   return `${encodeUint(byteLength(value))},${value}`;
 }
+var DEC_PARTS = /^(-?\d)(?:\.(\d+))?e([+-]\d+)$/;
+function splitDecimal(num) {
+  const match = num.toExponential().match(DEC_PARTS);
+  if (!match)
+    throw new Error(`Failed to split decimal for ${num}`);
+  const [, b1, b2 = "", e1] = match;
+  const base = Number.parseInt(b1 + b2, 10);
+  const exp = Number.parseInt(e1, 10) - b2.length;
+  return { base, exp };
+}
+function encodeDecimal(significand, power) {
+  return `${encodeZigzag(power)}*${encodeInt(significand)}`;
+}
 function encodeNumberNode(node) {
   const numberValue = node.value;
-  if (!Number.isFinite(numberValue))
-    throw new Error(`Cannot encode non-finite number: ${node.raw}`);
-  if (Number.isInteger(numberValue))
-    return encodeInt(numberValue);
+  if (Number.isNaN(numberValue))
+    return "5'";
+  if (numberValue === Infinity)
+    return "6'";
+  if (numberValue === -Infinity)
+    return "7'";
+  if (Number.isInteger(numberValue)) {
+    const { base, exp } = splitDecimal(numberValue);
+    if (exp >= 0 && exp <= 4)
+      return encodeInt(numberValue);
+    return encodeDecimal(base, exp);
+  }
   const raw = node.raw.toLowerCase();
   const sign = raw.startsWith("-") ? -1 : 1;
   const unsigned = sign < 0 ? raw.slice(1) : raw;
@@ -179,7 +200,7 @@ function encodeNumberNode(node) {
     significand /= 10;
     power += 1;
   }
-  return `${encodeZigzag(power)}*${encodeInt(significand)}`;
+  return encodeDecimal(significand, power);
 }
 function encodeOpcode(opcode) {
   return `${encodeUint(OPCODE_IDS[opcode])}%`;
@@ -439,6 +460,18 @@ function isPlainObject(value) {
 function isBareKeyName(key) {
   return /^[A-Za-z_][A-Za-z0-9_-]*$/.test(key);
 }
+function isNumericKey(key) {
+  if (key === "")
+    return false;
+  return String(Number(key)) === key && Number.isFinite(Number(key));
+}
+function stringifyKey(key) {
+  if (isBareKeyName(key))
+    return key;
+  if (isNumericKey(key))
+    return key;
+  return stringifyString(key);
+}
 function stringifyString(value) {
   return JSON.stringify(value);
 }
@@ -450,8 +483,12 @@ function stringifyInline(value) {
   if (typeof value === "boolean")
     return value ? "true" : "false";
   if (typeof value === "number") {
-    if (!Number.isFinite(value))
-      throw new Error("Rex stringify() cannot encode non-finite numbers");
+    if (Number.isNaN(value))
+      return "nan";
+    if (value === Infinity)
+      return "inf";
+    if (value === -Infinity)
+      return "-inf";
     return String(value);
   }
   if (typeof value === "string")
@@ -465,7 +502,7 @@ function stringifyInline(value) {
     const entries = Object.entries(value);
     if (entries.length === 0)
       return "{}";
-    const body = entries.map(([key, item]) => `${isBareKeyName(key) ? key : stringifyString(key)}: ${stringifyInline(item)}`).join(" ");
+    const body = entries.map(([key, item]) => `${stringifyKey(key)}: ${stringifyInline(item)}`).join(" ");
     return `{${body}}`;
   }
   throw new Error(`Rex stringify() cannot encode value of type ${typeof value}`);
@@ -502,7 +539,7 @@ ${indent}]`;
     if (entries.length === 0)
       return "{}";
     const lines = entries.map(([key, item]) => {
-      const keyText = isBareKeyName(key) ? key : stringifyString(key);
+      const keyText = stringifyKey(key);
       const rendered = stringifyPretty(item, depth + 1, indentSize, maxWidth);
       return `${childIndent}${keyText}: ${rendered}`;
     });
@@ -1159,7 +1196,16 @@ function inlineAdjacentPureAssignments(block) {
   return out;
 }
 function toNumberNode(value) {
-  return { type: "number", raw: String(value), value };
+  let raw;
+  if (Number.isNaN(value))
+    raw = "nan";
+  else if (value === Infinity)
+    raw = "inf";
+  else if (value === -Infinity)
+    raw = "-inf";
+  else
+    raw = String(value);
+  return { type: "number", raw, value };
 }
 function toStringNode(value) {
   return { type: "string", raw: JSON.stringify(value) };
@@ -1171,7 +1217,7 @@ function toLiteralNode(value) {
     return { type: "null" };
   if (typeof value === "boolean")
     return { type: "boolean", value };
-  if (typeof value === "number" && Number.isFinite(value))
+  if (typeof value === "number")
     return toNumberNode(value);
   if (typeof value === "string")
     return toStringNode(value);
@@ -1992,6 +2038,12 @@ function compile(source, options) {
   });
 }
 function parseNumber(raw) {
+  if (raw === "nan")
+    return NaN;
+  if (raw === "inf")
+    return Infinity;
+  if (raw === "-inf")
+    return -Infinity;
   if (/^-?0x/i.test(raw))
     return parseInt(raw, 16);
   if (/^-?0b/i.test(raw)) {
@@ -2434,7 +2486,10 @@ class CursorInterpreter {
         1: ctx.refs?.[1] ?? true,
         2: ctx.refs?.[2] ?? false,
         3: ctx.refs?.[3] ?? null,
-        4: ctx.refs?.[4] ?? undefined
+        4: ctx.refs?.[4] ?? undefined,
+        5: ctx.refs?.[5] ?? NaN,
+        6: ctx.refs?.[6] ?? Infinity,
+        7: ctx.refs?.[7] ?? -Infinity
       }
     };
     this.selfStack = ctx.selfStack && ctx.selfStack.length > 0 ? [...ctx.selfStack] : [initialSelf];
@@ -2991,25 +3046,39 @@ class CursorInterpreter {
       case OPCODES.do:
         return args.length ? args[args.length - 1] : undefined;
       case OPCODES.add:
+        if (args[0] === undefined || args[1] === undefined)
+          return;
         if (typeof args[0] === "string" || typeof args[1] === "string") {
-          return String(args[0] ?? "") + String(args[1] ?? "");
+          return String(args[0]) + String(args[1]);
         }
-        return Number(args[0] ?? 0) + Number(args[1] ?? 0);
+        return Number(args[0]) + Number(args[1]);
       case OPCODES.sub:
-        return Number(args[0] ?? 0) - Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) - Number(args[1]);
       case OPCODES.mul:
-        return Number(args[0] ?? 0) * Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) * Number(args[1]);
       case OPCODES.div:
-        return Number(args[0] ?? 0) / Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) / Number(args[1]);
       case OPCODES.mod:
-        return Number(args[0] ?? 0) % Number(args[1] ?? 0);
+        if (args[0] === undefined || args[1] === undefined)
+          return;
+        return Number(args[0]) % Number(args[1]);
       case OPCODES.neg:
-        return -Number(args[0] ?? 0);
+        if (args[0] === undefined)
+          return;
+        return -Number(args[0]);
       case OPCODES.not: {
         const value = args[0];
+        if (value === undefined)
+          return;
         if (typeof value === "boolean")
           return !value;
-        return ~Number(value ?? 0);
+        return ~Number(value);
       }
       case OPCODES.and: {
         const [a, b] = args;
@@ -3258,7 +3327,7 @@ var C = {
   gray: "\x1B[90m",
   boldBlue: "\x1B[1;34m"
 };
-var TOKEN_RE = /(?<blockComment>\/\*[\s\S]*?(?:\*\/|$))|(?<lineComment>\/\/[^\n]*)|(?<dstring>"(?:[^"\\]|\\.)*"?)|(?<sstring>'(?:[^'\\]|\\.)*'?)|(?<keyword>\b(?:when|unless|while|for|do|end|in|of|and|or|else|break|continue|delete|self)(?![a-zA-Z0-9_-]))|(?<literal>\b(?:true|false|null|undefined)(?![a-zA-Z0-9_-]))|(?<typePred>\b(?:string|number|object|array|boolean)(?![a-zA-Z0-9_-]))|(?<num>\b(?:0x[0-9a-fA-F]+|0b[01]+|(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b)/g;
+var TOKEN_RE = /(?<blockComment>\/\*[\s\S]*?(?:\*\/|$))|(?<lineComment>\/\/[^\n]*)|(?<dstring>"(?:[^"\\]|\\.)*"?)|(?<sstring>'(?:[^'\\]|\\.)*'?)|(?<keyword>\b(?:when|unless|while|for|do|end|in|of|and|or|else|break|continue|delete|self)(?![a-zA-Z0-9_-]))|(?<literal>\b(?:true|false|null|undefined|nan)(?![a-zA-Z0-9_-])|-?\binf\b)|(?<typePred>\b(?:string|number|object|array|boolean)(?![a-zA-Z0-9_-]))|(?<num>\b(?:0x[0-9a-fA-F]+|0b[01]+|(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b)/g;
 function highlightLine(line) {
   let result = "";
   let lastIndex = 0;
@@ -3406,6 +3475,33 @@ function highlightRexc(text) {
   }
   return out;
 }
+var JSON_TOKEN_RE = /(?<key>"(?:[^"\\]|\\.)*")\s*:|(?<string>"(?:[^"\\]|\\.)*")|(?<number>-?(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b|(?<bool>true|false)|(?<null>null)|(?<brace>[{}[\]])|(?<punct>[:,])/g;
+function highlightJSON(json) {
+  let result = "";
+  let lastIndex = 0;
+  JSON_TOKEN_RE.lastIndex = 0;
+  for (const m of json.matchAll(JSON_TOKEN_RE)) {
+    result += json.slice(lastIndex, m.index);
+    const text = m[0];
+    const g = m.groups;
+    if (g.key) {
+      result += C.cyan + g.key + C.reset + ":";
+    } else if (g.string) {
+      result += C.green + text + C.reset;
+    } else if (g.number) {
+      result += C.yellow + text + C.reset;
+    } else if (g.bool) {
+      result += C.yellow + text + C.reset;
+    } else if (g.null) {
+      result += C.dim + text + C.reset;
+    } else {
+      result += text;
+    }
+    lastIndex = m.index + text.length;
+  }
+  result += json.slice(lastIndex);
+  return result;
+}
 function stripStringsAndComments(source) {
   return source.replace(/\/\*[\s\S]*?\*\/|\/\/[^\n]*|"(?:[^"\\]|\\.)*"|'(?:[^'\\]|\\.)*'/g, (m) => " ".repeat(m.length));
 }
@@ -3504,6 +3600,8 @@ var KEYWORDS = [
   "false",
   "null",
   "undefined",
+  "nan",
+  "inf",
   "string",
   "number",
   "object",
@@ -3680,7 +3778,7 @@ async function startRepl() {
       const ir = parseToIR(source);
       const lowered = state.optimize ? optimizeIR(ir) : ir;
       if (state.showIR) {
-        console.log(`${C.dim}  IR: ${JSON.stringify(lowered)}${C.reset}`);
+        console.log(`${C.dim}  IR:${C.reset} ${highlightJSON(JSON.stringify(lowered))}`);
       }
       const rexc = compile(source, { optimize: state.optimize });
       if (state.showRexc) {
@@ -3717,5 +3815,6 @@ export {
   isIncomplete,
   highlightRexc,
   highlightLine,
+  highlightJSON,
   formatVarState
 };

package/rex-repl.ts

@@ -24,7 +24,7 @@ const C = {
 // ── Syntax highlighting ───────────────────────────────────────
 
 const TOKEN_RE =
-	/(?<blockComment>\/\*[\s\S]*?(?:\*\/|$))|(?<lineComment>\/\/[^\n]*)|(?<dstring>"(?:[^"\\]|\\.)*"?)|(?<sstring>'(?:[^'\\]|\\.)*'?)|(?<keyword>\b(?:when|unless|while|for|do|end|in|of|and|or|else|break|continue|delete|self)(?![a-zA-Z0-9_-]))|(?<literal>\b(?:true|false|null|undefined)(?![a-zA-Z0-9_-]))|(?<typePred>\b(?:string|number|object|array|boolean)(?![a-zA-Z0-9_-]))|(?<num>\b(?:0x[0-9a-fA-F]+|0b[01]+|(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b)/g;
+	/(?<blockComment>\/\*[\s\S]*?(?:\*\/|$))|(?<lineComment>\/\/[^\n]*)|(?<dstring>"(?:[^"\\]|\\.)*"?)|(?<sstring>'(?:[^'\\]|\\.)*'?)|(?<keyword>\b(?:when|unless|while|for|do|end|in|of|and|or|else|break|continue|delete|self)(?![a-zA-Z0-9_-]))|(?<literal>\b(?:true|false|null|undefined|nan)(?![a-zA-Z0-9_-])|-?\binf\b)|(?<typePred>\b(?:string|number|object|array|boolean)(?![a-zA-Z0-9_-]))|(?<num>\b(?:0x[0-9a-fA-F]+|0b[01]+|(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b)/g;
 
 export function highlightLine(line: string): string {
 	let result = "";
@@ -180,6 +180,39 @@ export function highlightRexc(text: string): string {
 	return out;
 }
 
+// ── JSON IR highlighting ─────────────────────────────────────
+
+const JSON_TOKEN_RE =
+	/(?<key>"(?:[^"\\]|\\.)*")\s*:|(?<string>"(?:[^"\\]|\\.)*")|(?<number>-?(?:0|[1-9]\d*)(?:\.\d+)?(?:[eE][+-]?\d+)?)\b|(?<bool>true|false)|(?<null>null)|(?<brace>[{}[\]])|(?<punct>[:,])/g;
+
+export function highlightJSON(json: string): string {
+	let result = "";
+	let lastIndex = 0;
+	JSON_TOKEN_RE.lastIndex = 0;
+
+	for (const m of json.matchAll(JSON_TOKEN_RE)) {
+		result += json.slice(lastIndex, m.index);
+		const text = m[0];
+		const g = m.groups!;
+		if (g.key) {
+			result += C.cyan + g.key + C.reset + ":";
+		} else if (g.string) {
+			result += C.green + text + C.reset;
+		} else if (g.number) {
+			result += C.yellow + text + C.reset;
+		} else if (g.bool) {
+			result += C.yellow + text + C.reset;
+		} else if (g.null) {
+			result += C.dim + text + C.reset;
+		} else {
+			result += text;
+		}
+		lastIndex = m.index! + text.length;
+	}
+	result += json.slice(lastIndex);
+	return result;
+}
+
 // ── Multi-line detection ──────────────────────────────────────
 
 /** Strip string literals and comments, replacing them with spaces. */
@@ -277,7 +310,7 @@ export function formatVarState(vars: Record<string, unknown>): string {
 const KEYWORDS = [
 	"when", "unless", "while", "for", "do", "end", "in", "of",
 	"and", "or", "else", "break", "continue", "delete",
-	"self", "true", "false", "null", "undefined",
+	"self", "true", "false", "null", "undefined", "nan", "inf",
 	"string", "number", "object", "array", "boolean",
 ];
 
@@ -504,7 +537,7 @@ export async function startRepl(): Promise<void> {
 			const lowered = state.optimize ? optimizeIR(ir) : ir;
 
 			if (state.showIR) {
-				console.log(`${C.dim}  IR: ${JSON.stringify(lowered)}${C.reset}`);
+				console.log(`${C.dim}  IR:${C.reset} ${highlightJSON(JSON.stringify(lowered))}`);
 			}
 
 			const rexc = compile(source, { optimize: state.optimize });