- CoolifyWebhook parsea el payload y envía mensaje formateado a Telegram - Detecta estado (✅ success / ❌ fail / 🚀 deploying / ⏹ stop / 🔄 restart) - Ruta /webhooks/coolify/:config_id identifica de qué instancia viene - Notifica a todos los chats autorizados del agente bot - GetAgentTelegramConfig() en models para obtener el bot activo Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
98 lines
3.3 KiB
Go
98 lines
3.3 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/sujit-baniya/fiber-boilerplate/app"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// TelegramAgentHistory guarda el historial de conversación por chat_id.
|
|
// Se usan los últimos N mensajes como contexto en cada llamada al AI.
|
|
type TelegramAgentHistory struct {
|
|
gorm.Model
|
|
ChatID int64 `json:"chat_id" gorm:"column:chat_id;index;not null"`
|
|
Role string `json:"role" gorm:"column:role;not null"` // user | assistant | tool
|
|
Content string `json:"content" gorm:"column:content;type:text;not null"`
|
|
// ToolName y ToolResult se usan para mensajes de tipo "tool"
|
|
ToolName string `json:"tool_name" gorm:"column:tool_name"`
|
|
ToolResult string `json:"tool_result" gorm:"column:tool_result;type:text"`
|
|
}
|
|
|
|
func (TelegramAgentHistory) TableName() string { return "telegram_agent_history" }
|
|
|
|
// TelegramAgentChatID guarda qué chat_ids están autorizados a usar el agente.
|
|
type TelegramAgentAuth struct {
|
|
gorm.Model
|
|
ChatID int64 `json:"chat_id" gorm:"column:chat_id;uniqueIndex;not null"`
|
|
Nombre string `json:"nombre" gorm:"column:nombre"`
|
|
Activo bool `json:"activo" gorm:"column:activo;default:true"`
|
|
}
|
|
|
|
func (TelegramAgentAuth) TableName() string { return "telegram_agent_auth" }
|
|
|
|
// GetAgentHistory retorna los últimos n mensajes del historial para un chat_id.
|
|
func GetAgentHistory(chatID int64, n int) ([]TelegramAgentHistory, error) {
|
|
var items []TelegramAgentHistory
|
|
err := app.Http.Database.DB.
|
|
Where("chat_id = ?", chatID).
|
|
Order("created_at DESC").
|
|
Limit(n).
|
|
Find(&items).Error
|
|
// Invertir para orden cronológico
|
|
for i, j := 0, len(items)-1; i < j; i, j = i+1, j-1 {
|
|
items[i], items[j] = items[j], items[i]
|
|
}
|
|
return items, err
|
|
}
|
|
|
|
func SaveAgentMessage(chatID int64, role, content, toolName, toolResult string) error {
|
|
msg := &TelegramAgentHistory{
|
|
ChatID: chatID,
|
|
Role: role,
|
|
Content: content,
|
|
ToolName: toolName,
|
|
ToolResult: toolResult,
|
|
}
|
|
return app.Http.Database.DB.Create(msg).Error
|
|
}
|
|
|
|
// ClearAgentHistory borra el historial de un chat (comando /reset).
|
|
func ClearAgentHistory(chatID int64) error {
|
|
return app.Http.Database.DB.Where("chat_id = ?", chatID).Delete(&TelegramAgentHistory{}).Error
|
|
}
|
|
|
|
// IsAgentAuthChat verifica si un chat_id está autorizado.
|
|
func IsAgentAuthChat(chatID int64) bool {
|
|
var auth TelegramAgentAuth
|
|
err := app.Http.Database.DB.Where("chat_id = ? AND activo = ?", chatID, true).First(&auth).Error
|
|
return err == nil
|
|
}
|
|
|
|
func GetAllAgentAuth() ([]TelegramAgentAuth, error) {
|
|
var items []TelegramAgentAuth
|
|
err := app.Http.Database.DB.Order("created_at DESC").Find(&items).Error
|
|
return items, err
|
|
}
|
|
|
|
func CreateAgentAuth(chatID int64, nombre string) error {
|
|
auth := &TelegramAgentAuth{ChatID: chatID, Nombre: nombre, Activo: true}
|
|
return app.Http.Database.DB.Create(auth).Error
|
|
}
|
|
|
|
func DeleteAgentAuth(id uint) error {
|
|
return app.Http.Database.DB.Delete(&TelegramAgentAuth{}, id).Error
|
|
}
|
|
|
|
// GetAgentTelegramConfig retorna el TelegramConfig vinculado al agente bot activo.
|
|
func GetAgentTelegramConfig() (*TelegramConfig, error) {
|
|
var ai AiConfig
|
|
if err := app.Http.Database.DB.Where("es_agente_bot = ? AND is_active = ?", true, true).First(&ai).Error; err != nil {
|
|
return nil, fmt.Errorf("no hay agente bot configurado")
|
|
}
|
|
if ai.TelegramConfigID == nil {
|
|
return nil, fmt.Errorf("el agente no tiene Telegram configurado")
|
|
}
|
|
return GetTelegramConfigByID(*ai.TelegramConfigID)
|
|
}
|