import type { NostrEvent } from "nostr-tools";
import { nip57 } from "nostr-tools";
import type { NDKTag } from "../events";
import { NDKEvent } from "../events";
import type { NDK } from "../ndk";
import type { NDKSigner } from "../signers";
import type { NDKUser } from "../user";
import type { NDKLnUrlData } from "./ln.js";

export async function generateZapRequest(
    target: NDKEvent | NDKUser,
    ndk: NDK,
    data: NDKLnUrlData,
    pubkey: string,
    amount: number, // amount to zap in millisatoshis
    relays: string[],
    comment?: string,
    tags?: NDKTag[],
    signer?: NDKSigner
): Promise<NDKEvent | null> {
    const zapEndpoint = data.callback;
    const zapRequest = nip57.makeZapRequest({
        profile: pubkey,

        // set the event to null since nostr-tools doesn't support nip-33 zaps
        event: null,
        amount,
        comment: comment || "",
        relays: relays.slice(0, 4),
    });

    // add the event tag if it exists; this supports both 'e' and 'a' tags
    if (target instanceof NDKEvent) {
        const tags = target.referenceTags();
        const nonPTags = tags.filter((tag) => tag[0] !== "p");
        zapRequest.tags.push(...nonPTags);
    }

    zapRequest.tags.push(["lnurl", zapEndpoint]);

    const event = new NDKEvent(ndk, zapRequest as NostrEvent);
    if (tags) {
        event.tags = event.tags.concat(tags);
    }

    const eTaggedEvents = new Set<string>();
    const aTaggedEvents = new Set<string>();

    for (const tag of event.tags) {
        if (tag[0] === "e") {
            eTaggedEvents.add(tag[1]);
        } else if (tag[0] === "a") {
            aTaggedEvents.add(tag[1]);
        }
    }

    if (eTaggedEvents.size > 1) throw new Error("Only one e-tag is allowed");
    if (aTaggedEvents.size > 1) throw new Error("Only one a-tag is allowed");

    // make sure we only have one `p` tag
    event.tags = event.tags.filter((tag) => tag[0] !== "p");
    event.tags.push(["p", pubkey]);

    await event.sign(signer);

    return event;
}