Add: Simple user login/register function

pkg/api/api.go

@@ -70,6 +70,9 @@ func NewAPI(config Config) (*API, error) {
 	apiMux.HandleFunc("/get_file_info", api.HandleGetFileInfo)
 	apiMux.HandleFunc("/get_file_stream_direct", api.HandleGetFileStreamDirect)
 	apiMux.HandleFunc("/prepare_file_stream_direct", api.HandlePrepareFileStreamDirect)
+	// user
+	apiMux.HandleFunc("/login", api.HandleLogin)
+	apiMux.HandleFunc("/register", api.HandleRegister)
 	// below needs token
 	apiMux.HandleFunc("/walk", api.HandleWalk)
 	apiMux.HandleFunc("/reset", api.HandleReset)

pkg/api/handle_error.go

@@ -6,6 +6,10 @@ import (
 	"net/http"
 )
 
+type Error struct {
+  Error string `json:"error,omitempty"`
+}
+
 func (api *API) HandleError(w http.ResponseWriter, r *http.Request, err error) {
 	api.HandleErrorString(w, r, err.Error())
 }
@@ -20,8 +24,8 @@ func (api *API) HandleErrorString(w http.ResponseWriter, r *http.Request, errorS
 
 func (api *API) HandleErrorStringCode(w http.ResponseWriter, r *http.Request, errorString string, code int) {
 	log.Println("[api] [Error]", code, errorString)
-	errStatus := &Status{
-		Status: errorString,
+	errStatus := &Error{
+		Error: errorString,
 	}
 	w.WriteHeader(code)
 	json.NewEncoder(w).Encode(errStatus)

pkg/api/handle_user.go

@@ -0,0 +1,83 @@
+package api
+
+import (
+	"encoding/json"
+	"log"
+	"msw-open-music/pkg/database"
+	"net/http"
+)
+
+type LoginRequest struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
+type LoginResponse struct {
+	User *database.User `json:"user"`
+}
+
+func (api *API) HandleLogin(w http.ResponseWriter, r *http.Request) {
+	// Get method will login as anonymous user
+	if r.Method == "GET" {
+		log.Println("Login as anonymous user")
+		user, err := api.Db.LoginAsAnonymous()
+		if err != nil {
+			api.HandleError(w, r, err)
+			return
+		}
+		resp := &LoginResponse{
+			User: user,
+		}
+		err = json.NewEncoder(w).Encode(resp)
+		return
+	}
+
+	var request LoginRequest
+	err := json.NewDecoder(r.Body).Decode(&request)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	log.Println("Login as user", request.Username)
+
+	user, err := api.Db.Login(request.Username, request.Password)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	resp := &LoginResponse{
+		User: user,
+	}
+	err = json.NewEncoder(w).Encode(resp)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+}
+
+type RegisterRequest struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+	Role     int64  `json:"role"`
+}
+
+func (api *API) HandleRegister(w http.ResponseWriter, r *http.Request) {
+	var request RegisterRequest
+	err := json.NewDecoder(r.Body).Decode(&request)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	log.Println("Register user", request.Username)
+
+	err = api.Db.Register(request.Username, request.Password, request.Role)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+
+	api.HandleOK(w, r)
+}