diff --git a/pkg/models/servidor.go b/pkg/models/servidor.go index b9189dd..6f1770b 100755 --- a/pkg/models/servidor.go +++ b/pkg/models/servidor.go @@ -1,6 +1,8 @@ package models import ( + "crypto/rand" + "encoding/hex" "log" "time" @@ -8,6 +10,12 @@ import ( "gorm.io/gorm" ) +func generateAgentToken() string { + b := make([]byte, 16) + rand.Read(b) + return "agent_" + hex.EncodeToString(b) +} + type Servidor struct { gorm.Model Nombre string `json:"nombre" gorm:"column:nombre"` @@ -73,6 +81,9 @@ func GetAllServidoresSelect() ([]Servidor, error) { } func CreateServidor(servidor Servidor) error { + if servidor.AgentToken == "" { + servidor.AgentToken = generateAgentToken() + } if err := app.Http.Database.DB.Create(&servidor).Error; err != nil { return err }