package main

import (
	"ZhenTuLocalPassiveAdapter/api"
	"ZhenTuLocalPassiveAdapter/config"
	"ZhenTuLocalPassiveAdapter/core"
	"ZhenTuLocalPassiveAdapter/dto"
	"ZhenTuLocalPassiveAdapter/logger"
	"ZhenTuLocalPassiveAdapter/telemetry"
	"context"
	"go.opentelemetry.io/otel"
	"go.opentelemetry.io/otel/attribute"
	"go.opentelemetry.io/otel/codes"
	"go.uber.org/zap"
	"os"
	"time"
)

var tracer = otel.Tracer("vpt")

func startTask(device config.DeviceMapping, task dto.Task) {
	ctx, span := tracer.Start(context.Background(), "startTask")
	defer span.End()
	span.SetAttributes(attribute.String("deviceNo", device.DeviceNo))
	span.SetAttributes(attribute.String("taskId", task.TaskID))
	span.SetAttributes(attribute.String("scenicId", task.ScenicID))
	span.SetAttributes(attribute.String("startTime", task.StartTime.Format("2006-01-02 15:04:05")))
	span.SetAttributes(attribute.String("endTime", task.EndTime.Format("2006-01-02 15:04:05")))
	fo, err := core.HandleTask(ctx, device, task)
	if err != nil {
		span.SetStatus(codes.Error, "处理任务失败")
		logger.Error("处理任务失败",
			zap.String("taskID", task.TaskID),
			zap.String("deviceNo", task.DeviceNo),
			zap.Error(err))
		api.ReportTaskFailure(ctx, task.TaskID)
		return
	}
	span.SetAttributes(attribute.String("fileUrl", fo.URL))
	logger.Info("处理任务成功",
		zap.String("taskID", task.TaskID),
		zap.String("deviceNo", task.DeviceNo))
	err = api.UploadTaskFile(ctx, task, *fo)
	if err != nil {
		span.SetStatus(codes.Error, "上传文件失败")
		logger.Error("上传文件失败",
			zap.String("taskID", task.TaskID),
			zap.String("deviceNo", task.DeviceNo),
			zap.Error(err))
		api.ReportTaskFailure(ctx, task.TaskID)
		return
	}
	result := api.ReportTaskSuccess(ctx, task.TaskID, fo)
	if !result {
		span.SetStatus(codes.Error, "上报任务成功失败")
		logger.Error("上报任务成功失败",
			zap.String("taskID", task.TaskID),
			zap.String("deviceNo", task.DeviceNo),
			zap.Error(err))
		return
	}
	span.SetStatus(codes.Ok, "上传文件成功")
	logger.Info("上传文件成功",
		zap.String("taskID", task.TaskID),
		zap.String("deviceNo", task.DeviceNo))
}

func main() {
	// 初始化日志
	err := logger.Init()
	if err != nil {
		panic(err)
	}
	defer logger.Sync()

	err = config.LoadConfig()
	if err != nil {
		logger.Fatal("加载配置文件失败", zap.Error(err))
		return
	}
	ctx := context.Background()
	shutdown, err := telemetry.InitTelemetry(ctx)
	if err != nil {
		logger.Fatal("Failed to initialize telemetry", zap.Error(err))
		return
	}
	defer shutdown(ctx)
	if config.Config.Record.Storage.Type == "local" {
		_, err = os.Stat(config.Config.Record.Storage.Path)
		if err != nil {
			logger.Error("录像文件夹配置失败", zap.Error(err))
			return
		} else {
			logger.Info("录像文件夹配置有效")
		}
	} else {
		logger.Info("录像文件夹配置为OSS")
	}
	// 每两秒定时执行
	for {
		// 执行任务
		tasks, err := api.SyncTask()
		if err == nil {
			for _, task := range tasks {
				logger.Info("开始处理任务",
					zap.String("taskID", task.TaskID),
					zap.String("deviceNo", task.DeviceNo),
					zap.String("startTime", task.StartTime.Format("2006-01-02 15:04:05")),
					zap.String("endTime", task.EndTime.Format("2006-01-02 15:04:05")))
				// 处理任务
				for _, device := range config.Config.Devices {
					if device.DeviceNo == task.DeviceNo {
						// 处理任务
						go startTask(device, task)
						break // 提前返回，避免不必要的循环
					}
				}
			}
		} else {
			logger.Error("同步任务失败", zap.Error(err))
		}
		// 等待两秒
		<-time.After(2 * time.Second)
	}
}