64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package goapi
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type API struct {
|
|
Mux *http.ServeMux
|
|
Perms map[string]string
|
|
}
|
|
|
|
func (a *API) GetPerm(r *http.Request) string {
|
|
return a.Perms[r.Method+"_"+strings.Split(r.RequestURI, "?")[0]]
|
|
}
|
|
|
|
func Init() API {
|
|
mux := &http.ServeMux{}
|
|
|
|
ret := API{
|
|
Mux: mux,
|
|
Perms: make(map[string]string),
|
|
}
|
|
|
|
mux.HandleFunc("/someapi", func(w http.ResponseWriter, r *http.Request) {
|
|
switch r.Method {
|
|
case "POST":
|
|
h_SomeAPI2(w, r)
|
|
default:
|
|
http.Error(w, "Method not allowed", 500)
|
|
}
|
|
})
|
|
return ret
|
|
}
|
|
|
|
func h_SomeAPI2(w http.ResponseWriter, r *http.Request) {
|
|
|
|
ctx := r.Context()
|
|
ctx = context.WithValue(r.Context(), "REQ", r)
|
|
ctx = context.WithValue(ctx, "RES", w)
|
|
req := &AStr{}
|
|
if r.Method != http.MethodGet && r.Method != http.MethodHead {
|
|
err := json.NewDecoder(r.Body).Decode(&req)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
return
|
|
}
|
|
}
|
|
|
|
res, err := SomeAPI2(ctx, req)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
return
|
|
}
|
|
w.Header().Add("Content-Type", "Application/json")
|
|
err = json.NewEncoder(w).Encode(res)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
return
|
|
}
|
|
}
|