import { NDKEvent } from '@nostr-dev-kit/ndk';
import { useQuery } from '@tanstack/react-query';

import { useNDK } from '@libs/ndk/provider';
import { useStorage } from '@libs/storage/provider';

import { toRawEvent } from '@utils/rawEvent';

export function useEvent(id: string, embed?: string) {
  const { db } = useStorage();
  const { ndk } = useNDK();
  const { status, data } = useQuery(
    ['event', id],
    async () => {
      // return embed event (nostr.band api) or repost
      if (embed) {
        const event: NDKEvent = JSON.parse(embed);
        return event;
      }
      // get event from db
      const dbEvent = await db.getEventByID(id);
      if (dbEvent) return dbEvent;

      // get event from relay if event in db not present
      const event = await ndk.fetchEvent(id);
      if (!event) throw new Error(`Event not found: ${id.toString()}`);

      let root: string;
      let reply: string;

      if (event.tags?.[0]?.[0] === 'e' && !event.tags?.[0]?.[3]) {
        root = event.tags[0][1];
      } else {
        root = event.tags.find((el) => el[3] === 'root')?.[1];
        reply = event.tags.find((el) => el[3] === 'reply')?.[1];
      }

      const rawEvent = toRawEvent(event);
      await db.createEvent(
        event.id,
        JSON.stringify(rawEvent),
        event.pubkey,
        event.kind,
        root,
        reply,
        event.created_at
      );

      return event;
    },
    {
      enabled: !!ndk,
      refetchOnMount: false,
      refetchOnWindowFocus: false,
    }
  );

  return { status, data };
}