New summarizer: nWhere? about tidy HOT 3 CLOSED

I too have made this helper myself! but called it countIf. nWhere is nicer I think, doesn't conflict with the summarizeIf style of things... my implementations avoided creating another array by using sum + predicate, more SQL style in a way. I'd be happy to add these as nWhere and sumWhere if you're up for it!

export function countIf<T extends object>(predicate: (d: T) => boolean) {
  return (items: T[]) => sum((d: T) => (predicate(d) ? 1 : 0))(items)
}

export function sumIf<T extends object>(
  predicate: (d: T) => boolean,
  key: keyof T | ((d: T) => number)
) {
  const keyFn =
    typeof key === 'function' ? key : (d: T) => d[key] as unknown as number

  return (items: T[]) => sum((d: T) => (predicate(d) ? keyFn(d) : 0))(items)
}

from tidy.

Sounds good – go for it!

from tidy.

I actually ended up doing this as an option on sum and n now which matches a couple other places, available in v2.5.0.

const data = [
  { str: 'foo', value: 3 },
  { str: 'foo', value: 1 },
  { str: 'bar', value: 3 },
  { str: 'bar', value: 1 },
  { str: 'bar', value: 7 },
];

tidy(data, summarize({
  count: n(),
  countFoo: n({ predicate: d => d.str === 'foo' })
})
// output:
[{ count: 5, countFoo: 2 }]

from tidy.