mirror of
https://github.com/hay-kot/homebox.git
synced 2025-08-02 15:50:27 +00:00
pdf generation code clean up
This commit is contained in:
parent
22db91185d
commit
f4d05a7f97
1 changed files with 33 additions and 40 deletions
|
@ -5,16 +5,15 @@ import (
|
|||
"fmt"
|
||||
"github.com/go-playground/validator/v10"
|
||||
"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/yeqown/go-qrcode/v2"
|
||||
"github.com/yeqown/go-qrcode/writer/standard"
|
||||
"image/png"
|
||||
"io"
|
||||
"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"
|
||||
)
|
||||
|
||||
|
@ -31,6 +30,7 @@ type pageQuery struct {
|
|||
PrintType string `schema:"printType" validate:"required,oneof=items locations both"`
|
||||
BaseAddr string `schema:"baseAddr" validate:"required"`
|
||||
}
|
||||
|
||||
type nopCloser struct {
|
||||
io.Writer
|
||||
}
|
||||
|
@ -114,7 +114,7 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
|
|||
return err
|
||||
}
|
||||
|
||||
sus := make([]string, 10)
|
||||
var URLs []string
|
||||
|
||||
if *data.Nested {
|
||||
|
||||
|
@ -122,34 +122,41 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
|
|||
|
||||
if data.PrintType == "items" {
|
||||
for _, thing := range locations.Items {
|
||||
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID
|
||||
sus = append(sus, encodeStr)
|
||||
encodeStr := data.BaseAddr + "/item/" + thing.ID.String() //concat the url and then UUID
|
||||
URLs = append(URLs, encodeStr)
|
||||
}
|
||||
} else if data.PrintType == "locations" {
|
||||
for _, thing := range locations.Children {
|
||||
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID
|
||||
sus = append(sus, encodeStr)
|
||||
encodeStr := data.BaseAddr + "/location/" + thing.ID.String() //concat the url and then UUID
|
||||
URLs = append(URLs, encodeStr)
|
||||
}
|
||||
} else if data.PrintType == "both" {
|
||||
for _, thing := range locations.Items {
|
||||
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID
|
||||
sus = append(sus, encodeStr)
|
||||
encodeStr := data.BaseAddr + "/item/" + thing.ID.String() //concat the url and then UUID
|
||||
URLs = append(URLs, encodeStr)
|
||||
}
|
||||
for _, thing := range locations.Children {
|
||||
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID
|
||||
sus = append(sus, encodeStr)
|
||||
encodeStr := data.BaseAddr + "/location/" + thing.ID.String() //concat the url and then UUID
|
||||
URLs = append(URLs, encodeStr)
|
||||
}
|
||||
}
|
||||
|
||||
pdf := gopdf.GoPdf{}
|
||||
pdf.Start(gopdf.Config{PageSize: *gopdf.PageSizeA4})
|
||||
pageCounter := 0
|
||||
|
||||
for i, thing := range sus {
|
||||
if i%4 == 0 {
|
||||
pdf.AddPage()
|
||||
pageCounter++
|
||||
rowCounter := 0
|
||||
|
||||
const qrPerRow = 4
|
||||
const qrPerPage = 24
|
||||
|
||||
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
|
||||
qrc, err := qrcode.New(thing)
|
||||
if err != nil {
|
||||
|
@ -158,11 +165,12 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
|
|||
|
||||
logo, err := png.Decode(bytes.NewReader(qrcodeLogo))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
return err
|
||||
}
|
||||
|
||||
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)
|
||||
if err != nil {
|
||||
return err
|
||||
|
@ -175,19 +183,20 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
|
|||
}
|
||||
|
||||
x := (i % 4) * 150
|
||||
y := (pageCounter-1)*300 + 50
|
||||
y := rowCounter * 150
|
||||
err = pdf.ImageByHolder(imgBytes, float64(x), float64(y), nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// Return the concatenated QR code images as a response
|
||||
w.Header().Set("Content-Type", "application/pdf")
|
||||
w.Header().Set("Content-Disposition", "inline; filename=qrCodes.pdf")
|
||||
|
||||
fmt.Printf("%v", sus)
|
||||
//_, err = w.Write([]byte(fmt.Sprintf("%v", sus)))
|
||||
fmt.Printf("%v", URLs)
|
||||
//_, err = w.Write([]byte(fmt.Sprintf("%v", URLs)))
|
||||
_, err = pdf.WriteTo(w)
|
||||
if err != nil {
|
||||
return err
|
||||
|
@ -195,19 +204,3 @@ func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFun
|
|||
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())
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue