object-scan/lib/core/find-result.js

Traverse object hierarchies using matching and callbacks.

export default (kwargs, ctx) => {
  if (ctx.rtn === 'context') {
    return {
      onMatch: () => {},
      get: () => kwargs.context
    };
  }
  if (ctx.rtn === 'bool') {
    let result = false;
    return {
      onMatch: () => {
        result = true;
      },
      get: () => result
    };
  }
  if (ctx.rtn === 'count') {
    let result = 0;
    return {
      onMatch: () => {
        result += 1;
      },
      get: () => result
    };
  }
  if (ctx.rtn === 'sum') {
    let result = 0;
    return {
      onMatch: ({ value }) => {
        result += value;
      },
      get: () => result
    };
  }

  const result = [];
  return {
    onMatch: (() => {
      if (typeof ctx.rtn === 'function') {
        return () => result.push(ctx.rtn(kwargs));
      }
      if (Array.isArray(ctx.rtn)) {
        return () => result.push(ctx.rtn.map((rtn) => kwargs[rtn]));
      }
      return () => result.push(kwargs[ctx.rtn]);
    })(),
    get: () => (ctx.abort ? result[0] : result)
  };
};