package com.zy.ai.service.impl;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.zy.ai.entity.AiDataAnalysisReport;
import com.zy.ai.service.DataAnalysisFileStorageService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Map;

@Slf4j
@Service("dataAnalysisFileStorageService")
public class DataAnalysisFileStorageServiceImpl implements DataAnalysisFileStorageService {

    @Value("${dataAnalysisStorage.loggingPath:../stock/out/wcs/aiAnalysis}")
    private String basePath;

    @Override
    public String saveReport(AiDataAnalysisReport report) {
        try {
            SimpleDateFormat dirFormat = new SimpleDateFormat("yyyyMMdd");
            SimpleDateFormat fileFormat = new SimpleDateFormat("yyyyMMdd_HHmmss");
            String dateDir = dirFormat.format(report.getCreateTime());
            String timestamp = fileFormat.format(report.getCreateTime());

            File dir = new File(basePath, dateDir);
            if (!dir.exists() && !dir.mkdirs()) {
                log.warn("Failed to create analysis storage directory: {}", dir.getAbsolutePath());
                return null;
            }

            String fileName = "analysis_" + report.getPeriodType() + "_" + timestamp + ".json";
            File file = new File(dir, fileName);

            Map<String, Object> content = new LinkedHashMap<>();
            content.put("periodType", report.getPeriodType());
            content.put("periodStart", report.getPeriodStart());
            content.put("periodEnd", report.getPeriodEnd());
            content.put("triggerType", report.getTriggerType());
            content.put("status", report.getStatus());
            content.put("summary", report.getSummary());
            content.put("structuredData", report.getStructuredData());
            content.put("llmCallCount", report.getLlmCallCount());
            content.put("totalTokens", report.getTotalTokens());
            content.put("createTime", report.getCreateTime());
            content.put("finishTime", report.getFinishTime());

            try (OutputStreamWriter writer = new OutputStreamWriter(
                    new FileOutputStream(file), StandardCharsets.UTF_8)) {
                writer.write(JSON.toJSONString(content, SerializerFeature.PrettyFormat,
                        SerializerFeature.WriteDateUseDateFormat));
            }

            String relativePath = dateDir + "/" + fileName;
            log.info("Data analysis report saved to file: {}", relativePath);
            return relativePath;
        } catch (Exception e) {
            log.error("Failed to save data analysis report to file", e);
            return null;
        }
    }
}