nocodb/packages/nc-gui/composables/useExtensionHelper.ts

import type { ColumnType, ViewType } from 'nocodb-sdk'
import type { ExtensionType } from '#imports'
import { useInjectionState } from '#imports'

const [useProvideExtensionHelper, useExtensionHelper] = useInjectionState((extension: Ref<ExtensionType>) => {
  const { $api } = useNuxtApp()

  const basesStore = useBases()

  const { activeProjectId: baseId } = storeToRefs(basesStore)

  const tableStore = useTablesStore()

  const { activeTables: tables } = storeToRefs(tableStore)

  const viewStore = useViewsStore()

  const { viewsByTable } = storeToRefs(viewStore)

  const { getMeta } = useMetas()

  const { eventBus } = useSmartsheetStoreOrThrow()

  const fullscreen = ref(false)

  const collapsed = computed({
    get: () => extension.value?.meta?.collapsed ?? false,
    set: (value) => {
      extension.value?.setMeta('collapsed', value)
    },
  })

  const getViewsForTable = async (tableId: string) => {
    if (viewsByTable.value.has(tableId)) {
      return viewsByTable.value.get(tableId) as ViewType[]
    }

    await viewStore.loadViews({ tableId, ignoreLoading: true })
    return viewsByTable.value.get(tableId) as ViewType[]
  }

  const getData = async (params: {
    tableId: string
    viewId?: string
    eachPage: (records: Record<string, any>[], nextPage: () => void) => Promise<void> | void
    done: () => Promise<void> | void
  }) => {
    const { tableId, viewId, eachPage, done } = params

    let page = 1

    const nextPage = async () => {
      const { list: records, pageInfo } = await $api.dbViewRow.list(
        'noco',
        baseId.value!,
        tableId,
        viewId as string,
        {
          offset: (page - 1) * 100,
          limit: 100,
        } as any,
      )

      if (pageInfo?.isLastPage) {
        await eachPage(records, () => {})
        await done()
      } else {
        page++
        await eachPage(records, nextPage)
      }
    }

    await nextPage()
  }

  const getTableMeta = async (tableId: string) => {
    return getMeta(tableId)
  }

  const insertData = async (params: { tableId: string; data: Record<string, any> }) => {
    const { tableId, data } = params

    const chunks = []

    let inserted = 0

    // chunk data into 100 records
    for (let i = 0; i < data.length; i += 100) {
      chunks.push(data.slice(i, i + 100))
    }

    for (const chunk of chunks) {
      inserted += chunk.length
      await $api.dbDataTableRow.create(tableId, chunk)
    }

    return {
      inserted,
    }
  }

  const updateData = async (params: { tableId: string; data: Record<string, any> }) => {
    const { tableId, data } = params

    const chunks = []

    let updated = 0

    // chunk data into 100 records
    for (let i = 0; i < data.length; i += 100) {
      chunks.push(data.slice(i, i + 100))
    }

    for (const chunk of chunks) {
      updated += chunk.length
      await $api.dbDataTableRow.update(tableId, chunk)
    }

    return {
      updated,
    }
  }

  const upsertData = async (params: { tableId: string; data: Record<string, any>; upsertField: ColumnType }) => {
    const { tableId, data, upsertField } = params

    const chunkSize = 100

    const tableMeta = await getMeta(tableId)

    if (!tableMeta?.columns) throw new Error('Table not found')

    const chunks = []

    for (let i = 0; i < data.length; i += chunkSize) {
      chunks.push(data.slice(i, i + chunkSize))
    }

    const insert = []
    const update = []

    let insertCounter = 0
    let updateCounter = 0

    for (const chunk of chunks) {
      // select chunk of data to determine if it's an insert or update
      const { list } = await $api.dbDataTableRow.list(tableId, {
        where: `(${upsertField.title},in,${chunk.map((record: Record<string, any>) => record[upsertField.title!]).join(',')})`,
        limit: chunkSize,
      })

      insert.push(
        ...chunk.filter(
          (record: Record<string, any>) =>
            !list.some((r: Record<string, any>) => `${r[upsertField.title!]}` === `${record[upsertField.title!]}`),
        ),
      )

      update.push(
        ...chunk
          .filter((record: Record<string, any>) =>
            list.some((r: Record<string, any>) => `${r[upsertField.title!]}` === `${record[upsertField.title!]}`),
          )
          .map((record: Record<string, any>) => {
            const existingRecord = list.find(
              (r: Record<string, any>) => `${r[upsertField.title!]}` === `${record[upsertField.title!]}`,
            )
            return {
              ...rowPkData(existingRecord!, tableMeta.columns!),
              ...record,
            }
          }),
      )
    }

    if (insert.length) {
      insertCounter += insert.length
      while (insert.length) {
        await $api.dbDataTableRow.create(tableId, insert.splice(0, chunkSize))
      }
    }

    if (update.length) {
      updateCounter += update.length
      while (update.length) {
        await $api.dbDataTableRow.update(tableId, update.splice(0, chunkSize))
      }
    }

    return { inserted: insertCounter, updated: updateCounter }
  }

  const reloadData = () => {
    eventBus.emit(SmartsheetStoreEvents.DATA_RELOAD)
  }

  const reloadMeta = () => {
    eventBus.emit(SmartsheetStoreEvents.FIELD_RELOAD)
  }

  return {
    fullscreen,
    collapsed,
    extension,
    tables,
    getViewsForTable,
    getData,
    getTableMeta,
    $api,
    insertData,
    updateData,
    upsertData,
    reloadData,
    reloadMeta,
  }
}, 'extension-helper')

export { useProvideExtensionHelper }

export function useExtensionHelperOrThrow() {
  const extensionStore = useExtensionHelper()
  if (extensionStore == null) throw new Error('Please call `useProvideExtensionHelper` on the appropriate parent component')
  return extensionStore
}