session-router.ts

import { chromeLocalStore } from 'chrome/helpers/chrome-local-store'
import { ResultAsync, ok } from 'neverthrow'

export type SessionRouter = ReturnType<typeof SessionRouter>

export type SessionId = string
export type WalletPublicKey = string

export const SessionRouter = () => {
  const store = new Map<SessionId, WalletPublicKey>()

  const getWalletPublicKey = (sessionId?: SessionId) =>
    sessionId ? store.get(sessionId) : undefined

  const refreshStore = (data: Record<SessionId, WalletPublicKey>) => {
    store.clear()
    Object.entries(data).forEach(([sessionId, walletPublicKey]) => {
      store.set(sessionId, walletPublicKey)
    })
  }

  return {
    refreshStore,
    getWalletPublicKey,
    store,
  }
}

export const sessionRouter = SessionRouter()

export const getSessionRouterData = (): ResultAsync<
  Record<string, string>,
  never
> =>
  chromeLocalStore
    .getItem('sessionRouter')
    .map(({ sessionRouter }) => sessionRouter || {})
    .orElse(() => ok({}))