package main

import (
    "github.com/dormoron/mist"
)

func main() {
    // 初始化服务器
    server := mist.InitHTTPServer()
    
    // 注册路由
    server.GET("/", func(ctx *mist.Context) {
        ctx.RespondWithJSON(200, map[string]string{
            "message": "Hello, Mist!",
        })
    })
    
    // 添加中间件
    server.Use(middlewares.Recovery())
    server.Use(middlewares.AccessLog())
    
    // 启动服务器
    server.Start(":8080")
}

// 参数路由
server.GET("/users/:id", func(ctx *mist.Context) {
    id := ctx.PathValue("id").String()
    ctx.RespondWithJSON(200, map[string]string{
        "user_id": id,
    })
})

// 正则表达式路由
server.GET("/posts/{id:[0-9]+}", func(ctx *mist.Context) {
    id := ctx.PathValue("id").Int()
    ctx.RespondWithJSON(200, map[string]string{
        "post_id": fmt.Sprintf("%d", id),
    })
})

// 路由分组
api := server.Group("/api")
api.Use(middlewares.Auth.JWT())

// 用户相关路由
users := api.Group("/users")
users.GET("/", listUsers)
users.POST("/", createUser)
users.GET("/:id", getUser)
users.PUT("/:id", updateUser)
users.DELETE("/:id", deleteUser)

server := mist.InitHTTPServer()
// 配置 HTTP/3
config := mist.DefaultHTTP3Config()
config.MaxIdleTimeout = 60 * time.Second

// 启动 HTTP/3 服务器
err := server.StartHTTP3(":443", "cert.pem", "key.pem", config)
if err != nil {
    log.Fatalf("HTTP/3 启动失败: %v", err)
}

server.GET("/download/:file", func(ctx *mist.Context) {
    filePath := "path/to/files/" + ctx.PathValue("file").String()
    zr := mist.NewZeroCopyResponse(ctx.ResponseWriter)
    err := zr.ServeFile(filePath)
    if err != nil {
        ctx.RespondWithJSON(500, map[string]string{
            "error": err.Error(),
        })
    }
})

monitor := mist.NewMemoryMonitor(10*time.Second, 60)
monitor.AddAlertCallback(func(stats mist.MemStats, message string) {
    log.Printf("内存警告: %s, 已分配: %d 字节", message, stats.Alloc)
})
monitor.Start()
defer monitor.Stop()

// 创建速率限制中间件
limiter := middlewares.RateLimit.NewFixedWindow(middlewares.RateLimit.Config{
    MaxRequests: 100,        // 每窗口最大请求数
    Window:      time.Minute, // 窗口大小
    KeyFunc: func(ctx *mist.Context) string {
        return ctx.ClientIP() // 基于客户端IP限流
    },
})

// 应用到所有路由
server.Use(limiter)

// 或仅应用到特定路由组
apiGroup := server.Group("/api")
apiGroup.Use(limiter)

// 使用默认 CORS 配置
server.Use(middlewares.CORS.Default())

// 或使用自定义配置
corsConfig := middlewares.CORS.Config{
    AllowOrigins:     []string{"https://example.com"},
    AllowMethods:     []string{"GET", "POST", "PUT", "DELETE"},
    AllowHeaders:     []string{"Origin", "Content-Type", "Authorization"},
    ExposeHeaders:    []string{"Content-Length"},
    AllowCredentials: true,
    MaxAge:           12 * time.Hour,
}
server.Use(middlewares.CORS.New(corsConfig))

// 全局限制请求体大小为1MB
server.Use(middlewares.BodyLimit.New("1MB"))

// 或使用自定义配置
config := middlewares.BodyLimit.Config{
    MaxSize:        2 * 1024 * 1024, // 2MB
    WhitelistPaths: []string{"/upload"},
}
server.Use(middlewares.BodyLimit.NewWithConfig(config))