From 47ee5c4776f3856dc0e977ccfa24b96544632e73 Mon Sep 17 00:00:00 2001 From: Masahiko AMANO Date: Fri, 27 Jan 2023 01:21:18 +0300 Subject: [PATCH] feat(web): add token validation handler --- web/server/web-server.go | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/web/server/web-server.go b/web/server/web-server.go index 1d78da3..fa7e0e0 100644 --- a/web/server/web-server.go +++ b/web/server/web-server.go @@ -95,6 +95,29 @@ func HandlerAuth(w http.ResponseWriter, r *http.Request) { } } +func HandlerToken(w http.ResponseWriter, r *http.Request) { + var request JSON + var response = JSON{Status: false} + var err error + r.Body = http.MaxBytesReader(w, r.Body, 1048576) + json_decoder := json.NewDecoder(r.Body) + json_decoder.DisallowUnknownFields() + err = json_decoder.Decode(&request) + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + return + } + if TokenValidate(request.Token) { + response.Status = true + } + jsonData, err := json.Marshal(response) + w.Header().Set("Content-Type", "application/json; charset=utf-8") + _, err = w.Write(jsonData) + if err != nil { + log.Println(err) + } +} + func HandlerTDBMS(w http.ResponseWriter, r *http.Request) { var request JSON var response []byte @@ -148,6 +171,7 @@ func main() { public_fs.ServeHTTP(w, r) }) http.HandleFunc("/AUTH", HandlerAuth) + http.HandleFunc("/token", HandlerToken) http.HandleFunc("/TDBMS", HandlerTDBMS) tfm_fs := http.FileServer(http.Dir("/srv/data/tfm")) http.Handle("/tfm/", http.StripPrefix("/tfm", tfm_fs))