61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package controllers
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"time"
|
|
|
|
"apiserver/common"
|
|
"apiserver/models"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func generateAPIKey() (string, error) {
|
|
bytes := make([]byte, 32)
|
|
if _, err := rand.Read(bytes); err != nil {
|
|
return "", err
|
|
}
|
|
return hex.EncodeToString(bytes), nil
|
|
}
|
|
|
|
type APIKeyController struct{}
|
|
|
|
func (a *APIKeyController) GenerateAPIKey(ctx *gin.Context) {
|
|
// Check authorization
|
|
token := ctx.GetHeader("Authorization")
|
|
if token == "" {
|
|
ReturnError(ctx, common.ERROR_AUTH_CHECK_TOKEN_CODE, "Unauthorized: No token provided", nil)
|
|
return
|
|
}
|
|
|
|
// Parse request
|
|
var req models.APIKeyRequest
|
|
if err := ctx.ShouldBindJSON(&req); err != nil {
|
|
ReturnError(ctx, common.INVALID_PARAMS_CODE, "Invalid request format: "+err.Error(), nil)
|
|
return
|
|
}
|
|
|
|
// Generate API key
|
|
apiKey, err := generateAPIKey()
|
|
if err != nil {
|
|
ReturnError(ctx, common.ERROR_CODE, "Failed to generate API key", nil)
|
|
return
|
|
}
|
|
|
|
// Set default expiry if not specified
|
|
expiryDays := 30
|
|
if req.ExpiryDays > 0 {
|
|
expiryDays = req.ExpiryDays
|
|
}
|
|
|
|
// Create response
|
|
response := models.APIKeyResponse{
|
|
APIKey: apiKey,
|
|
Username: req.Username,
|
|
CreateTime: time.Now(),
|
|
ExpiryTime: time.Now().AddDate(0, 0, expiryDays),
|
|
}
|
|
|
|
ReturnSuccess(ctx, 0, "API key generated successfully", response)
|
|
}
|