Feat:finish task

sh4ll0t/api/admin/admin.go

@@ -0,0 +1,28 @@
+package admin
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"hduhelp_text/api/user"
+	"hduhelp_text/db"
+	"net/http"
+)
+
+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 []user.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,49 @@
+package admin
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+	"hduhelp_text/db"
+	"net/http"
+	"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 // 假设你在 Answer 结构体中添加了 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"
+	"hduhelp_text/db"
+	"net/http"
+	"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,39 @@
+package answer
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"hduhelp_text/db"
+	"net/http"
+	"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") // 获取问题 ID
+	answerText := c.PostForm("Answer")
+
+	// 将问题 ID 从 string 转换为 uint
+	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), // 转换为 uint
+		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"
+	"hduhelp_text/db"
+	"net/http"
+)
+
+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"
+	"hduhelp_text/db"
+	"net/http"
+)
+
+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 question db.Question
+	if err := db.DB.First(&question, id).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	if session.Get("username") != question.Questioner {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "只有本作者才可以删除！"})
+		return
+	}
+
+	db.DB.Delete(&db.Answer{}, "question_id = ?", id)
+	db.DB.Delete(&question)
+	c.JSON(http.StatusOK, gin.H{"message": "删除成功"})
+}

sh4ll0t/api/question/ask.go

@@ -0,0 +1,58 @@
+package question
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"hduhelp_text/db"
+	"hduhelp_text/utils"
+	"net/http"
+)
+
+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, // 确保这里是 uint 类型
+		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"
+	"hduhelp_text/db"
+	"net/http"
+)
+
+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": "修改成功"})
+}

sh4ll0t/api/question/delete_question.go

@@ -0,0 +1,37 @@
+package question
+
+import (
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"hduhelp_text/db"
+	"net/http"
+)
+
+func DeleteQuestion(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 username string
+	if err := db.DB.Model(&db.Question{}).Select("questioner").Where("id = ?", id).Scan(&username).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	if session.Get("username") != username {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "只有本作者才可以删除！"})
+		return
+	}
+	if err := db.DB.Delete(&db.Answer{}, "question_id = ?", id).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	if err := db.DB.Delete(&db.Question{}, id).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "删除成功"})
+}