mailgun dep

vendor/github.com/mailgun/mailgun-go/v4/mock_routes.go

@@ -0,0 +1,139 @@
+package mailgun
+
+import (
+	"fmt"
+	"github.com/go-chi/chi"
+	"net/http"
+	"time"
+)
+
+type routeResponse struct {
+	Route Route `json:"route"`
+}
+
+func (ms *MockServer) addRoutes(r chi.Router) {
+	r.Post("/routes", ms.createRoute)
+	r.Get("/routes", ms.listRoutes)
+	r.Get("/routes/{id}", ms.getRoute)
+	r.Put("/routes/{id}", ms.updateRoute)
+	r.Delete("/routes/{id}", ms.deleteRoute)
+
+	for i := 0; i < 10; i++ {
+		ms.routeList = append(ms.routeList, Route{
+			Id:          randomString(10, "ID-"),
+			Priority:    0,
+			Description: fmt.Sprintf("Sample Route %d", i),
+			Actions: []string{
+				`forward("http://myhost.com/messages/")`,
+				`stop()`,
+			},
+			Expression: `match_recipient(".*@samples.mailgun.org")`,
+		})
+	}
+}
+
+func (ms *MockServer) listRoutes(w http.ResponseWriter, r *http.Request) {
+	skip := stringToInt(r.FormValue("skip"))
+	limit := stringToInt(r.FormValue("limit"))
+	if limit == 0 {
+		limit = 100
+	}
+
+	if skip > len(ms.routeList) {
+		skip = len(ms.routeList)
+	}
+
+	end := limit + skip
+	if end > len(ms.routeList) {
+		end = len(ms.routeList)
+	}
+
+	// If we are at the end of the list
+	if skip == end {
+		toJSON(w, routesListResponse{
+			TotalCount: len(ms.routeList),
+			Items:      []Route{},
+		})
+		return
+	}
+
+	toJSON(w, routesListResponse{
+		TotalCount: len(ms.routeList),
+		Items:      ms.routeList[skip:end],
+	})
+}
+
+func (ms *MockServer) getRoute(w http.ResponseWriter, r *http.Request) {
+	for _, item := range ms.routeList {
+		if item.Id == chi.URLParam(r, "id") {
+			toJSON(w, routeResponse{Route: item})
+			return
+		}
+	}
+	w.WriteHeader(http.StatusNotFound)
+	toJSON(w, okResp{Message: "route not found"})
+}
+
+func (ms *MockServer) createRoute(w http.ResponseWriter, r *http.Request) {
+	if r.FormValue("action") == "" {
+		w.WriteHeader(http.StatusBadRequest)
+		toJSON(w, okResp{Message: "'action' parameter is required"})
+		return
+	}
+
+	ms.routeList = append(ms.routeList, Route{
+		CreatedAt:   RFC2822Time(time.Now().UTC()),
+		Id:          randomString(10, "ID-"),
+		Priority:    stringToInt(r.FormValue("priority")),
+		Description: r.FormValue("description"),
+		Expression:  r.FormValue("expression"),
+		Actions:     r.Form["action"],
+	})
+	toJSON(w, createRouteResp{
+		Message: "Route has been created",
+		Route:   ms.routeList[len(ms.routeList)-1],
+	})
+}
+
+func (ms *MockServer) updateRoute(w http.ResponseWriter, r *http.Request) {
+	for i, item := range ms.routeList {
+		if item.Id == chi.URLParam(r, "id") {
+
+			if r.FormValue("action") != "" {
+				ms.routeList[i].Actions = r.Form["action"]
+			}
+			if r.FormValue("priority") != "" {
+				ms.routeList[i].Priority = stringToInt(r.FormValue("priority"))
+			}
+			if r.FormValue("description") != "" {
+				ms.routeList[i].Description = r.FormValue("description")
+			}
+			if r.FormValue("expression") != "" {
+				ms.routeList[i].Expression = r.FormValue("expression")
+			}
+			toJSON(w, ms.routeList[i])
+			return
+		}
+	}
+	w.WriteHeader(http.StatusNotFound)
+	toJSON(w, okResp{Message: "route not found"})
+}
+
+func (ms *MockServer) deleteRoute(w http.ResponseWriter, r *http.Request) {
+	result := ms.routeList[:0]
+	for _, item := range ms.routeList {
+		if item.Id == chi.URLParam(r, "id") {
+			continue
+		}
+		result = append(result, item)
+	}
+
+	if len(result) != len(ms.domainList) {
+		toJSON(w, okResp{Message: "success"})
+		ms.routeList = result
+		return
+	}
+
+	w.WriteHeader(http.StatusNotFound)
+	toJSON(w, okResp{Message: "route not found"})
+}