util.ts

import { CommandType } from "../types";

const digitRe = /^\d+$/;
const digitDotRe = /^\d+\..*$/;
const isDigit = (v: any) => digitRe.test(v);
const firstIsIndex = (s: string) => digitDotRe.test(s);

// Based on: https://www.30secondsofcode.org/js/s/unflatten-object
export const unflatten = (obj: any) =>
  Object.keys(obj).reduce(
    (res, k) => {
      k.split(".")
        .map((v) => (isDigit(v) ? parseInt(v) - 1 : v))
        .reduce(
          (acc: any, e, i, keys) =>
            acc[e] ||
            (acc[e] = isNaN(Number(keys[i + 1]))
              ? keys.length - 1 === i
                ? obj[k]
                : {}
              : []),
          res
        );
      return res;
    },
    firstIsIndex(Object.keys(obj)[0]) ? [] : {}
  );

export const search = (obj: any, key: string): any => {
  if (typeof obj !== "object") return null;
  if (obj[key]) return obj[key];
  for (const val of Object.values(obj)) {
    const s = search(val, key);
    if (s) return s;
  }
};

export const searchParent = (obj: any, key: string): Record<string, any> | undefined => {
  if (typeof obj !== "object") return;
  if (obj[key]) return obj;
  for (const val of Object.values(obj)) {
    const s = searchParent(val, key);
    if (s) return s;
  }
};

const _searchAll = (obj: any, key: string): any[] =>
  typeof obj !== "object"
    ? []
    : Object.entries(obj).reduce(
        (acc, [k, v]) => [...acc, k === key ? v : _searchAll(v, key)],
        [] as any[]
      );

export const searchAll = (obj: any, key: string) =>
  _searchAll(obj, key).flat(Infinity);

export const extractCommand = (msg: { [key: string]: any }): string | undefined => {
  const msgType: string | undefined = search(msg, "msgType");
  if (!msgType) {
    const id: string | undefined = search(msg, "msgId")
    const [frst] = id ? id.split("@") : [""]
    return frst.toUpperCase()
  }
  return msgType;
};

/** Unwraps object with single key */
export const unwrapObject = (data: any): any =>
  !Array.isArray(data) &&
  typeof data === "object" &&
  Object.keys(data).length === 1
    ? Object.values(data)[0]
    : data;

/** Unwraps array with single item */
export const unwrapArray = (arr: any) =>
  Array.isArray(arr) && arr.length === 1 ? arr[0] : arr;

export function makeBuffer(
  rootRecord: protobuf.Root,
  payload: any,
  options: Record<string, string>
) {
  const NoSessionContextRecord = rootRecord.lookupType(
    "usp_record.NoSessionContextRecord"
  );
  const noSessionContextRecordMsg = NoSessionContextRecord.create({
    payload,
  });
  const record: any = rootRecord.lookupType("usp_record.Record");
  const recordMsg = record.create({
    version: "1.0",
    PayloadSecurity: record.PayloadSecurity.PLAINTEXT,
    noSessionContext: noSessionContextRecordMsg,
    ...options,
  });
  const buffer = record.encode(recordMsg).finish();
  return buffer;
}

export const uniq = (initial?: string): string =>
  (initial || "") +
  (
    Date.now().toString(36) + Math.random().toString(36).substr(2, 5)
  ).toUpperCase();