conex-builder/server/main.go

package main

import (
	"encoding/json"
	"log"
	"net/http"
	"os"
	"os/signal"
	"strings"
	"syscall"
)

const (
	msgServerStart    = "Msg: main.go: Starting server"
	msgServerShutdown = "Msg: main.go: Server shutdown gracefully"
	errServerStart    = "Fatal: main.go: Start server"
	errReadBody       = "Error: main.go: Read request body"
	errParseBody      = "Error: main.go: Parse request body"
	errGetOrderID     = "Error: main.go: Get orderID from client URL"
	errCaptureOrder   = "Error: main.go: Capture order"
	errRegisterSite   = "Error: main.go: Register site in database"
	errEncodeResponse = "Error: main.go: Encode response"
	errCreateOrder    = "Error: main.go: Obtain orderID"
	errAuthGen        = "Error: main.go: Gen and register auth"
	errAuthEmail      = "Error: main.go: Send auth email"
	errAuthValidate   = "Error: main.go: Validate changes"
	errUpdateSite     = "Error: main.go: Updating site data"
)

func main() {
	initialize()

	http.HandleFunc("/api/orders", CreateOrderHandler)
	http.HandleFunc("/api/orders/", CaptureOrderHandler)
	http.HandleFunc("/api/update", UpdateSiteHandler)
	http.HandleFunc("/api/confirm", ConfirmChangesHandler)
	http.Handle("/", http.FileServer(http.Dir("./public")))

	stop := make(chan os.Signal, 1)
	signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
	port := os.Getenv("PORT")

	go func() {
		msg(msgServerStart + ": " + port + "...")
		if err := http.ListenAndServe(":"+port, nil); err != nil {
			fatal(err, errServerStart)
		}
	}()

	<-stop

	shutdown()
	msg(msgServerShutdown)
}

func msg(notice string) {
	log.Println(notice)
}

func httpErrorAndLog(w http.ResponseWriter,
	err error, notice string, client string,
) {
	log.Printf("%s: %v", notice, err)
	http.Error(w, client, http.StatusInternalServerError)
}

func fatal(err error, notice string) {
	shutdown()
	log.Fatalf("%s: %v", notice, err)
}

func CreateOrderHandler(w http.ResponseWriter, r *http.Request) {
	orderID, err := CreateOrder()
	if err != nil {
		httpErrorAndLog(w, err, errCreateOrder, "Error creating order")
		return
	}

	var response struct {
		ID string `json:"id"`
	}
	response.ID = orderID

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(response)
	return
}

func CaptureOrderHandler(w http.ResponseWriter, r *http.Request) {
	errClientNotice := "Error capturing order"

	var cart struct {
		Directory  string          `json:"directory"`
		EditorData json.RawMessage `json:"editor_data"`
	}
	if err := json.NewDecoder(r.Body).Decode(&cart); err != nil {
		httpErrorAndLog(w, err, errReadBody, errClientNotice)
		return
	}

	path := strings.TrimPrefix(r.URL.Path, "/api/orders/")
	parts := strings.Split(path, "/")
	orderID := parts[0]
	if orderID == "" {
		httpErrorAndLog(w, nil, errGetOrderID, errClientNotice)
		return
	}

	capture, receipt, err := CaptureOrder(orderID)
	if err != nil {
		httpErrorAndLog(w, err, errCaptureOrder, errClientNotice)
		return
	}

	if err := RegisterSitePayment(capture, cart.Directory, cart.EditorData); err != nil {
		httpErrorAndLog(w, err, errRegisterSite+": "+cart.Directory, errClientNotice)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	if err := json.NewEncoder(w).Encode(receipt); err != nil {
		httpErrorAndLog(w, err, errEncodeResponse, errClientNotice)
		return
	}

	return
}

func UpdateSiteHandler(w http.ResponseWriter, r *http.Request) {
	errClientNotice := "Error handling update request"

	var cart struct {
		Directory string `json:"directory"`
	}
	if err := json.NewDecoder(r.Body).Decode(&cart); err != nil {
		httpErrorAndLog(w, err, errReadBody, errClientNotice)
		return
	}

	code := GenerateCode()

	email, err := UpdateSiteAuth(cart.Directory, code)
	if err != nil {
		httpErrorAndLog(w, err, errAuthGen, errClientNotice)
		return
	}

	if err := SendAuthEmail(email, code); err != nil {
		httpErrorAndLog(w, err, errAuthEmail, errClientNotice)
		return
	}

	return
}

func ConfirmChangesHandler(w http.ResponseWriter, r *http.Request) {
	errClientNotice := "Error handling confirm changes request"

	var cart struct {
		Directory  string          `json:"directory"`
		Code       string          `json:"auth_code"`
		EditorData json.RawMessage `json:"editor_data"`
	}
	if err := json.NewDecoder(r.Body).Decode(&cart); err != nil {
		httpErrorAndLog(w, err, errReadBody, errClientNotice)
		return
	}

	pkey, err := ValidateSiteAuth(cart.Directory, cart.Code)
	if err != nil {
		httpErrorAndLog(w, err, errAuthValidate, errClientNotice)
		return
	}

	if err := UpdateSite(pkey, cart.EditorData); err != nil {
		httpErrorAndLog(w, err, errUpdateSite, errClientNotice)
		return
	}

	return
}