package internal

import (
	"github.com/boombuler/barcode/code128"
	"github.com/boombuler/barcode/qr"
	"github.com/johnfercher/maroto/internal/fpdf"
	"github.com/johnfercher/maroto/pkg/props"
	"github.com/jung-kurt/gofpdf/contrib/barcode"
)

// Code is the abstraction which deals of how to add QrCodes or Barcode in a PDF.
type Code interface {
	AddQr(code string, cell Cell, prop props.Rect)
	AddBar(code string, cell Cell, prop props.Barcode) (err error)
	AddDataMatrix(code string, cell Cell, prop props.Rect)
}

type code struct {
	pdf  fpdf.Fpdf
	math Math
}

// NewCode create a Code.
func NewCode(pdf fpdf.Fpdf, math Math) *code {
	return &code{
		pdf,
		math,
	}
}

// AddDataMatrix creates a DataMatrix code inside a cell.
func (s *code) AddDataMatrix(code string, cell Cell, prop props.Rect) {
	key := barcode.RegisterDataMatrix(s.pdf, code)

	var x, y, w, h float64
	if prop.Center {
		x, y, w, h = s.math.GetRectCenterColProperties(cell.Width, cell.Width, cell.Width, cell.Height, cell.X, prop.Percent)
	} else {
		x, y, w, h = s.math.GetRectNonCenterColProperties(cell.Width, cell.Width, cell.Width, cell.Height, cell.X, prop)
	}
	barcode.Barcode(s.pdf, key, x, y+cell.Y, w, h, false)
}

// AddQr create a QrCode inside a cell.
func (s *code) AddQr(code string, cell Cell, prop props.Rect) {
	key := barcode.RegisterQR(s.pdf, code, qr.H, qr.Unicode)

	var x, y, w, h float64
	if prop.Center {
		x, y, w, h = s.math.GetRectCenterColProperties(cell.Width, cell.Width, cell.Width, cell.Height, cell.X, prop.Percent)
	} else {
		x, y, w, h = s.math.GetRectNonCenterColProperties(cell.Width, cell.Width, cell.Width, cell.Height, cell.X, prop)
	}

	barcode.Barcode(s.pdf, key, x, y+cell.Y, w, h, false)
}

// AddBar create a Barcode inside a cell.
func (s *code) AddBar(code string, cell Cell, prop props.Barcode) (err error) {
	bcode, err := code128.Encode(code)
	if err != nil {
		return
	}

	heightPercentFromWidth := prop.Proportion.Height / prop.Proportion.Width
	var x, y, w, h float64
	if prop.Center {
		x, y, w, h = s.math.GetRectCenterColProperties(cell.Width, cell.Width*heightPercentFromWidth, cell.Width,
			cell.Height, cell.X, prop.Percent)
	} else {
		rectProps := props.Rect{Left: prop.Left, Top: prop.Top, Center: prop.Center, Percent: prop.Percent}
		x, y, w, h = s.math.GetRectNonCenterColProperties(cell.Width, cell.Width*heightPercentFromWidth, cell.Width,
			cell.Height, cell.X, rectProps)
	}

	barcode.Barcode(s.pdf, barcode.Register(bcode), x, y+cell.Y, w, h, false)
	return
}