1
0
Forkuj 0
email-collector/email-collector.go

94 wiersze
2.6 KiB
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))
}