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

import type { TableType } from 'nocodb-sdk'
import { UITypes } from 'nocodb-sdk'
import { useToast } from 'vue-toastification'
import { useProject } from './useProject'
import { useNuxtApp } from '#app'
import { extractSdkResponseErrorMsg } from '~/utils/errorUtils'

export function useTableCreate(onTableCreate?: (tableMeta: TableType) => void) {
  const table = reactive<{ title: string; table_name: string; columns: string[] }>({
    title: '',
    table_name: '',
    columns: {
      id: true,
      title: true,
      created_at: true,
      updated_at: true,
    },
  })

  const { sqlUi, project, tables } = useProject()

  const toast = useToast()

  const { $api } = useNuxtApp()

  const createTable = async () => {
    try {
      if (!sqlUi?.value) return
      const columns = sqlUi?.value?.getNewTableColumns().filter((col) => {
        if (col.column_name === 'id' && table.columns.id_ag) {
          Object.assign(col, sqlUi?.value?.getDataTypeForUiType({ uidt: UITypes.ID }, 'AG'))
          col.dtxp = sqlUi?.value?.getDefaultLengthForDatatype(col.dt)
          col.dtxs = sqlUi?.value?.getDefaultScaleForDatatype(col.dt)
          return true
        }
        return !!table.columns[col.column_name]
      })

      const tableMeta = await $api.dbTable.create(project?.value?.id as string, {
        ...table,
        columns,
      })

      onTableCreate?.(tableMeta)
    } catch (e: any) {
      toast.error(await extractSdkResponseErrorMsg(e))
    }
  }

  watch(
    () => table.title,
    (title) => {
      table.table_name = `${project?.value?.prefix || ''}${title}`
    },
  )

  const generateUniqueTitle = () => {
    let c = 1
    while (tables?.value?.some((t) => t.title === `Sheet${c}`)) {
      c++
    }
    table.title = `Sheet${c}`
  }

  return { table, createTable, generateUniqueTitle, tables, project }
}