smesh/src/services/storage.service.ts

import { StorageKey } from '@/constants'
import { isSameAccount } from '@/lib/account'
import { randomString } from '@/lib/random'
import {
  TAccount,
  TAccountPointer,
  TFeedType,
  TNoteListMode,
  TRelaySet,
  TThemeSetting
} from '@/types'
import { Event } from 'nostr-tools'

const DEFAULT_RELAY_SETS: TRelaySet[] = [
  {
    id: randomString(),
    name: 'Global',
    relayUrls: ['wss://relay.damus.io/', 'wss://nos.lol/']
  },
  {
    id: randomString(),
    name: 'Algo',
    relayUrls: ['wss://algo.utxo.one']
  }
]

class StorageService {
  static instance: StorageService

  private relaySets: TRelaySet[] = []
  private activeRelaySetId: string | null = null
  private feedType: TFeedType = 'relays'
  private themeSetting: TThemeSetting = 'system'
  private accounts: TAccount[] = []
  private currentAccount: TAccount | null = null
  private noteListMode: TNoteListMode = 'posts'
  private accountRelayListEventMap: Record<string, Event | undefined> = {} // pubkey -> relayListEvent
  private accountFollowListEventMap: Record<string, Event | undefined> = {} // pubkey -> followListEvent
  private accountMuteListEventMap: Record<string, Event | undefined> = {} // pubkey -> muteListEvent
  private accountMuteDecryptedTagsMap: Record<
    string,
    { id: string; tags: string[][] } | undefined
  > = {} // pubkey -> { id, tags }
  private accountProfileEventMap: Record<string, Event | undefined> = {} // pubkey -> profileEvent

  constructor() {
    if (!StorageService.instance) {
      this.init()
      StorageService.instance = this
    }
    return StorageService.instance
  }

  init() {
    this.themeSetting =
      (window.localStorage.getItem(StorageKey.THEME_SETTING) as TThemeSetting) ?? 'system'
    const accountsStr = window.localStorage.getItem(StorageKey.ACCOUNTS)
    this.accounts = accountsStr ? JSON.parse(accountsStr) : []
    const currentAccountStr = window.localStorage.getItem(StorageKey.CURRENT_ACCOUNT)
    this.currentAccount = currentAccountStr ? JSON.parse(currentAccountStr) : null
    const feedTypeStr = window.localStorage.getItem(StorageKey.FEED_TYPE)
    this.feedType = feedTypeStr ? JSON.parse(feedTypeStr) : 'relays'
    const noteListModeStr = window.localStorage.getItem(StorageKey.NOTE_LIST_MODE)
    this.noteListMode =
      noteListModeStr && ['posts', 'postsAndReplies', 'pictures'].includes(noteListModeStr)
        ? (noteListModeStr as TNoteListMode)
        : 'posts'

    const accountRelayListEventMapStr = window.localStorage.getItem(
      StorageKey.ACCOUNT_RELAY_LIST_EVENT_MAP
    )
    this.accountRelayListEventMap = accountRelayListEventMapStr
      ? JSON.parse(accountRelayListEventMapStr)
      : {}
    const accountFollowListEventMapStr = window.localStorage.getItem(
      StorageKey.ACCOUNT_FOLLOW_LIST_EVENT_MAP
    )
    this.accountFollowListEventMap = accountFollowListEventMapStr
      ? JSON.parse(accountFollowListEventMapStr)
      : {}
    const accountMuteListEventMapStr = window.localStorage.getItem(
      StorageKey.ACCOUNT_MUTE_LIST_EVENT_MAP
    )
    this.accountMuteListEventMap = accountMuteListEventMapStr
      ? JSON.parse(accountMuteListEventMapStr)
      : {}
    const accountMuteDecryptedTagsMapStr = window.localStorage.getItem(
      StorageKey.ACCOUNT_MUTE_DECRYPTED_TAGS_MAP
    )
    this.accountMuteDecryptedTagsMap = accountMuteDecryptedTagsMapStr
      ? JSON.parse(accountMuteDecryptedTagsMapStr)
      : {}
    const accountProfileEventMapStr = window.localStorage.getItem(
      StorageKey.ACCOUNT_PROFILE_EVENT_MAP
    )
    this.accountProfileEventMap = accountProfileEventMapStr
      ? JSON.parse(accountProfileEventMapStr)
      : {}

    const relaySetsStr = window.localStorage.getItem(StorageKey.RELAY_SETS)
    if (!relaySetsStr) {
      let relaySets: TRelaySet[] = []
      const legacyRelayGroupsStr = window.localStorage.getItem('relayGroups')
      if (legacyRelayGroupsStr) {
        const legacyRelayGroups = JSON.parse(legacyRelayGroupsStr)
        relaySets = legacyRelayGroups.map((group: any) => {
          return {
            id: randomString(),
            name: group.groupName,
            relayUrls: group.relayUrls
          }
        })
      }
      if (!relaySets.length) {
        relaySets = DEFAULT_RELAY_SETS
      }
      const activeRelaySetId = relaySets[0].id
      window.localStorage.setItem(StorageKey.RELAY_SETS, JSON.stringify(relaySets))
      window.localStorage.setItem(StorageKey.ACTIVE_RELAY_SET_ID, activeRelaySetId)
      this.relaySets = relaySets
      this.activeRelaySetId = activeRelaySetId
    } else {
      this.relaySets = JSON.parse(relaySetsStr)
      const activeRelaySetIdStr = window.localStorage.getItem(StorageKey.ACTIVE_RELAY_SET_ID)
      this.activeRelaySetId = activeRelaySetIdStr ? JSON.parse(activeRelaySetIdStr) : null
    }
  }

