clear pdf

internal/modules/pdf/interface.go

@@ -0,0 +1,7 @@
+package pdf
+
+import "evening_detective/internal/models"
+
+type IPDFGenerator interface {
+	CreateTeamsPDF(teams []*models.Team) ([]byte, error)
+}

internal/modules/pdf/service.go

@@ -1,12 +1,11 @@
-package pdf_service
+package pdf
 
 import (
 	"bytes"
+	"embed"
 	"evening_detective/internal/models"
 	"strings"
 
-	"embed"
-
 	"github.com/signintech/gopdf"
 	"github.com/skip2/go-qrcode"
 )
@@ -14,7 +13,13 @@ import (
 //go:embed JetBrainsMono-Medium.ttf
 var f embed.FS
 
-func CreateTeamsPdf(teams []*models.Team) ([]byte, error) {
+type service struct{}
+
+func NewPDFGenerator() IPDFGenerator {
+	return &service{}
+}
+
+func (s *service) CreateTeamsPDF(teams []*models.Team) ([]byte, error) {
 	pdf := &gopdf.GoPdf{}
 	pdf.Start(gopdf.Config{PageSize: *gopdf.PageSizeA4}) // W: 595, H: 842
 	file, err := f.Open("JetBrainsMono-Medium.ttf")

internal/services/services.go

@@ -7,7 +7,7 @@ import (
 	"evening_detective/internal/models"
 	"evening_detective/internal/modules/link"
 	"evening_detective/internal/modules/password"
-	"evening_detective/internal/services/pdf_service"
+	"evening_detective/internal/modules/pdf"
 	"evening_detective/internal/services/story_service"
 	"evening_detective/proto"
 	"fmt"
@@ -23,6 +23,7 @@ type Services struct {
 	storyService      *story_service.StoryService
 	linkService       link.ILinkService
 	passwordGenerator password.IPasswordGenerator
+	pdfGenerator      pdf.IPDFGenerator
 	clientHost        string
 }
 
@@ -31,6 +32,7 @@ func NewServices(
 	storyService *story_service.StoryService,
 	linkService link.ILinkService,
 	passwordGenerator password.IPasswordGenerator,
+	pdfGenerator pdf.IPDFGenerator,
 	clientHost string,
 ) *Services {
 	return &Services{
@@ -38,6 +40,7 @@ func NewServices(
 		storyService:      storyService,
 		linkService:       linkService,
 		passwordGenerator: passwordGenerator,
+		pdfGenerator:      pdfGenerator,
 		clientHost:        clientHost,
 	}
 }
@@ -203,7 +206,7 @@ func (s *Services) DownloadTeamsQrCodesFile(ctx context.Context, req *proto.Down
 	for _, team := range teams {
 		team.Link = s.linkService.GetTeamClientLink(s.clientHost, team.Name, team.Password)
 	}
-	b, err := pdf_service.CreateTeamsPdf(teams)
+	b, err := s.pdfGenerator.CreateTeamsPDF(teams)
 	if err != nil {
 		return nil, err
 	}