137 lines
3.6 KiB
Go
137 lines
3.6 KiB
Go
package pdf
|
|
|
|
import (
|
|
"bytes"
|
|
"math"
|
|
"strings"
|
|
|
|
"github.com/jung-kurt/gofpdf"
|
|
|
|
"oss.terrastruct.com/d2/d2renderers/d2fonts"
|
|
"oss.terrastruct.com/d2/d2themes"
|
|
"oss.terrastruct.com/d2/d2themes/d2themescatalog"
|
|
"oss.terrastruct.com/d2/lib/color"
|
|
)
|
|
|
|
type GoFPDF struct {
|
|
pdf *gofpdf.Fpdf
|
|
}
|
|
|
|
func Init() *GoFPDF {
|
|
newGofPDF := gofpdf.NewCustom(&gofpdf.InitType{
|
|
UnitStr: "in",
|
|
})
|
|
|
|
newGofPDF.AddUTF8FontFromBytes("source", "", d2fonts.FontFaces[d2fonts.SourceSansPro.Font(0, d2fonts.FONT_STYLE_REGULAR)])
|
|
newGofPDF.AddUTF8FontFromBytes("source", "B", d2fonts.FontFaces[d2fonts.SourceSansPro.Font(0, d2fonts.FONT_STYLE_BOLD)])
|
|
newGofPDF.SetAutoPageBreak(false, 0)
|
|
newGofPDF.SetLineWidth(0.05)
|
|
newGofPDF.SetMargins(0, 0, 0)
|
|
|
|
fpdf := GoFPDF{
|
|
pdf: newGofPDF,
|
|
}
|
|
|
|
return &fpdf
|
|
}
|
|
|
|
func (g *GoFPDF) GetFillRGB(themeID int64, fill string) (color.RGB, error) {
|
|
if fill == "" || strings.ToLower(fill) == "transparent" {
|
|
return color.RGB{
|
|
Red: 255,
|
|
Green: 255,
|
|
Blue: 255,
|
|
}, nil
|
|
}
|
|
|
|
rgb := color.Name2RGB(fill)
|
|
if (rgb != color.RGB{}) {
|
|
return rgb, nil
|
|
}
|
|
|
|
if color.IsThemeColor(fill) {
|
|
theme := d2themescatalog.Find(themeID)
|
|
fill = d2themes.ResolveThemeColor(theme, fill)
|
|
}
|
|
|
|
return color.Hex2RGB(fill)
|
|
}
|
|
|
|
func (g *GoFPDF) AddPDFPage(png []byte, boardPath []string, themeID int64, fill string) error {
|
|
var opt gofpdf.ImageOptions
|
|
opt.ImageType = "png"
|
|
imageInfo := g.pdf.RegisterImageOptionsReader(strings.Join(boardPath, "/"), opt, bytes.NewReader(png))
|
|
if g.pdf.Err() {
|
|
return g.pdf.Error()
|
|
}
|
|
imageWidth := imageInfo.Width() / 2
|
|
imageHeight := imageInfo.Height() / 2
|
|
|
|
// calculate page dimensions
|
|
var pageWidth float64
|
|
var pageHeight float64
|
|
|
|
g.pdf.SetFont("source", "B", 14)
|
|
pathString := strings.Join(boardPath, " / ")
|
|
headerMargin := 0.3
|
|
headerWidth := g.pdf.GetStringWidth(pathString) + 2*headerMargin
|
|
|
|
minPageDimension := 6.0
|
|
pageWidth = math.Max(math.Max(minPageDimension, imageWidth), headerWidth)
|
|
pageHeight = math.Max(minPageDimension, imageHeight)
|
|
|
|
fillRGB, err := g.GetFillRGB(themeID, fill)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Add page
|
|
headerHeight := 0.75
|
|
g.pdf.AddPageFormat("", gofpdf.SizeType{Wd: pageWidth, Ht: pageHeight + headerHeight})
|
|
|
|
// Draw header
|
|
g.pdf.SetFillColor(int(fillRGB.Red), int(fillRGB.Green), int(fillRGB.Blue))
|
|
g.pdf.Rect(0, 0, pageWidth, pageHeight+headerHeight, "F")
|
|
if fillRGB.IsLight() {
|
|
g.pdf.SetTextColor(10, 15, 37) // steel-900
|
|
} else {
|
|
g.pdf.SetTextColor(255, 255, 255)
|
|
}
|
|
g.pdf.SetFont("source", "", 14)
|
|
|
|
// Draw board path prefix
|
|
var prefixWidth float64
|
|
prefixPath := boardPath[:len(boardPath)-1]
|
|
if len(prefixPath) > 0 {
|
|
prefix := strings.Join(boardPath[:len(boardPath)-1], " / ") + " / "
|
|
prefixWidth = g.pdf.GetStringWidth(prefix)
|
|
|
|
g.pdf.SetXY(headerMargin, 0)
|
|
g.pdf.CellFormat(prefixWidth, headerHeight, prefix, "", 0, "", false, 0, "")
|
|
}
|
|
|
|
// Draw board name
|
|
boardName := boardPath[len(boardPath)-1]
|
|
g.pdf.SetFont("source", "B", 14)
|
|
g.pdf.SetXY(prefixWidth+headerMargin, 0)
|
|
g.pdf.CellFormat(pageWidth-prefixWidth-headerMargin, headerHeight, boardName, "", 0, "", false, 0, "")
|
|
|
|
// Draw image
|
|
g.pdf.ImageOptions(strings.Join(boardPath, "/"), (pageWidth-imageWidth)/2, headerHeight+(pageHeight-imageHeight)/2, imageWidth, imageHeight, false, opt, 0, "")
|
|
|
|
// Draw header/img seperator
|
|
g.pdf.SetXY(headerMargin, headerHeight)
|
|
g.pdf.SetLineWidth(0.01)
|
|
if fillRGB.IsLight() {
|
|
g.pdf.SetDrawColor(10, 15, 37) // steel-900
|
|
} else {
|
|
g.pdf.SetDrawColor(255, 255, 255)
|
|
}
|
|
g.pdf.CellFormat(pageWidth-(headerMargin*2), 0.01, "", "T", 0, "", false, 0, "")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (g *GoFPDF) Export(outputPath string) error {
|
|
return g.pdf.OutputFileAndClose(outputPath)
|
|
}
|