import {
  getConnection,
  watchConnection,
  type Config,
  type GetConnectionReturnType,
  type ResolvedRegister,
} from "@wagmi/core";
import { createEffect, createSignal, onCleanup, type Accessor } from "solid-js";

import type { ConfigParameter } from "../types/index.js";
import { useConfig } from "./useConfig.js";

export type SolidConnectionParameters<config extends Config = Config> =
  ConfigParameter<config>;

export type UseConnectionParameters<config extends Config = Config> = Accessor<
  SolidConnectionParameters<config>
>;

export type SolidConnectionReturnType<config extends Config = Config> =
  GetConnectionReturnType<config>;

export type UseConnectionReturnType<config extends Config = Config> = Accessor<
  SolidConnectionReturnType<config>
>;

/** https://wagmi.sh/solid/api/primitives/useConnection */
export function useConnection<
  config extends Config = ResolvedRegister["config"]
>(
  parameters: UseConnectionParameters<config> = () => ({})
): UseConnectionReturnType<config> {
  const config = useConfig(parameters);

  const [connection, setConnection] = createSignal(getConnection(config()));

  createEffect(() => {
    const _config = config();
    setConnection(() => getConnection(_config));

    const unsubscribe = watchConnection(_config, {
      onChange(data) {
        setConnection(() => data);
      },
    });
    onCleanup(() => unsubscribe());
  });

  return connection;
}