nocodb/packages/nc-gui-v2/composables/useSmartsheetRowStore.ts

import { notification } from 'ant-design-vue'
import { UITypes } from 'nocodb-sdk'
import type { ColumnType, LinkToAnotherRecordType, RelationTypes, TableType } from 'nocodb-sdk'
import type { Ref } from 'vue'
import { useNuxtApp } from '#app'
import { useInjectionState, useMetas, useProject, useVirtualCell } from '#imports'
import type { Row } from '~/composables/useViewData'
import { NOCO } from '~/lib'
import { extractPkFromRow, extractSdkResponseErrorMsg } from '~/utils'

const [useProvideSmartsheetRowStore, useSmartsheetRowStore] = useInjectionState((meta: Ref<TableType>, row: Ref<Row>) => {
  const { $api } = useNuxtApp()
  const { project } = useProject()
  const { metas } = useMetas()

  // state
  const state = ref<Record<string, Record<string, any> | Record<string, any>[] | null>>({})

  // getters
  const isNew = computed(() => row.value?.rowMeta?.new ?? false)

  // actions
  const addLTARRef = async (value: Record<string, any>, column: ColumnType) => {
    const { isHm, isMm, isBt } = $(useVirtualCell(ref(column)))
    if (isHm || isMm) {
      state.value[column.title!] = state.value[column.title!] || []
      state.value[column.title!]!.push(value)
    } else if (isBt) {
      state.value[column.title!] = value
    }
  }

  // actions
  const removeLTARRef = async (value: Record<string, any>, column: ColumnType) => {
    const { isHm, isMm, isBt } = $(useVirtualCell(ref(column)))
    if (isHm || isMm) {
      state.value[column.title!]?.splice(state.value[column.title!]?.indexOf(value), 1)
    } else if (isBt) {
      state.value[column.title!] = null
    }
  }

  const linkRecord = async (rowId: string, relatedRowId: string, column: ColumnType, type: RelationTypes) => {
    try {
      await $api.dbTableRow.nestedAdd(
        NOCO,
        project.value.title as string,
        meta.value.title as string,
        rowId,
        type,
        column.title as string,
        relatedRowId,
      )
    } catch (e: any) {
      notification.error({
        message: 'Linking failed',
        description: await extractSdkResponseErrorMsg(e),
      })
    }
  }

  /** sync LTAR relations kept in local state */
  const syncLTARRefs = async (row: Record<string, any>) => {
    const id = extractPkFromRow(row, meta.value.columns as ColumnType[])
    for (const column of meta?.value?.columns ?? []) {
      if (column.uidt !== UITypes.LinkToAnotherRecord) continue
      const colOptions = column?.colOptions as LinkToAnotherRecordType

      const { isHm, isMm, isBt } = $(useVirtualCell(ref(column)))
      const relatedTableMeta = metas.value?.[colOptions?.fk_related_model_id as string]

      if (isHm || isMm) {
        const relatedRows = (state.value?.[column.title!] ?? []) as Record<string, any>[]
        for (const relatedRow of relatedRows) {
          await linkRecord(id, extractPkFromRow(relatedRow, relatedTableMeta.columns as ColumnType[]), column, colOptions.type)
        }
      } else if (isBt && state?.value?.[column.title!]) {
        await linkRecord(
          id,
          extractPkFromRow(state.value?.[column.title!] as Record<string, any>, relatedTableMeta.columns as ColumnType[]),
          column,
          colOptions.type,
        )
      }
    }
  }

  return {
    row,
    state,
    isNew,
    // todo: use better name
    addLTARRef,
    removeLTARRef,
    syncLTARRefs,
  }
}, 'smartsheet-row-store')

export { useProvideSmartsheetRowStore }

export function useSmartsheetRowStoreOrThrow() {
  const smartsheetRowStore = useSmartsheetRowStore()
  if (smartsheetRowStore == null) throw new Error('Please call `useSmartsheetRowStore` on the appropriate parent component')
  return smartsheetRowStore
}