mock_subaccounts.go

package mailgun

import (
	"net/http"

	"github.com/go-chi/chi/v5"
)

func (ms *mockServer) addSubaccountRoutes(r chi.Router) {
	ms.subaccountList = append(ms.subaccountList, Subaccount{
		Id:     "enabled.subaccount",
		Name:   "mailgun.test",
		Status: "enabled",
	}, Subaccount{
		Id:     "disabled.subaccount",
		Name:   "mailgun.test",
		Status: "disabled",
	})

	r.Get("/accounts/subaccounts", ms.listSubaccounts)
	r.Post("/accounts/subaccounts", ms.createSubaccount)

	r.Get("/accounts/subaccounts/{subaccountID}", ms.getSubaccount)
	r.Post("/accounts/subaccounts/{subaccountID}/enable", ms.enableSubaccount)
	r.Post("/accounts/subaccounts/{subaccountID}/disable", ms.disableSubaccount)
}

func (ms *mockServer) listSubaccounts(w http.ResponseWriter, r *http.Request) {
	defer ms.mutex.Unlock()
	ms.mutex.Lock()

	var list subaccountsListResponse
	for _, subaccount := range ms.subaccountList {
		list.Items = append(list.Items, subaccount)
	}

	skip := stringToInt(r.FormValue("skip"))
	limit := stringToInt(r.FormValue("limit"))
	if limit == 0 {
		limit = 100
	}

	if skip > len(list.Items) {
		skip = len(list.Items)
	}

	end := limit + skip
	if end > len(list.Items) {
		end = len(list.Items)
	}

	// If we are at the end of the list
	if skip == end {
		toJSON(w, subaccountsListResponse{
			Total: len(list.Items),
			Items: []Subaccount{},
		})
		return
	}

	toJSON(w, subaccountsListResponse{
		Total: len(list.Items),
		Items: list.Items[skip:end],
	})
}

func (ms *mockServer) getSubaccount(w http.ResponseWriter, r *http.Request) {
	defer ms.mutex.Unlock()
	ms.mutex.Lock()

	for _, s := range ms.subaccountList {
		if s.Id == chi.URLParam(r, "subaccountID") {
			toJSON(w, SubaccountResponse{Item: s})
			return
		}
	}
	w.WriteHeader(http.StatusNotFound)
	toJSON(w, okResp{Message: "Not Found"})
}

func (ms *mockServer) createSubaccount(w http.ResponseWriter, r *http.Request) {
	defer ms.mutex.Unlock()
	ms.mutex.Lock()

	ms.subaccountList = append(ms.subaccountList, Subaccount{
		Id:     "test",
		Name:   r.FormValue("name"),
		Status: "active",
	})
	toJSON(w, okResp{Message: "Subaccount has been created"})
}

func (ms *mockServer) enableSubaccount(w http.ResponseWriter, r *http.Request) {
	defer ms.mutex.Unlock()
	ms.mutex.Lock()

	for _, subaccount := range ms.subaccountList {
		if subaccount.Id == chi.URLParam(r, "subaccountID") && subaccount.Status == "disabled" {
			subaccount.Status = "enabled"
			toJSON(w, SubaccountResponse{Item: subaccount})
			return
		}
		if subaccount.Id == chi.URLParam(r, "subaccountID") && subaccount.Status == "enabled" {
			toJSON(w, okResp{Message: "subaccount is already enabled"})
			return
		}
	}
	toJSON(w, okResp{Message: "Not Found"})
}

func (ms *mockServer) disableSubaccount(w http.ResponseWriter, r *http.Request) {
	defer ms.mutex.Unlock()
	ms.mutex.Lock()

	for _, subaccount := range ms.subaccountList {
		if subaccount.Id == chi.URLParam(r, "subaccountID") && subaccount.Status == "enabled" {
			subaccount.Status = "disabled"
			toJSON(w, SubaccountResponse{Item: subaccount})
			return
		}
		if subaccount.Id == chi.URLParam(r, "subaccountID") && subaccount.Status == "disabled" {
			toJSON(w, okResp{Message: "subaccount is already disabled"})
			return
		}
	}
	toJSON(w, okResp{Message: "Not Found"})
}