mail.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/google/uuid"
	"math/rand"
	"net/http"
	"strings"
	"time"
)

const (
	BASE_URL    = "https://www.1secmail.com/api/v1/"
	CHECK_EMAIL = BASE_URL + "?action=getMessages&login=%s&domain=%s"
	READ_EMAIL  = BASE_URL + "?action=readMessage&login=%s&domain=%s&id=%d"
)

type Mail struct {
	Id      int    `json:"id"`
	From    string `json:"from"`
	Subject string `json:"subject"`
	Date    string `json:"date"`
}

type Attachment struct {
	Filename string `json:"filename"`
	Size     int    `json:"size"`
	Type     string `json:"contentType"`
}

type MailBody struct {
	Mail
	Attachments []Attachment `json:"attachments"`
	Body        string       `json:"body"`
	TextBody    string       `json:"textBody"`
	HtmlBody    string       `json:"htmlBody"`
}

var Domains = []string{
	"1secmail.com",
	"1secmail.net",
	"1secmail.org",
	"kzcvv.com",
	"qiott.com",
	"wuuvo.com",
	"icznn.com",
	"ezztt.com",
}

func GetEmail() string {
	return uuid.New().String() + "@" + Domains[rand.Intn(len(Domains))]
}

func GetMessages(email string) []*Mail {
	var mails []*Mail

	req, err := http.NewRequest("GET", fmt.Sprintf(CHECK_EMAIL, strings.Split(email, "@")[0], strings.Split(email, "@")[1]), nil)
	if err != nil {
		return mails
	}

	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		return mails
	}

	err = json.NewDecoder(resp.Body).Decode(&mails)
	if err != nil {
		return mails
	}

	return mails
}

// ReadMessage will read the first message that matches the given function
// If no messages match the function, it will wait a second and try again
// This mean that this function will block until a message is found
func ReadMessage(ctx context.Context, t time.Duration, email string, fn func(*Mail) bool) *MailBody {
	timeout, cancel := context.WithTimeout(ctx, t)
	defer cancel()

	for {
		select {
		case <-timeout.Done():
			return nil
		default:
			mails := GetMessages(email)
			for _, mail := range mails {
				if fn(mail) {
					req, _ := http.NewRequest("GET", fmt.Sprintf(READ_EMAIL, strings.Split(email, "@")[0], strings.Split(email, "@")[1], mail.Id), nil)
					resp, _ := http.DefaultClient.Do(req)

					var mailBody *MailBody
					json.NewDecoder(resp.Body).Decode(&mailBody)
					return mailBody
				}
			}
			time.Sleep(time.Second)
		}
	}
}