import { defineStore } from 'pinia'
import { ref, computed } from 'vue'
import type { WalletState } from '@/interface'

export const useNeptuneStore = defineStore('neptune', () => {
    const defaultNetwork = (import.meta.env.VITE_NODE_NETWORK || 'mainnet') as 'mainnet' | 'testnet'

    // ===== STATE =====
    const wallet = ref<WalletState>({
        seedPhrase: null,
        password: null,
        receiverId: null,
        viewKey: null,
        spendingKey: null,
        address: null,
        network: defaultNetwork,
        balance: null,
        pendingBalance: null,
        utxos: [],
    })

    const keystorePath = ref<null | string>(null)

    // ===== SETTERS =====

    const setSeedPhrase = (seedPhrase: string[] | null) => {
        wallet.value.seedPhrase = seedPhrase
    }

    const setPassword = (password: string | null) => {
        wallet.value.password = password
    }

    const setReceiverId = (receiverId: string | null) => {
        wallet.value.receiverId = receiverId
    }

    const setViewKey = (viewKey: string | null) => {
        wallet.value.viewKey = viewKey
    }

    const setSpendingKey = (spendingKey: string | null) => {
        wallet.value.spendingKey = spendingKey
    }

    const setAddress = (address: string | null) => {
        wallet.value.address = address
    }

    const setNetwork = (network: 'mainnet' | 'testnet') => {
        wallet.value.network = network
    }

    const setBalance = (balance: string | null) => {
        wallet.value.balance = balance
    }

    const setPendingBalance = (pendingBalance: string | null) => {
        wallet.value.pendingBalance = pendingBalance
    }

    const setUtxos = (utxos: any[]) => {
        wallet.value.utxos = utxos
    }

    const setWallet = (walletData: Partial<WalletState>) => {
        wallet.value = { ...wallet.value, ...walletData }
    }

    const setKeystorePath = (path: string | null) => {
        keystorePath.value = path
    }

    const clearWallet = () => {
        wallet.value = {
            seedPhrase: null,
            password: null,
            receiverId: null,
            viewKey: null,
            spendingKey: null,
            address: null,
            network: defaultNetwork,
            balance: null,
            pendingBalance: null,
            utxos: [],
        }
    }

    // ===== GETTERS =====
    const getWallet = computed(() => wallet.value)
    const getSeedPhrase = computed(() => wallet.value.seedPhrase)
    const getSeedPhraseString = computed(() => wallet.value.seedPhrase?.join(' ') || '')
    const getPassword = computed(() => wallet.value.password)
    const getReceiverId = computed(() => wallet.value.receiverId)
    const getViewKey = computed(() => wallet.value.viewKey)
    const getSpendingKey = computed(() => wallet.value.spendingKey)
    const getAddress = computed(() => wallet.value.address)
    const getNetwork = computed(() => wallet.value.network)
    const getBalance = computed(() => wallet.value.balance)
    const getPendingBalance = computed(() => wallet.value.pendingBalance)
    const getUtxos = computed(() => wallet.value.utxos)
    const hasWallet = computed(() => wallet.value.address !== null)
    const getKeystorePath = computed(() => keystorePath.value)
    return {
        getWallet,
        getSeedPhrase,
        getSeedPhraseString,
        getPassword,
        getReceiverId,
        getViewKey,
        getSpendingKey,
        getAddress,
        getNetwork,
        getBalance,
        getPendingBalance,
        getUtxos,
        hasWallet,
        getKeystorePath,
        setSeedPhrase,
        setPassword,
        setReceiverId,
        setViewKey,
        setSpendingKey,
        setAddress,
        setNetwork,
        setBalance,
        setPendingBalance,
        setUtxos,
        setWallet,
        setKeystorePath,
        clearWallet,
    }
})