package main

import (
	"log"
	"net/http"
	"opencatd-open/router"
	"opencatd-open/store"
	"os"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
)

func main() {
	args := os.Args[1:]
	if len(args) > 0 && args[0] == "reset_root" {
		log.Println("reset root token...")
		ntoken := uuid.NewString()
		if err := store.UpdateUser(uint(1), ntoken); err != nil {
			log.Fatalln(err)
			return
		}
		log.Println("new root token:", ntoken)
		return
	}
	port := os.Getenv("PORT")
	r := gin.Default()
	group := r.Group("/1")
	{
		group.Use(router.AuthMiddleware())

		// 获取当前用户信息
		group.GET("/me", router.HandleMe)

		// 获取所有Key
		group.GET("/keys", router.HandleKeys)

		// 获取所有用户信息
		group.GET("/users", router.HandleUsers)

		group.GET("/usages", router.HandleUsage)

		// 添加Key
		group.POST("/keys", router.HandleAddKey)

		// 删除Key
		group.DELETE("/keys/:id", router.HandleDelKey)

		// 添加用户
		group.POST("/users", router.HandleAddUser)

		// 删除用户
		group.DELETE("/users/:id", router.HandleDelUser)

		// 重置用户Token
		group.POST("/users/:id/reset", router.HandleResetUserToken)
	}

	// 初始化用户
	r.POST("/1/users/init", router.Handleinit)

	r.POST("/v1/chat/completions", router.HandleProy)
	r.GET("/v1/models", router.HandleProy)
	r.GET("/v1/dashboard/billing/subscription", router.HandleProy)
	r.GET("/", func(c *gin.Context) {
		c.Writer.WriteHeader(http.StatusOK)
		c.Writer.WriteString(`<h1><a href="https://github.com/mirrors2/opencatd-open" >opencatd-open</a> available</h1>Api-Keys:<a href=https://platform.openai.com/account/api-keys >https://platform.openai.com/account/api-keys</a>`)
	})

	if port == "" {
		port = "80"
	}
	r.Run(":" + port)
}