kodadot/nft-gallery

import { defineStore } from 'pinia'
import {
  EntityWithId,
  NFTMetadata,
  TokenId,
} from '@/components/rmrk/service/scheme'
import { ComputedRef } from 'vue'
import type { Prefix } from '@kodadot1/static'

export type ListCartItemMediaUrl = { image: string; mimeType?: string }

type ListCartItemInternal = {
  id: string
  name: string
  urlPrefix: string
  price: string
  listPrice?: number
  sn: string
  collection: EntityWithId
  meta?: NFTMetadata
  metadata?: string
  discarded?: boolean
  mediaUrl?: ListCartItemMediaUrl // used inside the drop page to show the actual nft image before the metadata is updated
}

export type ListCartItem = ListCartItemInternal & TokenId

type ID = string

interface State {
  items: ListCartItem[]
  allUnlistedItems: ListCartItem[]
  chain: ComputedRef<Prefix>
  decimals: ComputedRef<number>
}

export const DEFAULT_FLOOR_PRICE_RATE = 1
const localStorage = useLocalStorage<ListCartItem[]>('listingCart', [])
export const useListingCartStore = defineStore('listingCart', {
  state: (): State => ({
    items: localStorage.value,
    allUnlistedItems: [],
    chain: usePrefix().urlPrefix,
    decimals: useChain().decimals,
  }),
  getters: {
    allItemsInChain: (state): ListCartItem[] =>
      state.items.filter((item) => item.urlPrefix === state.chain),
    itemsInChain(): ListCartItem[] {
      return this.allItemsInChain.filter((item) => !item.discarded)
    },
    count() {
      return this.itemsInChain.length
    },
    incompleteListPrices(): number {
      return this.itemsInChain.filter((item) => !item.listPrice).length
    },
  },
  actions: {
    getItem(id: ID) {
      return this.items.find((item) => item.id === id)
    },
    isItemInCart(id: ID) {
      return existInItemIndex(id, this.items) !== -1
    },
    setItem(payload: ListCartItem) {
      const itemIndex = existInItemIndex(payload.id, this.items)
      if (itemIndex === -1) {
        this.items.push(payload)
        localStorage.value = this.items
      }
    },
    setItemPrice({ id, price }: { id: ID; price?: number }) {
      const itemIndex = existInItemIndex(id, this.items)
      if (itemIndex !== -1) {
        this.items[itemIndex].listPrice = price
        localStorage.value = this.items
      }
    },
    setUnlistedItem(payload: ListCartItem) {
      const itemIndex = existInItemIndex(payload.id, this.allUnlistedItems)
      if (itemIndex === -1) {
        this.allUnlistedItems.push(payload)
      }
    },
    setUnlistedItems(payload: ListCartItem[]) {
      this.allUnlistedItems = payload
    },
    addAllToCart() {
      this.allUnlistedItems.forEach((item) => this.setItem(item))
    },
    setFixedPrice(price: number) {
      this.itemsInChain.forEach((item) => {
        item.listPrice = price
      })
    },
    setFloorPrice(rate = DEFAULT_FLOOR_PRICE_RATE) {
      this.itemsInChain.forEach((item) => {
        const floor = (Number(item.collection.floor) || 0) * +rate.toFixed(2)
        item.listPrice = Number(
          (floor / Math.pow(10, this.decimals)).toFixed(4),
        )
      })
    },
    removeItem(id: ID) {
      const itemIndex = existInItemIndex(id, this.items)
      if (itemIndex !== -1) {
        this.items.splice(itemIndex, 1)
        localStorage.value = this.items
      }
    },
    setItemDiscardedState({ id, discarded }: { id: ID; discarded: boolean }) {
      const itemIndex = existInItemIndex(id, this.items)
      if (itemIndex !== -1) {
        this.items[itemIndex].discarded = discarded
        localStorage.value = this.items
      }
    },
    clearDiscardedItems() {
      this.items = this.items.filter((item) => !item.discarded)
    },
    clearListedItems() {
      localStorage.value = []
      this.items = []
    },
    clear() {
      localStorage.value = []
      this.items = []
      this.allUnlistedItems = []
    },
  },
})

function existInItemIndex(id: string, items: ListCartItem[]) {
  return items.findIndex((item) => item.id === id || item.token?.id === id)
}