package main

import (
	"context"
	"fmt"
	"log"
	"time"

	"github.com/msuny-c/mailtm"
)

func main() {
	ctx := context.Background()

	cli, err := mailtm.New(
		mailtm.WithUserAgent("mailtm-go/1.0"),
	)
	if err != nil { log.Fatal(err) }

	doms, err := cli.ListDomains(ctx, 1)
	if err != nil { log.Fatal(err) }
	addr := fmt.Sprintf("tester@%s", doms.Member[0].Domain)

	acc, err := cli.CreateAccount(ctx, addr, "strong-password-123")
	if err != nil { log.Fatal(err) }

	tok, err := cli.Token(ctx, acc.Address, "strong-password-123")
	if err != nil { log.Fatal(err) }

	authed := cli.WithToken(tok.Token)

	// Wait up to 30s for the first message via polling
	msg, err := authed.WaitForFirstMessage(ctx, mailtm.WaitOptions{
		Timeout:      30 * time.Second,
		PollInterval: 2 * time.Second,
	})
	if err != nil { log.Fatal(err) }
	fmt.Println("Got message:", msg.Subject)

	// Download raw EML
	f, _ := os.Create("message.eml")
	defer f.Close()
	_ = authed.DownloadByURL(ctx, msg.DownloadURL, f)
}