From 1dee0cbc3a66593952d23f04cd676dcf1c25b513 Mon Sep 17 00:00:00 2001
From: maru <maru@myyahoo.com>
Date: Fri, 29 Dec 2023 14:30:47 -0500
Subject: [PATCH] Add debug mode bool

---
 api/account.go      |  8 ++++----
 api/generic.go      | 35 +++++++++++++++++++++++++++++++++++
 api/savedata.go     |  6 +++---
 pokerogue-server.go | 15 +++------------
 4 files changed, 45 insertions(+), 19 deletions(-)

diff --git a/api/account.go b/api/account.go
index ee0ed2f..33a3933 100644
--- a/api/account.go
+++ b/api/account.go
@@ -29,7 +29,7 @@ type AccountInfoResponse struct{
 	Username string `json:"string"`
 }
 
-func HandleAccountInfo(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleAccountInfo(w http.ResponseWriter, r *http.Request) {
 	if r.Header.Get("Authorization") == "" {
 		http.Error(w, "missing token", http.StatusBadRequest)
 		return
@@ -60,7 +60,7 @@ func HandleAccountInfo(w http.ResponseWriter, r *http.Request) {
 
 type AccountRegisterRequest GenericAuthRequest
 
-func HandleAccountRegister(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleAccountRegister(w http.ResponseWriter, r *http.Request) {
 	var request AccountRegisterRequest
 	err := json.NewDecoder(r.Body).Decode(&request)
 	if err != nil {
@@ -108,7 +108,7 @@ func HandleAccountRegister(w http.ResponseWriter, r *http.Request) {
 type AccountLoginRequest GenericAuthRequest
 type AccountLoginResponse GenericAuthResponse
 
-func HandleAccountLogin(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleAccountLogin(w http.ResponseWriter, r *http.Request) {
 	var request AccountLoginRequest
 	err := json.NewDecoder(r.Body).Decode(&request)
 	if err != nil {
@@ -167,7 +167,7 @@ func HandleAccountLogin(w http.ResponseWriter, r *http.Request) {
 
 // /api/account/logout - log out of account
 
-func HandleAccountLogout(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleAccountLogout(w http.ResponseWriter, r *http.Request) {
 	token, err := base64.StdEncoding.DecodeString(r.Header.Get("Authorization"))
 	if err != nil {
 		http.Error(w, fmt.Sprintf("failed to decode token: %s", err), http.StatusBadRequest)
diff --git a/api/generic.go b/api/generic.go
index bcbbe01..dbb2668 100644
--- a/api/generic.go
+++ b/api/generic.go
@@ -1,5 +1,40 @@
 package api
 
+import "net/http"
+
+type Server struct {
+	Debug bool
+}
+
+func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	if s.Debug {
+		w.Header().Add("Access-Control-Allow-Origin", "*")
+
+		if r.Method == "OPTIONS" {
+			w.WriteHeader(http.StatusOK)
+			return
+		}
+	}
+
+	switch r.URL.Path {
+	case "/account/info":
+		s.HandleAccountInfo(w, r)
+	case "/account/register":
+		s.HandleAccountRegister(w, r)
+	case "/account/login":
+		s.HandleAccountLogin(w, r)
+	case "/account/logout":
+		s.HandleAccountLogout(w, r)
+
+	case "/savedata/get":
+		s.HandleSavedataGet(w, r)
+	case "/savedata/update":
+		s.HandleSavedataUpdate(w, r)
+	case "/savedata/delete":
+		s.HandleSavedataDelete(w, r)
+	}
+}
+
 // auth
 
 type GenericAuthRequest struct {
diff --git a/api/savedata.go b/api/savedata.go
index 174337d..06209c4 100644
--- a/api/savedata.go
+++ b/api/savedata.go
@@ -7,7 +7,7 @@ import "net/http"
 type SavedataGetRequest struct{}
 type SavedataGetResponse struct{}
 
-func HandleSavedataGet(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleSavedataGet(w http.ResponseWriter, r *http.Request) {
 
 }
 
@@ -16,7 +16,7 @@ func HandleSavedataGet(w http.ResponseWriter, r *http.Request) {
 type SavedataUpdateRequest struct{}
 type SavedataUpdateResponse struct{}
 
-func HandleSavedataUpdate(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleSavedataUpdate(w http.ResponseWriter, r *http.Request) {
 
 }
 
@@ -25,6 +25,6 @@ func HandleSavedataUpdate(w http.ResponseWriter, r *http.Request) {
 type SavedataDeleteRequest struct{}
 type SavedataDeleteResponse struct{}
 
-func HandleSavedataDelete(w http.ResponseWriter, r *http.Request) {
+func (s *Server) HandleSavedataDelete(w http.ResponseWriter, r *http.Request) {
 
 }
diff --git a/pokerogue-server.go b/pokerogue-server.go
index 3a9774e..7f747a8 100644
--- a/pokerogue-server.go
+++ b/pokerogue-server.go
@@ -12,6 +12,8 @@ import (
 )
 
 func main() {
+	debug := flag.Bool("debug", false, "debug mode")
+
 	proto := flag.String("proto", "tcp", "protocol for api to use (tcp, unix)")
 	addr := flag.String("addr", "0.0.0.0", "network address for api to listen on")
 
@@ -41,18 +43,7 @@ func main() {
 		os.Chmod(*addr, 0777)
 	}
 
-	// account
-	http.HandleFunc("/account/info", api.HandleAccountInfo)
-	http.HandleFunc("/account/register", api.HandleAccountRegister)
-	http.HandleFunc("/account/login", api.HandleAccountLogin)
-	http.HandleFunc("/account/logout", api.HandleAccountLogout)
-
-	// savedata
-	http.HandleFunc("/savedata/get", api.HandleSavedataGet)
-	http.HandleFunc("/savedata/update", api.HandleSavedataUpdate)
-	http.HandleFunc("/savedata/delete", api.HandleSavedataDelete)
-	
-	err = http.Serve(listener, nil)
+	err = http.Serve(listener, &api.Server{Debug: *debug})
 	if err != nil {
 		log.Fatalf("failed to create http server or server errored: %s", err)
 	}