feat:finish task

Rrrrr226/Account_System/forum/answer/answer.go

@@ -0,0 +1 @@
+package answer

Rrrrr226/Account_System/forum/create/create.go

@@ -0,0 +1,36 @@
+package create
+
+import (
+	"github.com/gin-gonic/gin"
+	"goexample/Account_System/forum/initial"
+	"net/http"
+)
+
+func Create(c *gin.Context) {
+	if c.Request.Method != "POST" {
+		c.AbortWithStatusJSON(http.StatusMethodNotAllowed, gin.H{"error": "Method not allowed"})
+		return
+	}
+	//Bind方法解析表单数据到结构体
+	var newQuestion initial.Question
+	if err := c.Bind(&newQuestion.QuestionInfo); err != nil {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "Invalid request body"})
+		return
+	}
+
+	newQuestion.QuestionInfo.Content = c.PostForm("question")
+	if newQuestion.QuestionInfo.Content == "" {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "Question is required"})
+		return
+	}
+
+	newQuestion.QuestionInfo.UserName = c.PostForm("username")
+
+	result := initial.Dbq.Create(&newQuestion)
+	if result.Error != nil {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": result.Error.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "Your Question successfully created"})
+}

Rrrrr226/Account_System/forum/distribute/distribute.go

@@ -0,0 +1,58 @@
+package distribute
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"goexample/Account_System/forum/initial"
+	"goexample/Account_System/forum/readall"
+)
+
+/*func Tosql() *gorm.DB {
+	db, err := gorm.Open("mysql", "root:123456789@tcp(127.0.0.1:3306)/dbquestion?charset=utf8&parseTime=True&loc=Local")
+	if err != nil {
+		panic(err)
+	}
+	return db
+}*/
+
+func Finduser(Username string) (*initial.QuestionInfo, error) {
+	db := initial.Dbq
+	var user initial.QuestionInfo
+	err := db.Where("username=?", Username).First(&user).Error
+	if err != nil {
+		return nil, err
+	}
+	if user.UserName != "" {
+		return &user, nil
+	}
+	return nil, fmt.Errorf("user not found")
+}
+
+func FindQuestion(Username string) *initial.Question {
+	db := initial.Dbq
+	var question initial.Question
+	db.Where("username=?", Username).First(&question)
+	return &question
+}
+
+func Distribute(c *gin.Context) {
+	username := c.Param("username")
+	if user, _ := Finduser(username); user != nil {
+		readall.GetAllQuestions()
+
+		c.JSON(200, gin.H{
+			"user":     "found",
+			"username": username,
+			"question": readall.GetAllQuestions(),
+		})
+
+		c.JSON(200, gin.H{
+			"question": FindQuestion(username),
+		})
+
+		return
+	}
+
+	c.JSON(200, gin.H{"message": "User not found:" + username})
+
+}

Rrrrr226/Account_System/forum/initial/initial.go

@@ -0,0 +1,34 @@
+package initial
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/jinzhu/gorm"
+)
+
+var Dbq *gorm.DB
+
+type Question struct {
+	gorm.Model
+	QuestionInfo QuestionInfo
+	AnswerInfo   AnswerInfo
+}
+
+type QuestionInfo struct {
+	Content  string
+	UserName string
+}
+
+type AnswerInfo struct {
+	Content  string
+	UserName string
+}
+
+func Initial() error {
+	var err error
+	Dbq, err = gorm.Open("mysql", "root:123456789@tcp(127.0.0.1:3306)/dbquestion?charset=utf8&parseTime=True&loc=Local")
+	if err != nil {
+		return err
+	}
+	Dbq.AutoMigrate(&Question{})
+	return nil
+}

Rrrrr226/Account_System/forum/login/login.go

@@ -0,0 +1,103 @@
+package login
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/mysql"
+	"net/http"
+)
+
+type User struct {
+	gorm.Model
+	Username string
+	Password string
+}
+
+var db *gorm.DB
+
+/*func InitDB() {
+	var err error
+	db, err = gorm.Open("mysql", "user:password@tcp(127.0.0.1:24306)/main_db?charset=utf8&parseTime=True&loc=Local")
+	if err != nil {
+		panic(err)
+	}
+	db.AutoMigrate(&User{})
+	/*u0 := User{Username: "Admin", Password: "hdu123"}
+	result := db.Create(&u0)
+	if result.Error != nil {
+		fmt.Println(result.Error)
+	} else {
+		fmt.Printf("User created with ID: %v\n", u0.ID)
+	}
+}*/
+
+// 检查是否匹配
+func findUser(Username string) *User {
+	var user User
+	db.Where("username=?", Username).First(&user)
+	if user.Username != "" {
+		return &user
+	}
+	return nil
+}
+
+func authenticate(Username, Password string) bool {
+	user := findUser(Username)
+	return user != nil && user.Password == Password
+}
+
+func LoginHandler(c *gin.Context) {
+	if c.Request.Method != "POST" {
+		c.AbortWithStatusJSON(http.StatusMethodNotAllowed, gin.H{"error": "Method not allowed"})
+		return
+	}
+	username := c.PostForm("Username")
+	password := c.PostForm("Password")
+	if username == "" || password == "" {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "Username or password is empty"})
+		return
+	}
+	if authenticate(username, password) {
+		//http.Redirect(w, r, "/forum/:username", http.StatusFound)
+		_, _ = fmt.Println("Welcome to forum!")
+		//c.Redirect(http.StatusFound, "/forum/"+username)
+	} else {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid username or password"})
+	}
+}
+
+/*func sayHello(w http.ResponseWriter, r *http.Request) {
+	b, err := ioutil.ReadFile("./Hello.txt")
+	if err != nil {
+		//成功
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	fmt.Fprintln(w, string(b))
+}*/
+
+func Registerhandler(c *gin.Context) {
+	if c.Request.Method != "POST" {
+		c.AbortWithStatusJSON(http.StatusMethodNotAllowed, gin.H{"error": "Method not allowed"})
+		return
+	}
+	username := c.PostForm("Username")
+	password := c.PostForm("Password")
+	if username == "" || password == "" {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "Username or password is empty"})
+		return
+	}
+	if findUser(username) != nil {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "Username already exists"})
+		return
+	}
+
+	newuser := User{Username: username, Password: password}
+	result := db.Create(&newuser)
+	if result.Error != nil {
+		c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": result.Error.Error()})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"message": "User registered successfully", "username": newuser.Username})
+}

