import { createMemo, type Accessor } from 'solid-js';
import { type Event as NostrEvent } from 'nostr-tools';
import { createQuery, type CreateQueryResult } from '@tanstack/solid-query';
import timeout from '@/utils/timeout';

import useBatchedEvent from '@/clients/useBatchedEvent';

export type UseEventProps = {
  // TODO リレーURLを考慮したい
  relayUrls: string[];
  eventId: string;
};

export type UseEvent = {
  event: Accessor<NostrEvent | undefined>;
  query: CreateQueryResult<NostrEvent | undefined>;
};

const { exec } = useBatchedEvent<UseEventProps>(() => ({
  generateKey: ({ eventId }: UseEventProps) => eventId,
  mergeFilters: (args: UseEventProps[]) => {
    const eventIds = args.map((arg) => arg.eventId);
    return [{ kinds: [1], ids: eventIds }];
  },
  extractKey: (event: NostrEvent) => event.id,
}));

const useEvent = (propsProvider: () => UseEventProps | null): UseEvent => {
  const props = createMemo(propsProvider);
  const query = createQuery(
    () => ['useEvent', props()] as const,
    ({ queryKey, signal }) => {
      const [, currentProps] = queryKey;
      if (currentProps == null) return undefined;
      return timeout(15000, `useEvent: ${currentProps.eventId}`)(exec(currentProps, signal));
    },
    {
      // a hour
      staleTime: 60 * 60 * 1000,
      cacheTime: 60 * 60 * 1000,
    },
  );

  const event = () => query.data;

  return { event, query };
};

export default useEvent;