supabase/apps/studio/components/interfaces/Storage/BucketFilePickerDialog/BucketFilePickerState.tsx

import { createContext, PropsWithChildren, useContext, useMemo } from 'react'
import { proxy, useSnapshot } from 'valtio'

import { StorageItemWithColumn } from '@/components/interfaces/Storage/Storage.types'
import type { Bucket } from '@/data/storage/buckets-query'

function createBucketFilePickerState({
  bucket,
  maxFiles,
  acceptedFileExtensions,
}: {
  bucket: Bucket
  maxFiles: number
  acceptedFileExtensions?: string[]
}) {
  const state = proxy({
    bucket: bucket,
    maxFiles: maxFiles,
    acceptedFileExtensions: acceptedFileExtensions,

    columns: [] as string[],
    popColumn: () => {
      const lastColumnIndex = state.columns.length - 1
      state.columns = state.columns.slice(0, lastColumnIndex)
    },
    popColumnAtIndex: (index: number) => {
      state.columns = state.columns.slice(0, index)
    },
    pushColumnAtIndex: (column: string, index: number) => {
      state.columns = state.columns.slice(0, index).concat([column])
    },

    selectedItems: [] as StorageItemWithColumn[],
    setSelectedItems: (items: StorageItemWithColumn[]) => (state.selectedItems = items),
    clearSelectedItems: (columnIndex?: number) => {
      if (columnIndex !== undefined) {
        state.selectedItems = state.selectedItems.filter((item) => item.columnIndex !== columnIndex)
      } else {
        state.selectedItems = []
      }
    },

    itemSearchString: '',
    setItemSearchString: (value: string) => (state.itemSearchString = value),

    selectedFilePreview: undefined as StorageItemWithColumn | undefined,
    setSelectedFilePreview: (file?: StorageItemWithColumn) => (state.selectedFilePreview = file),
  })

  return state
}

export type BucketFilePickerState = ReturnType<typeof createBucketFilePickerState>

const DEFAULT_STATE_CONFIG = {
  bucket: {} as Bucket,
  maxFiles: 1 as const,
}

const BucketFilePickerStateContext = createContext<BucketFilePickerState>(
  createBucketFilePickerState(DEFAULT_STATE_CONFIG)
)

export const BucketFilePickerStateContextProvider = ({
  bucket,
  maxFiles,
  acceptedFileExtensions,
  children,
}: PropsWithChildren<{ bucket: Bucket; maxFiles: number; acceptedFileExtensions?: string[] }>) => {
  const state = useMemo(
    () => createBucketFilePickerState({ bucket, maxFiles, acceptedFileExtensions }),
    // eslint-disable-next-line react-hooks/exhaustive-deps
    [bucket, maxFiles, acceptedFileExtensions?.join(',')]
  )

  return (
    <BucketFilePickerStateContext.Provider value={state}>
      {children}
    </BucketFilePickerStateContext.Provider>
  )
}

export function useBucketFilePickerStateSnapshot(options?: Parameters<typeof useSnapshot>[1]) {
  const state = useContext(BucketFilePickerStateContext)
  return useSnapshot(state, options) as BucketFilePickerState
}