feat(courier): expose setters again

ory · Mar 7, 2022 · 598dc3a · 598dc3a
1 parent fa1f234
commit 598dc3a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/courier/courier.go b/courier/courier.go
@@ -30,6 +30,8 @@ type (
 		SmtpDialer() *gomail.Dialer
 		DispatchQueue(ctx context.Context) error
 		DispatchMessage(ctx context.Context, msg Message) error
+		SetGetEmailTemplateType(f func(t EmailTemplate) (TemplateType, error))
+		SetNewEmailTemplateFromMessage(f func(d Dependencies, msg Message) (EmailTemplate, error))
 	}
 
 	Provider interface {

diff --git a/courier/smtp.go b/courier/smtp.go
@@ -70,6 +70,14 @@ func newSMTP(ctx context.Context, deps Dependencies) *smtpClient {
 	}
 }
 
+func (c *courier) SetGetEmailTemplateType(f func(t EmailTemplate) (TemplateType, error)) {
+	c.smtpClient.GetTemplateType = f
+}
+
+func (c *courier) SetNewEmailTemplateFromMessage(f func(d Dependencies, msg Message) (EmailTemplate, error)) {
+	c.smtpClient.NewTemplateFromMessage = f
+}
+
 func (c *courier) SmtpDialer() *gomail.Dialer {
 	return c.smtpClient.Dialer
 }