invoice.go

package main

import (
	"crypto/tls"
	"fmt"
	"os"
	"strconv"
	"strings"
	"time"

	"github.com/SebastiaanKlippert/go-wkhtmltopdf"
	"github.com/Shopify/gomail"
)

func sendInvoice(id int) error {
	dog, err := getDog(id)
	if err != nil {
		return err
	}

	_, err = generatePdf(dog)
	if err != nil {
		return err
	}

	err = sendEmail(dog)
	if err != nil {
		return err
	}

	return nil
}

func sendInvoices() (status string, err error) {
	emails, err := getEmailQueue()
	if err != nil {
		return "", err
	}

	err = markEmailsInProcess(emails)
	if err != nil {
		return "", err
	}

	for _, email := range emails {
		if err != nil {
			return "", err
		}
		err = sendInvoice(email.DogID)
		if err != nil {
			return "", err
		}

		err = markEmailSent(email.ID)
		if err != nil {
			return "", err
		}
	}

	return "Processed " + strconv.Itoa(len(emails)) + " emails", nil
}

func generatePdf(dog Dog) (string, error) {
	pdfGenerator, err := wkhtmltopdf.NewPDFGenerator()
	if err != nil {
		return "", err
	}

	page := wkhtmltopdf.NewPage(os.Getenv("BASE_URL") + "/invoice/" + strconv.Itoa(dog.ID))

	pdfGenerator.AddPage(page)

	err = pdfGenerator.Create()
	if err != nil {
		return "", err
	}

	invoiceFile := fmt.Sprintf("./public/%s.pdf", getInvoiceNumber(dog))

	err = pdfGenerator.WriteFile(invoiceFile)
	if err != nil {
		return "", err
	}

	return invoiceFile, nil
}

func sendEmail(dog Dog) error {
	invoiceFile := fmt.Sprintf("./public/%s.pdf", getInvoiceNumber(dog))
	smtpPort, err := strconv.Atoi(os.Getenv("SMTP_PORT"))
	if err != nil {
		return fmt.Errorf("error converting SMTP_PORT to int: %s", err)
	}

	d := gomail.NewDialer(
		os.Getenv("SMTP_HOST"),
		smtpPort,
		os.Getenv("SMTP_USER"),
		os.Getenv("SMTP_PASS"),
	)
	d.Timeout = 30 * time.Second
	d.TLSConfig = &tls.Config{InsecureSkipVerify: true}

	ownerFirstName := strings.Split(dog.OwnerName, " ")[0]
	fromFirstName := strings.Split(os.Getenv("FROM_NAME"), " ")[0]

	walkOrWalks := "walk"
	if dog.Quantity > 1 {
		walkOrWalks = "walks"
	}

	m := gomail.NewMessage()
	m.SetHeader("From", fmt.Sprintf("Canine Club<%s>", os.Getenv("SMTP_USER")))
	m.SetHeader("To", fmt.Sprintf("%s <%s>", dog.OwnerName, dog.Email))
	m.SetHeader("Subject", "Canine Club - Invoice for "+dog.Name)
	m.SetBody(
		"text/html",
		"Hi "+ownerFirstName+",<br><br>Please find attached the invoice for "+dog.Name+"'s "+walkOrWalks+" this week.<p style='font-weight:lighter;'>Please use '<b>"+dog.Name+"</b>' as the reference when making payment. Also note that payment is due by "+nextMonday(
			time.Now(),
		).Format("Monday, 2 January 2006")+
			".</p><br>Any questions let me know,<br>Thank you!<br><br>"+fromFirstName+"<br>Canine Club",
	)
	m.Attach(invoiceFile)

	err = d.DialAndSend(m)
	if err != nil {
		return err
	} else {
		return nil
	}
}

func getInvoiceNumber(dog Dog) string {
	prefix := strings.ToUpper(dog.Name[0:3])
	return prefix + time.Now().Format("20060102")
}

func nextMonday(t time.Time) time.Time {
	if t.Weekday() == time.Monday {
		return t.AddDate(0, 0, 7)
	}
	for t.Weekday() != time.Monday {
		t = t.AddDate(0, 0, 1)
	}
	return t
}