340 lines
9.4 KiB
Go
340 lines
9.4 KiB
Go
// Copyright (c) 2025-2026 Probo Inc <hello@probo.com>.
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
// PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
package trust
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
|
|
"go.gearno.de/kit/pg"
|
|
"go.probo.inc/probo/pkg/coredata"
|
|
"go.probo.inc/probo/pkg/docgen"
|
|
"go.probo.inc/probo/pkg/gid"
|
|
"go.probo.inc/probo/pkg/html2pdf"
|
|
"go.probo.inc/probo/pkg/mail"
|
|
"go.probo.inc/probo/pkg/page"
|
|
"go.probo.inc/probo/pkg/pdfutils"
|
|
)
|
|
|
|
type (
|
|
DocumentService struct {
|
|
svc *Service
|
|
html2pdfConverter *html2pdf.Converter
|
|
}
|
|
|
|
ErrDocumentArchived struct{}
|
|
)
|
|
|
|
func (e ErrDocumentArchived) Error() string {
|
|
return "cannot access an archived document"
|
|
}
|
|
|
|
func (s *DocumentService) ListForOrganizationId(
|
|
ctx context.Context,
|
|
scope coredata.Scoper,
|
|
organizationID gid.GID,
|
|
cursor *page.Cursor[coredata.DocumentOrderField],
|
|
) (*page.Page[*coredata.Document, coredata.DocumentOrderField], error) {
|
|
var documents coredata.Documents
|
|
|
|
err := s.svc.pg.WithConn(
|
|
ctx,
|
|
func(ctx context.Context, conn pg.Querier) error {
|
|
filter := coredata.NewDocumentTrustCenterFilter()
|
|
|
|
if err := documents.LoadPublishedByOrganizationID(ctx, conn, scope, organizationID, cursor, filter); err != nil {
|
|
return fmt.Errorf("cannot load published documents: %w", err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return page.NewPage(documents, cursor), nil
|
|
}
|
|
|
|
func (s *DocumentService) ExportPDF(
|
|
ctx context.Context,
|
|
scope coredata.Scoper,
|
|
documentID gid.GID,
|
|
email mail.Addr,
|
|
) ([]byte, error) {
|
|
pdfData, err := s.exportPDFData(ctx, scope, documentID)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot export document PDF: %w", err)
|
|
}
|
|
|
|
watermarkedPDF, err := pdfutils.AddConfidentialWithTimestamp(pdfData, email)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot add watermark to PDF: %w", err)
|
|
}
|
|
|
|
return watermarkedPDF, nil
|
|
}
|
|
|
|
func (s *DocumentService) ExportPDFWithoutWatermark(
|
|
ctx context.Context,
|
|
scope coredata.Scoper,
|
|
documentID gid.GID,
|
|
) ([]byte, error) {
|
|
return s.exportPDFData(ctx, scope, documentID)
|
|
}
|
|
|
|
func (s DocumentService) Get(
|
|
ctx context.Context,
|
|
scope coredata.Scoper,
|
|
organizationID gid.GID,
|
|
documentID gid.GID,
|
|
) (*coredata.Document, error) {
|
|
document := &coredata.Document{}
|
|
|
|
err := s.svc.pg.WithConn(
|
|
ctx,
|
|
func(ctx context.Context, conn pg.Querier) error {
|
|
err := document.LoadByID(ctx, conn, scope, documentID)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot load document: %w", err)
|
|
}
|
|
|
|
if document.ArchivedAt != nil {
|
|
return &ErrDocumentArchived{}
|
|
}
|
|
|
|
return nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if document.OrganizationID != organizationID {
|
|
return nil, ErrDocumentNotFound
|
|
}
|
|
|
|
if document.TrustCenterVisibility == coredata.TrustCenterVisibilityNone {
|
|
return nil, ErrDocumentNotVisible
|
|
}
|
|
|
|
return document, nil
|
|
}
|
|
|
|
func (s *DocumentService) exportPDFData(
|
|
ctx context.Context,
|
|
scope coredata.Scoper,
|
|
documentID gid.GID,
|
|
) ([]byte, error) {
|
|
document := &coredata.Document{}
|
|
version := &coredata.DocumentVersion{}
|
|
fileRecord := &coredata.File{}
|
|
|
|
err := s.svc.pg.WithConn(
|
|
ctx,
|
|
func(ctx context.Context, conn pg.Querier) error {
|
|
if err := document.LoadByID(ctx, conn, scope, documentID); err != nil {
|
|
return fmt.Errorf("cannot load document: %w", err)
|
|
}
|
|
|
|
if document.ArchivedAt != nil {
|
|
return &ErrDocumentArchived{}
|
|
}
|
|
|
|
if document.TrustCenterVisibility == coredata.TrustCenterVisibilityNone {
|
|
return fmt.Errorf("document not visible on trust center")
|
|
}
|
|
|
|
if err := version.LoadLatestPublishedVersion(ctx, conn, scope, documentID); err != nil {
|
|
return fmt.Errorf("cannot load latest published document version: %w", err)
|
|
}
|
|
|
|
if version.FileID == nil {
|
|
return nil
|
|
}
|
|
|
|
if err := fileRecord.LoadByID(ctx, conn, scope, *version.FileID); err != nil {
|
|
return fmt.Errorf("cannot load document version file: %w", err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if version.FileID != nil {
|
|
pdfData, err := s.svc.fileManager.GetFileBytes(ctx, fileRecord)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot fetch document PDF file: %w", err)
|
|
}
|
|
|
|
return pdfData, nil
|
|
}
|
|
|
|
// TODO: remove on-the-fly fallback once all published versions have a stored PDF.
|
|
pdfData, err := s.generatePDFOnTheFly(ctx, scope, document, version)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot generate PDF on the fly: %w", err)
|
|
}
|
|
|
|
return pdfData, nil
|
|
}
|
|
|
|
// generatePDFOnTheFly generates a PDF from scratch for versions that don't have
|
|
// a stored file yet. Can be removed once all published versions have been
|
|
// processed by the document PDF worker.
|
|
func (s *DocumentService) generatePDFOnTheFly(
|
|
ctx context.Context,
|
|
scope coredata.Scoper,
|
|
document *coredata.Document,
|
|
version *coredata.DocumentVersion,
|
|
) ([]byte, error) {
|
|
organization := &coredata.Organization{}
|
|
|
|
var approverNames []string
|
|
|
|
err := s.svc.pg.WithConn(
|
|
ctx,
|
|
func(ctx context.Context, conn pg.Querier) error {
|
|
lastQuorum := &coredata.DocumentVersionApprovalQuorum{}
|
|
if err := lastQuorum.LoadLastByDocumentVersionID(ctx, conn, scope, version.ID); err != nil {
|
|
if !errors.Is(err, coredata.ErrResourceNotFound) {
|
|
return fmt.Errorf("cannot load last approval quorum: %w", err)
|
|
}
|
|
} else if lastQuorum.Status == coredata.DocumentVersionApprovalQuorumStatusApproved {
|
|
approvedDecisions := &coredata.DocumentVersionApprovalDecisions{}
|
|
|
|
approvedFilter := coredata.NewDocumentVersionApprovalDecisionFilter(
|
|
coredata.DocumentVersionApprovalDecisionStates{coredata.DocumentVersionApprovalDecisionStateApproved},
|
|
)
|
|
if err := approvedDecisions.LoadByQuorumID(
|
|
ctx,
|
|
conn,
|
|
scope,
|
|
lastQuorum.ID,
|
|
page.NewCursor(
|
|
100,
|
|
nil,
|
|
page.Head,
|
|
page.OrderBy[coredata.DocumentVersionApprovalDecisionOrderField]{
|
|
Field: coredata.DocumentVersionApprovalDecisionOrderFieldCreatedAt,
|
|
Direction: page.OrderDirectionAsc,
|
|
},
|
|
),
|
|
approvedFilter,
|
|
); err != nil {
|
|
return fmt.Errorf("cannot load approved decisions: %w", err)
|
|
}
|
|
|
|
approverProfileIDs := make([]gid.GID, 0, len(*approvedDecisions))
|
|
for _, d := range *approvedDecisions {
|
|
approverProfileIDs = append(approverProfileIDs, d.ApproverID)
|
|
}
|
|
|
|
if len(approverProfileIDs) > 0 {
|
|
profiles := coredata.MembershipProfiles{}
|
|
if err := profiles.LoadByIDs(ctx, conn, scope, approverProfileIDs); err != nil {
|
|
return fmt.Errorf("cannot load approver profiles: %w", err)
|
|
}
|
|
|
|
for _, p := range profiles {
|
|
approverNames = append(approverNames, p.FullName)
|
|
}
|
|
}
|
|
}
|
|
|
|
if err := organization.LoadByID(ctx, conn, scope, document.OrganizationID); err != nil {
|
|
return fmt.Errorf("cannot load organization: %w", err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
classification := docgen.ClassificationSecret
|
|
|
|
switch version.Classification {
|
|
case coredata.DocumentClassificationPublic:
|
|
classification = docgen.ClassificationPublic
|
|
case coredata.DocumentClassificationInternal:
|
|
classification = docgen.ClassificationInternal
|
|
case coredata.DocumentClassificationConfidential:
|
|
classification = docgen.ClassificationConfidential
|
|
}
|
|
|
|
horizontalLogoBase64 := ""
|
|
|
|
if organization.HorizontalLogoFileID != nil {
|
|
fileRecord := &coredata.File{}
|
|
|
|
fileErr := s.svc.pg.WithConn(ctx, func(ctx context.Context, conn pg.Querier) error {
|
|
return fileRecord.LoadByID(ctx, conn, scope, *organization.HorizontalLogoFileID)
|
|
})
|
|
if fileErr == nil {
|
|
base64Data, mimeType, logoErr := s.svc.fileManager.GetFileBase64(ctx, fileRecord)
|
|
if logoErr == nil {
|
|
horizontalLogoBase64 = fmt.Sprintf("data:%s;base64,%s", mimeType, base64Data)
|
|
}
|
|
}
|
|
}
|
|
|
|
docData := docgen.DocumentData{
|
|
Title: version.Title,
|
|
Content: json.RawMessage([]byte(version.Content)),
|
|
Major: version.Major,
|
|
Minor: version.Minor,
|
|
Classification: classification,
|
|
Approvers: approverNames,
|
|
PublishedAt: version.PublishedAt,
|
|
CompanyHorizontalLogoBase64: horizontalLogoBase64,
|
|
}
|
|
|
|
htmlContent, err := docgen.RenderHTML(docData)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot generate HTML: %w", err)
|
|
}
|
|
|
|
cfg := html2pdf.RenderConfig{
|
|
PageFormat: html2pdf.PageFormatA4,
|
|
Orientation: html2pdf.OrientationPortrait,
|
|
MarginTop: html2pdf.NewMarginInches(1.0),
|
|
MarginBottom: html2pdf.NewMarginInches(1.0),
|
|
MarginLeft: html2pdf.NewMarginInches(1.0),
|
|
MarginRight: html2pdf.NewMarginInches(1.0),
|
|
PrintBackground: true,
|
|
Scale: 1.0,
|
|
}
|
|
|
|
pdfReader, err := s.html2pdfConverter.GeneratePDF(ctx, htmlContent, cfg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot generate PDF: %w", err)
|
|
}
|
|
|
|
pdfData, err := io.ReadAll(pdfReader)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("cannot read PDF data: %w", err)
|
|
}
|
|
|
|
return pdfData, nil
|
|
}
|