SamWaf/innerbean/message_info.go

package innerbean

import (
	"SamWaf/utils/wechat"
)

type BaseInfo interface {
	ToFormat() string
}

/***

信息类型:
服务器:
IP地址:

*/

type BaseMessageInfo struct {
	OperaType string `json:"operatype"`
	Server    string `json:"server"`
}

/*
**
域名信息:
触发规则:
*/
type RuleMessageInfo struct {
	BaseMessageInfo
	Domain   string `json:"domain"`
	RuleInfo string `json:"ruleinfo"`
	Ip       string `json:"ip"`
}

/*
*
升级结果
*/
type UpdateResultMessageInfo struct {
	BaseMessageInfo
	Msg     string `json:"msg"`
	Success string `json:"success"`
}

/*
*
导出日志结果
*/
type ExportResultMessageInfo struct {
	BaseMessageInfo
	Msg     string `json:"msg"`
	Success string `json:"success"`
}

// 日志统一导出任务完成通知：仅携带任务定位信息，前端从任务接口申请下载票据。
type LogExportReadyMessageInfo struct {
	BaseMessageInfo
	TaskID string `json:"task_id"`
	Msg    string `json:"msg"`
}

/*
*
实时操作反馈
*/
type OpResultMessageInfo struct {
	BaseMessageInfo
	Msg     string `json:"msg"`
	Success string `json:"success"`
}

/*
*
用户登录信息
*/
type UserLoginMessageInfo struct {
	BaseMessageInfo
	Username string `json:"username"` // 用户名
	Ip       string `json:"ip"`       // 登录IP
	Time     string `json:"time"`     // 登录时间

	// 登录来源变化告警（本次 IP 或归属地与该账号上次登录不一致）
	// 走独立的消息类型 manage_login_abnormal，让只想收告警的人不被每次正常登录打扰。
	Abnormal     bool   `json:"abnormal"`      // 是否与上次登录来源不一致
	LastIp       string `json:"last_ip"`       // 上次登录IP
	LastLocation string `json:"last_location"` // 上次登录归属地
	LastTime     string `json:"last_time"`     // 上次登录时间
}

/*
*
攻击信息
*/
type AttackInfoMessageInfo struct {
	BaseMessageInfo
	AttackType string `json:"attack_type"` // 攻击类型
	Url        string `json:"url"`         // 攻击URL
	Ip         string `json:"ip"`          // 攻击IP
	Time       string `json:"time"`        // 攻击时间
}

/*
*
周报信息
*/
type WeeklyReportMessageInfo struct {
	BaseMessageInfo
	TotalRequests   int64  `json:"total_requests"`   // 总请求数
	BlockedRequests int64  `json:"blocked_requests"` // 拦截请求数
	WeekRange       string `json:"week_range"`       // 周期范围
}

/*
*
SSL证书过期信息
*/
type SSLExpireMessageInfo struct {
	BaseMessageInfo
	Domain     string `json:"domain"`      // 域名
	ExpireTime string `json:"expire_time"` // 过期时间
	DaysLeft   int    `json:"days_left"`   // 剩余天数
}

/*
*
系统错误信息
*/
type SystemErrorMessageInfo struct {
	BaseMessageInfo
	ErrorType string `json:"error_type"` // 错误类型
	ErrorMsg  string `json:"error_msg"`  // 错误信息
	Time      string `json:"time"`       // 发生时间
}

/*
*
IP封禁信息
*/
type IPBanMessageInfo struct {
	BaseMessageInfo
	Ip               string `json:"ip"`                // IP地址
	Reason           string `json:"reason"`            // 封禁原因
	Duration         int    `json:"duration"`          // 封禁时长（分钟）
	RemainingSeconds int    `json:"remaining_seconds"` // 剩余封禁时间（秒）
	Time             string `json:"time"`              // 封禁时间
}

/*
*
统一访问认证事件（登录成功 / 安全异常）
*/
type AccessMessageInfo struct {
	BaseMessageInfo
	Event       string `json:"event"`        // 事件码，见 model.AccessEvent*
	EventName   string `json:"event_name"`   // 事件中文名
	AccountName string `json:"account_name"` // 访问账号（未认证事件可能为空）
	Host        string `json:"host"`         // 被访问的域名
	Url         string `json:"url"`          // 被访问的地址
	Ip          string `json:"ip"`           // 来源IP
	Location    string `json:"location"`     // IP归属地
	Message     string `json:"message"`      // 事件详情
	Time        string `json:"time"`         // 发生时间
	Abnormal    bool   `json:"abnormal"`     // true=安全告警 false=日常告知，决定走哪个订阅类型
}

func (r RuleMessageInfo) ToFormat() map[string]*wechat.DataItem {
	Data := map[string]*wechat.DataItem{}
	Data["domain"] = &wechat.DataItem{
		Value: r.Domain,
		Color: "#808080",
	}
	Data["ruleinfo"] = &wechat.DataItem{
		Value: r.RuleInfo,
		Color: "#808080",
	}
	Data["ip"] = &wechat.DataItem{
		Value: r.Ip,
		Color: "#808080",
	}
	Data["server"] = &wechat.DataItem{
		Value: r.Server,
		Color: "#808080",
	}
	Data["operatype"] = &wechat.DataItem{
		Value: r.OperaType,
		Color: "#808080",
	}
	return Data
}

func (r OperatorMessageInfo) ToFormat() map[string]*wechat.DataItem {
	Data := map[string]*wechat.DataItem{}
	Data["operacnt"] = &wechat.DataItem{
		Value: r.OperaCnt,
		Color: "#808080",
	}
	Data["server"] = &wechat.DataItem{
		Value: r.Server,
		Color: "#808080",
	}
	Data["operatype"] = &wechat.DataItem{
		Value: r.OperaType,
		Color: "#808080",
	}
	return Data
}

/*
**
信息内容:
*/
type OperatorMessageInfo struct {
	BaseMessageInfo
	OperaCnt string `json:"operacnt"`
}

// SystemStatsData 系统统计数据结构，用于ECharts展示
type SystemStatsData struct {
	BaseMessageInfo
	Timestamp       int64       `json:"timestamp"`         // 时间戳
	QPS             uint64      `json:"qps"`               // 当前QPS
	LogQPS          uint64      `json:"log_qps"`           // 日志处理QPS
	MainQueue       int         `json:"main_queue"`        // 主数据队列数量
	LogQueue        int         `json:"log_queue"`         // 日志队列数量
	StatsQueue      int         `json:"stats_queue"`       // 统计队列数量
	MessageQueue    int         `json:"message_queue"`     // 消息队列数量
	CPUPercent      float64     `json:"cpu_percent"`       // CPU使用率
	MemoryPercent   float64     `json:"memory_percent"`    // 内存使用率
	NetworkRecv     uint64      `json:"network_recv"`      // 网络接收字节数(累计)
	NetworkSent     uint64      `json:"network_sent"`      // 网络发送字节数(累计)
	NetworkRecvRate uint64      `json:"network_recv_rate"` // 网络接收速率(字节/秒)
	NetworkSentRate uint64      `json:"network_sent_rate"` // 网络发送速率(字节/秒)
	SystemMonitor   interface{} `json:"system_monitor"`    // 完整的系统监控信息
}