Function useDeleteColumn

apps/sim/hooks/queries/tables.ts:1775–1841

({ workspaceId, tableId }: RowMutationContext)

}

export function useDeleteColumn({ workspaceId, tableId }: RowMutationContext) {
  const queryClient = useQueryClient()

  return useMutation({
    mutationFn: async (columnName: string) => {
      return requestJson(deleteTableColumnContract, {
        params: { tableId },
        body: { workspaceId, columnName },
      })
    },
    onMutate: async (columnName) => {
      await queryClient.cancelQueries({ queryKey: tableKeys.detail(tableId) })

      const lower = columnName.toLowerCase()
      const previousDetail = queryClient.getQueryData<TableDefinition>(tableKeys.detail(tableId))
      // The grid deletes by stable id; legacy callers may pass a name. Resolve
      // the column's storage id once from either form, then strip schema,
      // widths, and row data by that single id — all three are id-keyed, so a
      // name arg with a distinct id must never be used as the strip key directly.
      const target = previousDetail?.schema.columns.find(
        (c) => getColumnId(c) === columnName || c.name.toLowerCase() === lower
      )
      const stripKey = target ? getColumnId(target) : columnName

      if (previousDetail) {
        const nextColumns = previousDetail.schema.columns.filter((c) => getColumnId(c) !== stripKey)
        const prevWidths = previousDetail.metadata?.columnWidths
        const nextMetadata = prevWidths
          ? {
              ...previousDetail.metadata,
              columnWidths: Object.fromEntries(
                Object.entries(prevWidths).filter(([k]) => k !== stripKey)
              ),
            }
          : previousDetail.metadata
        queryClient.setQueryData<TableDefinition>(tableKeys.detail(tableId), {
          ...previousDetail,
          schema: { ...previousDetail.schema, columns: nextColumns },
          metadata: nextMetadata,
        })
      }

      const rowSnapshots = await snapshotAndMutateRows(queryClient, tableId, (row) => {
        if (!(stripKey in row.data)) return null
        const { [stripKey]: _removed, ...rest } = row.data
        return { ...row, data: rest }
      })

      return { previousDetail, rowSnapshots }
    },
    onError: (error, _columnName, context) => {
      if (context?.previousDetail) {
        queryClient.setQueryData(tableKeys.detail(tableId), context.previousDetail)
      }
      if (context?.rowSnapshots) {
        for (const [key, data] of context.rowSnapshots) {
          queryClient.setQueryData(key, data)
        }