@fraczak/k 1.3.4 → 2.0.1

package/pretty.mjs

@@ -0,0 +1,95 @@
+function compareAs(fn) {
+  return function (a, b) {
+    [a, b] = [a, b].map(fn);
+    if (a < b) return -1;
+    if (a > b) return 1;
+    return 0;
+  };
+}
+
+const nameRE = /^[a-zA-Z0-9_][a-zA-Z0-9_?!]*$/;
+
+function prettyCode_labels (codes, representatives, label_ref_map) {
+  return Object.keys(label_ref_map)
+    .sort( compareAs((x) => x) )
+    .map( (label) => {
+      const plabel = nameRE.test(label) ? label : `'${label}'`;
+      return `${prettyCode(codes, representatives, {
+        code: "ref",
+        ref: label_ref_map[label],
+      })} ${plabel}`;
+    })
+    .join(", ");
+};
+
+function prettyCode (codes, representatives, codeExp) {
+  switch (codeExp.code) {
+    case "ref":
+      const name = representatives[codeExp.ref] || codeExp.ref;
+      if (name.startsWith(":")) {
+        return prettyCode(codes, representatives, codes[name]);
+      } else {
+        return name;
+      }
+    case "product":
+      return `{${prettyCode_labels(codes, representatives, codeExp.product)}}`;
+    case "union":
+      return `<${prettyCode_labels(codes, representatives, codeExp.union)}>`;
+    case "vector":
+      return `[${representatives[codeExp.vector] || codeExp.vector}]`;
+    default:
+      return ":error";
+  }
+};
+
+function prettyRel (prettyCode, exp) {
+  "use strict";
+  const pretty = (exp) => {
+    switch (exp.op) {
+      case "vector":
+        return `[${exp.vector.map(pretty).join(", ")}]`;
+      case "union":
+        return `<${exp.union.map(pretty).join(", ")}>`;
+      case "ref":
+        return exp.ref;
+      case "identity":
+        return "()";
+      case "comp":
+        return exp.comp.map(pretty).join(" ");
+      case "str":
+        return `'${exp.str}'`;
+      case "int":
+        return exp.int;
+      case "dot":
+        if ("number" === typeof exp.dot) {
+          return `.${exp.dot}`;
+        } else if (nameRE.test(exp.dot)) {
+          return `.${exp.dot}`;
+        } else {
+          return `'${exp.dot}'`;
+        }
+        break;
+      case "code":
+        return `$${prettyCode({
+          code: "ref",
+          ref: exp.code,
+        })}`;
+      case "product":
+        return (function (labelled) {
+          return `{${labelled.join(", ")}}`;
+        })(
+          exp.product.map(function ({ label, exp }) {
+            if (nameRE.test(label)) {
+              return `${pretty(exp)} ${label}`;
+            } else {
+              return `${pretty(exp)} '${label}'`;
+            }
+          })
+        );
+    }
+  };
+  return pretty(exp);
+};
+
+export default { prettyCode, prettyRel };
+export { prettyCode, prettyRel };

package/repl.mjs

