package di

import (
	"context"

	"cellbot/internal/engine"
	"cellbot/internal/protocol"
	"cellbot/pkg/net"
	"go.uber.org/fx"
	"go.uber.org/zap"
)

// RegisterLifecycleHooks 注册应用生命周期钩子
func RegisterLifecycleHooks(
	logger *zap.Logger,
	eventBus *engine.EventBus,
	dispatcher *engine.Dispatcher,
	botManager *protocol.BotManager,
	server *net.Server,
) fx.Option {
	return fx.Invoke(
		func(lc fx.Lifecycle) {
			lc.Append(fx.Hook{
				OnStart: func(ctx context.Context) error {
					logger.Info("Starting CellBot application...")

					// 启动事件总线
					eventBus.Start()

					// 启动分发器
					dispatcher.Start(ctx)

					// 启动所有机器人
					if err := botManager.StartAll(ctx); err != nil {
						logger.Error("Failed to start bots", zap.Error(err))
					}

					// 启动HTTP服务器
					if err := server.Start(); err != nil {
						logger.Error("Failed to start server", zap.Error(err))
						return err
					}

					logger.Info("CellBot application started successfully")
					return nil
				},
				OnStop: func(ctx context.Context) error {
					logger.Info("Stopping CellBot application...")

					// 停止HTTP服务器
					if err := server.Stop(); err != nil {
						logger.Error("Failed to stop server", zap.Error(err))
					}

					// 停止所有机器人
					if err := botManager.StopAll(ctx); err != nil {
						logger.Error("Failed to stop bots", zap.Error(err))
					}

					// 停止分发器
					dispatcher.Stop()

					// 停止事件总线
					eventBus.Stop()

					logger.Info("CellBot application stopped successfully")
					return nil
				},
			})
		},
	)
}

// Lifecycle 生命周期管理选项
var Lifecycle = fx.Options(
	fx.Invoke(RegisterLifecycleHooks),
)