Serve email static assets from object store

Signed-off-by: Émile Ré <emile@getprobo.com>
This commit is contained in:
Émile Ré
2026-02-04 14:32:33 +04:00
parent 09b7f2bb24
commit 27821632c6
16 changed files with 240 additions and 103 deletions

View File

Before

Width:  |  Height:  |  Size: 1.8 KiB

After

Width:  |  Height:  |  Size: 1.8 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 74 KiB

View File

@@ -16,22 +16,46 @@ package emails
import ( import (
"bytes" "bytes"
"context"
"embed" "embed"
"errors"
"fmt" "fmt"
htmltemplate "html/template" htmltemplate "html/template"
"io/fs"
"mime"
"net/url" "net/url"
"path/filepath"
texttemplate "text/template" texttemplate "text/template"
"time" "time"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/service/s3"
"go.probo.inc/probo/pkg/baseurl" "go.probo.inc/probo/pkg/baseurl"
"go.probo.inc/probo/pkg/filevalidation"
) )
//go:embed dist //go:embed dist
var Templates embed.FS var Templates embed.FS
var (
//go:embed assets
staticAssets embed.FS
staticAssetsValidator = filevalidation.NewValidator(
filevalidation.WithMaxFileSize(5*1024*1024),
filevalidation.WithCategories(
filevalidation.CategoryImage,
filevalidation.CategoryVideo,
),
)
)
type StaticAssetURLs map[string]string
type ( type (
PresenterConfig struct { PresenterConfig struct {
BaseURL string BaseURL string
PoweredByLogoURL string
SenderCompanyName string SenderCompanyName string
SenderCompanyWebsiteURL string SenderCompanyWebsiteURL string
SenderCompanyLogoURL string SenderCompanyLogoURL string
@@ -40,8 +64,8 @@ type (
PresenterVariables struct { PresenterVariables struct {
// Static variables // Static variables
BaseOrigin string
BaseURL string BaseURL string
PoweredByLogoURL string
SenderCompanyName string SenderCompanyName string
SenderCompanyWebsiteURL string SenderCompanyWebsiteURL string
SenderCompanyLogoURL string SenderCompanyLogoURL string
@@ -58,27 +82,22 @@ type (
} }
) )
func DefaultPresenterConfig(baseURL string) PresenterConfig { func DefaultPresenterConfig(baseURL string, staticAssetURLs StaticAssetURLs) PresenterConfig {
return PresenterConfig{ return PresenterConfig{
BaseURL: baseURL, BaseURL: baseURL,
PoweredByLogoURL: staticAssetURLs["probo-gray-small.png"],
SenderCompanyName: "Probo", SenderCompanyName: "Probo",
SenderCompanyWebsiteURL: "https://www.getprobo.com", SenderCompanyWebsiteURL: "https://www.getprobo.com",
SenderCompanyLogoURL: baseurl.MustParse(baseURL).AppendPath("/logos/probo.png").MustString(), SenderCompanyLogoURL: staticAssetURLs["probo.png"],
SenderCompanyHeadquarterAddress: "Probo Inc, 490 Post St, STE 640, San Francisco, CA, 94102, US", SenderCompanyHeadquarterAddress: "Probo Inc, 490 Post St, STE 640, San Francisco, CA, 94102, US",
} }
} }
func NewPresenterFromConfig(cfg PresenterConfig, fullName string) *Presenter { func NewPresenterFromConfig(cfg PresenterConfig, fullName string) *Presenter {
baseURL := baseurl.MustParse(cfg.BaseURL)
baseOrigin := url.URL{
Scheme: baseURL.Scheme(),
Host: baseURL.Host(),
}
return &Presenter{ return &Presenter{
variables: PresenterVariables{ variables: PresenterVariables{
BaseOrigin: baseOrigin.String(),
BaseURL: cfg.BaseURL, BaseURL: cfg.BaseURL,
PoweredByLogoURL: cfg.PoweredByLogoURL,
SenderCompanyName: cfg.SenderCompanyName, SenderCompanyName: cfg.SenderCompanyName,
SenderCompanyWebsiteURL: cfg.SenderCompanyWebsiteURL, SenderCompanyWebsiteURL: cfg.SenderCompanyWebsiteURL,
SenderCompanyLogoURL: cfg.SenderCompanyLogoURL, SenderCompanyLogoURL: cfg.SenderCompanyLogoURL,
@@ -89,13 +108,102 @@ func NewPresenterFromConfig(cfg PresenterConfig, fullName string) *Presenter {
} }
} }
func NewPresenter(baseURL string, fullName string) *Presenter { func NewPresenter(baseURL string, staticAssetURLs StaticAssetURLs, fullName string) *Presenter {
return NewPresenterFromConfig( return NewPresenterFromConfig(
DefaultPresenterConfig(baseURL), DefaultPresenterConfig(baseURL, staticAssetURLs),
fullName, fullName,
) )
} }
func GenerateStaticAssetURLs(ctx context.Context, s3Client *s3.Client, bucket string) (StaticAssetURLs, error) {
assetURLs := make(map[string]string)
subFS, err := fs.Sub(staticAssets, "assets")
if err != nil {
return nil, fmt.Errorf("cannot create subtree file system: %w", err)
}
err = fs.WalkDir(subFS, ".", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if d.IsDir() {
return nil
}
info, err := d.Info()
if err != nil {
if errors.Is(err, fs.ErrNotExist) {
return nil
}
return fmt.Errorf("cannot get dir entry info: %w", err)
}
ext := filepath.Ext(info.Name())
mimeType := mime.TypeByExtension(ext)
if err := staticAssetsValidator.Validate(info.Name(), mimeType, info.Size()); err != nil {
return fmt.Errorf("cannot validate file: %w", err)
}
file, err := subFS.Open(path)
if err != nil {
return err
}
defer func() { _ = file.Close() }()
_, err = s3Client.PutObject(
ctx,
&s3.PutObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(path),
Body: file,
Metadata: map[string]string{
"type": "static-email-asset",
},
ContentType: aws.String(mimeType),
},
)
if err != nil {
return fmt.Errorf("cannot upload file to S3: %w", err)
}
presignClient := s3.NewPresignClient(s3Client)
encodedFilename := url.QueryEscape(info.Name())
contentDisposition := fmt.Sprintf("attachment; filename=%q; filename*=UTF-8''%s",
encodedFilename, encodedFilename)
presignedReq, err := presignClient.PresignGetObject(
ctx,
&s3.GetObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(path),
ResponseCacheControl: aws.String("max-age=3600, public"),
ResponseContentDisposition: &contentDisposition,
},
func(opts *s3.PresignOptions) {
opts.Expires = 7 * 24 * time.Hour
},
)
if err != nil {
return fmt.Errorf("cannot presign GetObject request: %w", err)
}
assetURLs[path] = presignedReq.URL
return nil
})
if err != nil {
return nil, fmt.Errorf("cannot generate asset URLs: %w", err)
}
return assetURLs, nil
}
const ( const (
subjectConfirmEmail = "Confirm your email address" subjectConfirmEmail = "Confirm your email address"
subjectPasswordReset = "Reset your password" subjectPasswordReset = "Reset your password"

View File

@@ -5,7 +5,7 @@ export function ProboLogo() {
return ( return (
<Img <Img
style={{verticalAlign: "middle", height: "100%"}} style={{verticalAlign: "middle", height: "100%"}}
src="{{.BaseOrigin}}/logos/probo-gray-small.png" src="{{.PoweredByLogoURL}}"
alt="Probo" alt="Probo"
height="16" height="16"
/> />

View File

@@ -116,7 +116,7 @@ func (s AccountService) ChangeEmail(ctx context.Context, identityID gid.GID, req
return fmt.Errorf("cannot update identity: %w", err) return fmt.Errorf("cannot update identity: %w", err)
} }
emailPresenter := emails.NewPresenter(s.baseURL, identity.FullName) emailPresenter := emails.NewPresenter(s.baseURL, s.emailStaticAssetURLs, identity.FullName)
subject, textBody, htmlBody, err := emailPresenter.RenderConfirmEmail("/auth/verify-email", confirmationToken) subject, textBody, htmlBody, err := emailPresenter.RenderConfirmEmail("/auth/verify-email", confirmationToken)
if err != nil { if err != nil {

View File

@@ -290,7 +290,7 @@ func (s AuthService) SendPasswordResetInstructionByEmail(
return fmt.Errorf("cannot load identity: %w", err) return fmt.Errorf("cannot load identity: %w", err)
} }
emailPresenter := emails.NewPresenter(s.baseURL, identity.FullName) emailPresenter := emails.NewPresenter(s.baseURL, s.emailStaticAssetURLs, identity.FullName)
subject, textBody, htmlBody, err := emailPresenter.RenderPasswordReset( subject, textBody, htmlBody, err := emailPresenter.RenderPasswordReset(
"/auth/reset-password", "/auth/reset-password",
@@ -406,7 +406,7 @@ func (s AuthService) CreateIdentityWithPassword(
return nil, nil, fmt.Errorf("cannot generate confirmation token: %w", err) return nil, nil, fmt.Errorf("cannot generate confirmation token: %w", err)
} }
emailPresenter := emails.NewPresenter(s.baseURL, req.FullName) emailPresenter := emails.NewPresenter(s.baseURL, s.emailStaticAssetURLs, req.FullName)
subject, textBody, htmlBody, err := emailPresenter.RenderConfirmEmail("/auth/verify-email", confirmationToken) subject, textBody, htmlBody, err := emailPresenter.RenderConfirmEmail("/auth/verify-email", confirmationToken)
if err != nil { if err != nil {
@@ -569,7 +569,7 @@ func (s AuthService) SendMagicLink(ctx context.Context, req *SendMagicLinkReques
return fmt.Errorf("cannot load organization: %w", err) return fmt.Errorf("cannot load organization: %w", err)
} }
emailPresenterCfg := emails.DefaultPresenterConfig(s.baseURL) emailPresenterCfg := emails.DefaultPresenterConfig(s.baseURL, s.emailStaticAssetURLs)
if req.CompliancePageID != nil { if req.CompliancePageID != nil {
var err error var err error
emailPresenterCfg, err = s.CompliancePageService.EmailPresenterConfig(ctx, *req.CompliancePageID) emailPresenterCfg, err = s.CompliancePageService.EmailPresenterConfig(ctx, *req.CompliancePageID)

View File

@@ -91,7 +91,7 @@ func (s *CompliancePageService) EmailPresenterConfig(ctx context.Context, compli
organization = &coredata.Organization{} organization = &coredata.Organization{}
customDomain *coredata.CustomDomain customDomain *coredata.CustomDomain
logoFile = &coredata.File{} logoFile = &coredata.File{}
emailPresenterCfg = emails.DefaultPresenterConfig(s.baseURL) emailPresenterCfg = emails.DefaultPresenterConfig(s.baseURL, s.emailStaticAssetURLs)
) )
scope := coredata.NewScopeFromObjectID(compliancePageID) scope := coredata.NewScopeFromObjectID(compliancePageID)

View File

@@ -464,7 +464,7 @@ func (s *OrganizationService) InviteMember(
return fmt.Errorf("cannot generate invitation token: %w", err) return fmt.Errorf("cannot generate invitation token: %w", err)
} }
emailPresenter := emails.NewPresenter(s.baseURL, identity.FullName) emailPresenter := emails.NewPresenter(s.baseURL, s.emailStaticAssetURLs, identity.FullName)
subject, textBody, htmlBody, err := emailPresenter.RenderInvitation( subject, textBody, htmlBody, err := emailPresenter.RenderInvitation(
"/auth/signup-from-invitation", "/auth/signup-from-invitation",

View File

@@ -11,6 +11,7 @@ import (
"go.gearno.de/kit/log" "go.gearno.de/kit/log"
"go.gearno.de/kit/pg" "go.gearno.de/kit/pg"
"go.opentelemetry.io/otel/trace" "go.opentelemetry.io/otel/trace"
"go.probo.inc/probo/packages/emails"
"go.probo.inc/probo/pkg/baseurl" "go.probo.inc/probo/pkg/baseurl"
"go.probo.inc/probo/pkg/connector" "go.probo.inc/probo/pkg/connector"
"go.probo.inc/probo/pkg/coredata" "go.probo.inc/probo/pkg/coredata"
@@ -28,6 +29,7 @@ type (
pg *pg.Client pg *pg.Client
fm *filemanager.Service fm *filemanager.Service
hp *passwdhash.Profile hp *passwdhash.Profile
emailStaticAssetURLs emails.StaticAssetURLs
encryptionKey cipher.EncryptionKey encryptionKey cipher.EncryptionKey
baseURL string baseURL string
tokenSecret string tokenSecret string
@@ -82,6 +84,7 @@ func NewService(
pgClient *pg.Client, pgClient *pg.Client,
fm *filemanager.Service, fm *filemanager.Service,
hp *passwdhash.Profile, hp *passwdhash.Profile,
emailStaticAssetURLs emails.StaticAssetURLs,
cfg Config, cfg Config,
) (*Service, error) { ) (*Service, error) {
if cfg.Bucket == "" { if cfg.Bucket == "" {
@@ -104,6 +107,7 @@ func NewService(
pg: pgClient, pg: pgClient,
fm: fm, fm: fm,
hp: hp, hp: hp,
emailStaticAssetURLs: emailStaticAssetURLs,
baseURL: cfg.BaseURL.String(), baseURL: cfg.BaseURL.String(),
tokenSecret: cfg.TokenSecret, tokenSecret: cfg.TokenSecret,
disableSignup: cfg.DisableSignup, disableSignup: cfg.DisableSignup,

View File

@@ -543,7 +543,7 @@ func (s *DocumentService) SendSigningNotifications(
return fmt.Errorf("cannot create signing request token: %w", err) return fmt.Errorf("cannot create signing request token: %w", err)
} }
emailPresenter := emails.NewPresenter(s.svc.baseURL, people.FullName) emailPresenter := emails.NewPresenter(s.svc.baseURL, s.svc.emailStaticAssetURLs, people.FullName)
subject, textBody, htmlBody, err := emailPresenter.RenderDocumentSigning( subject, textBody, htmlBody, err := emailPresenter.RenderDocumentSigning(
"/documents/signing-requests", "/documents/signing-requests",
@@ -1823,7 +1823,7 @@ func (s *DocumentService) SendExportEmail(
return fmt.Errorf("cannot generate download URL: %w", err) return fmt.Errorf("cannot generate download URL: %w", err)
} }
emailPresenter := emails.NewPresenter(s.svc.baseURL, recipientName) emailPresenter := emails.NewPresenter(s.svc.baseURL, s.svc.emailStaticAssetURLs, recipientName)
subject, textBody, htmlBody, err := emailPresenter.RenderDocumentExport( subject, textBody, htmlBody, err := emailPresenter.RenderDocumentExport(
downloadURL, downloadURL,

View File

@@ -254,17 +254,17 @@ func (s FrameworkService) Export(
return fmt.Errorf("cannot load evidence file: %w", err) return fmt.Errorf("cannot load evidence file: %w", err)
} }
object, err := s.svc.s3.GetObject( object, err := s.svc.s3.GetObject(
ctx, ctx,
&s3.GetObjectInput{ &s3.GetObjectInput{
Bucket: aws.String(s.svc.bucket), Bucket: aws.String(s.svc.bucket),
Key: aws.String(evidence_file.FileKey), Key: aws.String(evidence_file.FileKey),
}, },
) )
if err != nil { if err != nil {
return fmt.Errorf("cannot download evidence: %w", err) return fmt.Errorf("cannot download evidence: %w", err)
} }
defer func() { _ = object.Body.Close() }() defer func() { _ = object.Body.Close() }()
w, err := archive.Create(fmt.Sprintf("%s/%s/%s/%s", framework.Name, control.SectionTitle, measure.Name, evidence_file.FileName)) w, err := archive.Create(fmt.Sprintf("%s/%s/%s/%s", framework.Name, control.SectionTitle, measure.Name, evidence_file.FileName))
if err != nil { if err != nil {
@@ -800,7 +800,7 @@ func (s FrameworkService) SendExportEmail(
return fmt.Errorf("cannot generate download URL: %w", err) return fmt.Errorf("cannot generate download URL: %w", err)
} }
emailPresenter := emails.NewPresenter(s.svc.baseURL, recipientName) emailPresenter := emails.NewPresenter(s.svc.baseURL, s.svc.emailStaticAssetURLs, recipientName)
subject, textBody, htmlBody, err := emailPresenter.RenderFrameworkExport( subject, textBody, htmlBody, err := emailPresenter.RenderFrameworkExport(
downloadURL, downloadURL,

View File

@@ -23,6 +23,7 @@ import (
"go.gearno.de/kit/log" "go.gearno.de/kit/log"
"go.gearno.de/kit/pg" "go.gearno.de/kit/pg"
"go.gearno.de/x/ref" "go.gearno.de/x/ref"
"go.probo.inc/probo/packages/emails"
"go.probo.inc/probo/pkg/agents" "go.probo.inc/probo/pkg/agents"
"go.probo.inc/probo/pkg/certmanager" "go.probo.inc/probo/pkg/certmanager"
"go.probo.inc/probo/pkg/coredata" "go.probo.inc/probo/pkg/coredata"
@@ -49,18 +50,19 @@ type ExportService interface {
type ( type (
Service struct { Service struct {
pg *pg.Client pg *pg.Client
s3 *s3.Client s3 *s3.Client
bucket string bucket string
encryptionKey cipher.EncryptionKey encryptionKey cipher.EncryptionKey
baseURL string baseURL string
tokenSecret string tokenSecret string
agentConfig agents.Config agentConfig agents.Config
html2pdfConverter *html2pdf.Converter html2pdfConverter *html2pdf.Converter
acmeService *certmanager.ACMEService acmeService *certmanager.ACMEService
fileManager *filemanager.Service fileManager *filemanager.Service
logger *log.Logger logger *log.Logger
slack *slack.Service slack *slack.Service
emailStaticAssetURLs emails.StaticAssetURLs
} }
TenantService struct { TenantService struct {
@@ -73,6 +75,7 @@ type (
tokenSecret string tokenSecret string
agent *agents.Agent agent *agents.Agent
fileManager *filemanager.Service fileManager *filemanager.Service
emailStaticAssetURLs emails.StaticAssetURLs
Frameworks *FrameworkService Frameworks *FrameworkService
Measures *MeasureService Measures *MeasureService
Tasks *TaskService Tasks *TaskService
@@ -106,7 +109,7 @@ type (
ProcessingActivities *ProcessingActivityService ProcessingActivities *ProcessingActivityService
DataProtectionImpactAssessments *DataProtectionImpactAssessmentService DataProtectionImpactAssessments *DataProtectionImpactAssessmentService
TransferImpactAssessments *TransferImpactAssessmentService TransferImpactAssessments *TransferImpactAssessmentService
StatesOfApplicability *StateOfApplicabilityService StatesOfApplicability *StateOfApplicabilityService
Files *FileService Files *FileService
CustomDomains *CustomDomainService CustomDomains *CustomDomainService
SlackMessages *slack.SlackMessageService SlackMessages *slack.SlackMessageService
@@ -128,6 +131,7 @@ func NewService(
logger *log.Logger, logger *log.Logger,
slackService *slack.Service, slackService *slack.Service,
iamService *iam.Service, iamService *iam.Service,
emailStaticAssetURLs emails.StaticAssetURLs,
) (*Service, error) { ) (*Service, error) {
if bucket == "" { if bucket == "" {
return nil, fmt.Errorf("bucket is required") return nil, fmt.Errorf("bucket is required")
@@ -136,18 +140,19 @@ func NewService(
iamService.Authorizer.RegisterPolicySet(ProboPolicySet()) iamService.Authorizer.RegisterPolicySet(ProboPolicySet())
svc := &Service{ svc := &Service{
pg: pgClient, pg: pgClient,
s3: s3Client, s3: s3Client,
bucket: bucket, bucket: bucket,
encryptionKey: encryptionKey, encryptionKey: encryptionKey,
baseURL: baseURL, baseURL: baseURL,
tokenSecret: tokenSecret, tokenSecret: tokenSecret,
agentConfig: agentConfig, agentConfig: agentConfig,
html2pdfConverter: html2pdfConverter, html2pdfConverter: html2pdfConverter,
acmeService: acmeService, acmeService: acmeService,
fileManager: fileManagerService, fileManager: fileManagerService,
logger: logger, logger: logger,
slack: slackService, slack: slackService,
emailStaticAssetURLs: emailStaticAssetURLs,
} }
return svc, nil return svc, nil
@@ -155,15 +160,16 @@ func NewService(
func (s *Service) WithTenant(tenantID gid.TenantID) *TenantService { func (s *Service) WithTenant(tenantID gid.TenantID) *TenantService {
tenantService := &TenantService{ tenantService := &TenantService{
pg: s.pg, pg: s.pg,
s3: s.s3, s3: s.s3,
bucket: s.bucket, bucket: s.bucket,
encryptionKey: s.encryptionKey, encryptionKey: s.encryptionKey,
baseURL: s.baseURL, baseURL: s.baseURL,
scope: coredata.NewScope(tenantID), scope: coredata.NewScope(tenantID),
tokenSecret: s.tokenSecret, tokenSecret: s.tokenSecret,
agent: agents.NewAgent(nil, s.agentConfig), agent: agents.NewAgent(nil, s.agentConfig),
fileManager: s.fileManager, fileManager: s.fileManager,
emailStaticAssetURLs: s.emailStaticAssetURLs,
} }
tenantService.Frameworks = &FrameworkService{ tenantService.Frameworks = &FrameworkService{

View File

@@ -609,7 +609,7 @@ func (s *TrustCenterService) EmailPresenterConfig(ctx context.Context, complianc
organization = &coredata.Organization{} organization = &coredata.Organization{}
customDomain *coredata.CustomDomain customDomain *coredata.CustomDomain
logoFile = &coredata.File{} logoFile = &coredata.File{}
emailPresenterCfg = emails.DefaultPresenterConfig(s.svc.baseURL) emailPresenterCfg = emails.DefaultPresenterConfig(s.svc.baseURL, s.svc.emailStaticAssetURLs)
) )
scope := coredata.NewScopeFromObjectID(compliancePageID) scope := coredata.NewScopeFromObjectID(compliancePageID)

View File

@@ -29,6 +29,7 @@ import (
"sync" "sync"
"time" "time"
"go.probo.inc/probo/packages/emails"
pemutil "go.probo.inc/probo/pkg/crypto/pem" pemutil "go.probo.inc/probo/pkg/crypto/pem"
"github.com/aws/aws-sdk-go-v2/service/s3" "github.com/aws/aws-sdk-go-v2/service/s3"
@@ -292,11 +293,21 @@ func (impl *Implm) Run(
} }
} }
emailStaticAssetURLs, err := emails.GenerateStaticAssetURLs(
ctx,
s3Client,
impl.cfg.AWS.Bucket,
)
if err != nil {
return fmt.Errorf("cannot generate email static asset URLs: %w", err)
}
iamService, err := iam.NewService( iamService, err := iam.NewService(
ctx, ctx,
pgClient, pgClient,
fileManagerService, fileManagerService,
hp, hp,
emailStaticAssetURLs,
iam.Config{ iam.Config{
DisableSignup: impl.cfg.Auth.DisableSignup, DisableSignup: impl.cfg.Auth.DisableSignup,
InvitationTokenValidity: time.Duration(impl.cfg.Auth.InvitationConfirmationTokenValidity) * time.Second, InvitationTokenValidity: time.Duration(impl.cfg.Auth.InvitationConfirmationTokenValidity) * time.Second,
@@ -376,6 +387,7 @@ func (impl *Implm) Run(
l.Named("probo"), l.Named("probo"),
slackService, slackService,
iamService, iamService,
emailStaticAssetURLs,
) )
if err != nil { if err != nil {
return fmt.Errorf("cannot create probo service: %w", err) return fmt.Errorf("cannot create probo service: %w", err)
@@ -393,6 +405,7 @@ func (impl *Implm) Run(
fileManagerService, fileManagerService,
l, l,
slackService, slackService,
emailStaticAssetURLs,
) )
serverHandler, err := server.NewServer( serverHandler, err := server.NewServer(

View File

@@ -22,6 +22,7 @@ import (
"github.com/aws/aws-sdk-go-v2/service/s3" "github.com/aws/aws-sdk-go-v2/service/s3"
"go.gearno.de/kit/log" "go.gearno.de/kit/log"
"go.gearno.de/kit/pg" "go.gearno.de/kit/pg"
"go.probo.inc/probo/packages/emails"
"go.probo.inc/probo/pkg/coredata" "go.probo.inc/probo/pkg/coredata"
"go.probo.inc/probo/pkg/crypto/cipher" "go.probo.inc/probo/pkg/crypto/cipher"
"go.probo.inc/probo/pkg/filemanager" "go.probo.inc/probo/pkg/filemanager"
@@ -35,18 +36,19 @@ import (
type ( type (
Service struct { Service struct {
pg *pg.Client pg *pg.Client
s3 *s3.Client s3 *s3.Client
bucket string bucket string
proboSvc *probo.Service proboSvc *probo.Service
encryptionKey cipher.EncryptionKey encryptionKey cipher.EncryptionKey
slackSigningSecret string slackSigningSecret string
baseURL string baseURL string
iam *iam.Service iam *iam.Service
html2pdfConverter *html2pdf.Converter html2pdfConverter *html2pdf.Converter
fileManager *filemanager.Service fileManager *filemanager.Service
logger *log.Logger logger *log.Logger
slack *slack.Service slack *slack.Service
emailStaticAssetURLs emails.StaticAssetURLs
} }
TenantService struct { TenantService struct {
@@ -61,6 +63,7 @@ type (
html2pdfConverter *html2pdf.Converter html2pdfConverter *html2pdf.Converter
fileManager *filemanager.Service fileManager *filemanager.Service
logger *log.Logger logger *log.Logger
emailStaticAssetURLs emails.StaticAssetURLs
TrustCenters *TrustCenterService TrustCenters *TrustCenterService
Documents *DocumentService Documents *DocumentService
Audits *AuditService Audits *AuditService
@@ -87,35 +90,38 @@ func NewService(
fileManagerService *filemanager.Service, fileManagerService *filemanager.Service,
logger *log.Logger, logger *log.Logger,
slack *slack.Service, slack *slack.Service,
emailStaticAssetURLs emails.StaticAssetURLs,
) *Service { ) *Service {
return &Service{ return &Service{
pg: pgClient, pg: pgClient,
s3: s3Client, s3: s3Client,
bucket: bucket, bucket: bucket,
encryptionKey: encryptionKey, encryptionKey: encryptionKey,
slackSigningSecret: slackSigningSecret, slackSigningSecret: slackSigningSecret,
baseURL: baseURL, baseURL: baseURL,
iam: iam, iam: iam,
html2pdfConverter: html2pdfConverter, html2pdfConverter: html2pdfConverter,
fileManager: fileManagerService, fileManager: fileManagerService,
logger: logger, logger: logger,
slack: slack, slack: slack,
emailStaticAssetURLs: emailStaticAssetURLs,
} }
} }
func (s *Service) WithTenant(tenantID gid.TenantID) *TenantService { func (s *Service) WithTenant(tenantID gid.TenantID) *TenantService {
tenantService := &TenantService{ tenantService := &TenantService{
pg: s.pg, pg: s.pg,
s3: s.s3, s3: s.s3,
bucket: s.bucket, bucket: s.bucket,
scope: coredata.NewScope(tenantID), scope: coredata.NewScope(tenantID),
proboSvc: s.proboSvc, proboSvc: s.proboSvc,
encryptionKey: s.encryptionKey, encryptionKey: s.encryptionKey,
baseURL: s.baseURL, baseURL: s.baseURL,
iam: s.iam, iam: s.iam,
html2pdfConverter: s.html2pdfConverter, html2pdfConverter: s.html2pdfConverter,
fileManager: s.fileManager, fileManager: s.fileManager,
logger: s.logger, logger: s.logger,
emailStaticAssetURLs: s.emailStaticAssetURLs,
} }
tenantService.TrustCenters = &TrustCenterService{svc: tenantService} tenantService.TrustCenters = &TrustCenterService{svc: tenantService}

View File

@@ -226,7 +226,7 @@ func (s *TrustCenterService) EmailPresenterConfig(ctx context.Context, complianc
organization = &coredata.Organization{} organization = &coredata.Organization{}
customDomain *coredata.CustomDomain customDomain *coredata.CustomDomain
logoFile = &coredata.File{} logoFile = &coredata.File{}
emailPresenterCfg = emails.DefaultPresenterConfig(s.svc.baseURL) emailPresenterCfg = emails.DefaultPresenterConfig(s.svc.baseURL, s.svc.emailStaticAssetURLs)
) )
scope := coredata.NewScopeFromObjectID(compliancePageID) scope := coredata.NewScopeFromObjectID(compliancePageID)