import useSWRSubscription from "swr/subscription"
import { getDynamicBuilder, getLookupFn } from "@polkadot-api/metadata-builders"
import { distinct, filter, map, mergeMap } from "rxjs"

import { useUnstableProvider } from "./UnstableProvider"
import { useMetadata } from "./MetadataProvider"

export const useErasTotalStake = ({ eraIndex }) => {
    const { chainHead$, chainId } = useUnstableProvider()
    const metadata = useMetadata()
    const { data: eraTotalStake } = useSWRSubscription(
        chainHead$ && chainId && metadata
            ? ["eraTotalStake", chainHead$, eraIndex, chainId, metadata]
            : null,
        ([_, chainHead$, eraIndex, chainId, metadata], { next }) => {
            const { finalized$, storage$ } = chainHead$
            const subscription = finalized$.pipe(
                filter(Boolean),
                mergeMap((blockInfo) => {
                    const builder = getDynamicBuilder(getLookupFn(metadata))
                    const eraTotalStake = builder.buildStorage("Staking", "ErasTotalStake")
                    return storage$(blockInfo?.hash, "value", () =>
                        eraTotalStake?.keys.enc(eraIndex)
                    ).pipe(
                        filter(Boolean),
                        distinct(),
                        map((value) => eraTotalStake?.value.dec(value))
                    )
                }),
            )
            .subscribe({
                next(eraTotalStake) {
                    next(null, eraTotalStake)
                },
                error: next,
            })
            return () => subscription.unsubscribe()
        }
    )
    return eraTotalStake;
}