first commit

Author: hunterlong

.gitignore

@@ -0,0 +1,2 @@
+.idea
+*.db

README.md

@@ -0,0 +1,52 @@
+# Statping Email Notifier
+This repo contains a dedicated service for Statping that allows users to receive notifications to their email address using our SMTP servers.
+
+# Required Environment Variables
+- `HOST` - SMTP host
+- `USERNAME` - SMTP username
+- `PASSWORD` - SMTP password
+- `PORT` - SMTP port
+
+> Server runs on port :8080
+
+# API Endpoints
+
+#### POST `/request`
+This endpoint will send an email to the user with a confirmation link/key. Email's won't be sent until this confirmation link is clicked.
+```json
+{
+  "email": "info@myaddress.com",
+  "version": "0.90.33"
+}
+```
+
+#### GET `/confirm/{id}`
+Confirm an email address with the key provided in the request email.a
+
+#### GET `/check?email=info@myaddress.com`
+Check the status of your email address.
+
+#### GET `/resend?email=info@myaddress.com`
+Attempt to resend the confirmation request email.
+
+#### GET `/unsubscribe?email=info@myaddress.com`
+Unsubscribe the email address, and remove them from database.
+
+#### POST `/send`
+```json
+{
+  "email": "info@myaddress.com",
+  "key": "<secret_key_here>",
+  "version": "0.90.33",
+  "service": {
+    ...service JSON
+  },
+  "failure": {
+      ...failure JSON
+  }
+}
+```
+The main endpoint that will send an email about uptime or downtime.
+
+
+

emailer.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"crypto/tls"
+	"fmt"
+	"gopkg.in/gomail.v2"
+	"os"
+	"strconv"
+)
+
+var (
+	config *Config
+)
+
+type Config struct {
+	Host     string
+	Username string
+	Password string
+	Port     int
+}
+
+func InitConfig() {
+	config = &Config{
+		Host:     os.Getenv("HOST"),
+		Username: os.Getenv("USERNAME"),
+		Password: os.Getenv("PASSWORD"),
+		Port:     toInt(os.Getenv("PORT")),
+	}
+}
+
+func SendEmail(u *User) error {
+	m := gomail.NewMessage()
+	m.SetHeader("From", "info@betatude.com")
+	m.SetHeader("To", u.Email)
+	m.SetHeader("Subject", "Hello!")
+	m.SetBody("text/html", "Hello <b>Bob</b> and <i>Cora</i>!")
+
+	d := gomail.NewDialer(config.Host, config.Port, config.Username, config.Password)
+	d.TLSConfig = &tls.Config{InsecureSkipVerify: true}
+
+	fmt.Printf("Sending email to %s...\n", u.Email)
+
+	if err := d.DialAndSend(m); err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	fmt.Printf("Sent email to %s\n", u.Email)
+
+	return nil
+}
+
+func toInt(s string) int {
+	num, _ := strconv.ParseInt(s, 10, 64)
+	return int(num)
+}

handlers.go

