import type {BrowserContext} from "@playwright/test"
import type {StampedEvent} from "@welshman/util"
import type {TestUser} from "../keys"

// The function playwright installs on window for the shim below to call into. A binding is the
// only way across: the keys live in node, and a signer built in the page would be a different
// thing from the one seeding signs with.
const TEST_NIP07_KEY = "__TEST_NIP07__"

type Nip07Call =
  | {method: "getPublicKey"}
  | {method: "signEvent"; template: StampedEvent}
  | {method: "encrypt" | "decrypt"; scheme: "nip04" | "nip44"; pubkey: string; message: string}

/**
 * A NIP-07 provider backed by a test identity's real signer, so an extension login produces
 * signatures the relays accept — including the NIP-42 auth events a members-only relay demands.
 *
 * Both halves have to be installed before the page navigates: LogIn.svelte reads `window.nostr`
 * while it renders to decide whether to offer the button at all.
 */
export const injectNip07 = async (context: BrowserContext, user: TestUser) => {
  await context.exposeBinding(TEST_NIP07_KEY, (source, call: Nip07Call) => {
    if (call.method === "getPublicKey") {
      return user.pubkey
    }

    if (call.method === "signEvent") {
      return user.signer.sign(call.template)
    }

    const {encrypt, decrypt} = call.scheme === "nip04" ? user.signer.nip04 : user.signer.nip44

    return call.method === "encrypt"
      ? encrypt(call.pubkey, call.message)
      : decrypt(call.pubkey, call.message)
  })

  await context.addInitScript(key => {
    const call = (request: Nip07Call) => Reflect.get(window, key)(request)

    Object.assign(window, {
      nostr: {
        getPublicKey: () => call({method: "getPublicKey"}),
        signEvent: (template: StampedEvent) => call({method: "signEvent", template}),
        nip04: {
          encrypt: (pubkey: string, message: string) =>
            call({method: "encrypt", scheme: "nip04", pubkey, message}),
          decrypt: (pubkey: string, message: string) =>
            call({method: "decrypt", scheme: "nip04", pubkey, message}),
        },
        nip44: {
          encrypt: (pubkey: string, message: string) =>
            call({method: "encrypt", scheme: "nip44", pubkey, message}),
          decrypt: (pubkey: string, message: string) =>
            call({method: "decrypt", scheme: "nip44", pubkey, message}),
        },
      },
    })
  }, TEST_NIP07_KEY)
}