mirror of
https://github.com/hay-kot/homebox.git
synced 2025-08-03 00:00:27 +00:00
initial location based pdf generation
This commit is contained in:
parent
449650e72c
commit
22db91185d
1 changed files with 99 additions and 31 deletions
|
@ -2,8 +2,9 @@ package v1
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"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/rs/zerolog/log"
|
|
||||||
"github.com/signintech/gopdf"
|
"github.com/signintech/gopdf"
|
||||||
"image/png"
|
"image/png"
|
||||||
"io"
|
"io"
|
||||||
|
@ -25,8 +26,17 @@ type query struct {
|
||||||
Data string `schema:"data" validate:"required,max=4296"`
|
Data string `schema:"data" validate:"required,max=4296"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type nestedQuery struct {
|
type pageQuery struct {
|
||||||
Nested bool `schema:"nested" validate:"required,"`
|
Nested *bool `schema:"nested" validate:"required"`
|
||||||
|
PrintType string `schema:"printType" validate:"required,oneof=items locations both"`
|
||||||
|
BaseAddr string `schema:"baseAddr" validate:"required"`
|
||||||
|
}
|
||||||
|
type nopCloser struct {
|
||||||
|
io.Writer
|
||||||
|
}
|
||||||
|
|
||||||
|
func (nopCloser) Close() error {
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// HandleGenerateQRCode godoc
|
// HandleGenerateQRCode godoc
|
||||||
|
@ -79,67 +89,125 @@ func (ctrl *V1Controller) HandleGenerateQRCode() errchain.HandlerFunc {
|
||||||
|
|
||||||
func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFunc {
|
func (ctrl *V1Controller) HandleGenerateQRCodeForLocations() errchain.HandlerFunc {
|
||||||
return func(w http.ResponseWriter, r *http.Request) error {
|
return func(w http.ResponseWriter, r *http.Request) error {
|
||||||
data, err := adapters.DecodeQuery[nestedQuery](r)
|
|
||||||
|
data, err := adapters.DecodeQuery[pageQuery](r) //decode the query with the pageQuery struct
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
routeUUID, err := ctrl.routeUUID(r, "location_id")
|
routeUUID, err := ctrl.routeUUID(r, "location_id") //from /qrcode/{location_id}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
err = validator.New().Var(routeUUID.String(), "required,uuid") //validate the UUID, so .Get below won't throw errors
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println(routeUUID)
|
||||||
|
|
||||||
auth := services.NewContext(r.Context())
|
auth := services.NewContext(r.Context())
|
||||||
locations, err := ctrl.repo.Locations.GetOneByGroup(auth, auth.GID, routeUUID)
|
locations, err := ctrl.repo.Locations.Get(auth, routeUUID)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
image, err := png.Decode(bytes.NewReader(qrcodeLogo))
|
sus := make([]string, 10)
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
if *data.Nested {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
toWriteCloser := struct {
|
if data.PrintType == "items" {
|
||||||
io.Writer
|
for _, thing := range locations.Items {
|
||||||
io.Closer
|
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID
|
||||||
}{
|
sus = append(sus, encodeStr)
|
||||||
Writer: w,
|
}
|
||||||
Closer: io.NopCloser(nil),
|
} 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)
|
||||||
|
}
|
||||||
|
} 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)
|
||||||
|
}
|
||||||
|
for _, thing := range locations.Children {
|
||||||
|
encodeStr := data.BaseAddr + "/" + thing.ID.String() //concat the url and then UUID
|
||||||
|
sus = append(sus, encodeStr)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var qrCodeBuffer bytes.Buffer
|
|
||||||
|
|
||||||
pdf := gopdf.GoPdf{}
|
pdf := gopdf.GoPdf{}
|
||||||
pdf.Start(gopdf.Config{PageSize: *gopdf.PageSizeA4})
|
pdf.Start(gopdf.Config{PageSize: *gopdf.PageSizeA4})
|
||||||
|
pageCounter := 0
|
||||||
|
|
||||||
if data.Nested {
|
for i, thing := range sus {
|
||||||
|
if i%4 == 0 {
|
||||||
}
|
pdf.AddPage()
|
||||||
|
pageCounter++
|
||||||
for _, location := range locations.Items {
|
}
|
||||||
encodeStr := r.URL.String() + location.ID.String() //concat the url and then UUID
|
//turn each thing into a image buffer that gopdf accepts
|
||||||
log.Debug().Msg(encodeStr)
|
qrc, err := qrcode.New(thing)
|
||||||
|
|
||||||
qrc, err := qrcode.New(encodeStr) //create QR code obj from screen
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
qrWriter := standard.NewWithWriter(toWriteCloser, standard.WithLogoImage(image))
|
logo, err := png.Decode(bytes.NewReader(qrcodeLogo))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
err = qrc.Save(qrWriter)
|
img := bytes.NewBuffer(nil)
|
||||||
|
wr := standard.NewWithWriter(nopCloser{Writer: img}, standard.WithQRWidth(40), standard.WithLogoImage(logo))
|
||||||
|
err = qrc.Save(wr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = w.Write(qrCodeBuffer.Bytes())
|
//add the image buffer to the pdf
|
||||||
|
imgBytes, err := gopdf.ImageHolderByBytes(img.Bytes())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
x := (i % 4) * 150
|
||||||
|
y := (pageCounter-1)*300 + 50
|
||||||
|
err = pdf.ImageByHolder(imgBytes, float64(x), float64(y), nil)
|
||||||
|
if err != nil {
|
||||||
|
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", "image/png")
|
w.Header().Set("Content-Type", "application/pdf")
|
||||||
w.Header().Set("Content-Disposition", "attachment; filename=qrcodes.png")
|
w.Header().Set("Content-Disposition", "inline; filename=qrCodes.pdf")
|
||||||
|
|
||||||
|
fmt.Printf("%v", sus)
|
||||||
|
//_, err = w.Write([]byte(fmt.Sprintf("%v", sus)))
|
||||||
|
_, err = pdf.WriteTo(w)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
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())
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue