/**
 * Copyright (c) TonTech.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 */

import { useEffect } from 'react';
import { useQueryClient } from '@tanstack/react-query';
import { watchBalanceByAddress, hasStreamingProvider, resolveNetwork } from '@ton/appkit';
import type { WatchBalanceByAddressOptions } from '@ton/appkit';
import { handleBalanceUpdate } from '@ton/appkit/queries';

import { useAppKit } from '../../settings';

export type UseWatchBalanceByAddressParameters = Partial<WatchBalanceByAddressOptions>;

/**
 * Hook to watch balance of a specific address in real-time.
 * Automatically updates the TanStack Query cache for `useBalanceByAddress`.
 */
export const useWatchBalanceByAddress = (parameters: UseWatchBalanceByAddressParameters): void => {
    const { address, network, onChange } = parameters;
    const appKit = useAppKit();
    const queryClient = useQueryClient();

    useEffect(() => {
        if (!address) return;

        const resolvedNetwork = resolveNetwork(appKit, network);
        if (!resolvedNetwork || !hasStreamingProvider(appKit, resolvedNetwork)) {
            // eslint-disable-next-line no-console
            console.warn(
                resolvedNetwork
                    ? `No streaming provider available for network: ${resolvedNetwork?.chainId}`
                    : 'No network provided',
            );

            return;
        }

        const addressString = address.toString();

        return watchBalanceByAddress(appKit, {
            address,
            network: resolvedNetwork,
            onChange: (balance) => {
                onChange?.(balance);
                handleBalanceUpdate(queryClient, { address: addressString, network: resolvedNetwork }, balance);
            },
        });
    }, [address, network, appKit, queryClient, onChange]);
};