contact.go

package generator

import (
	"bytes"
	b64 "encoding/base64"
	"image"

	"github.com/go-pdf/fpdf"
)

// Contact contact a company informations
type Contact struct {
	Name    string   `json:"name,omitempty" validate:"required,min=1,max=256"`
	Logo    []byte   `json:"logo,omitempty"` // Logo byte array
	Address *Address `json:"address,omitempty"`

	// AddtionnalInfo to append after contact informations. You can use basic html here (bold, italic tags).
	AddtionnalInfo []string `json:"additional_info,omitempty"`
}

// appendContactTODoc append the contact to the document
func (c *Contact) appendContactTODoc(
	x float64,
	y float64,
	fill bool,
	logoAlign string,
	doc *Document,
) float64 {
	doc.pdf.SetXY(x, y)

	// Logo
	if c.Logo != nil {
		// Create filename
		fileName := b64.StdEncoding.EncodeToString([]byte(c.Name))

		// Create reader from logo bytes
		ioReader := bytes.NewReader(c.Logo)

		// Get image format
		_, format, _ := image.DecodeConfig(bytes.NewReader(c.Logo))

		// Register image in pdf
		imageInfo := doc.pdf.RegisterImageOptionsReader(fileName, fpdf.ImageOptions{
			ImageType: format,
		}, ioReader)

		if imageInfo != nil {
			var imageOpt fpdf.ImageOptions
			imageOpt.ImageType = format
			doc.pdf.ImageOptions(fileName, doc.pdf.GetX(), y, 0, 30, false, imageOpt, 0, "")
			doc.pdf.SetY(y + 30)
		}
	}

	// Name
	if fill {
		doc.pdf.SetFillColor(
			doc.Options.GreyBgColor[0],
			doc.Options.GreyBgColor[1],
			doc.Options.GreyBgColor[2],
		)
	} else {
		doc.pdf.SetFillColor(255, 255, 255)
	}

	// Reset x
	doc.pdf.SetX(x)

	// Name rect
	doc.pdf.Rect(x, doc.pdf.GetY(), 70, 8, "F")

	// Set name
	doc.pdf.SetFont(doc.Options.BoldFont, "B", 10)
	doc.pdf.Cell(40, 8, doc.encodeString(c.Name))
	doc.pdf.SetFont(doc.Options.Font, "", 10)

	if c.Address != nil {
		// Address rect
		var addrRectHeight float64 = 17

		if len(c.Address.Address2) > 0 {
			addrRectHeight = addrRectHeight + 5
		}

		if len(c.Address.Country) == 0 {
			addrRectHeight = addrRectHeight - 5
		}

		doc.pdf.Rect(x, doc.pdf.GetY()+9, 70, addrRectHeight, "F")

		// Set address
		doc.pdf.SetFont(doc.Options.Font, "", 10)
		doc.pdf.SetXY(x, doc.pdf.GetY()+10)
		doc.pdf.MultiCell(70, 5, doc.encodeString(c.Address.ToString()), "0", "L", false)
	}

	// Addtionnal info
	if c.AddtionnalInfo != nil {
		doc.pdf.SetXY(x, doc.pdf.GetY())
		doc.pdf.SetFontSize(SmallTextFontSize)
		doc.pdf.SetXY(x, doc.pdf.GetY()+2)

		for _, line := range c.AddtionnalInfo {
			doc.pdf.SetXY(x, doc.pdf.GetY())
			doc.pdf.MultiCell(70, 3, doc.encodeString(line), "0", "L", false)
		}

		doc.pdf.SetXY(x, doc.pdf.GetY())
		doc.pdf.SetFontSize(BaseTextFontSize)
	}

	return doc.pdf.GetY()
}

// appendCompanyContactToDoc append the company contact to the document
func (c *Contact) appendCompanyContactToDoc(doc *Document) float64 {
	x, y, _, _ := doc.pdf.GetMargins()
	return c.appendContactTODoc(x, y, true, "L", doc)
}

// appendCustomerContactToDoc append the customer contact to the document
func (c *Contact) appendCustomerContactToDoc(doc *Document) float64 {
	return c.appendContactTODoc(130, BaseMarginTop+25, true, "R", doc)
}