mirrors/homebox
1
0
Fork 1
mirror of https://github.com/hay-kot/homebox.git synced 2025-08-02 15:50:27 +00:00
Code Issues Projects Releases Packages Wiki Activity

pdf generation code clean up

Browse source
This commit is contained in:
icanotc 2023-08-23 04:48:38 -04:00
parent 22db91185d
commit f4d05a7f97
1 changed files with 33 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

73
backend/app/api/handlers/v1/v1_ctrl_qrcode.go
View file

@ -5,16 +5,15 @@ import (
"fmt" "fmt"
"github.com/go-playground/validator/v10" "github.com/go-playground/validator/v10"
"github.com/hay-kot/homebox/backend/internal/core/services" "github.com/hay-kot/homebox/backend/internal/core/services"
"github.com/hay-kot/homebox/backend/internal/web/adapters"
"github.com/hay-kot/httpkit/errchain"
"github.com/signintech/gopdf" "github.com/signintech/gopdf"
"github.com/yeqown/go-qrcode/v2"
"github.com/yeqown/go-qrcode/writer/standard"
"image/png" "image/png"
"io" "io"
"net/http" "net/http"
"github.com/hay-kot/homebox/backend/internal/web/adapters"
"github.com/hay-kot/httpkit/errchain"
"github.com/yeqown/go-qrcode/v2"
"github.com/yeqown/go-qrcode/writer/standard"
_ "embed" _ "embed"
) )
@ -31,6 +30,7 @@ type pageQuery struct {
PrintType string `schema:"printType" validate:"required,oneof=items locations both"` PrintType string `schema:"printType" validate:"required,oneof=items locations both"`
BaseAddr string `schema:"baseAddr" validate:"required"` BaseAddr string `schema:"baseAddr" validate:"required"`
} }
type nopCloser struct { type nopCloser struct {
io.Writer io.Writer
} }
@ -114,7 +114,7 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
return err return err
} }
sus := make([]string, 10) var URLs []string
if *data.Nested { if *data.Nested {
@ -122,34 +122,41 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
if data.PrintType == "items" { if data.PrintType == "items" {
for _, thing := range locations.Items { for _, thing := range locations.Items {
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID encodeStr := data.BaseAddr + "/item/" + thing.ID.String() //concat the url and then UUID
sus = append(sus, encodeStr) URLs = append(URLs, encodeStr)
} }
} else if data.PrintType == "locations" { } else if data.PrintType == "locations" {
for _, thing := range locations.Children { for _, thing := range locations.Children {
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID encodeStr := data.BaseAddr + "/location/" + thing.ID.String() //concat the url and then UUID
sus = append(sus, encodeStr) URLs = append(URLs, encodeStr)
} }
} else if data.PrintType == "both" { } else if data.PrintType == "both" {
for _, thing := range locations.Items { for _, thing := range locations.Items {
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID encodeStr := data.BaseAddr + "/item/" + thing.ID.String() //concat the url and then UUID
sus = append(sus, encodeStr) URLs = append(URLs, encodeStr)
} }
for _, thing := range locations.Children { for _, thing := range locations.Children {
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID encodeStr := data.BaseAddr + "/location/" + thing.ID.String() //concat the url and then UUID
sus = append(sus, encodeStr) URLs = append(URLs, encodeStr)
} }
} }
pdf := gopdf.GoPdf{} pdf := gopdf.GoPdf{}
pdf.Start(gopdf.Config{PageSize: *gopdf.PageSizeA4}) pdf.Start(gopdf.Config{PageSize: *gopdf.PageSizeA4})
pageCounter := 0 pdf.AddPage()
rowCounter := 0
for i, thing := range sus { const qrPerRow = 4
if i%4 == 0 { const qrPerPage = 24
pdf.AddPage()
pageCounter++ for i, thing := range URLs {
if i%qrPerRow == 0 && i != 0 {
rowCounter++
if i%qrPerPage == 0 {
pdf.AddPage()
}
} }
//turn each thing into a image buffer that gopdf accepts //turn each thing into a image buffer that gopdf accepts
qrc, err := qrcode.New(thing) qrc, err := qrcode.New(thing)
if err != nil { if err != nil {
@ -158,11 +165,12 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
logo, err := png.Decode(bytes.NewReader(qrcodeLogo)) logo, err := png.Decode(bytes.NewReader(qrcodeLogo))
if err != nil { if err != nil {
panic(err) return err
} }
img := bytes.NewBuffer(nil) img := bytes.NewBuffer(nil)
wr := standard.NewWithWriter(nopCloser{Writer: img}, standard.WithQRWidth(40), standard.WithLogoImage(logo)) wr := standard.NewWithWriter(nopCloser{Writer: img}, standard.WithLogoImage(logo), standard.WithBorderWidth(5), standard.WithQRWidth(7))
err = qrc.Save(wr) err = qrc.Save(wr)
if err != nil { if err != nil {
return err return err
@ -175,19 +183,20 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
} }
x := (i % 4) * 150 x := (i % 4) * 150
y := (pageCounter-1)*300 + 50 y := rowCounter * 150
err = pdf.ImageByHolder(imgBytes, float64(x), float64(y), nil) err = pdf.ImageByHolder(imgBytes, float64(x), float64(y), nil)
if err != nil { if err != nil {
return err return err
} }
} }
// Return the concatenated QR code images as a response // Return the concatenated QR code images as a response
w.Header().Set("Content-Type", "application/pdf") w.Header().Set("Content-Type", "application/pdf")
w.Header().Set("Content-Disposition", "inline; filename=qrCodes.pdf") w.Header().Set("Content-Disposition", "inline; filename=qrCodes.pdf")
fmt.Printf("%v", sus) fmt.Printf("%v", URLs)
//_, err = w.Write([]byte(fmt.Sprintf("%v", sus))) //_, err = w.Write([]byte(fmt.Sprintf("%v", URLs)))
_, err = pdf.WriteTo(w) _, err = pdf.WriteTo(w)
if err != nil { if err != nil {
return err return err
@ -195,19 +204,3 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
return err return err
} }
} }
//log.Debug().Msg(encodeStr)
//
//qrc, err := qrcode.New(encodeStr) //create QR code obj from screen
//if err != nil {
// return err
//}
//
//qrWriter := standard.NewWithWriter(toWriteCloser, standard.WithLogoImage(image))
//
//err = qrc.Save(qrWriter)
//if err != nil {
// return err
//}
//
//_, err = w.Write(qrCodeBuffer.Bytes())