import { CustomError } from '../../utils/error'

export default async (
  req: Record<string, any>,
  $firestore
): Promise<Record<string, any>> => {
  try {
    const entity = req.params.entity
    const children = req.params.children
    const entityDefinition = await $firestore.getDoc({
      collection: 'entities',
      id: entity,
    })

    if (!children) return entityDefinition

    return entityDefinition.relationships[children]
  } catch (error: any) {
    throw new CustomError(`[getEntityDefinition] ${error.message}`, 400)
  }
}