mirror of
https://github.com/usual2970/certimate.git
synced 2025-06-08 13:39:53 +00:00
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package certificate
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/usual2970/certimate/internal/app"
|
|
"github.com/usual2970/certimate/internal/domain"
|
|
"github.com/usual2970/certimate/internal/notify"
|
|
"github.com/usual2970/certimate/internal/repository"
|
|
)
|
|
|
|
const (
|
|
defaultExpireSubject = "您有 ${COUNT} 张证书即将过期"
|
|
defaultExpireMessage = "有 ${COUNT} 张证书即将过期，域名分别为 ${DOMAINS}，请保持关注！"
|
|
)
|
|
|
|
type CertificateRepository interface {
|
|
GetExpireSoon(ctx context.Context) ([]domain.Certificate, error)
|
|
}
|
|
|
|
type certificateService struct {
|
|
repo CertificateRepository
|
|
}
|
|
|
|
func NewCertificateService(repo CertificateRepository) *certificateService {
|
|
return &certificateService{
|
|
repo: repo,
|
|
}
|
|
}
|
|
|
|
func (s *certificateService) InitSchedule(ctx context.Context) error {
|
|
scheduler := app.GetScheduler()
|
|
|
|
err := scheduler.Add("certificate", "0 0 * * *", func() {
|
|
certs, err := s.repo.GetExpireSoon(context.Background())
|
|
if err != nil {
|
|
app.GetApp().Logger().Error("failed to get expire soon certificate", "err", err)
|
|
return
|
|
}
|
|
msg := buildMsg(certs)
|
|
if err := notify.SendToAllChannels(msg.Subject, msg.Message); err != nil {
|
|
app.GetApp().Logger().Error("failed to send expire soon certificate", "err", err)
|
|
}
|
|
})
|
|
if err != nil {
|
|
app.GetApp().Logger().Error("failed to add schedule", "err", err)
|
|
return err
|
|
}
|
|
scheduler.Start()
|
|
app.GetApp().Logger().Info("certificate schedule started")
|
|
return nil
|
|
}
|
|
|
|
func buildMsg(records []domain.Certificate) *domain.NotifyMessage {
|
|
if len(records) == 0 {
|
|
return nil
|
|
}
|
|
|
|
// 查询模板信息
|
|
settingRepo := repository.NewSettingRepository()
|
|
setting, err := settingRepo.GetByName(context.Background(), "notifyTemplates")
|
|
|
|
subject := defaultExpireSubject
|
|
message := defaultExpireMessage
|
|
|
|
if err == nil {
|
|
var templates *domain.NotifyTemplates
|
|
|
|
json.Unmarshal([]byte(setting.Content), &templates)
|
|
|
|
if templates != nil && len(templates.NotifyTemplates) > 0 {
|
|
subject = templates.NotifyTemplates[0].Title
|
|
message = templates.NotifyTemplates[0].Content
|
|
}
|
|
}
|
|
|
|
// 替换变量
|
|
count := len(records)
|
|
domains := make([]string, count)
|
|
|
|
for i, record := range records {
|
|
domains[i] = record.SAN
|
|
}
|
|
|
|
countStr := strconv.Itoa(count)
|
|
domainStr := strings.Join(domains, ";")
|
|
|
|
subject = strings.ReplaceAll(subject, "${COUNT}", countStr)
|
|
subject = strings.ReplaceAll(subject, "${DOMAINS}", domainStr)
|
|
|
|
message = strings.ReplaceAll(message, "${COUNT}", countStr)
|
|
message = strings.ReplaceAll(message, "${DOMAINS}", domainStr)
|
|
|
|
// 返回消息
|
|
return &domain.NotifyMessage{
|
|
Subject: subject,
|
|
Message: message,
|
|
}
|
|
}
|