@@ -0,0 +1,112 @@
+#!/usr/bin/env node
+
+import k from "./index.mjs";
+import run from "./run.mjs";
+import { prettyCode, prettyRel } from "./pretty.mjs";
+
+import fs from "node:fs";
+
+const splice = [].splice;
+
+console.log("Very! experimental repl shell for 'k-language'...");
+
+const help = function () {
+  console.log(" --h          print help");
+  console.log(" --a          print codes and relations");
+  console.log(" --c          print codes");
+  console.log(" --r          print rels");
+  return console.log(" --l file.k   load 'file.k'");
+};
+
+help();
+
+const re = /^[ \n\t]*(?:--l[ ]+)(.+[^ ])[ ]*?$/;
+
+(function (val, buffer) {
+  return process.stdin.on("data", function (data) {
+    var e, file, first, i, kScript, last, len, line, ref, rest, results, todo;
+    [first, ...rest] = data.toString("utf8").split("\n");
+    buffer.push(first);
+    if (rest.length > 0) {
+      todo = buffer.join("");
+      (ref = rest), ([...rest] = ref), ([last] = splice.call(rest, -1));
+      buffer = [last];
+      todo = [todo, ...rest];
+      results = [];
+      for (i = 0, len = todo.length; i < len; i++) {
+        line = todo[i];
+        if (val === void 0) {
+          val = {};
+        }
+        if (line.match(/^[ \n\t]*(?:--h)?$/)) {
+          results.push(help());
+        } else if (line.match(re)) {
+          file = line.match(re)[1];
+          console.log(`-- loading file: ${file} ...`);
+          kScript = fs.readFileSync(file).toString();
+          val = k.compile(kScript)(val);
+          results.push(console.log(`=> ${JSON.stringify(val)}`));
+        } else if (line.match(/^[ \n\t]*(?:--a)?$/)) {
+          results.push(console.log(JSON.stringify(run.defs, " ", 2)));
+        } else if (line.match(/^[ \n\t]*(?:--r)?$/)) {
+          results.push(
+            console.log(
+              (function (defs, result) {
+                if (defs == null) {
+                  return result;
+                }
+                return (function (prettyRel) {
+                  var ref1, relExps, relName;
+                  ref1 = defs.rels;
+                  for (relName in ref1) {
+                    relExps = ref1[relName];
+                    result[relName] = relExps.map(prettyRel);
+                  }
+                  return result;
+                })(
+                  prettyRel.bind(
+                    null,
+                    prettyCode.bind(null, defs.codes, defs.representatives)
+                  )
+                );
+              })(run.defs, {})
+            )
+          );
+        } else if (line.match(/^[ \n\t]*(?:--c)?$/)) {
+          results.push(
+            console.log(
+              (function (defs, result) {
+                var codeExp, codeName, ref1;
+                if (defs == null) {
+                  return result;
+                }
+                ref1 = defs.codes;
+                for (codeName in ref1) {
+                  codeExp = ref1[codeName];
+                  result[codeName] = prettyCode(
+                    defs.codes,
+                    defs.representatives,
+                    codeExp
+                  );
+                }
+                return result;
+              })(run.defs, {})
+            )
+          );
+        } else if (!line.match(/^[ \n\t]*(?:#.*)?$/)) {
+          try {
+            val = k.run(`${line} ()`, val);
+            results.push(console.log(`=> ${JSON.stringify(val)}`));
+          } catch (error) {
+            e = error;
+            console.error("ERROR:");
+            results.push(console.error(e));
+          }
+        } else {
+          results.push(void 0);
+        }
+      }
+      return results;
+    }
+  });
+})({}, []);

package/run.mjs

@@ -0,0 +1,158 @@
+const modulo = (a, b) => ((+a % (b = +b)) + b) % b;
+const valid = (x) => (isNaN(x) ? undefined : x);
+
+const builtin = {
+  "_log!": (arg) => {
+    console.error(`_log!: ${JSON.stringify(arg)}`);
+    return arg;
+  },
+  GT: (args) => {
+    const [last, ...rest] = args;
+    const [ok, _] = rest.reduce(
+      ([ok, last], x) => [ok && last > x, x],
+      [true, last]
+    );
+    if (ok) return args;
+  },
+  EQ: (args) => {
+    const [first, ...rest] = args;
+    const ok = rest.reduce((ok, x) => ok && first === x, true);
+    if (ok) return args;
+  },
+  PLUS: (args) => valid(args.reduce((res, x) => res + x, 0)),
+  TIMES: (args) => valid(args.reduce((res, x) => res * x, 1)),
+  DIV: ([x, y]) => {
+    const div = Math.floor(x / y);
+    const rem = modulo(x, y);
+    if (x === div * y + rem) return { div, rem };
+  },
+  FDIV: ([x, y]) => x / y,
+  CONCAT: (strs) => strs.join(""),
+  true: () => true,
+  false: () => false,
+  null: () => null,
+  toJSON: (x) => JSON.stringify(x),
+  fromJSON: (x) => JSON.parse(x),
+  CONS: ([x, y]) => [x, ...y],
+  SNOC: (x) => (x.length > 1 ? [x[0], x.slice(1)] : undefined),
+  toDateMsec: (x) => new Date(x).getTime(),
+  toDateStr: (x) => new Date(x).toISOString(),
+};
+
+const codes = {
+  int: (x) => Number.isInteger(x),
+  string: (x) => x instanceof String || "string" === typeof x,
+  bool: (x) => x === true || x === false,
+};
+
+function verify(code, value) {
+  if (code == null) {
+    // representatives = run.defs.representatives
+    // defCodes = JSON.stringify run.defs.codes
+    // console.log {code,value, representatives, defCodes}
+    return false;
+  }
+  switch (code.code) {
+    case "vector":
+      if (!Array.isArray(value)) return false;
+      return value.every((x) => verify(code.vector, x));
+    case "product":
+      if ("object" !== typeof value) return false;
+      else {
+        const fields = Object.keys(value);
+        if (fields.length !== Object.keys(code.product).length) return false;
+        return fields.every((label) =>
+          verify(code.product[label], value[label])
+        );
+      }
+    case "union":
+      if ("object" !== typeof value) return false;
+      else {
+        const fields = Object.keys(value);
+        if (fields.length !== 1) return false;
+        return verify(code.union[fields[0]], value[fields[0]]);
+      }
+    default: {
+      const c = run.defs.codes[run.defs.representatives[code]];
+      if (c != null) return verify(c, value);
+      return codes[code](value);
+    }
+  }
+}
+
+function run(exp, value) {
+  "use strict";
+  if (value === undefined) return;
+  switch (exp.op) {
+    case "code":
+      if (verify(exp.code, value)) {
+        return value;
+      }
+      break;
+    case "identity":
+      return value;
+    case "str":
+    case "int":
+      return exp[exp.op];
+    case "ref":
+      const defn = run.defs.rels[exp.ref];
+      if (defn != null) {
+        return run(defn[defn.length - 1], value);
+      }
+      return builtin[exp.ref](value);
+    case "dot":
+      if (typeof value === "string") {
+        if (value == exp.dot) return {};
+        return;
+      }
+      if (/^(0|([1-9]\d*))$/.test(value)) {
+        if (value == exp.dot) return {};
+        return;
+      } 
+      if (value === true) {
+        if (exp.dot == "true") return {};
+        return;
+      }
+      if (value === false) {
+        if (exp.dot == "false") return {};
+        return;
+      }
+      return value[exp.dot];
+    case "comp":
+      return exp.comp.reduce((value, exp) => {
+        if (value !== undefined) return run(exp, value);
+      }, value);
+    case "union":
+      for (let i = 0, len = exp.union.length; i < len; i++) {
+        const result = run(exp.union[i], value);
+        if (result !== undefined) {
+          return result;
+        }
+      }
+      return;
+    case "vector": {
+      const result = [];
+      for (let i = 0, len = exp.vector.length; i < len; i++) {
+        const r = run(exp.vector[i], value);
+        if (r === undefined) return;
+        result.push(r);
+      }
+      return result;
+    }
+    case "product": {
+      const result = {};
+      for (let i = 0, len = exp.product.length; i < len; i++) {
+        const { label, exp: e } = exp.product[i];
+        const r = run(e, value);
+        if (r === undefined) return;
+        result[label] = r;
+      }
+      return result;
+    }
+    default:
+      return console.error(exp.op);
+  }
+}
+
+export default run;
+export { run };

package/symbol-table.mjs

@@ -0,0 +1,133 @@
+const rels = {};
+const codes = {
+  "{}": {
+    code: "product",
+    product: {},
+  },
+};
+
+const identity = {
+  op: "identity",
+};
+
+function is_identity_rel(rel) {
+  return rel.op === "identity";
+}
+
+function is_constant_rel(rel) {
+  switch (rel.op) {
+    case "int":
+    case "str":
+      return true;
+    case "vector":
+      return Object.values(rel[rel.op]).every(is_constant_rel);
+    case "product":
+      return Object.values(rel[rel.op]).every(({ exp }) =>
+        is_constant_rel(exp)
+      );
+    default:
+      return false;
+  }
+}
+
+function is_empty_rel(rel) {
+  return rel.op === "union" && rel.union.length === 0;
+}
+
+function is_full_rel(rel) {
+  if (is_constant_rel(rel)) return true;
+  switch (rel.op) {
+    case "int":
+    case "str":
+    case "identity":
+      return true;
+    case "comp":
+      return rel.comp.every(is_full_rel);
+    case "vector":
+      return Object.values(rel[rel.op]).every(is_full_rel);
+    case "product":
+      return Object.values(rel[rel.op]).every(({ exp }) => is_full_rel(exp));
+    default:
+      return false;
+  }
+}
+
+function comp_first(e1, e2) {
+  if (is_identity_rel(e1)) return e2;
+  if (is_identity_rel(e2)) return e1;
+  if (is_empty_rel(e1)) return e1;
+  if (is_empty_rel(e2)) return e2;
+  if (e1.op === "comp" && e2.op === "comp")
+    return {
+      op: "comp",
+      comp: [].concat(e1.comp, e2.comp),
+    };
+  if (e1.op === "comp")
+    return {
+      op: "comp",
+      comp: [].concat(e1.comp, [e2]),
+    };
+  if (e2.op === "comp")
+    return {
+      op: "comp",
+      comp: [].concat([e1], e2.comp),
+    };
+  return {
+    op: "comp",
+    comp: [e1, e2],
+  };
+}
+
+function comp (e1, e2)  {
+  const result = comp_first(e1, e2);
+  if (result.op !== "comp") return result;
+  result.comp = result.comp.reduceRight(
+    (c, e) => (c[0] && is_constant_rel(c[0]) && is_full_rel(e) ? c : [e, ...c]),
+    []
+  );
+  return result;
+}
+
+function union  (rels) {
+  const list = [];
+  label: for (const rel of rels) {
+    const new_rels = rel.op === "union" ? rel.union : [rel];
+    for (const new_rel of new_rels) {
+      list.push(new_rel);
+      if (is_full_rel(new_rel)) break label;
+    }
+  }
+  if (list.length === 1) return list[0];
+  return {
+    op: "union",
+    union: list,
+  };
+};
+
+function as_ref (codeExp) {
+  if (codeExp.code === "ref") return codeExp.ref;
+  const newName = `:${Object.keys(codes).length}`;
+  codes[newName] = codeExp;
+  return newName;
+};
+
+function add_rel (name, rel) {
+  if (rels[name] == null) rels[name] = [];
+  rels[name].push(rel);
+};
+
+function add_code (name, code) {
+  codes[name] = code;
+}
+
+export default {
+  identity,
+  comp,
+  union,
+  rels,
+  codes,
+  add_rel,
+  add_code,
+  as_ref,
+};
+export { identity, comp, union, rels, codes, add_rel, add_code, as_ref };

package/test.mjs

@@ -0,0 +1,143 @@
+import k from "./index.mjs";
+
+[
+  ["ANYTHING...", "()"],
+  ["ANYTHING...", '{"ala" name, 23 age}'],
+  [
+    {
+      year: 2002,
+      age: 19,
+    },
+    "[.year, .age]",
+  ],
+  ["duplicate me", "[(), ()]"],
+  ["nesting", "[[[()]]]"],
+  ["nesting and accessing", "[[()]] {() nested, .0.0 val}"],
+  [
+    {
+      test: "parse integer",
+    },
+    "0000",
+  ],
+  [
+    {
+      x: 3,
+      y: 4,
+    },
+    "[.y,.x] PLUS",
+  ],
+].map(function ([data, script]) {
+  console.log(`k_expression = '${script}';`);
+  console.log(`k.run(k_expression,${JSON.stringify(data)});`);
+  console.log(`// RETURNS: ${JSON.stringify(k.run(script, data))}`);
+  return console.log("");
+});
+
+let k_expression =
+  '{.name nom, <[.age, 18] GT .0, [.age, 12] GT "ado", "enfant"> age}';
+
+let k_fn = k.compile(k_expression);
+
+console.log(`var k_fn = k.compile('${k_expression}');`);
+
+console.log("");
+
+[
+  {
+    age: 23,
+    name: "Emily",
+  },
+  {
+    age: 16,
+    name: "Katrina",
+  },
+  {
+    age: 2,
+    name: "Mark",
+  },
+].map(function (data) {
+  console.log(`k_fn(${JSON.stringify(data)});`);
+  console.log(`// RETURNS: ${JSON.stringify(k_fn(data))}`);
+  return console.log("");
+});
+
+k_expression = "$t = < i: int, t: [ t ] > ; <$t, $int>";
+
+k_fn = k.compile(k_expression);
+
+console.log(`var k_fn = k.compile('${k_expression}');`);
+
+console.log("");
+
+[
+  1,
+  {
+    i: 1,
+  },
+  [
+    {
+      i: 2,
+    },
+    {
+      i: 3,
+    },
+    {
+      t: [],
+    },
+  ],
+  {
+    t: [
+      {
+        i: 2,
+      },
+      {
+        i: 3,
+      },
+      {
+        t: [],
+      },
+    ],
+  },
+].map(function (data) {
+  console.log(`k_fn(${JSON.stringify(data)});`);
+  console.log(`// RETURNS: ${JSON.stringify(k_fn(data))}`);
+  return console.log("");
+});
+
+k_expression = "$ < < [ int ] ints, [ bool ] bools > list, string None>";
+
+k_fn = k.compile(k_expression);
+
+console.log(`var k_fn = k.compile('${k_expression}');`);
+
+console.log("");
+
+[
+  {
+    None: "None",
+  },
+  {
+    list: {
+      ints: [],
+    },
+  },
+  {
+    list: {
+      ints: [1, 2, 3],
+    },
+  },
+  {
+    list: {
+      bools: [true, false],
+    },
+  },
+  {
+    list: {
+      bools: [true, 0],
+    },
+  },
+].map(function (data) {
+  console.log(`k_fn(${JSON.stringify(data)});`);
+  console.log(`// RETURNS: ${JSON.stringify(k_fn(data))}`);
+  return console.log("");
+});