import { cloneDeep } from "lodash/fp"
import { fetchTableData } from "./tables"
import { fetchViewData } from "./views"
import { fetchRelationshipData } from "./relationships"
import { executeQuery } from "./queries"

/**
 * Fetches all rows for a particular Budibase data source.
 */
export const fetchDatasource = async dataSource => {
  if (!dataSource || !dataSource.type) {
    return []
  }

  // Fetch all rows in data source
  const { type, tableId, fieldName } = dataSource
  let rows = []
  if (type === "table") {
    rows = await fetchTableData(tableId)
  } else if (type === "view") {
    rows = await fetchViewData(dataSource)
  } else if (type === "query") {
    // Set the default query params
    let parameters = cloneDeep(dataSource.queryParams || {})
    for (let param of dataSource.parameters) {
      if (!parameters[param.name]) {
        parameters[param.name] = param.default
      }
    }
    rows = await executeQuery({ queryId: dataSource._id, parameters })
  } else if (type === "link") {
    rows = await fetchRelationshipData({
      rowId: dataSource.rowId,
      tableId: dataSource.rowTableId,
      fieldName,
    })
  }

  // Enrich the result is always an array
  return Array.isArray(rows) ? rows : []
}