jan/web/helpers/atoms/Extension.atom.ts

import { atom } from 'jotai'
import { atomWithStorage } from 'jotai/utils'

type ExtensionId = string

export type InstallingExtensionState = {
  extensionId: ExtensionId
  percentage: number
  localPath?: string
}

export const installingExtensionAtom = atom<InstallingExtensionState[]>([])

export const setInstallingExtensionAtom = atom(
  null,
  (get, set, extensionId: string, state: InstallingExtensionState) => {
    const current = get(installingExtensionAtom)

    const isExists = current.some((e) => e.extensionId === extensionId)
    if (isExists) {
      const newCurrent = current.map((e) => {
        if (e.extensionId === extensionId) {
          return state
        }
        return e
      })
      set(installingExtensionAtom, newCurrent)
    } else {
      set(installingExtensionAtom, [...current, state])
    }
  }
)

export const removeInstallingExtensionAtom = atom(
  null,
  (get, set, extensionId: string) => {
    const current = get(installingExtensionAtom)
    const newCurrent = current.filter((e) => e.extensionId !== extensionId)
    set(installingExtensionAtom, newCurrent)
  }
)

const INACTIVE_ENGINE_PROVIDER = 'inActiveEngineProvider'
export const inActiveEngineProviderAtom = atomWithStorage<string[]>(
  INACTIVE_ENGINE_PROVIDER,
  [],
  undefined,
  { getOnInit: true }
)

const SHOW_SETTING_ACTIVE_LOCAL_ENGINE = 'showSettingActiveLocalEngine'
export const showSettingActiveLocalEngineAtom = atomWithStorage<string[]>(
  SHOW_SETTING_ACTIVE_LOCAL_ENGINE,
  [],
  undefined,
  { getOnInit: true }
)

const SHOW_SETTING_ACTIVE_REMOTE_ENGINE = 'showSettingActiveRemoteEngine'
export const showSettingActiveRemoteEngineAtom = atomWithStorage<string[]>(
  SHOW_SETTING_ACTIVE_REMOTE_ENGINE,
  [],
  undefined,
  { getOnInit: true }
)