49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
|
package mailgun
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"github.com/go-chi/chi"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (ms *MockServer) addExportRoutes(r chi.Router) {
|
||
|
|
r.Post("/exports", ms.postExports)
|
||
|
|
r.Get("/exports", ms.listExports)
|
||
|
|
r.Get("/exports/{id}", ms.getExport)
|
||
|
|
r.Get("/exports/{id}/download_url", ms.getExportLink)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MockServer) postExports(w http.ResponseWriter, r *http.Request) {
|
||
|
|
e := Export{
|
||
|
|
ID: strconv.Itoa(len(ms.exportList)),
|
||
|
|
URL: r.FormValue("url"),
|
||
|
|
Status: "complete",
|
||
|
|
}
|
||
|
|
|
||
|
|
ms.exportList = append(ms.exportList, e)
|
||
|
|
toJSON(w, okResp{Message: "success"})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MockServer) listExports(w http.ResponseWriter, _ *http.Request) {
|
||
|
|
toJSON(w, ExportList{
|
||
|
|
Items: ms.exportList,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MockServer) getExport(w http.ResponseWriter, r *http.Request) {
|
||
|
|
for _, export := range ms.exportList {
|
||
|
|
if export.ID == chi.URLParam(r, "id") {
|
||
|
|
toJSON(w, export)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
w.WriteHeader(http.StatusNotFound)
|
||
|
|
toJSON(w, okResp{Message: "export not found"})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ms *MockServer) getExportLink(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.Header().Set("Location", "/some/s3/url")
|
||
|
|
w.WriteHeader(http.StatusFound)
|
||
|
|
}
|