certificate-management/ca-server/internal/handlers/login_handler.go

package handlers

import (
	"ca-mini/pkg/utils"
	"encoding/json"
	"fmt"
	"net/http"
	"time"
)

type LoginRequest struct {
	Username string `json:"username"`
	Password string `json:"password"`
}

type TokenInfo struct {
	Token string `json:"token"`
}

type UserInfo struct {
	Name         string   `json:"name"`
	Introduction string   `json:"introduction"`
	Avatar       string   `json:"avatar"`
	Roles        []string `json:"roles"`
}

type LoginResponse struct {
	Code    int       `json:"code"`
	Message string    `json:"message"`
	Date    string    `json:"date"`
	Data    TokenInfo `json:"data"`
}
type LogutResponse struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
}

type UserInfoResponse struct {
	Code    int      `json:"code"`
	Message string   `json:"message"`
	Date    string   `json:"date"`
	Data    UserInfo `json:"data"`
}

// IssueCertificate 处理证书签发请求
func Login(w http.ResponseWriter, r *http.Request) {

	// 解析CSR请求
	var loginRequest LoginRequest
	if err := json.NewDecoder(r.Body).Decode(&loginRequest); err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	// 检查用户
	if !CheckUser(loginRequest) {
		http.Error(w, "Unauthorized", http.StatusUnauthorized)
		return
	}

	// 生成Token
	token, err := utils.GenerateRandomString(32)
	if err != nil {
		http.Error(w, fmt.Sprintf("Failed to generate token: %v", err), http.StatusInternalServerError)
		return
	}
	// 返回Token信息
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)
	response := LoginResponse{
		Code:    http.StatusOK,
		Message: "Login successful",
		Date:    time.Now().Format(time.RFC3339),
		Data:    TokenInfo{Token: token},
	}
	json.NewEncoder(w).Encode(response)
}

func Logout(w http.ResponseWriter, r *http.Request) {
	// 这里可以添加注销逻辑，例如清除用户会话等
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)
	response := LogutResponse{
		Code:    http.StatusOK,
		Message: "Logout successful",
	}
	json.NewEncoder(w).Encode(response)
}

// CheckBlacklist 处理黑名单查询请求
func CheckUser(login LoginRequest) bool {
	if login.Username == "admin" && login.Password == "111111" {
		return true
	}
	return false
}

func GetUserInfo(w http.ResponseWriter, r *http.Request) {
	// 假设用户信息存储在某个地方，这里直接返回一个示例用户信息
	userInfo := UserInfo{
		Name:         "Admin User",
		Introduction: "This is an admin",
		Avatar:       "https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif",
		Roles:        []string{"admin"},
	}
	// 返回Token信息
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)
	response := UserInfoResponse{
		Code:    http.StatusOK,
		Message: "Login successful",
		Date:    time.Now().Format(time.RFC3339),
		Data:    userInfo,
	}
	json.NewEncoder(w).Encode(response)
}