@sendbird/uikit-utils/src/shared/arrayToMap.ts

A collection of utility functions and constants for building chat UI components with Sendbird UIKit.

import type { FilterByValueType } from '../types';

export default function arrayToMap<T extends Record<K, unknown>, K extends keyof T = keyof T>(
  arr: T[],
  selector: keyof FilterByValueType<T, string | number>,
): Record<string, T>;

export default function arrayToMap<T extends Record<K, unknown>, K extends keyof T = keyof T>(
  arr: T[],
  selector: string,
  fallbackSelector: K,
): Record<string, T>;

export default function arrayToMap<T extends Record<K, unknown>, K extends keyof T = keyof T>(
  arr: T[],
  selector: string,
  selector2: string,
  combineKey: true,
): Record<string, T>;

/**
 * Convert Array of the object to Object
 * @param {object[]} arr
 * @param {keyof object} selector
 * @param {keyof object} [fallbackSelector]
 * @param {boolean} [combine]
 * @returns {Record<keyof object, object>}
 * @example
 * ```
 *  const arr = [{name: 'john'}, {name: 'chris'}, {name: 'sravan'}, {name: 'hoon'}];
 *  const obj = arrayToMap(arr, 'name');
 *
 *  console.log(obj);
 *  // {
 *  //  'john': {name: 'john'},
 *  //  'chris': {name: 'chris'},
 *  //  'sravan': {name: 'sravan'},
 *  //  'hoon': {name: 'hoon'}
 *  // }
 * ```
 * */
export default function arrayToMap<T extends Record<K, unknown>, K extends keyof T = keyof T>(
  arr: T[],
  selector: K,
  fallbackSelector?: K,
  combine?: boolean,
) {
  return arr.reduce((accum, curr) => {
    if (combine && fallbackSelector) {
      const _key = curr[selector] + (curr[fallbackSelector] as string);
      accum[_key] = curr;
    } else {
      const _key = (curr[selector] || curr[fallbackSelector as K]) as string;
      accum[_key] = curr;
    }

    return accum;
  }, {} as Record<string, T>);
}

export function arrayToMapWithGetter<T>(arr: T[], getSelector: (item: T) => string) {
  return arr.reduce((accum, curr) => {
    const _key = getSelector(curr);
    accum[_key] = curr;
    return accum;
  }, {} as Record<string, T>);
}