be.ems/lib/config/psap.go

package config

import (
	"fmt"
	"os"

	"be.ems/lib/email"
	"gopkg.in/yaml.v3"
)

// PsapConfig PSAP配置结构体
type PsapConfig struct {
	Ticket struct {
		Notification struct {
			SMTP email.EmailConfig `yaml:"smtp"`
		} `yaml:"notifcation"` // 注意：配置文件中是 "notifcation"，保持一致
		Timeout struct {
			// 时间单位为分钟
			New           int `yaml:"new"`           // NEW状态超时时间（分钟）
			InProgress    int `yaml:"inProgress"`    // IN_PROGRESS状态超时时间（分钟）
			NoAnswer1     int `yaml:"noAnswer1"`     // NO_ANSWER_1状态超时时间（分钟）
			NoAnswer2     int `yaml:"noAnswer2"`     // NO_ANSWER_2状态超时时间（分钟）
			NearlyTimeout int `yaml:"nearlyTimeout"` // 提前提醒时间（分钟）
		} `yaml:"timeout"`
	} `yaml:"ticket"`
}

var psapConfig *PsapConfig

// ReadPsapConfig 读取PSAP配置文件
func ReadPsapConfig(configFile string) (*PsapConfig, error) {
	yamlFile, err := os.ReadFile(configFile)
	if err != nil {
		return nil, fmt.Errorf("read psap config file error: %w", err)
	}

	var config PsapConfig
	err = yaml.Unmarshal(yamlFile, &config)
	if err != nil {
		return nil, fmt.Errorf("unmarshal psap config error: %w", err)
	}

	psapConfig = &config
	return &config, nil
}

// GetPsapConfig 获取PSAP配置
func GetPsapConfig() *PsapConfig {
	if psapConfig == nil {
		// 如果配置未加载，尝试从默认位置加载
		config, err := ReadPsapConfig("./etc/psap.yaml")
		if err != nil {
			fmt.Printf("Failed to load PSAP config: %v\n", err)
			return nil
		}
		psapConfig = config
	}
	return psapConfig
}

// GetSMTPConfig 获取SMTP配置
func GetSMTPConfig() *email.EmailConfig {
	config := GetPsapConfig()
	if config == nil {
		return nil
	}
	return &config.Ticket.Notification.SMTP
}

// GetTimeoutConfig 获取超时配置
func GetTimeoutConfig() *struct {
	New           int `yaml:"new"`
	InProgress    int `yaml:"inProgress"`
	NoAnswer1     int `yaml:"noAnswer1"`
	NoAnswer2     int `yaml:"noAnswer2"`
	NearlyTimeout int `yaml:"nearlyTimeout"`
} {
	config := GetPsapConfig()
	if config == nil {
		return nil
	}
	return &config.Ticket.Timeout
}

// 以下是具体的配置获取方法

// GetNewTicketTimeoutMicros 获取NEW状态超时时间（微秒）
func GetNewTicketTimeoutMicros() int64 {
	config := GetTimeoutConfig()
	if config == nil {
		return 60 * 60 * 1000000 // 默认60分钟
	}
	return int64(config.New) * 60 * 1000000 // 分钟转微秒
}

// GetInProgressTicketTimeoutMicros 获取IN_PROGRESS状态超时时间（微秒）
func GetInProgressTicketTimeoutMicros() int64 {
	config := GetTimeoutConfig()
	if config == nil {
		return 60 * 60 * 1000000 // 默认60分钟
	}
	return int64(config.InProgress) * 60 * 1000000
}

// GetNoAnswer1TicketTimeoutMicros 获取NO_ANSWER_1状态超时时间（微秒）
func GetNoAnswer1TicketTimeoutMicros() int64 {
	config := GetTimeoutConfig()
	if config == nil {
		return 4 * 60 * 60 * 1000000 // 默认4小时
	}
	return int64(config.NoAnswer1) * 60 * 1000000
}

// GetNoAnswer2TicketTimeoutMicros 获取NO_ANSWER_2状态超时时间（微秒）
func GetNoAnswer2TicketTimeoutMicros() int64 {
	config := GetTimeoutConfig()
	if config == nil {
		return 8 * 60 * 60 * 1000000 // 默认8小时
	}
	return int64(config.NoAnswer2) * 60 * 1000000
}

// GetNearlyTimeoutMicros 获取提前提醒时间（微秒）
func GetNearlyTimeoutMicros() int64 {
	config := GetTimeoutConfig()
	if config == nil {
		return 20 * 60 * 1000000 // 默认20分钟
	}
	return int64(config.NearlyTimeout) * 60 * 1000000
}

// IsSMTPEnabled 检查SMTP是否启用
func IsSMTPEnabled() bool {
	config := GetSMTPConfig()
	if config == nil {
		return false
	}
	return config.Enabled
}

// GetSMTPHost 获取SMTP主机
func GetSMTPHost() string {
	config := GetSMTPConfig()
	if config == nil {
		return ""
	}
	return config.Host
}

// GetSMTPPort 获取SMTP端口
func GetSMTPPort() int {
	config := GetSMTPConfig()
	if config == nil {
		return 25
	}
	return config.Port
}

// GetSMTPUser 获取SMTP用户名
func GetSMTPUsername() string {
	config := GetSMTPConfig()
	if config == nil {
		return ""
	}
	return config.Username
}

// GetSMTPPassword 获取SMTP密码
func GetSMTPPassword() string {
	config := GetSMTPConfig()
	if config == nil {
		return ""
	}
	return config.Password
}

// GetSMTPFrom 获取SMTP发件人
func GetSMTPFrom() string {
	config := GetSMTPConfig()
	if config == nil {
		return ""
	}
	return config.From
}