email-collector.go

package main // import "github.com/productionwentdown/email-collector"

import (
	"bytes"
	"encoding/csv"
	"encoding/json"
	"flag"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"sync"
	"time"

	// modified from @badoux's checkmail
	"github.com/productionwentdown/email-collector/checkmail"
)

var filename string
var listen string
var redirect string
var slack string

func main() {
	flag.StringVar(&filename, "file", "list.csv", "file to append records to")
	flag.StringVar(&listen, "listen", ":8080", "address to listen to")
	flag.StringVar(&redirect, "redirect", "/subscribed", "path to redirect to upon success")
	flag.StringVar(&slack, "slack", "", "optional slack webhook url")
	flag.Parse()

	csvMutex := &sync.Mutex{}
	csvFile, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	defer csvFile.Close()
	if err != nil {
		log.Fatal(err)
	}
	csvWriter := csv.NewWriter(csvFile)

	http.HandleFunc("/subscribe", func(w http.ResponseWriter, r *http.Request) {
		err := r.ParseForm()
		if err != nil {
			log.Println(err)
			http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
			return
		}
		email := r.Form.Get("email")
		if err := checkmail.ValidateFormat(email); err != nil {
			log.Println(email, err)
			http.Error(w, "Email is not valid", http.StatusBadRequest)
			return
		}
		err = checkmail.ValidateHost(email)
		if _, ok := err.(checkmail.SmtpError); !ok && err != nil {
			log.Println(email, err)
			http.Error(w, "Email is not valid", http.StatusBadRequest)
			return
		}
		log.Println(email, "success")
		csvMutex.Lock()
		csvWriter.Write([]string{email, time.Now().String()})
		csvWriter.Flush()
		csvMutex.Unlock()
		if len(slack) > 4 {
			object := map[string]string{
				"text": "A human (" + email + ") subscribed to the list! Wohoo!",
			}
			payload, err := json.Marshal(object)
			if err != nil {
				log.Println("json.Marshal failed, not supposed to happen!")
				http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
				return
			}
			resp, err := http.Post(slack, "application/json", bytes.NewBuffer(payload))
			if err != nil {
				log.Println(err)
				http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
				return
			}
			defer resp.Body.Close()
			if resp.StatusCode != http.StatusOK {
				body, _ := ioutil.ReadAll(resp.Body)
				log.Println(body)
				http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
				return
			}
		}
		w.Header().Add("Location", redirect)
		w.WriteHeader(303)
	})

	log.Fatal(http.ListenAndServe(listen, nil))

}