New directory layout - DOES NOT COMPILE

web/mailbox.go

@@ -0,0 +1,163 @@
+package controllers
+
+import (
+	"github.com/jhillyerd/inbucket/app/inbucket"
+	"github.com/robfig/revel"
+	"html/template"
+)
+
+type Mailbox struct {
+	*rev.Controller
+}
+
+func (c Mailbox) Index(name string) rev.Result {
+	c.Validation.Required(name).Message("Account name is required")
+
+	if c.Validation.HasErrors() {
+		c.Validation.Keep()
+		c.FlashParams()
+		return c.Redirect(Application.Index)
+	}
+
+	return c.Render(name)
+}
+
+func (c Mailbox) List(name string) rev.Result {
+	c.Validation.Required(name).Message("Account name is required")
+
+	if c.Validation.HasErrors() {
+		c.Validation.Keep()
+		c.FlashParams()
+		return c.Redirect(Application.Index)
+	}
+
+	ds := inbucket.NewDataStore()
+	mb, err := ds.MailboxFor(name)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	messages, err := mb.GetMessages()
+	if err != nil {
+		return c.RenderError(err)
+	}
+	rev.INFO.Printf("Got %v messsages", len(messages))
+
+	c.Response.Out.Header().Set("Expires", "-1")
+	return c.Render(name, messages)
+}
+
+func (c Mailbox) Show(name string, id string) rev.Result {
+	c.Validation.Required(name).Message("Account name is required")
+	c.Validation.Required(id).Message("Message ID is required")
+
+	if c.Validation.HasErrors() {
+		c.Validation.Keep()
+		c.FlashParams()
+		return c.Redirect(Application.Index)
+	}
+
+	ds := inbucket.NewDataStore()
+	mb, err := ds.MailboxFor(name)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	message, err := mb.GetMessage(id)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	_, mime, err := message.ReadBody()
+	if err != nil {
+		return c.RenderError(err)
+	}
+	body := template.HTML(inbucket.TextToHtml(mime.Text))
+	htmlAvailable := mime.Html != ""
+
+	c.Response.Out.Header().Set("Expires", "-1")
+	return c.Render(name, message, body, htmlAvailable)
+}
+
+func (c Mailbox) Delete(name string, id string) rev.Result {
+	c.Validation.Required(name).Message("Account name is required")
+	c.Validation.Required(id).Message("Message ID is required")
+
+	if c.Validation.HasErrors() {
+		c.Validation.Keep()
+		c.FlashParams()
+		return c.Redirect(Application.Index)
+	}
+
+	ds := inbucket.NewDataStore()
+	mb, err := ds.MailboxFor(name)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	message, err := mb.GetMessage(id)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	err = message.Delete()
+	if err != nil {
+		return c.RenderError(err)
+	}
+	c.Response.Out.Header().Set("Expires", "-1")
+	return c.RenderText("OK")
+}
+
+func (c Mailbox) Html(name string, id string) rev.Result {
+	c.Validation.Required(name).Message("Account name is required")
+	c.Validation.Required(id).Message("Message ID is required")
+
+	if c.Validation.HasErrors() {
+		c.Validation.Keep()
+		c.FlashParams()
+		return c.Redirect(Application.Index)
+	}
+
+	ds := inbucket.NewDataStore()
+	mb, err := ds.MailboxFor(name)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	message, err := mb.GetMessage(id)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	_, mime, err := message.ReadBody()
+	if err != nil {
+		return c.RenderError(err)
+	}
+	// Mark as safe to render HTML
+	// TODO: It is not really safe to render, need to sanitize.
+	body := template.HTML(mime.Html)
+
+	c.Response.Out.Header().Set("Expires", "-1")
+	return c.Render(name, message, body)
+}
+
+func (c Mailbox) Source(name string, id string) rev.Result {
+	c.Validation.Required(name).Message("Account name is required")
+	c.Validation.Required(id).Message("Message ID is required")
+
+	if c.Validation.HasErrors() {
+		c.Validation.Keep()
+		c.FlashParams()
+		return c.Redirect(Application.Index)
+	}
+
+	ds := inbucket.NewDataStore()
+	mb, err := ds.MailboxFor(name)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	message, err := mb.GetMessage(id)
+	if err != nil {
+		return c.RenderError(err)
+	}
+	raw, err := message.ReadRaw()
+	if err != nil {
+		return c.RenderError(err)
+	}
+
+	c.Response.Out.Header().Set("Expires", "-1")
+	return c.RenderText(*raw)
+}