import { RelationTypes, isLinksOrLTAR } from 'nocodb-sdk' import type { ColumnType, LinkToAnotherRecordType, TableType } from 'nocodb-sdk' import type { Ref } from 'vue' import type { MaybeRef } from '@vueuse/core' import { NOCO, computed, deepCompare, extractPkFromRow, extractSdkResponseErrorMsg, isBt, isHm, isMm, message, ref, storeToRefs, unref, useI18n, useInjectionState, useMetas, useNuxtApp, useProject, } from '#imports' import type { Row } from '~/lib' const [useProvideSmartsheetRowStore, useSmartsheetRowStore] = useInjectionState( (meta: Ref, row: MaybeRef) => { const { $api } = useNuxtApp() const { t } = useI18n() const { project } = storeToRefs(useProject()) const { metas } = useMetas() const currentRow = ref(row) // state const state = ref | Record[] | null>>({}) // getters const isNew = computed(() => unref(row).rowMeta.new ?? false) // actions const addLTARRef = async (value: Record, column: ColumnType) => { if (isHm(column) || isMm(column)) { if (!state.value[column.title!]) state.value[column.title!] = [] if (state.value[column.title!]!.find((ln: Record) => deepCompare(ln, value))) { // This value is already in the list return message.info(t('msg.info.valueAlreadyInList')) } if (Array.isArray(value)) { state.value[column.title!]!.push(...value) } else { state.value[column.title!]!.push(value) } } else if (isBt(column)) { state.value[column.title!] = value } } // actions const removeLTARRef = async (value: Record, column: ColumnType) => { if (isHm(column) || isMm(column)) { state.value[column.title!]?.splice(state.value[column.title!]?.indexOf(value), 1) } else if (isBt(column)) { state.value[column.title!] = null } } const linkRecord = async ( rowId: string, relatedRowId: string, column: ColumnType, type: RelationTypes, { metaValue = meta.value }: { metaValue?: TableType } = {}, ) => { try { await $api.dbTableRow.nestedAdd( NOCO, project.value.id as string, metaValue?.id as string, rowId, type as 'mm' | 'hm', column.id as string, relatedRowId, ) } catch (e: any) { message.error(await extractSdkResponseErrorMsg(e)) } } /** sync LTAR relations kept in local state */ const syncLTARRefs = async (row: Record, { metaValue = meta.value }: { metaValue?: TableType } = {}) => { const id = extractPkFromRow(row, metaValue?.columns as ColumnType[]) for (const column of metaValue?.columns ?? []) { if (!isLinksOrLTAR(column)) continue const colOptions = column.colOptions as LinkToAnotherRecordType const relatedTableMeta = metas.value?.[colOptions?.fk_related_model_id as string] if (isHm(column) || isMm(column)) { const relatedRows = (state.value?.[column.title!] ?? []) as Record[] for (const relatedRow of relatedRows) { await linkRecord( id, extractPkFromRow(relatedRow, relatedTableMeta.columns as ColumnType[]), column, colOptions.type as RelationTypes, { metaValue }, ) } } else if (isBt(column) && state.value?.[column.title!]) { await linkRecord( id, extractPkFromRow(state.value?.[column.title!] as Record, relatedTableMeta.columns as ColumnType[]), column, colOptions.type as RelationTypes, { metaValue }, ) } // clear LTAR refs after sync state.value[column.title!] = null } } // clear LTAR cell const clearLTARCell = async (column: ColumnType) => { try { if (!column || !isLinksOrLTAR(column)) return const relatedTableMeta = metas.value?.[(column?.colOptions)?.fk_related_model_id as string] if (isNew.value) { state.value[column.title!] = null } else if (currentRow.value) { if ((column.colOptions)?.type === RelationTypes.BELONGS_TO) { if (!currentRow.value.row[column.title!]) return await $api.dbTableRow.nestedRemove( NOCO, project.value.id as string, meta.value?.id as string, extractPkFromRow(currentRow.value.row, meta.value?.columns as ColumnType[]), 'bt' as any, column.id as string, extractPkFromRow(currentRow.value.row[column.title!], relatedTableMeta?.columns as ColumnType[]), ) currentRow.value.row[column.title!] = null } else { for (const link of (currentRow.value.row[column.title!] as Record[]) || []) { await $api.dbTableRow.nestedRemove( NOCO, project.value.id as string, meta.value?.id as string, extractPkFromRow(currentRow.value.row, meta.value?.columns as ColumnType[]), (column?.colOptions).type as 'hm' | 'mm', column.id as string, extractPkFromRow(link, relatedTableMeta?.columns as ColumnType[]), ) } currentRow.value.row[column.title!] = [] } } } catch (e: any) { message.error(await extractSdkResponseErrorMsg(e)) } } const loadRow = async () => { const record = await $api.dbTableRow.read( NOCO, project.value?.id as string, meta.value?.title as string, extractPkFromRow(unref(row)?.row, meta.value?.columns as ColumnType[]), ) Object.assign(unref(row), { row: record, oldRow: { ...record }, rowMeta: {}, }) } return { row, state, isNew, // todo: use better name addLTARRef, removeLTARRef, syncLTARRefs, loadRow, currentRow, clearLTARCell, } }, '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 }