main.go

package main

import (
    "log"
    "os"
    "time"

    "rss2telegram/internal/bot"
    "rss2telegram/internal/config"
    "rss2telegram/internal/rss"
    "rss2telegram/internal/storage"
    "rss2telegram/internal/stats"
)

type App struct {
    bot        *bot.Bot
    rssManager *rss.Manager
    config     *config.Config
    db         *storage.Storage
}

func NewApp(cfg *config.Config, db *storage.Storage, stats *stats.Stats) (*App, error) {
    bot, err := bot.NewBot(cfg.Telegram.BotToken, cfg.Telegram.Users, cfg.Telegram.Channels, db, cfg, "/app/config/config.yaml", stats)
    if err != nil {
        return nil, err
    }

    rssConfigs := make([]rss.Config, len(cfg.RSS))
    for i, rssCfg := range cfg.RSS {
        rssConfigs[i] = rss.Config{
            URL:      rssCfg.URL,
            Interval: rssCfg.Interval,
            Keywords: rssCfg.Keywords,
            Group:    rssCfg.Group,
        }
    }

    rssManager := rss.NewManager(rssConfigs, db)

    app := &App{
        bot:        bot,
        rssManager: rssManager,
        config:     cfg,
        db:         db,
    }

    bot.SetMessageHandler(app.handleMessage)
    bot.SetUpdateRSSHandler(app.updateRSS)
    rssManager.SetMessageHandler(app.handleMessage)

    return app, nil
}

func (app *App) handleMessage(title, url, group string, pubDate time.Time, matchedKeywords []string) error {
    return app.bot.SendMessage(title, url, group, pubDate, matchedKeywords)
}

func (app *App) updateRSS() {
    rssConfigs := make([]rss.Config, len(app.config.RSS))
    for i, rssCfg := range app.config.RSS {
        rssConfigs[i] = rss.Config{
            URL:      rssCfg.URL,
            Interval: rssCfg.Interval,
            Keywords: rssCfg.Keywords,
            Group:    rssCfg.Group,
        }
    }
    app.rssManager.UpdateFeeds(rssConfigs)
    log.Println("RSS订阅已更新")
}

func (app *App) Start() {
    go app.bot.Start()
    go app.rssManager.Start()
    go app.watchConfig()
}

func (app *App) watchConfig() {
    ticker := time.NewTicker(1 * time.Minute)
    defer ticker.Stop()

    for {
        select {
        case <-ticker.C:
            newCfg, err := config.Load("/app/config/config.yaml")
            if err != nil {
                log.Printf("加载配置失败: %v", err)
                continue
            }

            if !app.config.Equal(newCfg) {
                log.Println("检测到配置变更，正在更新...")
                app.config = newCfg
                app.bot.UpdateConfig(newCfg)
                app.updateRSS()
            }
        }
    }
}

func main() {
    log.SetFlags(log.LstdFlags | log.Lshortfile)
    log.SetOutput(os.Stdout)

    log.Println("启动 RSS 到 Telegram 机器人")

    var cfg *config.Config
    var err error

    // 首先尝试从环境变量加载配置
    cfg = config.LoadFromEnv()

    // 如果环境变量中没有足够的配置信息，则尝试从配置文件加载
    if cfg.Telegram.BotToken == "" || len(cfg.Telegram.Users) == 0 {
        log.Println("环境变量中配置不完整，尝试从配置文件加载")
        cfg, err = config.Load("/app/config/config.yaml")
        if err != nil {
            log.Fatalf("加载配置失败: %v", err)
        }
    }

    // 打印加载的配置（注意不要打印敏感信息如 bot token）
    log.Printf("加载的配置: Users: %v, Channels: %v", cfg.Telegram.Users, cfg.Telegram.Channels)

    db := storage.NewStorage("/app/data/sent_items.txt")
    stats, err := stats.NewStats("/app/data/stats.yaml")
    if err != nil {
        log.Fatalf("创建统计失败: %v", err)
    }

    app, err := NewApp(cfg, db, stats)
    if err != nil {
        log.Fatalf("创建应用失败: %v", err)
    }

    app.Start()

    log.Println("机器人现在正在运行")

    // 保持应用运行
    select {}
}