hoppscotch/packages/hoppscotch-common/src/newstore/localstate.ts

import { Ref } from "vue"
import { distinctUntilChanged, pluck } from "rxjs"
import DispatchingStore, { defineDispatchers } from "./DispatchingStore"
import { useStream } from "@composables/stream"

type LocalState = {
  REMEMBERED_TEAM_ID: string | undefined
}

const defaultLocalState: LocalState = {
  REMEMBERED_TEAM_ID: undefined,
}

type ApplyLocalState = {
  [K in keyof LocalState]: {
    key: K
    value: LocalState[K]
  }
}[keyof LocalState]

const dispatchers = defineDispatchers({
  bulkApplyState(_currentState: LocalState, payload: Partial<LocalState>) {
    return payload
  },
  applyState(_currentState: LocalState, { key, value }: ApplyLocalState) {
    const result: Partial<LocalState> = {
      [key]: value,
    }

    return result
  },
})

export const localStateStore = new DispatchingStore(
  defaultLocalState,
  dispatchers
)

export const localState$ = localStateStore.subject$.asObservable()

export function bulkApplyLocalState(obj: Partial<LocalState>) {
  localStateStore.dispatch({
    dispatcher: "bulkApplyState",
    payload: obj,
  })
}

export function applyLocalState<K extends keyof LocalState>(
  key: K,
  value: LocalState[K]
) {
  localStateStore.dispatch({
    dispatcher: "applyState",
    payload: { key, value },
  })
}

export function useLocalState<K extends keyof LocalState>(
  key: K
): Ref<LocalState[K]> {
  return useStream(
    localStateStore.subject$.pipe(pluck(key), distinctUntilChanged()),
    localStateStore.value[key],
    (value: LocalState[K]) => {
      localStateStore.dispatch({
        dispatcher: "applyState",
        payload: { key, value },
      })
    }
  )
}