about summary refs log tree commit diff stats
path: root/widgets/compose.go
diff options
context:
space:
mode:
Diffstat (limited to 'widgets/compose.go')
-rw-r--r--widgets/compose.go37
1 files changed, 34 insertions, 3 deletions
diff --git a/widgets/compose.go b/widgets/compose.go
index 22c58da..a55d147 100644
--- a/widgets/compose.go
+++ b/widgets/compose.go
@@ -23,6 +23,7 @@ import (
 	"github.com/pkg/errors"
 
 	"git.sr.ht/~sircmpwn/aerc/config"
+	"git.sr.ht/~sircmpwn/aerc/lib/templates"
 	"git.sr.ht/~sircmpwn/aerc/lib/ui"
 	"git.sr.ht/~sircmpwn/aerc/worker/types"
 )
@@ -53,7 +54,7 @@ type Composer struct {
 }
 
 func NewComposer(aerc *Aerc, conf *config.AercConfig,
-	acct *config.AccountConfig, worker *types.Worker, defaults map[string]string) *Composer {
+	acct *config.AccountConfig, worker *types.Worker, template string, defaults map[string]string) (*Composer, error) {
 
 	if defaults == nil {
 		defaults = make(map[string]string)
@@ -62,13 +63,14 @@ func NewComposer(aerc *Aerc, conf *config.AercConfig,
 		defaults["From"] = acct.From
 	}
 
+	templateData := templates.ParseTemplateData(defaults)
 	layout, editors, focusable := buildComposeHeader(
 		conf.Compose.HeaderLayout, defaults)
 
 	email, err := ioutil.TempFile("", "aerc-compose-*.eml")
 	if err != nil {
 		// TODO: handle this better
-		return nil
+		return nil, err
 	}
 
 	c := &Composer{
@@ -86,11 +88,14 @@ func NewComposer(aerc *Aerc, conf *config.AercConfig,
 	}
 
 	c.AddSignature()
+	if err := c.AddTemplate(template, templateData); err != nil {
+		return nil, err
+	}
 
 	c.updateGrid()
 	c.ShowTerminal()
 
-	return c
+	return c, nil
 }
 
 func buildComposeHeader(layout HeaderLayout, defaults map[string]string) (
@@ -163,6 +168,32 @@ func (c *Composer) AppendContents(reader io.Reader) {
 	c.email.Sync()
 }
 
+func (c *Composer) AddTemplate(template string, data interface{}) error {
+	if template == "" {
+		return nil
+	}
+
+	templateText, err := templates.ParseTemplateFromFile(template, c.config.Templates.TemplateDirs, data)
+	if err != nil {
+		return err
+	}
+	c.PrependContents(bytes.NewReader(templateText))
+	return nil
+}
+
+func (c *Composer) AddTemplateFromString(template string, data interface{}) error {
+	if template == "" {
+		return nil
+	}
+
+	templateText, err := templates.ParseTemplate(template, data)
+	if err != nil {
+		return err
+	}
+	c.PrependContents(bytes.NewReader(templateText))
+	return nil
+}
+
 func (c *Composer) AddSignature() {
 	var signature []byte
 	if c.acct.SignatureCmd != "" {