package core

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"

	"git.huangwc.com/pig/pig-farm-controller/internal/app/api"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/config"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
)

// Application 是整个应用的核心，封装了所有组件和生命周期。
type Application struct {
	Config *config.Config
	Logger *logs.Logger
	API    *api.API

	Infra  *Infrastructure
	Domain *DomainServices
	App    *AppServices
}

// NewApplication 创建并初始化一个新的 Application 实例。
// 这是应用的“组合根”，所有依赖都在这里被创建和注入。
func NewApplication(configPath string) (*Application, error) {
	// 1. 初始化基本组件: 配置和日志
	cfg := config.NewConfig()
	if err := cfg.Load(configPath); err != nil {
		return nil, fmt.Errorf("无法加载配置: %w", err)
	}
	logger := logs.NewLogger(cfg.Log)

	// 2. 初始化所有分层服务
	infra, err := initInfrastructure(cfg, logger)
	if err != nil {
		return nil, fmt.Errorf("初始化基础设施失败: %w", err)
	}
	domain := initDomainServices(cfg, infra, logger)
	appServices := initAppServices(infra, domain, logger)

	// 3. 初始化 API 入口点
	apiServer := api.NewAPI(
		cfg.Server,
		logger,
		infra.Repos.UserRepo,
		infra.Repos.DeviceRepo,
		infra.Repos.AreaControllerRepo,
		infra.Repos.DeviceTemplateRepo,
		infra.Repos.PlanRepo,
		appServices.PigFarmService,
		appServices.PigBatchService,
		appServices.MonitorService,
		infra.TokenService,
		appServices.AuditService,
		infra.NotifyService,
		domain.GeneralDeviceService,
		infra.Lora.ListenHandler,
		domain.AnalysisPlanTaskManager,
	)

	// 4. 组装 Application 对象
	app := &Application{
		Config: cfg,
		Logger: logger,
		API:    apiServer,
		Infra:  infra,
		Domain: domain,
		App:    appServices,
	}

	return app, nil
}

// Start 启动应用的所有组件并阻塞，直到接收到关闭信号。
func (app *Application) Start() error {
	app.Logger.Info("应用启动中...")

	// 1. 启动底层监听器
	if err := app.Infra.Lora.LoraListener.Listen(); err != nil {
		return fmt.Errorf("启动 LoRa Mesh 监听器失败: %w", err)
	}

	// 2. 初始化应用状态 (清理、刷新任务等)
	if err := app.initializeState(); err != nil {
		return fmt.Errorf("初始化应用状态失败: %w", err)
	}

	// 3. 启动后台工作协程
	app.Domain.Scheduler.Start()
	app.Domain.TimedCollector.Start()

	// 4. 启动 API 服务器
	app.API.Start()

	// 5. 等待关闭信号
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit

	// 接收到信号后，执行优雅关闭
	return app.Stop()
}

// Stop 优雅地关闭应用的所有组件。
func (app *Application) Stop() error {
	app.Logger.Info("应用关闭中...")

	// 关闭 API 服务器
	app.API.Stop()

	// 关闭任务执行器
	app.Domain.Scheduler.Stop()

	// 关闭定时采集器
	app.Domain.TimedCollector.Stop()

	// 断开数据库连接
	if err := app.Infra.Storage.Disconnect(); err != nil {
		app.Logger.Errorw("数据库连接断开失败", "error", err)
	}

	// 关闭 LoRa Mesh 监听器
	if err := app.Infra.Lora.LoraListener.Stop(); err != nil {
		app.Logger.Errorw("LoRa Mesh 监听器关闭失败", "error", err)
	}

	// 刷新日志缓冲区
	_ = app.Logger.Sync()

	app.Logger.Info("应用已成功关闭")
	return nil
}

// initializeState 在应用启动时准备其初始数据状态。
// 这包括清理任何因上次异常关闭而留下的悬空任务或请求。
func (app *Application) initializeState() error {
	// 清理待采集任务 (非致命错误)
	if err := app.initializePendingCollections(); err != nil {
		app.Logger.Errorw("清理待采集任务时发生非致命错误", "error", err)
	}

	// 初始化待执行任务列表 (致命错误)
	if err := app.initializePendingTasks(); err != nil {
		return fmt.Errorf("初始化待执行任务列表失败: %w", err)
	}

	return nil
}

// initializePendingCollections 在应用启动时处理所有未完成的采集请求。
// 我们的策略是：任何在程序重启前仍处于“待处理”状态的请求，都应被视为已失败。
// 这保证了系统在每次启动时都处于一个干净、确定的状态。
func (app *Application) initializePendingCollections() error {
	app.Logger.Info("开始清理所有未完成的采集请求...")

	// 直接将所有 'pending' 状态的请求更新为 'timed_out'。
	count, err := app.Infra.Repos.PendingCollectionRepo.MarkAllPendingAsTimedOut()
	if err != nil {
		return fmt.Errorf("清理未完成的采集请求失败: %v", err)
	} else if count > 0 {
		app.Logger.Infof("成功将 %d 个未完成的采集请求标记为超时。", count)
	} else {
		app.Logger.Info("没有需要清理的采集请求。")
	}

	return nil
}

