import type { ColumnType, ViewType } from 'nocodb-sdk' import type { ExtensionType } from '#imports' import { useInjectionState } from '#imports' const [useProvideExtensionHelper, useExtensionHelper] = useInjectionState((extension: Ref) => { 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[], nextPage: () => void) => Promise | void done: () => Promise | 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 }) => { 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 }) => { 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; 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) => record[upsertField.title!]).join(',')})`, limit: chunkSize, }) insert.push( ...chunk.filter( (record: Record) => !list.some((r: Record) => `${r[upsertField.title!]}` === `${record[upsertField.title!]}`), ), ) update.push( ...chunk .filter((record: Record) => list.some((r: Record) => `${r[upsertField.title!]}` === `${record[upsertField.title!]}`), ) .map((record: Record) => { const existingRecord = list.find( (r: Record) => `${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 }