Merge pull request #37 from TotalTechGeek/fix/strictly-honor-sugaring

This set of changes allows JSON Logic engine to more strictly honor sugaring

Author: TotalTechGeek

asLogic.js

@@ -21,7 +21,7 @@ function pick (keep, obj) {
 export function asLogicSync (functions, keep = ['var'], engine = new LogicEngine()) {
   engine.methods = pick(keep, engine.methods)
   engine.addMethod('list', i => [].concat(i))
-  Object.keys(functions).forEach(i => engine.addMethod(i, data => Array.isArray(data) ? functions[i](...data) : functions[i](data === null ? undefined : data)))
+  Object.keys(functions).forEach(i => engine.addMethod(i, data => functions[i](...data)))
   return engine.build.bind(engine)
 }
 

asLogic.test.js

@@ -1,7 +1,7 @@
 import { asLogicSync, asLogicAsync } from './asLogic.js'
 
 const module = {
-  hello: (name = 'World', last = '') => `Hello, ${name}${last.length ? ' ' : ''}${last}!`
+  hello: (name = 'World', last = '') => `Hello, ${name || 'World'}${last.length ? ' ' : ''}${last}!`
 }
 
 describe('asLogicSync', () => {

async.test.js

@@ -6,7 +6,7 @@ const modes = [
 ]
 
 for (const engine of modes) {
-  engine.addMethod('as1', async (n) => n + 1, { async: true, deterministic: true })
+  engine.addMethod('as1', async ([n]) => n + 1, { async: true, deterministic: true })
 }
 
 modes.forEach((logic) => {
@@ -676,7 +676,7 @@ modes.forEach((logic) => {
         length: ['hello']
       })
 
-      expect(answer).toStrictEqual(1)
+      expect(answer).toStrictEqual(5)
     })
 
     test('length object (2 keys)', async () => {
@@ -781,7 +781,7 @@ modes.forEach((logic) => {
 
   describe('addMethod', () => {
     test('adding a method works', async () => {
-      logic.addMethod('+1', (item) => item + 1, { sync: true })
+      logic.addMethod('+1', ([item]) => item + 1, { sync: true })
       expect(
         await logic.run({
           '+1': 7

asyncLogic.js

@@ -9,6 +9,7 @@ import { buildAsync } from './compiler.js'
 import omitUndefined from './utilities/omitUndefined.js'
 import { optimize } from './async_optimizer.js'
 import { applyPatches } from './compatibility.js'
+import { coerceArray } from './utilities/coerceArray.js'
 
 /**
  * An engine capable of running asynchronous JSON Logic.
@@ -75,16 +76,15 @@ class AsyncLogicEngine {
     if (!this.methods[func]) throw new Error(`Method '${func}' was not found in the Logic Engine.`)
 
     if (typeof this.methods[func] === 'function') {
-      const input = await this.run(data, context, { above })
-      const result = await this.methods[func](input, context, above, this)
+      const input = (!data || typeof data !== 'object') ? [data] : await this.run(data, context, { above })
+      const result = await this.methods[func](coerceArray(input), context, above, this)
       return Array.isArray(result) ? Promise.all(result) : result
     }
 
     if (typeof this.methods[func] === 'object') {
       const { asyncMethod, method, traverse } = this.methods[func]
       const shouldTraverse = typeof traverse === 'undefined' ? true : traverse
-      const parsedData = shouldTraverse ? await this.run(data, context, { above }) : data
-
+      const parsedData = shouldTraverse ? ((!data || typeof data !== 'object') ? [data] : coerceArray(await this.run(data, context, { above }))) : data
       const result = await (asyncMethod || method)(parsedData, context, above, this)
       return Array.isArray(result) ? Promise.all(result) : result
     }
@@ -96,12 +96,12 @@ class AsyncLogicEngine {
    *
    * @param {String} name The name of the method being added.
    * @param {((args: any, context: any, above: any[], engine: AsyncLogicEngine) => any) | { traverse?: Boolean, method?: (args: any, context: any, above: any[], engine: AsyncLogicEngine) => any, asyncMethod?: (args: any, context: any, above: any[], engine: AsyncLogicEngine) => Promise<any>, deterministic?: Function | Boolean }} method
-   * @param {{ deterministic?: Boolean, async?: Boolean, sync?: Boolean }} annotations This is used by the compiler to help determine if it can optimize the function being generated.
+   * @param {{ deterministic?: Boolean, async?: Boolean, sync?: Boolean, optimizeUnary?: boolean }} annotations This is used by the compiler to help determine if it can optimize the function being generated.
    */
   addMethod (
     name,
     method,
-    { deterministic, async, sync } = {}
+    { deterministic, async, sync, optimizeUnary } = {}
   ) {
     if (typeof async === 'undefined' && typeof sync === 'undefined') sync = false
     if (typeof sync !== 'undefined') async = !sync
@@ -112,7 +112,7 @@ class AsyncLogicEngine {
       else method = { method, traverse: true }
     } else method = { ...method }
 
-    Object.assign(method, omitUndefined({ deterministic }))
+    Object.assign(method, omitUndefined({ deterministic, optimizeUnary }))
     // @ts-ignore
     this.fallback.addMethod(name, method, { deterministic })
     this.methods[name] = declareSync(method, sync)
@@ -188,6 +188,8 @@ class AsyncLogicEngine {
   async build (logic, options = {}) {
     const { above = [], top = true } = options
     this.fallback.truthy = this.truthy
+    // @ts-ignore
+    this.fallback.allowFunctions = this.allowFunctions
     if (top) {
       const constructedFunction = await buildAsync(logic, { engine: this, above, async: true, state: {} })
 

async_optimizer.js

@@ -3,6 +3,7 @@ import { isDeterministic } from './compiler.js'
 import { map } from './async_iterators.js'
 import { isSync, Sync } from './constants.js'
 import declareSync from './utilities/declareSync.js'
+import { coerceArray } from './utilities/coerceArray.js'
 
 /**
  * Turns an expression like { '+': [1, 2] } into a function that can be called with data.
@@ -26,7 +27,8 @@ function getMethod (logic, engine, methodName, above) {
     return (data, abv) => called(args, data, abv || above, engine)
   }
 
-  const args = logic[methodName]
+  let args = logic[methodName]
+  if (!args || typeof args !== 'object') args = [args]
 
   if (Array.isArray(args)) {
     const optimizedArgs = args.map(l => optimize(l, engine, above))
@@ -48,11 +50,11 @@ function getMethod (logic, engine, methodName, above) {
 
     if (isSync(optimizedArgs) && (method.method || method[Sync])) {
       const called = method.method ? method.method : method
-      return declareSync((data, abv) => called(typeof optimizedArgs === 'function' ? optimizedArgs(data, abv) : optimizedArgs, data, abv || above, engine), true)
+      return declareSync((data, abv) => called(coerceArray(typeof optimizedArgs === 'function' ? optimizedArgs(data, abv) : optimizedArgs, method.optimizeUnary), data, abv || above, engine), true)
     }
 
     return async (data, abv) => {
-      return called(typeof optimizedArgs === 'function' ? await optimizedArgs(data, abv) : optimizedArgs, data, abv || above, engine)
+      return called(coerceArray(typeof optimizedArgs === 'function' ? await optimizedArgs(data, abv) : optimizedArgs, method.optimizeUnary), data, abv || above, engine)
     }
   }
 }
@@ -65,6 +67,7 @@ function getMethod (logic, engine, methodName, above) {
  * @returns A function that optimizes the logic for the engine in advance.
  */
 export function optimize (logic, engine, above = []) {
+  engine.fallback.allowFunctions = engine.allowFunctions
   if (Array.isArray(logic)) {
     const arr = logic.map(l => optimize(l, engine, above))
     if (isSync(arr)) return declareSync((data, abv) => arr.map(l => typeof l === 'function' ? l(data, abv) : l), true)

bench/incompatible.json

@@ -231,11 +231,11 @@ function timeout (n, x) {
       expect(await f({ x: 1, y: 2 })).toStrictEqual({ a: 1, b: 2 })
     })
 
-    test('Invalid eachKey', async () => {
-      expect(async () => await logic.build({ eachKey: 5 })).rejects.toThrow(
-        InvalidControlInput
-      )
-    })
+    // test('Invalid eachKey', async () => {
+    //   expect(async () => await logic.build({ eachKey: 5 })).rejects.toThrow(
+    //     InvalidControlInput
+    //   )
+    // })
 
     test('Simple deterministic eachKey', async () => {
       const f = await logic.build({ eachKey: { a: 1, b: { '+': [1, 1] } } })
@@ -246,7 +246,7 @@ function timeout (n, x) {
 })
 
 const logic = new AsyncLogicEngine()
-logic.addMethod('as1', async (n) => timeout(100, n + 1), { async: true })
+logic.addMethod('as1', async ([n]) => timeout(100, n + 1), { async: true })
 
 describe('Testing async build with full async', () => {
   test('Async +1', async () => {

build.test.js

@@ -1,6 +1,12 @@
 import fs from 'fs'
 import { LogicEngine, AsyncLogicEngine } from './index.js'
-const tests = JSON.parse(fs.readFileSync('./bench/compatible.json').toString())
+const tests = []
+
+// get all json files from "suites" directory
+const files = fs.readdirSync('./suites')
+for (const file of files) {
+  if (file.endsWith('.json')) tests.push(...JSON.parse(fs.readFileSync(`./suites/${file}`).toString()).filter(i => typeof i !== 'string'))
+}
 
 // eslint-disable-next-line no-labels
 inline: {

compatible.test.js

@@ -10,6 +10,7 @@ import declareSync from './utilities/declareSync.js'
 
 // asyncIterators is required for the compiler to operate as intended.
 import asyncIterators from './async_iterators.js'
+import { coerceArray } from './utilities/coerceArray.js'
 
 /**
  * Provides a simple way to compile logic into a function that can be run.
@@ -191,25 +192,32 @@ function buildString (method, buildState = {}) {
       }
     }
 
+    let lower = method[func]
+    if (!lower || typeof lower !== 'object') lower = [lower]
+
     if (engine.methods[func] && engine.methods[func].compile) {
-      let str = engine.methods[func].compile(method[func], buildState)
+      let str = engine.methods[func].compile(lower, buildState)
       if (str[Compiled]) str = str[Compiled]
 
       if ((str || '').startsWith('await')) buildState.asyncDetected = true
 
       if (str !== false) return str
     }
 
+    let coerce = engine.methods[func].optimizeUnary ? '' : 'coerceArray'
+    if (!coerce && Array.isArray(lower) && lower.length === 1) lower = lower[0]
+    else if (coerce && Array.isArray(lower)) coerce = ''
+
     if (typeof engine.methods[func] === 'function') {
       asyncDetected = !isSync(engine.methods[func])
-      return makeAsync(`engine.methods["${func}"](` + buildString(method[func], buildState) + ', context, above, engine)')
+      return makeAsync(`engine.methods["${func}"](${coerce}(` + buildString(lower, buildState) + '), context, above, engine)')
     } else {
       if (engine.methods[func] && (typeof engine.methods[func].traverse === 'undefined' ? true : engine.methods[func].traverse)) {
         asyncDetected = Boolean(async && engine.methods[func] && engine.methods[func].asyncMethod)
-        return makeAsync(`engine.methods["${func}"]${asyncDetected ? '.asyncMethod' : '.method'}(` + buildString(method[func], buildState) + ', context, above, engine)')
+        return makeAsync(`engine.methods["${func}"]${asyncDetected ? '.asyncMethod' : '.method'}(${coerce}(` + buildString(lower, buildState) + '), context, above, engine)')
       } else {
         asyncDetected = Boolean(async && engine.methods[func] && engine.methods[func].asyncMethod)
-        notTraversed.push(method[func])
+        notTraversed.push(lower)
         return makeAsync(`engine.methods["${func}"]${asyncDetected ? '.asyncMethod' : '.method'}(` + `notTraversed[${notTraversed.length - 1}]` + ', context, above, engine)')
       }
     }
@@ -294,12 +302,12 @@ function processBuiltString (method, str, buildState) {
     str = str.replace(`__%%%${x}%%%__`, item)
   })
 
-  const final = `(values, methods, notTraversed, asyncIterators, engine, above) => ${buildState.asyncDetected ? 'async' : ''} (context ${buildState.extraArguments ? ',' + buildState.extraArguments : ''}) => { const result = ${str}; return result }`
+  const final = `(values, methods, notTraversed, asyncIterators, engine, above, coerceArray) => ${buildState.asyncDetected ? 'async' : ''} (context ${buildState.extraArguments ? ',' + buildState.extraArguments : ''}) => { const result = ${str}; return result }`
 
   // console.log(str)
   // console.log(final)
   // eslint-disable-next-line no-eval
-  return declareSync((typeof globalThis !== 'undefined' ? globalThis : global).eval(final)(values, methods, notTraversed, asyncIterators, engine, above), !buildState.asyncDetected)
+  return declareSync((typeof globalThis !== 'undefined' ? globalThis : global).eval(final)(values, methods, notTraversed, asyncIterators, engine, above, coerceArray), !buildState.asyncDetected)
 }
 
 export { build }

compiler.js

@@ -178,7 +178,7 @@ const defaultMethods = {
     }
     return string.substr(from, end)
   },
-  length: (i) => {
+  length: ([i]) => {
     if (typeof i === 'string' || Array.isArray(i)) return i.length
     if (i && typeof i === 'object') return Object.keys(i).length
     return 0
@@ -398,7 +398,7 @@ const defaultMethods = {
     for (let i = 0; i < arr.length; i++) res += arr[i]
     return res
   },
-  keys: (obj) => typeof obj === 'object' ? Object.keys(obj) : [],
+  keys: ([obj]) => typeof obj === 'object' ? Object.keys(obj) : [],
   pipe: {
     traverse: false,
     [Sync]: (data, buildState) => isSyncDeep(data, buildState.engine, buildState),
@@ -799,13 +799,13 @@ defaultMethods.var.compile = function (data, buildState) {
     typeof data === 'number' ||
     (Array.isArray(data) && data.length <= 2)
   ) {
-    if (data === '../index' && buildState.iteratorCompile) return 'index'
-
     if (Array.isArray(data)) {
       key = data[0]
       defaultValue = typeof data[1] === 'undefined' ? null : data[1]
     }
 
+    if (key === '../index' && buildState.iteratorCompile) return 'index'
+
     // this counts the number of var accesses to determine if they're all just using this override.
     // this allows for a small optimization :)
     if (typeof key === 'undefined' || key === null || key === '') return 'context'
@@ -835,6 +835,9 @@ defaultMethods.var.compile = function (data, buildState) {
   return false
 }
 
+// @ts-ignore Allowing a optimizeUnary attribute that can be used for performance optimizations
+defaultMethods['+'].optimizeUnary = defaultMethods['-'].optimizeUnary = defaultMethods.var.optimizeUnary = defaultMethods['!'].optimizeUnary = defaultMethods['!!'].optimizeUnary = defaultMethods.cat.optimizeUnary = true
+
 export default {
   ...defaultMethods
 }