package wpdf import ( "bytes" "github.com/jung-kurt/gofpdf" ) type WPDFBuilder struct { b *gofpdf.Fpdf tr func(string) string cellHeight float64 cellSpacing float64 fontName PDFFontFamily fontStyle PDFFontStyle fontSize float64 } type PDFMargins struct { Left float64 Top float64 Right float64 } type PDFColor struct { R int G int B int } func NewPDFBuilder(orientation PDFOrientation, size PDFSize, unicode bool) *WPDFBuilder { fpdfbuilder := gofpdf.New(string(orientation), "mm", string(size), "") var tr func(string) string if unicode { tr = fpdfbuilder.UnicodeTranslatorFromDescriptor("") } else { tr = func(s string) string { return s } } b := &WPDFBuilder{ b: fpdfbuilder, tr: tr, cellHeight: 5, cellSpacing: 1, } b.SetMargins(PDFMargins{Left: 15, Top: 25, Right: 15}) // default values b.SetFont(FontHelvetica, Normal, 12) // ensures font is set return b } func (b *WPDFBuilder) FPDF() *gofpdf.Fpdf { return b.b } func (b *WPDFBuilder) SetMargins(v PDFMargins) { b.b.SetMargins(v.Left, v.Top, v.Right) } func (b *WPDFBuilder) AddPage() { b.b.AddPage() } func (b *WPDFBuilder) SetTextColor(cr, cg, cb int) { b.b.SetTextColor(cr, cg, cb) } func (b *WPDFBuilder) GetTextColor() (cr, cg, cb int) { return b.b.GetTextColor() } func (b *WPDFBuilder) SetDrawColor(cr, cg, cb int) { b.b.SetDrawColor(cr, cg, cb) } func (b *WPDFBuilder) GetDrawColor() (cr, cg, cb int) { return b.b.GetDrawColor() } func (b *WPDFBuilder) SetFillColor(cr, cg, cb int) { b.b.SetFillColor(cr, cg, cb) } func (b *WPDFBuilder) GetFillColor() (cr, cg, cb int) { return b.b.GetFillColor() } func (b *WPDFBuilder) SetLineWidth(w float64) { b.b.SetLineWidth(w) } func (b *WPDFBuilder) GetLineWidth() float64 { return b.b.GetLineWidth() } func (b *WPDFBuilder) SetFont(fontName PDFFontFamily, fontStyle PDFFontStyle, fontSize float64) { b.b.SetFont(string(fontName), string(fontStyle), fontSize) b.fontName = fontName b.fontStyle = fontStyle b.fontSize = fontSize b.cellHeight = b.b.PointConvert(fontSize) } func (b *WPDFBuilder) SetCellSpacing(h float64) { b.cellSpacing = h } func (b *WPDFBuilder) Ln(h float64) { b.b.Ln(h) } func (b *WPDFBuilder) Build() ([]byte, error) { buf := new(bytes.Buffer) err := b.b.Output(buf) if err != nil { return nil, err } return buf.Bytes(), nil } func (b *WPDFBuilder) SetX(x float64) { b.b.SetX(x) } func (b *WPDFBuilder) GetX() float64 { return b.b.GetX() } func (b *WPDFBuilder) SetY(y float64) { b.b.SetY(y) } func (b *WPDFBuilder) GetY() float64 { return b.b.GetY() } func (b *WPDFBuilder) SetXY(x float64, y float64) { b.b.SetXY(x, y) } func (b *WPDFBuilder) GetXY() (x float64, y float64) { return b.b.GetXY() } func (b *WPDFBuilder) GetMargins() (left, top, right, bottom float64) { return b.b.GetMargins() } func (b *WPDFBuilder) GetMarginLeft() float64 { v, _, _, _ := b.b.GetMargins() return v } func (b *WPDFBuilder) GetMarginTop() float64 { _, v, _, _ := b.b.GetMargins() return v } func (b *WPDFBuilder) GetMarginRight() float64 { _, _, v, _ := b.b.GetMargins() return v } func (b *WPDFBuilder) GetMarginBottom() float64 { _, _, _, v := b.b.GetMargins() return v } func (b *WPDFBuilder) GetPageSize() (width, height float64) { return b.b.GetPageSize() } func (b *WPDFBuilder) GetPageWidth() float64 { v, _ := b.b.GetPageSize() return v } func (b *WPDFBuilder) GetPageHeight() float64 { _, v := b.b.GetPageSize() return v } func (b *WPDFBuilder) GetWorkAreaWidth() float64 { return b.GetPageWidth() - b.GetMarginLeft() - b.GetMarginRight() } func (b *WPDFBuilder) GetStringWidth(str string) float64 { return b.b.GetStringWidth(str) }