medusa-store/packages/utils/src/common/get-selects-and-relations-from-object-array.ts

import { deduplicate } from "./deduplicate"
import { isObject } from "./is-object"

export function getSelectsAndRelationsFromObjectArray(
  dataArray: object[],
  prefix?: string
): {
  selects: string[]
  relations: string[]
} {
  const selects: string[] = []
  const relations: string[] = []

  for (const data of dataArray) {
    for (const [key, value] of Object.entries(data)) {
      if (isObject(value)) {
        relations.push(setKey(key, prefix))
        const res = getSelectsAndRelationsFromObjectArray(
          [value],
          setKey(key, prefix)
        )
        selects.push(...res.selects)
        relations.push(...res.relations)
      } else if (Array.isArray(value)) {
        relations.push(setKey(key, prefix))
        const res = getSelectsAndRelationsFromObjectArray(
          value,
          setKey(key, prefix)
        )
        selects.push(...res.selects)
        relations.push(...res.relations)
      } else {
        selects.push(setKey(key, prefix))
      }
    }
  }

  const uniqueSelects: string[] = deduplicate(selects)
  const uniqueRelations: string[] = deduplicate(relations)

  return {
    selects: uniqueSelects,
    relations: uniqueRelations,
  }
}

function setKey(key: string, prefix?: string) {
  if (prefix) {
    return `${prefix}.${key}`
  } else {
    return key
  }
}