task_track/backend/internal/handler/auth/test_login.go

package auth

import (
	"task-track-backend/model"
	"task-track-backend/pkg/auth"

	"github.com/gin-gonic/gin"
	"golang.org/x/crypto/bcrypt"
	"gorm.io/gorm"
)

func (h *AuthHandler) TestLogin(c *gin.Context) {
	// 创建测试用户（如果不存在）
	var user model.User
	err := h.db.Where("username = ?", "test").First(&user).Error
	if err == gorm.ErrRecordNotFound {
		// 创建测试用户
		hashedPassword, _ := bcrypt.GenerateFromPassword([]byte("test123"), bcrypt.DefaultCost)
		user = model.User{
			Username: "test",
			Email:    "test-user@example.com",
			Password: string(hashedPassword),
			RealName: "测试用户",
			Status:   1,
		}
		if err := h.db.Create(&user).Error; err != nil {
			c.JSON(500, gin.H{"code": 500, "message": "Failed to create test user", "error": err.Error()})
			return
		}
	}

	// 获取用户的默认组织ID（如果没有关联组织，使用0）
	var organizationID uint = 0
	var userOrg model.UserOrganization
	if err := h.db.Where("user_id = ?", user.ID).First(&userOrg).Error; err == nil {
		organizationID = userOrg.OrganizationID
	}

	// 生成 JWT token
	token, err := auth.GenerateToken(user.ID, user.Username, organizationID)
	if err != nil {
		c.JSON(500, gin.H{"code": 500, "message": "Failed to generate token", "error": err.Error()})
		return
	}

	c.JSON(200, gin.H{
		"code":    200,
		"message": "Test login successful",
		"data": gin.H{
			"token": token,
			"user":  user,
		},
	})
}