up

2023-03-28 22:46:22 +08:00
parent 58835ee0d4
commit 2ba63e226d
6 changed files with 492 additions and 5 deletions
--- a/opencatd/db/cat.go
+++ b/opencatd/db/cat.go
@@ -0,0 +1,132 @@
+package db
+
+import (
+	"database/sql"
+	"fmt"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+type User struct {
+	ID        int    `json:"id"`
+	Name      string `json:"name"`
+	Token     string `json:"token"`
+	CreatedAt string `json:"createdAt"`
+	UpdatedAt string `json:"updatedAt"`
+	IsDelete  bool   `json:"isDelete"`
+}
+
+func addUser(c *gin.Context) {
+	var user User
+	if err := c.ShouldBindJSON(&user); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	db, err := sql.Open("sqlite3", "test.db")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	stmt, err := db.Prepare("INSERT INTO users(name, token, created_at, updated_at, is_delete) values(?,?,?,?,?)")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	res, err := stmt.Exec(user.Name, user.Token, user.CreatedAt, user.UpdatedAt, user.IsDelete)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	id, err := res.LastInsertId()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	user.ID = int(id)
+
+	c.JSON(http.StatusOK, user)
+}
+
+func updateUser(c *gin.Context) {
+	id := c.Param("id")
+
+	var user User
+	if err := c.ShouldBindJSON(&user); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	db, err := sql.Open("sqlite3", "test.db")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	stmt, err := db.Prepare("UPDATE users SET name=?, token=?, created_at=?, updated_at=?, is_delete=? WHERE id=?")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	_, err = stmt.Exec(user.Name, user.Token, user.CreatedAt, user.UpdatedAt, user.IsDelete, id)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	c.JSON(http.StatusOK, user)
+}
+
+func deleteUser(c *gin.Context) {
+	id := c.Param("id")
+
+	db, err := sql.Open("sqlite3", "test.db")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	stmt, err := db.Prepare("DELETE FROM users WHERE id=?")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	_, err = stmt.Exec(id)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "ok"})
+}
+
+func initDB() {
+	db, err := sql.Open("sqlite3", "test.db")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer db.Close()
+
+	_, err = db.Exec("CREATE TABLE IF NOT EXISTS users(id INTEGER PRIMARY KEY, name TEXT, token TEXT, created_at DATETIME, updated_at DATETIME, is_delete BOOLEAN)")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	shouldBeDeleted := false
+	now := time.Now().Format(time.RFC3339)
+	stmt, err := db.Prepare("INSERT INTO users(name, token, created_at, updated_at, is_delete) values(?,?,?,?,?)")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	res, err := stmt.Exec("Test User", "123456789", now, now, shouldBeDeleted)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	id, err := res.LastInsertId()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Printf("New user ID: %d\n", id)
+}
--- a/opencatd/opencat.go
+++ b/opencatd/opencat.go
@@ -0,0 +1,172 @@
+package main
+
+import (
+	"net/http"
+	"strconv"
+
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+)
+
+type User struct {
+	IsDelete  bool   `json:"IsDelete"`
+	ID        int    `json:"id"`
+	UpdatedAt string `json:"updatedAt"`
+	Name      string `json:"name"`
+	Token     string `json:"token"`
+	CreatedAt string `json:"createdAt"`
+}
+
+type Key struct {
+	ID        int    `json:"id"`
+	Key       string `json:"key"`
+	UpdatedAt string `json:"updatedAt"`
+	Name      string `json:"name"`
+	CreatedAt string `json:"createdAt"`
+}
+
+var users = []User{
+	{false, 1, "2023-03-28T12:36:03.907590825Z", "root", "4930f144-7ffc-469b-8f2d-71a267358d87", "2023-03-28T11:52:45.593940257Z"},
+}
+
+var keys = []Key{
+	{1, "1234567890", "2023-03-28T12:53:41.124319079Z", "key1", "2023-03-28T12:53:41.124318928Z"},
+}
+
+func main() {
+	router := gin.Default()
+
+	// 初始化用户
+	router.POST("/1/users/init", func(c *gin.Context) {
+		resJSON := User{
+			false,
+			1,
+			"2023-03-28T14:13:54.969089197Z",
+			"root",
+			"3260dd53-463f-45d0-afa4-57acd04ee036",
+			"2023-03-28T14:13:54.969089107Z",
+		}
+
+		c.JSON(http.StatusOK, resJSON)
+	})
+
+	// 获取当前用户信息
+	router.GET("/1/me", func(c *gin.Context) {
+		resJSON := users[0]
+
+		c.JSON(http.StatusOK, resJSON)
+	})
+
+	// 获取所有Key
+	router.GET("/1/keys", func(c *gin.Context) {
+		c.JSON(http.StatusOK, keys)
+	})
+
+	// 获取所有用户信息
+	router.GET("/1/users", func(c *gin.Context) {
+		c.JSON(http.StatusOK, users)
+	})
+
+	// 添加Key
+	router.POST("/1/keys", func(c *gin.Context) {
+		var body Key
+		if err := c.BindJSON(&body); err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+			return
+		}
+
+		keys = append(keys, body)
+		c.JSON(http.StatusCreated, body)
+	})
+
+	// 删除Key
+	router.DELETE("/1/keys/:id", func(c *gin.Context) {
+		id := c.Param("id")
+		var keyIdx int = -1
+
+		for idx, key := range keys {
+			if strconv.Itoa(key.ID) == id {
+				keyIdx = idx
+				break
+			}
+		}
+
+		if keyIdx == -1 {
+			c.JSON(http.StatusNotFound, gin.H{"error": "Key not found"})
+			return
+		}
+
+		keys = append(keys[:keyIdx], keys[keyIdx+1:]...)
+		c.JSON(http.StatusOK, gin.H{"message": "ok"})
+	})
+
+	// 添加用户
+	router.POST("/1/users", func(c *gin.Context) {
+		var body User
+		if err := c.BindJSON(&body); err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+			return
+		}
+
+		newUser := User{
+			false,
+			len(users) + 1,
+			"2023-03-28T13:05:02.103135082Z",
+			body.Name,
+			GenerateToken(),
+			"2023-03-28T12:38:57.513105431Z",
+		}
+
+		users = append(users, newUser)
+		c.JSON(http.StatusCreated, newUser)
+	})
+
+	// 删除用户
+	router.DELETE("/1/users/:id", func(c *gin.Context) {
+		id := c.Param("id")
+		var userIdx int = -1
+
+		for idx, user := range users {
+			if strconv.Itoa(user.ID) == id {
+				userIdx = idx
+				break
+			}
+		}
+
+		if userIdx == -1 {
+			c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
+			return
+		}
+
+		users = append(users[:userIdx], users[userIdx+1:]...)
+		c.JSON(http.StatusOK, gin.H{"message": "ok"})
+	})
+
+	// 重置用户Token
+	router.POST("/1/users/:id/reset", func(c *gin.Context) {
+		id := c.Param("id")
+		var userIdx int = -1
+
+		for idx, user := range users {
+			if strconv.Itoa(user.ID) == id {
+				userIdx = idx
+				break
+			}
+		}
+
+		if userIdx == -1 {
+			c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
+			return
+		}
+
+		users[userIdx].Token = GenerateToken()
+		c.JSON(http.StatusOK, users[userIdx])
+	})
+
+	router.Run(":8080")
+}
+
+func GenerateToken() string {
+	token := uuid.New()
+	return token.String()
+}
--- a/opencatd/opencatd.c
+++ b/opencatd/opencatd.c
@@ -0,0 +1,108 @@
+docker pull debian:bullseye
+
+opencat://team/join?domain=http://168.138.211.226&token=f1ad7afa-65fe-436f-bbbb-8df8a35c332e
+
+"/1/me"
+"/1/keys"
+"/1/users"
+
+POST "/1/users/init"
+
+{
+  "IsDelete" : false,
+  "id" : 1,
+  "updatedAt" : "2023-03-28T14:13:54.969089197Z",
+  "name" : "root",
+  "token" : "3260dd53-463f-45d0-afa4-57acd04ee036",
+  "createdAt" : "2023-03-28T14:13:54.969089107Z"
+}
+
+
+GET "1/me"
+
+{
+  "IsDelete" : false,
+  "id" : 1,
+  "updatedAt" : "2023-03-28T12:36:03.907590825Z",
+  "name" : "root",
+  "token" : "4930f144-7ffc-469b-8f2d-71a267358d87",
+  "createdAt" : "2023-03-28T11:52:45.593940257Z"
+}
+
+GET "1/keys"
+
+[
+  {
+    "id" : 1,
+    "key" : "1234567890",
+    "updatedAt" : "2023-03-28T12:53:41.124319079Z",
+    "name" : "key1",
+    "createdAt" : "2023-03-28T12:53:41.124318928Z"
+  }
+]
+
+GET "1/users"
+
+[
+  {
+    "IsDelete" : false,
+    "id" : 1,
+    "updatedAt" : "2023-03-28T12:36:03.907590825Z",
+    "name" : "root",
+    "token" : "4930f144-7ffc-469b-8f2d-71a267358d87",
+    "createdAt" : "2023-03-28T11:52:45.593940257Z"
+  }
+]
+
+//添加key
+POST "/1/keys" body {"key":"1234567890","name":"key1"}
+
+{
+  "id" : 1,
+  "key" : "1234567890",
+  "updatedAt" : "2023-03-28T12:53:41.124319079Z",
+  "name" : "key1",
+  "createdAt" : "2023-03-28T12:53:41.124318928Z"
+}
+
+//删除KEY
+DELETE "1/keys/1"
+
+{"message":"ok"}
+
+//添加user
+POST "1/users" body{"name":"user1"}
+Host: 168.138.211.226
+User-Agent: OpenCat/99 CFNetwork/1404.0.5 Darwin/22.3.0
+Authorization: Bearer 4930f144-7ffc-469b-8f2d-71a267358d87
+
+{
+  "IsDelete" : false,
+  "id" : 2,
+  "updatedAt" : "2023-03-28T12:38:57.513105571Z",
+  "name" : "user1",
+  "token" : "f5bbee90-5f10-4501-bbe4-4989d0c41a82",
+  "createdAt" : "2023-03-28T12:38:57.513105431Z"
+}
+
+//删除user3
+DELETE "1/users/3"
+
+{"message":"ok"}
+
+//重置user token
+POST "/1/users/2/reset"
+
+{
+  "IsDelete" : false,
+  "id" : 2,
+  "updatedAt" : "2023-03-28T13:05:02.103135082Z",
+  "name" : "user1",
+  "token" : "87f0af9c-ae80-4754-8de8-2d9522b24dee",
+  "createdAt" : "2023-03-28T12:38:57.513105431Z"
+}
+
+http://localhost/v1/chat/completions
+
+./opencatd reset_root
+