import { useEffect, useMemo, useState } from "react";
import { EventKind, NostrLink, RequestBuilder, NoteCollection } from "@snort/system";
import { useRequestBuilder } from "@snort/system-react";

import useLogin from "Hooks/useLogin";
import { useReactions } from "./FeedReactions";

export default function useThreadFeed(link: NostrLink) {
  const [allEvents, setAllEvents] = useState<Array<NostrLink>>([]);
  const pref = useLogin().preferences;

  const sub = useMemo(() => {
    const sub = new RequestBuilder(`thread:${link.id}`);
    sub.withOptions({
      leaveOpen: true,
    });
    sub.withFilter()
      .kinds([EventKind.TextNote])
      .link(link);
    if (allEvents.length > 0) {
      const f = sub
        .withFilter()
        .kinds([EventKind.TextNote]);
      allEvents.forEach(x => f.replyToLink(x));
    }
    return sub;
  }, [allEvents.length, pref]);

  const store = useRequestBuilder(NoteCollection, sub);

  useEffect(() => {
    if (store.data) {
      const mainNotes = store.data?.filter(a => a.kind === EventKind.TextNote || a.kind === EventKind.Polls) ?? [];
      const links = mainNotes.map(a => [
        NostrLink.fromEvent(a), 
        ...a.tags.filter(a => a[0] === "e" || a[0] === "a").map(v => NostrLink.fromTag(v))
      ]).flat();
      setAllEvents(links);
    }
  }, [store.data?.length]);

  const reactions = useReactions(`thread:${link.id}:reactions`, allEvents);

  return {
    thread: store.data ?? [],
    reactions: reactions.data ?? [],
  };
}