default

api/admin/challenge.go

@@ -0,0 +1,79 @@
+package admin
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"main.go/api"
+	db "main.go/database"
+	. "main.go/type"
+	"strings"
+)
+
+// CrDb 专门用作给数据替换。
+type CrDb struct {
+	Name        string   `json:"name" binding:"required"`
+	Score       int      `json:"score" binding:"required"`
+	Flag        string   `json:"flag"` // 暂时一个题只能一个flag
+	Description string   `json:"description"`
+	Attachment  string `json:"attachment"`
+	Category    string   `json:"category" binding:"required"`
+	Tags        string   `json:"tags"`
+	Hints       string `json:"hints"`
+	Visible     int      `json:"visible"`
+}
+
+
+//NewChallenge 新建一个题目
+func NewChallenge(c *gin.Context){
+	var request ChallengeRequest
+	if err := c.ShouldBindJSON(&request); err != nil {
+		fmt.Println(err)
+		c.JSON(400, gin.H{"code": 400, "msg": "Request format wrong!"})
+		return
+	}
+	challenge := &Challenge{
+		Name:        request.Name,
+		Score:       request.Score,
+		Flag:        request.Flag,
+		Description: request.Description,
+		Attachment:  request.Attachment,
+		Category:    request.Category,
+		Tags:        request.Tags,
+		Hints:       request.Hints,
+		Visible:     request.Visible,
+	}
+	if err := addChallenge(challenge); err != nil {
+		c.JSON(400, gin.H{"code": 400, "msg": "Add challenge failure!"})
+		return
+	}
+
+	c.JSON(200, gin.H{"code": 200, "msg": "Add challenge success!"})
+}
+
+//addChallenge 添加题目内容
+func addChallenge(c *Challenge) error {
+	api.Link()
+	DB := db.DBsnctf
+	// 使用逗号分隔字符串
+	attachmentString := strings.Join(c.Attachment, ",")
+	hintString := strings.Join(c.Hints, ",")
+	crdb := &CrDb{
+		Name: c.Name,
+		Score: c.Score,
+		Flag: c.Flag,
+		Description: c.Description,
+		Attachment: attachmentString,
+		Category: c.Category,
+		Tags: c.Tags,
+		Hints: hintString,
+		Visible: c.Visible,
+	}
+	//插入数据
+	err := DB.Table("challenge").Create(crdb).Error
+	//command := "INSERT INTO challenge (name,score,flag,description,attachment,category,tags,hints,visible) VALUES (?,?,?,?,?,?,?,?,?);"
+	//err := DB.Debug().Raw(command,c.Name, c.Score, c.Flag, c.Description, attachmentString, c.Category, c.Tags, hintString, c.Visible).Error
+	if err != nil {
+		return err
+	}
+	return nil
+}