@@ -0,0 +1,163 @@
+package main
+
+import (
+	"encoding/json"
+	"github.com/gorilla/mux"
+	"net/http"
+)
+
+func ConfirmHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+	vars := mux.Vars(r)
+	key := vars["id"]
+
+	user, err := FindUserKey(key)
+	if err != nil || user == nil {
+		sendError(w, notFound)
+		return
+	}
+	if user.Key == key && user.Confirmed {
+		response := requestResponse{
+			Email:   user.Email,
+			Message: "email address has already been confirmed",
+		}
+		sendJSON(w, response)
+		return
+	}
+
+	if user.Key == key {
+		user.Confirmed = true
+		user.Update()
+	}
+	response := requestResponse{
+		Email:   user.Email,
+		Message: "email address is now confirmed",
+	}
+	sendJSON(w, response)
+}
+
+func UnsubscribeHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+	email := r.URL.Query().Get("email")
+
+	user, err := FindUser(email)
+	if err != nil || user == nil {
+		sendError(w, notFound)
+		return
+	}
+
+	user.Delete()
+
+	status := &requestResponse{
+		Email:   user.Email,
+		Message: "email has been unsubscribed",
+	}
+
+	sendJSON(w, status)
+}
+
+func ResendHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+	email := r.URL.Query().Get("email")
+
+	user, err := FindUser(email)
+	if err != nil || user == nil {
+		sendError(w, notFound)
+		return
+	}
+	status := &requestResponse{
+		Email: user.Email,
+	}
+	if user.Confirmed {
+		status.Message = "email has already been confirmed"
+	} else {
+		status.Message = "email has not been confirmed yet"
+	}
+	sendJSON(w, status)
+}
+
+func CheckHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+	email := r.URL.Query().Get("email")
+
+	user, err := FindUser(email)
+	if err != nil || user == nil {
+		sendError(w, notFound)
+		return
+	}
+	status := &requestResponse{
+		Email: user.Email,
+	}
+	if user.Confirmed {
+		status.Message = "email has been confirmed"
+	} else {
+		status.Message = "email has not been confirmed yet " + user.Key
+	}
+	sendJSON(w, status)
+}
+
+func SendHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+
+	user, err := FindUser("info@socialeck.com")
+	if err != nil || user == nil {
+		sendError(w, notFound)
+		return
+	}
+
+	err = SendEmail(user)
+	if err != nil {
+		sendError(w, err)
+		return
+	}
+}
+
+func RequestHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+	var req requestJSON
+	decoder := json.NewDecoder(r.Body)
+	if err := decoder.Decode(&req); err != nil {
+		sendError(w, err)
+		return
+	}
+
+	user, err := FindUser(req.Email)
+	if err != nil {
+		sendError(w, notFound)
+		return
+	}
+
+	if user != nil {
+		response := &requestResponse{
+			Email:   user.Email,
+			Message: "this email need to be confirmed",
+		}
+		sendJSON(w, response)
+		return
+	}
+
+	user = &User{
+		Email: req.Email,
+		Key:   RandomString(32),
+		Sent:  0,
+	}
+
+	user.Create()
+
+	response := &requestResponse{
+		Email:   user.Email,
+		Message: "check email",
+	}
+
+	sendJSON(w, response)
+}
+
+func sendJSON(w http.ResponseWriter, obj interface{}) {
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(obj)
+}
+
+func sendError(w http.ResponseWriter, err error) {
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(&errorResponse{err.Error()})
+}

main.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"github.com/gorilla/mux"
+	"github.com/jinzhu/gorm"
+	"math/rand"
+	"net/http"
+	"time"
+
+	_ "github.com/jinzhu/gorm/dialects/sqlite"
+)
+
+var (
+	db *gorm.DB
+)
+
+func init() {
+	var err error
+	db, err = gorm.Open("sqlite3", "emailer.db")
+	if err != nil {
+		panic(err)
+	}
+	db.AutoMigrate(&User{})
+
+	InitConfig()
+}
+
+func main() {
+	r := mux.NewRouter()
+	r.HandleFunc("/request", RequestHandler).Methods("POST")
+	r.HandleFunc("/confirm/{id}", ConfirmHandler).Methods("GET")
+	r.HandleFunc("/check", CheckHandler).Methods("GET")
+	r.HandleFunc("/resend", ResendHandler).Methods("GET")
+	r.HandleFunc("/unsubscribe", UnsubscribeHandler).Methods("GET")
+	r.HandleFunc("/send", SendHandler).Methods("POST")
+	http.ListenAndServe(":8000", r)
+}
+
+var characterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
+
+// RandomString generates a random string of n length
+func RandomString(n int) string {
+	b := make([]rune, n)
+	rand.Seed(time.Now().UnixNano())
+	for i := range b {
+		b[i] = characterRunes[rand.Intn(len(characterRunes))]
+	}
+	return string(b)
+}

methods.go

@@ -0,0 +1,40 @@
+package main
+
+import "fmt"
+
+func FindUser(email string) (*User, error) {
+	var u User
+	err := db.Model(&User{}).Where("email = ?", email).Find(&u)
+	if err.Error != nil {
+		return nil, err.Error
+	}
+	return &u, nil
+}
+
+func FindUserKey(key string) (*User, error) {
+	var u User
+	err := db.Model(&User{}).Where("key = ?", key).Find(&u)
+	if err.Error != nil {
+		return nil, err.Error
+	}
+	return &u, nil
+}
+
+func (u *User) Create() error {
+	e := db.Model(&User{}).Create(u)
+	return e.Error
+}
+
+func (u *User) Update() error {
+	e := db.Model(&User{}).Update(u)
+	return e.Error
+}
+
+func (u *User) Delete() error {
+	e := db.Model(&User{}).Delete(u)
+	return e.Error
+}
+
+func (u *User) ConfirmLink() string {
+	return fmt.Sprintf("https://emailer.statping.com/confirm/%s", u.Key)
+}