import { atom } from 'jotai'

// download states
export const modelDownloadStateAtom = atom<Record<string, DownloadState>>({})

export const setDownloadStateAtom = atom(
  null,
  (get, set, state: DownloadState) => {
    const currentState = { ...get(modelDownloadStateAtom) }
    console.debug(
      `current download state for ${state.fileName} is ${JSON.stringify(state)}`
    )
    state.fileName = state.fileName.replace('models/', '')
    // TODO: Need somehow to not depend on filename
    currentState[state.fileName] = state
    set(modelDownloadStateAtom, currentState)
  }
)

export const setDownloadStateSuccessAtom = atom(
  null,
  (get, set, fileName: string) => {
    const currentState = { ...get(modelDownloadStateAtom) }
    fileName = fileName.replace('models/', '')
    const state = currentState[fileName]
    if (!state) {
      console.error(`Cannot find download state for ${fileName}`)
      return
    }

    delete currentState[fileName]
    set(modelDownloadStateAtom, currentState)
  }
)