Rrrrr226/Account_System/forum/readall/readall.go

@@ -0,0 +1,37 @@
+package readall
+
+import (
+	"github.com/gin-gonic/gin"
+	"goexample/Account_System/forum/initial"
+)
+
+/*func ReadAll() {
+//initial.Init()
+
+
+
+ReadAll() func(c *gin.Context) {
+	id := c.Param("id")
+
+	var questions []initial.Question
+	if err := initial.Dbq.Find(&questions).Error;err!=nil{
+		c.JSON(500,gin.H{"error":err.Error()})
+		return
+	}
+	c.JSON(200,questions)
+}*/
+
+func GetAllQuestions() gin.HandlerFunc {
+	/*db, err := gorm.Open("mysql", "root:123456789@tcp(127.0.0.1:3306)/dbquestion?charset=utf8&parseTime=True&loc=Local")
+	if err != nil {
+		panic(err)
+	}*/
+	return func(c *gin.Context) {
+		var questions []initial.Question
+		if err := initial.Dbq.Find(&questions).Error; err != nil {
+			c.JSON(500, gin.H{"error": err.Error()})
+			return
+		}
+		c.JSON(200, questions)
+	}
+}

Rrrrr226/Forum/AI/AI.go

@@ -0,0 +1,30 @@
+package AI
+
+import (
+	"github.com/gin-gonic/gin"
+	_ "github.com/go-sql-driver/mysql"
+	"goexample/Forum/InitDB"
+	"goexample/Forum/Models"
+	"net/http"
+)
+
+// 假设的AI API响应结构
+type AIResponse struct {
+	Content string `json:"answer"`
+}
+
+var aiAPIURL = "https://yiyan.baidu.com/" // 替换为你的AI API URL
+
+func Aiimport(c *gin.Context) {
+	id := c.Param("id")
+	result := InitDB.Db.Where("id=?", id).Find(&Models.Question{})
+	if result.Error != nil {
+		if result.RecordNotFound() {
+			c.JSON(http.StatusNotFound, gin.H{"error": "no questions found with this question_id"})
+		} else {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "database error"})
+		}
+		return
+	}
+
+}

Rrrrr226/Forum/InitDB/InitDB.go

@@ -0,0 +1,20 @@
+package InitDB
+
+import (
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/jinzhu/gorm"
+	"goexample/Forum/Models"
+	"log"
+)
+
+var Db *gorm.DB
+
+func InitDB() error {
+	var err error
+	Db, err = gorm.Open("mysql", "root:123456789@tcp(127.0.0.1:3306)/dbquestion?charset=utf8&parseTime=True&loc=Local")
+	if err != nil {
+		log.Fatal(err)
+	}
+	Db.AutoMigrate(&Models.Question{}, &Models.UserLogin{})
+	return nil
+}

Rrrrr226/Forum/Login/Login.go

@@ -0,0 +1,37 @@
+package Login
+
+import (
+	"github.com/gin-gonic/gin"
+	"goexample/Forum/InitDB"
+	"goexample/Forum/Models"
+	"goexample/Forum/Token"
+	"net/http"
+)
+
+func LoginHandler(c *gin.Context) {
+	if c.Request.Method != "POST" {
+		c.JSON(http.StatusMethodNotAllowed, gin.H{"error": "Method not allowed"})
+		return
+	}
+
+	var login Models.UserLogin
+	if err := c.ShouldBindJSON(&login); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	var user Models.UserLogin
+	result := InitDB.Db.Where("username = ? AND password = ?", login.Username, login.Password).First(&user)
+	if result.Error != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "unauthorized"})
+		return
+	}
+
+	token, err := Token.GenerateToken(user.ID, user.Username)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to generate token"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"status": "success", "token": token})
+}

Rrrrr226/Forum/Middleware/Middleware.go

@@ -0,0 +1,34 @@
+package Middleware
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"goexample/Forum/Token"
+	"net/http"
+	"strings"
+)
+
+func AuthMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		token := c.GetHeader("Authorization")
+		if token == "" {
+			c.JSON(http.StatusUnauthorized, gin.H{"error": "Authorization token is missing"})
+			c.Abort()
+			return
+		}
+		token = strings.Split(token, " ")[1]
+
+		claims, err := Token.ParseToken(token)
+		if err != nil {
+			c.JSON(http.StatusUnauthorized, gin.H{"error": "invalid or expired token"})
+			c.Abort()
+			return
+		}
+		userIDStr := claims.Id
+		fmt.Println("User ID from token:", userIDStr)
+
+		// 设置用户 ID 到上下文中，以便后续中间件或处理器使用
+		c.Set("userId", userIDStr)
+		c.Next()
+	}
+}