import { Modal, message } from 'ant-design-vue' import type { LinkToAnotherRecordType, TableType } from 'nocodb-sdk' import { UITypes } from 'nocodb-sdk' import { useI18n } from 'vue-i18n' import { useNuxtApp } from '#app' import { TabType } from '~/composables/useTabs' import { SYSTEM_COLUMNS, extractSdkResponseErrorMsg, useProject } from '#imports' export function useTable(onTableCreate?: (tableMeta: TableType) => void) { const table = reactive<{ title: string; table_name: string; columns: string[] }>({ title: '', table_name: '', columns: SYSTEM_COLUMNS, }) const { t } = useI18n() const { $e, $api } = useNuxtApp() const { getMeta, removeMeta } = useMetas() const { loadTables } = useProject() const { closeTab } = useTabs() const { sqlUi, project, tables } = useProject() const createTable = async () => { if (!sqlUi?.value) return const columns = sqlUi?.value?.getNewTableColumns().filter((col) => { if (col.column_name === 'id' && table.columns.includes('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.includes(col.column_name) }) try { const tableMeta = await $api.dbTable.create(project?.value?.id as string, { ...table, columns, }) $e('a:table:create') onTableCreate?.(tableMeta) } catch (e: any) { message.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}` } const deleteTable = (table: TableType) => { $e('c:table:delete') // 'Click Submit to Delete The table' Modal.confirm({ title: `${t('msg.info.deleteTableConfirmation')} : ${table.title}?`, okText: t('general.yes'), okType: 'danger', cancelText: t('general.no'), async onOk() { try { const meta = (await getMeta(table.id as string)) as TableType const relationColumns = meta?.columns?.filter((c) => c.uidt === UITypes.LinkToAnotherRecord) if (relationColumns?.length) { const refColMsgs = await Promise.all( relationColumns.map(async (c, i) => { const refMeta = (await getMeta( (c?.colOptions as LinkToAnotherRecordType)?.fk_related_model_id as string, )) as TableType return `${i + 1}. ${c.title} is a LinkToAnotherRecord of ${(refMeta && refMeta.title) || c.title}` }), ) message.info( h('div', { innerHTML: `
Unable to delete tables because of the following.

${refColMsgs.join('
')}

Delete them & try again
`, }), ) return } await $api.dbTable.delete(table?.id as string) await closeTab({ type: TabType.TABLE, id: table.id, title: table.title, }) await loadTables() removeMeta(table.id as string) // Deleted table successfully message.info(t('msg.info.tableDeleted')) $e('a:table:delete') } catch (e: any) { message.error(await extractSdkResponseErrorMsg(e)) } }, }) } return { table, createTable, generateUniqueTitle, tables, project, deleteTable } }