Feat:finish task

sh4ll0t/api/admin/admin.go

@@ -0,0 +1,27 @@
+package admin
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+)
+
+func Admin(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+	if session.Get("username") != "admin" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "管理员才可以访问！"})
+		return
+	}
+	var questions []db.Question
+	if err := db.DB.Preload("Answers").Find(&questions).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, questions)
+
+}

sh4ll0t/api/admin/check_answer.go

@@ -0,0 +1,47 @@
+package admin
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+	"net/http"
+	"sh4ll0t/db"
+	"strconv"
+)
+
+func CheckAnswer(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+	if session.Get("username") != "admin" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "管理员才可以访问！"})
+		return
+	}
+
+	idStr := c.PostForm("id")
+	checkStatusStr := c.PostForm("check")
+	checkStatus, err := strconv.Atoi(checkStatusStr)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "无效的审核状态"})
+		return
+	}
+
+	var answer db.Answer
+	if err := db.DB.First(&answer, idStr).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			c.JSON(http.StatusNotFound, gin.H{"error": "ID 不存在"})
+		} else {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		}
+		return
+	}
+	answer.CheckStatus = checkStatus
+	if err := db.DB.Save(&answer).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "审核状态更新成功"})
+}

sh4ll0t/api/admin/check_question.go

@@ -0,0 +1,47 @@
+package admin
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+	"net/http"
+	"sh4ll0t/db"
+	"strconv"
+)
+
+func CheckQuestion(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+	if session.Get("username") != "admin" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "管理员才可以访问！"})
+		return
+	}
+
+	checkStatusStr := c.PostForm("check")
+	checkStatus, err := strconv.Atoi(checkStatusStr)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "无效的审核状态"})
+		return
+	}
+
+	idStr := c.PostForm("id")
+	var question db.Question
+	if err := db.DB.First(&question, idStr).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			c.JSON(http.StatusNotFound, gin.H{"error": "ID 不存在"})
+		} else {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		}
+		return
+	}
+	question.CheckStatus = checkStatus
+	if err := db.DB.Save(&question).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "审核状态更新成功"})
+}

sh4ll0t/api/answer/answer.go

@@ -0,0 +1,37 @@
+package answer
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+	"strconv"
+)
+
+func Answer(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+	respondent := session.Get("username").(string)
+	idStr := c.PostForm("id")
+	answerText := c.PostForm("answer")
+	questionID, err := strconv.ParseUint(idStr, 10, 32)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "无效的问题 ID"})
+		return
+	}
+
+	answer := db.Answer{
+		QuestionID: uint(questionID),
+		AnswerText: answerText,
+		Respondent: respondent,
+	}
+
+	if err := db.DB.Create(&answer).Error; err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "回答失败"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"message": "回答成功"})
+}

sh4ll0t/api/answer/change_answer.go

@@ -0,0 +1,36 @@
+package answer
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+)
+
+func ChangeAnswer(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+	id := c.PostForm("id")
+	newAnswerText := c.PostForm("Answer")
+	var answer db.Answer
+	if err := db.DB.First(&answer, id).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	if session.Get("username") != answer.Respondent {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "只有本作者才可以修改！"})
+		return
+	}
+
+	answer.AnswerText = newAnswerText
+	if err := db.DB.Save(&answer).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "修改成功"})
+}

sh4ll0t/api/answer/delete_answer.go

@@ -0,0 +1,31 @@
+package answer
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+)
+
+func DeleteAnswer(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+
+	id := c.PostForm("id")
+	var answer db.Answer
+	if err := db.DB.First(&answer, id).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	if session.Get("username") != answer.Respondent {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "只有本作者才可以删除！"})
+		return
+	}
+
+	db.DB.Delete(&db.Answer{}, "id = ?", id)
+	db.DB.Delete(&answer)
+	c.JSON(http.StatusOK, gin.H{"message": "删除成功"})
+}

sh4ll0t/api/answer/search_answer.go

@@ -0,0 +1,37 @@
+package answer
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+)
+
+func SearchAnswer(c *gin.Context) {
+	answer := c.PostForm("answer")
+	var answers []db.Answer
+	var questions []db.Question
+	err := db.DB.Where("answer_text LIKE ? AND `check` = ?", "%"+answer+"%", 1).Find(&answers).Error
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err})
+		return
+	}
+	if len(answers) == 0 {
+		c.JSON(http.StatusNotFound, gin.H{"message": "未查询到相关答案"})
+		return
+	}
+	err = db.DB.Preload("Answers").Where("id IN ?", getQuestionIDs(answers)).Find(&questions).Error
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, questions)
+}
+
+func getQuestionIDs(answers []db.Answer) []uint {
+	var ids []uint
+	for _, answer := range answers {
+		ids = append(ids, answer.QuestionID)
+	}
+	return ids
+}

sh4ll0t/api/question/ask.go

@@ -0,0 +1,58 @@
+package question
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+	"sh4ll0t/utils"
+)
+
+func Ask(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+
+	username, ok := session.Get("username").(string)
+	if !ok || username == "" {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "用户名不存在"})
+		return
+	}
+
+	questionText := c.PostForm("question")
+	if questionText == "" {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "问题内容不能为空"})
+		return
+	}
+
+	newQuestion := db.Question{
+		QuestionText: questionText,
+		Questioner:   username,
+	}
+
+	if err := db.DB.Create(&newQuestion).Error; err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	answerText, err := utils.GenerateAIAnswer(questionText)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "无法生成答案"})
+		return
+	}
+
+	answer := db.Answer{
+		QuestionID: newQuestion.ID,
+		AnswerText: answerText,
+		Respondent: "ai",
+	}
+
+	if err := db.DB.Create(&answer).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "答案保存失败"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "提问成功"})
+}

sh4ll0t/api/question/change_question.go

@@ -0,0 +1,38 @@
+package question
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"sh4ll0t/db"
+)
+
+func ChangeQuestion(c *gin.Context) {
+	session := sessions.Default(c)
+	if auth, ok := session.Get("authenticated").(bool); !ok || !auth {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
+		return
+	}
+
+	id := c.PostForm("id")
+	questionText := c.PostForm("question")
+
+	var existingQuestion db.Question
+	if err := db.DB.First(&existingQuestion, "id = ?", id).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	if session.Get("username") != existingQuestion.Questioner {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "只有本作者才可以修改！"})
+		return
+	}
+
+	existingQuestion.QuestionText = questionText
+	if err := db.DB.Save(&existingQuestion).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "修改成功"})
+}