增加用户管理

app/cmd/main.go

@@ -2,6 +2,8 @@ package main
 
 import (
 	"context"
+	"doudizhu-server/internal/captcha"
+	"doudizhu-server/internal/db"
 	"doudizhu-server/internal/game"
 	"doudizhu-server/internal/handlers"
 	"doudizhu-server/internal/redis"
@@ -10,15 +12,13 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
+	"strconv"
 	"syscall"
 	"time"
 )
 
 func main() {
-	redisAddr := os.Getenv("REDIS_ADDR")
-	if redisAddr == "" {
-		redisAddr = "localhost:6379"
-	}
+	redisAddr := getEnv("REDIS_ADDR", "localhost:6379")
 
 	redisClient, err := redis.NewClient(redis.Config{
 		Addr:     redisAddr,
@@ -31,13 +31,74 @@ func main() {
 	defer redisClient.Close()
 	log.Println("Connected to Redis:", redisAddr)
 
+	dbHost := getEnv("DB_HOST", "localhost")
+	dbPort, _ := strconv.Atoi(getEnv("DB_PORT", "5432"))
+	dbUser := getEnv("DB_USER", "postgres")
+	dbPassword := getEnv("DB_PASSWORD", "postgres")
+	dbName := getEnv("DB_NAME", "doudizhu")
+
+	database, err := db.New(db.Config{
+		Host:     dbHost,
+		Port:     dbPort,
+		User:     dbUser,
+		Password: dbPassword,
+		Database: dbName,
+	})
+	if err != nil {
+		log.Fatalf("Failed to connect to PostgreSQL: %v", err)
+	}
+	defer database.Close()
+	log.Println("Connected to PostgreSQL:", dbHost)
+
 	gameMgr := game.NewGameManager(redisClient)
 	hub := ws.NewHub(gameMgr)
 	go hub.Run()
 
-	h := handlers.NewHandler(gameMgr, hub)
+	captchaMgr := captcha.NewManager(redisClient)
+	h := handlers.NewHandler(gameMgr, hub, redisClient, database, captchaMgr)
 
 	mux := http.NewServeMux()
+
+	mux.HandleFunc("/api/auth/captcha", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "GET" {
+			h.GetCaptcha(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
+	mux.HandleFunc("/api/auth/register", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "POST" {
+			h.Register(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
+	mux.HandleFunc("/api/auth/login", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "POST" {
+			h.Login(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
+	mux.HandleFunc("/api/auth/validate", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "GET" {
+			h.ValidateToken(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
+	mux.HandleFunc("/api/auth/logout", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "POST" {
+			h.Logout(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
 	mux.HandleFunc("/api/rooms", func(w http.ResponseWriter, r *http.Request) {
 		if r.Method == "POST" {
 			h.CreateRoom(w, r)
@@ -45,6 +106,23 @@ func main() {
 			http.NotFound(w, r)
 		}
 	})
+
+	mux.HandleFunc("/api/rooms/current", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "GET" {
+			h.GetCurrentRoom(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
+	mux.HandleFunc("/api/rooms/leave", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method == "POST" {
+			h.LeaveRoom(w, r)
+		} else {
+			http.NotFound(w, r)
+		}
+	})
+
 	mux.HandleFunc("/api/rooms/", func(w http.ResponseWriter, r *http.Request) {
 		if r.Method == "POST" {
 			h.JoinRoom(w, r)
@@ -52,6 +130,7 @@ func main() {
 			http.NotFound(w, r)
 		}
 	})
+
 	mux.HandleFunc("/api/ws", h.WebSocket)
 
 	srv := &http.Server{
@@ -82,3 +161,10 @@ func main() {
 	gameMgr.Stop()
 	log.Println("Server stopped")
 }
+
+func getEnv(key, defaultVal string) string {
+	if val := os.Getenv(key); val != "" {
+		return val
+	}
+	return defaultVal
+}