feat(ai-config): cifra la API key en reposo y la acota por tenant
Prepara el terreno para que el cliente cargue su propia cuenta de IA sin que su clave quede en texto plano ni que el selector le muestre las de los demás. - ClaveEnClaro() descifra con fallback a texto plano: las filas viejas se leen igual y quedan cifradas al primer guardado, sin script ni downtime. utils.Decrypt hace panic (no devuelve error) con entrada que no es un ciphertext válido, así que el fallback va sobre recover — eso mismo es el mecanismo de detección de "todavía está en claro". - Migrados TODOS los lectores: uMind (chat y embeddings), bot de Telegram, Whisper, Query Runner, streaming de IA y Landing Generator. Un lector sin migrar mandaría el ciphertext como API key. - AiConfig gana TenantID (null = global del staff) y GetAiConfigSelectPorTenants para acotar el selector. - Test de los 4 casos del fallback, incluido hex válido que no descifra. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Sonnet 5
parent
c69b904b03
commit
96cd24f17d
@@ -6,6 +6,7 @@ import (
|
||||
"strings"
|
||||
|
||||
"github.com/sujit-baniya/fiber-boilerplate/app"
|
||||
"github.com/sujit-baniya/fiber-boilerplate/utils"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
@@ -29,10 +30,52 @@ type AiConfig struct {
|
||||
// Solo debe haber una config activa como agente a la vez.
|
||||
EsAgenteBot bool `gorm:"default:false" json:"es_agente_bot"`
|
||||
TelegramConfigID *uint `gorm:"index" json:"telegram_config_id"`
|
||||
// TenantID acota la config a un tenant de uMind: null = config global del
|
||||
// staff (el comportamiento histórico). Sin esto, el selector de IA le
|
||||
// mostraría a cada cliente las claves de todos los demás.
|
||||
TenantID *uint `gorm:"index" json:"tenant_id"`
|
||||
}
|
||||
|
||||
func (AiConfig) TableName() string { return "ai_configs" }
|
||||
|
||||
// ClaveEnClaro devuelve la API key lista para usar. Las filas guardadas antes
|
||||
// de que se cifrara este campo están en texto plano y se devuelven tal cual;
|
||||
// al volver a guardarlas quedan cifradas, así que el parque se migra solo sin
|
||||
// script ni downtime.
|
||||
//
|
||||
// utils.Decrypt hace panic con entrada que no sea un ciphertext válido (no
|
||||
// devuelve error), de ahí el recover: es el mecanismo de detección de "esto
|
||||
// todavía está en texto plano".
|
||||
func (c *AiConfig) ClaveEnClaro() string {
|
||||
if c.ApiKey == "" || app.Http.Server.Key == "" {
|
||||
return c.ApiKey
|
||||
}
|
||||
return descifrarOTalCual(c.ApiKey)
|
||||
}
|
||||
|
||||
func descifrarOTalCual(valor string) (out string) {
|
||||
defer func() {
|
||||
if recover() != nil {
|
||||
out = valor
|
||||
}
|
||||
}()
|
||||
claro := utils.Decrypt(valor, app.Http.Server.Key)
|
||||
if claro == "" {
|
||||
return valor
|
||||
}
|
||||
return claro
|
||||
}
|
||||
|
||||
// CifrarClaveAi cifra una API key para guardarla. Si no hay APP_KEY
|
||||
// configurada devuelve el valor tal cual — preferible a romper el guardado en
|
||||
// un entorno sin la clave, y ClaveEnClaro lo lee igual.
|
||||
func CifrarClaveAi(clave string) string {
|
||||
if clave == "" || app.Http.Server.Key == "" {
|
||||
return clave
|
||||
}
|
||||
return utils.Encrypt(clave, app.Http.Server.Key)
|
||||
}
|
||||
|
||||
func GetAllAiConfigs(limit, offset int, search string) ([]AiConfig, int64, error) {
|
||||
var items []AiConfig
|
||||
var total int64
|
||||
@@ -111,6 +154,23 @@ func GetAiConfigSelect() ([]AiConfig, error) {
|
||||
return items, nil
|
||||
}
|
||||
|
||||
// GetAiConfigSelectPorTenants acota el selector a las configs propias de esos
|
||||
// tenants más las globales del staff (tenant_id IS NULL), que son las que se
|
||||
// ofrecen a todos. Sin este filtro el cliente vería las claves de los demás.
|
||||
func GetAiConfigSelectPorTenants(tenantIDs []uint) ([]AiConfig, error) {
|
||||
var items []AiConfig
|
||||
db := app.Http.Database.DB.Model(&AiConfig{}).Select("id, nombre, provider, tenant_id")
|
||||
if len(tenantIDs) == 0 {
|
||||
db = db.Where("tenant_id IS NULL")
|
||||
} else {
|
||||
db = db.Where("tenant_id IS NULL OR tenant_id IN ?", tenantIDs)
|
||||
}
|
||||
if err := db.Order("nombre ASC").Find(&items).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
// GetAgenteBotConfig retorna la config marcada como agente Telegram, con su TelegramConfig cargada.
|
||||
func GetAgenteBotConfig() (*AiConfig, *TelegramConfig, error) {
|
||||
var ai AiConfig
|
||||
|
||||
@@ -0,0 +1,62 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/sujit-baniya/fiber-boilerplate/app"
|
||||
"github.com/sujit-baniya/fiber-boilerplate/config"
|
||||
"github.com/sujit-baniya/fiber-boilerplate/utils"
|
||||
)
|
||||
|
||||
// El resto del proyecto inicializa app.Http en el arranque; en tests hay que
|
||||
// hacerlo a mano antes de tocar cualquier cosa que lea la config.
|
||||
func conAppKey(t *testing.T, key string) {
|
||||
t.Helper()
|
||||
anterior := app.Http
|
||||
app.Http = &config.AppConfig{}
|
||||
app.Http.Server.Key = key
|
||||
t.Cleanup(func() { app.Http = anterior })
|
||||
}
|
||||
|
||||
// La migración progresiva de AiConfig.ApiKey depende de que ClaveEnClaro
|
||||
// distinga una clave cifrada de una que todavía está en texto plano. Si esto
|
||||
// se rompe, el sistema empieza a mandar ciphertext como API key a OpenAI y
|
||||
// todos los agentes dejan de responder.
|
||||
func TestClaveEnClaroMigracionProgresiva(t *testing.T) {
|
||||
conAppKey(t, "6368616e676520746869732070617373776f726420746f206120736563726574") // 32 bytes en hex
|
||||
|
||||
casos := []struct {
|
||||
nombre string
|
||||
guardado string
|
||||
esperado string
|
||||
descripci string
|
||||
}{
|
||||
{"vacío", "", "", "sin clave no hay nada que descifrar"},
|
||||
{"texto plano", "sk-proj-abc123", "sk-proj-abc123", "fila vieja sin cifrar, se devuelve tal cual"},
|
||||
{"cifrada", utils.Encrypt("sk-proj-abc123", app.Http.Server.Key), "sk-proj-abc123", "fila nueva, se descifra"},
|
||||
{"hex que no es ciphertext", "deadbeef", "deadbeef", "hex válido pero no descifrable: no debe romper"},
|
||||
}
|
||||
|
||||
for _, cas := range casos {
|
||||
t.Run(cas.nombre, func(t *testing.T) {
|
||||
cfg := &AiConfig{ApiKey: cas.guardado}
|
||||
if got := cfg.ClaveEnClaro(); got != cas.esperado {
|
||||
t.Errorf("%s: ClaveEnClaro() = %q, esperaba %q", cas.descripci, got, cas.esperado)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// Sin APP_KEY el guardado no debe romperse: se guarda en claro y se lee en
|
||||
// claro, que es exactamente el comportamiento previo a esta migración.
|
||||
func TestCifrarClaveAiSinAppKey(t *testing.T) {
|
||||
conAppKey(t, "")
|
||||
|
||||
if got := CifrarClaveAi("sk-test"); got != "sk-test" {
|
||||
t.Errorf("CifrarClaveAi sin APP_KEY = %q, esperaba pasarla tal cual", got)
|
||||
}
|
||||
cfg := &AiConfig{ApiKey: "sk-test"}
|
||||
if got := cfg.ClaveEnClaro(); got != "sk-test" {
|
||||
t.Errorf("ClaveEnClaro sin APP_KEY = %q, esperaba pasarla tal cual", got)
|
||||
}
|
||||
}
|
||||
@@ -1314,7 +1314,7 @@ func callOpenAICompatibleAI(ai *models.AiConfig, messages []agentMessage, tools
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
req.Header.Set("Authorization", "Bearer "+ai.ApiKey)
|
||||
req.Header.Set("Authorization", "Bearer "+ai.ClaveEnClaro())
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
client := &http.Client{Timeout: 120 * time.Second}
|
||||
@@ -1461,7 +1461,7 @@ func callAnthropicAI(ai *models.AiConfig, messages []agentMessage, tools []agent
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
req.Header.Set("x-api-key", ai.ApiKey)
|
||||
req.Header.Set("x-api-key", ai.ClaveEnClaro())
|
||||
req.Header.Set("anthropic-version", "2023-06-01")
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
|
||||
@@ -46,7 +46,7 @@ func GenerarEmbeddings(ai *models.AiConfig, textos []string) ([][]float32, error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
req.Header.Set("Authorization", "Bearer "+ai.ApiKey)
|
||||
req.Header.Set("Authorization", "Bearer "+ai.ClaveEnClaro())
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
|
||||
resp, err := umindEmbeddingsHTTPClient.Do(req)
|
||||
|
||||
@@ -26,7 +26,7 @@ func TranscribirAudio(ai *models.AiConfig, audioPath string) (string, error) {
|
||||
if ai == nil {
|
||||
return "", fmt.Errorf("no hay una configuración de Whisper activa: ve a /app/ai-config, crea o edita una y márcale el módulo 'Transcripción de audio (Whisper)'")
|
||||
}
|
||||
if ai.ApiKey == "" {
|
||||
if ai.ClaveEnClaro() == "" {
|
||||
return "", fmt.Errorf("la configuración de Whisper '%s' no tiene API key", ai.Nombre)
|
||||
}
|
||||
|
||||
@@ -66,7 +66,7 @@ func TranscribirAudio(ai *models.AiConfig, audioPath string) (string, error) {
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
req.Header.Set("Authorization", "Bearer "+ai.ApiKey)
|
||||
req.Header.Set("Authorization", "Bearer "+ai.ClaveEnClaro())
|
||||
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||||
|
||||
resp, err := whisperHTTPClient.Do(req)
|
||||
|
||||
Reference in New Issue
Block a user