// initializePendingTasks 在应用启动时清理并刷新待执行任务列表。
func (app *Application) initializePendingTasks() error {
	logger := app.Logger
	planRepo := app.Infra.Repos.PlanRepo
	pendingTaskRepo := app.Infra.Repos.PendingTaskRepo
	executionLogRepo := app.Infra.Repos.ExecutionLogRepo
	analysisPlanTaskManager := app.Domain.AnalysisPlanTaskManager

	logger.Info("开始初始化待执行任务列表...")

	// 阶段一：修正因崩溃导致状态不一致的固定次数计划
	logger.Info("阶段一：开始修正因崩溃导致状态不一致的固定次数计划...")
	plansToCorrect, err := planRepo.FindPlansWithPendingTasks()
	if err != nil {
		return fmt.Errorf("查找需要修正的计划失败: %w", err)
	}

	for _, plan := range plansToCorrect {
		logger.Infof("发现需要修正的计划 #%d (名称: %s)。", plan.ID, plan.Name)

		// 更新计划的执行计数
		plan.ExecuteCount++
		logger.Infof("计划 #%d 执行计数已从 %d 更新为 %d。", plan.ID, plan.ExecuteCount-1, plan.ExecuteCount)

		if plan.ExecutionType == models.PlanExecutionTypeManual ||
			(plan.ExecutionType == models.PlanExecutionTypeAutomatic && plan.ExecuteCount >= plan.ExecuteNum) {
			// 更新计划状态为已停止
			plan.Status = models.PlanStatusStopped
			logger.Infof("计划 #%d 状态已更新为 '执行完毕'。", plan.ID)

		}
		// 保存更新后的计划
		if err := planRepo.UpdatePlan(plan); err != nil {
			logger.Errorf("修正计划 #%d 状态失败: %v", plan.ID, err)
			// 这是一个非阻塞性错误，继续处理其他计划
		}
	}
	logger.Info("阶段一：固定次数计划修正完成。")

	// 阶段二：清理所有待执行任务和相关日志
	logger.Info("阶段二：开始清理所有待执行任务和相关日志...")

	// --- 新增逻辑：处理因崩溃导致状态不一致的计划主表状态 ---
	// 1. 查找所有未完成的计划执行日志 (状态为 Started 或 Waiting)
	incompletePlanLogs, err := executionLogRepo.FindIncompletePlanExecutionLogs()
	if err != nil {
		return fmt.Errorf("查找未完成的计划执行日志失败: %w", err)
	}

	// 2. 收集所有受影响的唯一 PlanID
	affectedPlanIDs := make(map[uint]struct{})
	for _, log := range incompletePlanLogs {
		affectedPlanIDs[log.PlanID] = struct{}{}
	}

	// 3. 对于每个受影响的 PlanID，重置其 execute_count 并将其状态设置为 Failed
	for planID := range affectedPlanIDs {
		logger.Warnf("检测到计划 #%d 在应用崩溃前处于未完成状态，将重置其计数并标记为失败。", planID)
		// 使用 UpdatePlanStateAfterExecution 来更新主表状态，避免影响关联数据
		if err := planRepo.UpdatePlanStateAfterExecution(planID, 0, models.PlanStatusFailed); err != nil {
			logger.Errorf("重置计划 #%d 计数并标记为失败时出错: %v", planID, err)
			// 这是一个非阻塞性错误，继续处理其他计划
		}
	}
	logger.Info("阶段二：计划主表状态修正完成。")

	// 直接调用新的方法来更新计划执行日志状态为失败
	if err := executionLogRepo.FailAllIncompletePlanExecutionLogs(); err != nil {
		logger.Errorf("更新所有未完成计划执行日志状态为失败失败: %v", err)
		// 这是一个非阻塞性错误，继续执行
	}

	// 直接调用新的方法来更新任务执行日志状态为取消
	if err := executionLogRepo.CancelAllIncompleteTaskExecutionLogs(); err != nil {
		logger.Errorf("更新所有未完成任务执行日志状态为取消失败: %v", err)
		// 这是一个非阻塞性错误，继续执行
	}

	// 清空待执行列表
	if err := pendingTaskRepo.ClearAllPendingTasks(); err != nil {
		return fmt.Errorf("清空待执行任务列表失败: %w", err)
	}
	logger.Info("阶段二：待执行任务和相关日志清理完成。")

	// 阶段三：初始刷新
	logger.Info("阶段三：开始刷新待执行列表...")
	if err := analysisPlanTaskManager.Refresh(); err != nil {
		return fmt.Errorf("刷新待执行任务列表失败: %w", err)
	}
	logger.Info("阶段三：待执行任务列表初始化完成。")

	logger.Info("待执行任务列表初始化完成。")
	return nil
}