github.com/benoitkugler/goacve@v0.0.0-20201217100549-151ce6e55dc8/server/core/utils/pdf/pdfutils.go

package pdf

import (
	"fmt"
	"path/filepath"
	"strings"
	"time"

	rd "github.com/benoitkugler/goACVE/server/core/rawdata"

	"github.com/jung-kurt/gofpdf"
)

// ------------------------------------ Helpers ------------------------------------
type rgb [3]int

type CustomPdf struct {
	*gofpdf.Fpdf
	Tr func(string) string
}

// renvoie un pdf prêt à l'emploi (footer et page prête)
func InitPdfDetails(addPage bool, footerOnlyLastPage bool) CustomPdf {
	pdf := gofpdf.New("P", "mm", "A4", "")
	pdf.SetFont("Arial", "", 12)
	pdf.SetMargins(sideMargin, marginTop, -1)
	mypdf := CustomPdf{Fpdf: pdf, Tr: pdf.UnicodeTranslatorFromDescriptor("")}
	if footerOnlyLastPage {
		mypdf.SetFooterFuncLpi(func(lastPage bool) {
			if lastPage {
				mypdf.drawFooter()
			}
		})
	} else {
		mypdf.SetFooterFunc(mypdf.drawFooter)
	}
	if addPage {
		mypdf.AddPage()
	}
	return mypdf
}

func initPdf() CustomPdf {
	return InitPdfDetails(true, true)
}

func (pdf CustomPdf) SetFillColor(c rgb) {
	pdf.Fpdf.SetFillColor(c[0], c[1], c[2])
}

func (pdf CustomPdf) SetDrawColor(c rgb) {
	pdf.Fpdf.SetDrawColor(c[0], c[1], c[2])
}

func (pdf CustomPdf) insertImage(name string, w, h float64) {
	file := filepath.Join(IMAGES_PATH, name)
	pdf.ImageOptions(file, -1, -1, w, h, true, gofpdf.ImageOptions{}, 0, "")
}

func (pdf CustomPdf) AreaWidth() float64 {
	w, _ := pdf.GetPageSize()
	mL, _, mR, _ := pdf.GetMargins()
	return w - mL - mR
}

func (pdf CustomPdf) areaBottomLimit() float64 {
	_, h := pdf.GetPageSize()
	return h - heightFooter - 6
}

func (pdf CustomPdf) LineHeight() float64 {
	_, lineHeight := pdf.GetFontSize()
	return lineHeight + 1
}

func (pdf CustomPdf) CellFormat(w, h float64, txtStr, borderStr, alignStr string, fill bool) {
	pdf.Fpdf.CellFormat(w, h, txtStr, borderStr, 0, alignStr, fill, 0, "")
}

func (pdf CustomPdf) widthNeeded(lines []string) float64 {
	max, addPadding := 0., 5.
	for _, s := range lines {
		w := pdf.GetStringWidth(s)
		if w > max {
			max = w
		}
	}
	return max + addPadding
}

// ------------------------------------------ Header / Footer ------------------------------------------

// Attention, len(lignes) <= 5 sinon ca dépasse !
func (pdf CustomPdf) drawHeaderRight(expediteur []string, h float64) {
	lineHeight := pdf.LineHeight()
	padding := (h - lineHeight*float64(len(expediteur))) / 2
	pdf.SetY(pdf.GetY() + padding)
	pdf.MultiCell(pdf.AreaWidth(), lineHeight, pdf.Tr(strings.Join(expediteur, "\n")), "", "RM", false)
}

// DrawHeader ajoute un header
// `iconWidth` et `iconHeight` peuvent être non nul
// pour forcer une taille d'icône.
func (pdf CustomPdf) DrawHeader(rightHeaderContent []string, iconWidth, iconHeight float64) {
	pdf.SetY(pdf.GetY() + paddingHeaderTop)
	oldY := pdf.GetY()
	pdf.insertImage("app.png", iconWidth, iconHeight)
	y, h := pdf.GetY(), pdf.GetY()-oldY
	pdf.SetY(oldY)
	pdf.drawHeaderRight(rightHeaderContent, h)
	pdf.SetY(y)
}

func (pdf CustomPdf) drawFooter() {
	h := pdf.HTMLBasicNew()
	pdf.SetFontSize(10)
	_, _, mR, _ := pdf.GetMargins()
	rondsWidth, rondsHeight := 70., 6.

	pdf.SetY(-heightFooter)
	pdf.SetFillColor(secondaryColor)
	pdf.SetTextColor(0, 0, 0)
	pdf.CellFormat(0, 17, "", "", "", true)
	pdf.SetXY(-(rondsWidth + mR + 2), -heightFooter-(rondsHeight/2))
	pdf.insertImage("ronds.png", rondsWidth, rondsHeight)
	pdf.SetY(-15)
	pdf.SetLeftMargin(sideMargin + paddingFooter)
	pdf.SetRightMargin(sideMargin + paddingFooter)
	h.Write(4, pdf.Tr(rd.Asso.Title))
	pdf.Ln(-1)
	pdf.SetFontSize(9)
	h.Write(4, pdf.Tr(rd.Asso.Infos))
	pdf.Ln(-1)
	pdf.SetFontSize(7.5)
	h.Write(3, pdf.Tr(rd.Asso.Details))
}

// ------------------------ Zone titre / destinataire ------------------------
func formatAdresse(s string) []string {
	var outs []string
	for _, line := range strings.Split(s, "\n") {
		trim := strings.TrimSpace(line)
		if trim != "" {
			outs = append(outs, trim)
		}
	}
	return outs
}

func (pdf CustomPdf) drawBandeDestinataire(titre string, dest rd.Destinataire) {
	titreWidth, titreHeight, spacingDate := 60., 10., 2.
	y := pdf.GetY() + spacingHeader
	pdf.SetY(y)
	pdf.SetDrawColor(primaryColor)
	pdf.SetLineWidth(0.8)
	pdf.ClipRoundedRect(pdf.GetX(), y, titreWidth, titreHeight, 4, true)
	pdf.CellFormat(titreWidth, titreHeight, pdf.Tr(titre), "", "CM", false)
	pdf.ClipEnd()
	pdf.SetY(y + titreHeight + spacingDate)
	now := time.Now()
	pdf.SetFont("Arial", "I", 10)
	pdf.CellFormat(titreWidth, 5, fmt.Sprintf("Edition au %d/%d/%d", now.Day(), now.Month(), now.Year()), "", "CM", false)

	pdf.SetFont("Arial", "", 13)
	codePostalVille := strings.TrimSpace(dest.CodePostal.String()) + " " + strings.TrimSpace(dest.Ville.ToUpper())
	adresseLines := formatAdresse(dest.Adresse.String())
	lines := append(append([]string{dest.NomPrenom.String()}, adresseLines...), codePostalVille)
	neededWidth, neededHeight := pdf.widthNeeded(lines), pdf.LineHeight()*float64(len(lines))
	x, paddingTop, paddingRight := pdf.AreaWidth()-neededWidth, 2., 3.
	pdf.SetDrawColor(secondaryColor)
	pdf.ClipRoundedRect(x, y, neededWidth+paddingRight, neededHeight+3, 5, true)
	pdf.SetXY(x, y+paddingTop)
	pdf.MultiCell(neededWidth, pdf.LineHeight(), pdf.Tr(strings.Join(lines, "\n")), "", "R", false)
	pdf.ClipEnd()
	pdf.Ln(spacingBandeDestinataire)
	pdf.SetDrawColor(rgb{0, 0, 0})
	pdf.SetLineWidth(0.2)
}