  getRelaySets() {
    return this.relaySets
  }

  setRelaySets(relaySets: TRelaySet[]) {
    this.relaySets = relaySets
    window.localStorage.setItem(StorageKey.RELAY_SETS, JSON.stringify(this.relaySets))
  }

  getActiveRelaySetId() {
    return this.activeRelaySetId
  }

  setActiveRelaySetId(id: string | null) {
    this.activeRelaySetId = id
    if (id) {
      window.localStorage.setItem(
        StorageKey.ACTIVE_RELAY_SET_ID,
        JSON.stringify(this.activeRelaySetId)
      )
    } else {
      window.localStorage.removeItem(StorageKey.ACTIVE_RELAY_SET_ID)
    }
  }

  getFeedType() {
    return this.feedType
  }

  setFeedType(feedType: TFeedType) {
    this.feedType = feedType
    window.localStorage.setItem(StorageKey.FEED_TYPE, JSON.stringify(this.feedType))
  }

  getThemeSetting() {
    return this.themeSetting
  }

  setThemeSetting(themeSetting: TThemeSetting) {
    window.localStorage.setItem(StorageKey.THEME_SETTING, themeSetting)
    this.themeSetting = themeSetting
  }

  getNoteListMode() {
    return this.noteListMode
  }

  setNoteListMode(mode: TNoteListMode) {
    window.localStorage.setItem(StorageKey.NOTE_LIST_MODE, mode)
    this.noteListMode = mode
  }

  getAccounts() {
    return this.accounts
  }

  findAccount(account: TAccountPointer) {
    return this.accounts.find((act) => isSameAccount(act, account))
  }

  getCurrentAccount() {
    return this.currentAccount
  }

  getAccountNsec(pubkey: string) {
    const account = this.accounts.find((act) => act.pubkey === pubkey && act.signerType === 'nsec')
    return account?.nsec
  }

  getAccountNcryptsec(pubkey: string) {
    const account = this.accounts.find(
      (act) => act.pubkey === pubkey && act.signerType === 'ncryptsec'
    )
    return account?.ncryptsec
  }

  addAccount(account: TAccount) {
    if (this.accounts.find((act) => isSameAccount(act, account))) {
      return
    }
    this.accounts.push(account)
    window.localStorage.setItem(StorageKey.ACCOUNTS, JSON.stringify(this.accounts))
    return account
  }

  removeAccount(account: TAccount) {
    this.accounts = this.accounts.filter((act) => !isSameAccount(act, account))
    delete this.accountFollowListEventMap[account.pubkey]
    delete this.accountRelayListEventMap[account.pubkey]
    delete this.accountMuteListEventMap[account.pubkey]
    delete this.accountMuteDecryptedTagsMap[account.pubkey]
    delete this.accountProfileEventMap[account.pubkey]
    window.localStorage.setItem(StorageKey.ACCOUNTS, JSON.stringify(this.accounts))
    window.localStorage.setItem(
      StorageKey.ACCOUNT_FOLLOW_LIST_EVENT_MAP,
      JSON.stringify(this.accountFollowListEventMap)
    )
    window.localStorage.setItem(
      StorageKey.ACCOUNT_MUTE_LIST_EVENT_MAP,
      JSON.stringify(this.accountMuteListEventMap)
    )
    window.localStorage.setItem(
      StorageKey.ACCOUNT_MUTE_DECRYPTED_TAGS_MAP,
      JSON.stringify(this.accountMuteDecryptedTagsMap)
    )
    window.localStorage.setItem(
      StorageKey.ACCOUNT_RELAY_LIST_EVENT_MAP,
      JSON.stringify(this.accountRelayListEventMap)
    )
    window.localStorage.setItem(
      StorageKey.ACCOUNT_PROFILE_EVENT_MAP,
      JSON.stringify(this.accountProfileEventMap)
    )
  }

