import {
  EndpointInterface,
  Schema,
  FetchFunction,
  NI,
} from '@data-client/core';

import useController from './useController.js';
import useFocusEffect from './useFocusEffect.native.js';

/**
 * Keeps a resource fresh by subscribing to updates.
 * @see https://dataclient.io/docs/api/useSubscription
 */
export default function useSubscription<
  E extends EndpointInterface<
    FetchFunction,
    Schema | undefined,
    undefined | false
  >,
>(endpoint: E, ...args: readonly [...Parameters<E>] | readonly [null]) {
  const controller = useController();

  const key = args[0] !== null ? endpoint.key(...args) : '';

  useFocusEffect(
    () => {
      if (!key) return;
      // typescript cannot infer truthy key means args is not null
      const cleanedArgs = args as Parameters<E>;

      controller.subscribe(endpoint, ...cleanedArgs);
      return () => {
        controller.unsubscribe(endpoint, ...cleanedArgs);
      };
    },
    [controller, key],
    true,
  );
}