main.go

package main

import (
	"flag"
	"fmt"
	"github.com/gorilla/mux"
	"log"
	"net/http"
	"net/url"
	"os"
	"strings"
)

var (
	mandrillApiUrl string
	mandrillKey    string
)

func main() {
	port := flag.String("port", "8080", "HTTP Port to listen on")
	flag.Parse()

	mandrillApiUrl = "https://mandrillapp.com/api/1.0/"
	mandrillKey = os.Getenv("MANDRILL_KEY")
	if mandrillKey == "" {
		log.Fatal("MANDRILL_KEY envrionment variable is not set.")
		os.Exit(1)
	}

	log.Println("Starting Server on", *port)
	r := mux.NewRouter()
	r.HandleFunc("/", HomeHandler)
	r.HandleFunc("/{email}", EmailHandler)
	http.Handle("/", r)
	http.ListenAndServe(":"+*port, r)
}

func HomeHandler(rw http.ResponseWriter, r *http.Request) {
	http.Redirect(rw, r, "http://munrocape.github.io/staticcontact/", 301)
}

func EmailHandler(rw http.ResponseWriter, r *http.Request) {
	destination := r.URL.Path[1:]
	var query url.Values
	if r.Method == "GET" {
		query = r.URL.Query()
	} else if r.Method == "POST" {
		r.ParseForm()
		query = r.Form
	} else {
		fmt.Fprintln(rw, "Please submit via GET or POST. See www.staticcontact.com for further instructions.")
	}
	EmailSender(rw, r, destination, query)
}

func EmailSender(rw http.ResponseWriter, r *http.Request, destination string, query url.Values) {
	form := ParseQuery(query)
	if form.Email == "" {
		fmt.Fprintln(rw, "Please provide a sender email.")
		return
	}
	if sendEmail(destination, form) {
		fmt.Fprintln(rw, "Success! Your message has been delivered.")
	} else {
		fmt.Fprintln(rw, "Uh-oh! We were unable to deliver your message. Please confirm that you entered a valid email address.")
	}
}

func ParseQuery(query url.Values) *Form {
	form := new(Form)
	additionalFields := ""
	for k, v := range query {
		k = strings.ToLower(k)
		if (k == "email") {
			form.Email = v[0]
		} else if (k == "name") {
			form.Name = v[0]
		} else if (k == "subject") {
			form.Subject = v[0]
		} else if (k == "message") {
			form.Message = k + ": " + v[0] + "<br>\n"
		} else {
			additionalFields = additionalFields + k + ": " + v[0] + "<br>\n"
		}
	}
	if form.Subject == "" {
		form.Subject = "New Form Submission!"
	}
	if additionalFields != "" {
		if form.Message == "" {
			form.Message = form.Message + "The following fields were also entered:\n<br>" + additionalFields
		} else {
			form.Message = form.Message + "\n<br>The following additional fields were also entered:\n<br>" + additionalFields
		}
	}
	return form
}