  switchAccount(account: TAccount | null) {
    if (isSameAccount(this.currentAccount, account)) {
      return
    }
    const act = this.accounts.find((act) => isSameAccount(act, account))
    if (!act) {
      return
    }
    this.currentAccount = act
    window.localStorage.setItem(StorageKey.CURRENT_ACCOUNT, JSON.stringify(act))
  }

  getAccountRelayListEvent(pubkey: string) {
    return this.accountRelayListEventMap[pubkey]
  }

  setAccountRelayListEvent(relayListEvent: Event) {
    const pubkey = relayListEvent.pubkey
    if (
      this.accountRelayListEventMap[pubkey] &&
      this.accountRelayListEventMap[pubkey].created_at > relayListEvent.created_at
    ) {
      return false
    }
    this.accountRelayListEventMap[pubkey] = relayListEvent
    window.localStorage.setItem(
      StorageKey.ACCOUNT_RELAY_LIST_EVENT_MAP,
      JSON.stringify(this.accountRelayListEventMap)
    )
    return true
  }

  getAccountFollowListEvent(pubkey: string) {
    return this.accountFollowListEventMap[pubkey]
  }

  setAccountFollowListEvent(followListEvent: Event) {
    const pubkey = followListEvent.pubkey
    if (
      this.accountFollowListEventMap[pubkey] &&
      this.accountFollowListEventMap[pubkey].created_at > followListEvent.created_at
    ) {
      return false
    }
    this.accountFollowListEventMap[pubkey] = followListEvent
    window.localStorage.setItem(
      StorageKey.ACCOUNT_FOLLOW_LIST_EVENT_MAP,
      JSON.stringify(this.accountFollowListEventMap)
    )
    return true
  }

  getAccountMuteListEvent(pubkey: string) {
    return this.accountMuteListEventMap[pubkey]
  }

  setAccountMuteListEvent(muteListEvent: Event) {
    const pubkey = muteListEvent.pubkey
    if (
      this.accountMuteListEventMap[pubkey] &&
      this.accountMuteListEventMap[pubkey].created_at > muteListEvent.created_at
    ) {
      return false
    }
    this.accountMuteListEventMap[pubkey] = muteListEvent
    window.localStorage.setItem(
      StorageKey.ACCOUNT_MUTE_LIST_EVENT_MAP,
      JSON.stringify(this.accountMuteListEventMap)
    )
    return true
  }

  getAccountMuteDecryptedTags(muteListEvent: Event) {
    const stored = this.accountMuteDecryptedTagsMap[muteListEvent.pubkey]
    if (stored && stored.id === muteListEvent.id) {
      return stored.tags
    }
    return null
  }

  setAccountMuteDecryptedTags(muteListEvent: Event, tags: string[][]) {
    this.accountMuteDecryptedTagsMap[muteListEvent.pubkey] = { id: muteListEvent.id, tags }
    window.localStorage.setItem(
      StorageKey.ACCOUNT_MUTE_DECRYPTED_TAGS_MAP,
      JSON.stringify(this.accountMuteDecryptedTagsMap)
    )
  }

  getAccountProfileEvent(pubkey: string) {
    return this.accountProfileEventMap[pubkey]
  }

  setAccountProfileEvent(profileEvent: Event) {
    const pubkey = profileEvent.pubkey
    if (
      this.accountProfileEventMap[pubkey] &&
      this.accountProfileEventMap[pubkey].created_at > profileEvent.created_at
    ) {
      return false
    }
    this.accountProfileEventMap[pubkey] = profileEvent
    window.localStorage.setItem(
      StorageKey.ACCOUNT_PROFILE_EVENT_MAP,
      JSON.stringify(this.accountProfileEventMap)
    )
    return true
  }
}

const instance = new StorageService()

export default instance