package com.lingyue.graph.service;

import com.lingyue.common.event.DocumentParsedEvent;
import com.lingyue.common.exception.ServiceException;
import com.lingyue.graph.entity.TextStorage;
import com.lingyue.graph.repository.TextStorageRepository;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.stereotype.Service;

import java.io.File;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.security.MessageDigest;
import java.util.UUID;

/**
 * 文本存储服务
 * 
 * @author lingyue
 * @since 2026-01-14
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class TextStorageService {
    
    private final TextStorageRepository textStorageRepository;
    private final RAGService ragService;
    private final ApplicationEventPublisher eventPublisher;
    
    @Value("${rag.auto-index.enabled:true}")
    private boolean autoIndexEnabled;
    
    @Value("${ner.auto-extract.enabled:true}")
    private boolean nerAutoExtractEnabled;
    
    /**
     * 保存文本存储记录
     * 
     * @param documentId 文档ID
     * @param filePath TXT文件路径
     * @return 文本存储记录
     */
    public TextStorage saveTextStorage(String documentId, String filePath) {
        // 检查文件是否存在
        File file = new File(filePath);
        if (!file.exists()) {
            throw new ServiceException("文本文件不存在: " + filePath);
        }
        
        // 检查是否已存在记录
        TextStorage existing = textStorageRepository.findByDocumentId(documentId);
        if (existing != null) {
            // 更新现有记录
            existing.setFilePath(filePath);
            existing.setFileSize(file.length());
            existing.setChecksum(calculateChecksum(filePath));
            existing.setUpdateTime(new java.util.Date());
            textStorageRepository.updateById(existing);
            log.info("更新文本存储记录: documentId={}, filePath={}", documentId, filePath);
            return existing;
        }
        
        // 创建新记录
        TextStorage textStorage = new TextStorage();
        textStorage.setId(UUID.randomUUID().toString().replace("-", ""));
        textStorage.setDocumentId(documentId);
        textStorage.setFilePath(filePath);
        textStorage.setFileSize(file.length());
        textStorage.setChecksum(calculateChecksum(filePath));
        textStorage.setCreateTime(new java.util.Date());
        textStorage.setUpdateTime(new java.util.Date());
        
        textStorageRepository.insert(textStorage);
        log.info("创建文本存储记录: documentId={}, filePath={}", documentId, filePath);
        
        return textStorage;
    }
    
    /**
     * 根据文档ID查询文本存储记录
     * 
     * @param documentId 文档ID
     * @return 文本存储记录
     */
    public TextStorage getByDocumentId(String documentId) {
        return textStorageRepository.findByDocumentId(documentId);
    }
    
    /**
     * 保存文本并自动建立 RAG 索引
     * 
     * @param documentId 文档ID
     * @param filePath TXT文件路径
     * @return 文本存储记录
     */
    public TextStorage saveAndIndex(String documentId, String filePath) {
        return saveAndIndex(documentId, filePath, null);
    }
    
    /**
     * 保存文本并自动建立 RAG 索引（带用户ID）
     * 
     * @param documentId 文档ID
     * @param filePath TXT文件路径
     * @param userId 用户ID
     * @return 文本存储记录
     */
    public TextStorage saveAndIndex(String documentId, String filePath, String userId) {
        // 1. 保存文本存储记录
        TextStorage textStorage = saveTextStorage(documentId, filePath);
        
        // 2. 自动建立 RAG 索引
        if (autoIndexEnabled) {
            try {
                String text = Files.readString(Path.of(filePath), StandardCharsets.UTF_8);
                int chunkCount = ragService.indexDocument(documentId, textStorage.getId(), text);
                log.info("自动建立 RAG 索引完成: documentId={}, chunks={}", documentId, chunkCount);
            } catch (Exception e) {
                log.warn("自动建立 RAG 索引失败，不影响主流程: documentId={}", documentId, e);
                // 索引失败不影响主流程
            }
        }
        
        // 3. 发布文档解析完成事件，触发 NER 等后续处理
        if (nerAutoExtractEnabled) {
            try {
                DocumentParsedEvent event = new DocumentParsedEvent(
                        this, documentId, userId, filePath, textStorage.getId());
                eventPublisher.publishEvent(event);
                log.info("发布文档解析完成事件: documentId={}", documentId);
            } catch (Exception e) {
                log.warn("发布文档解析完成事件失败，不影响主流程: documentId={}", documentId, e);
            }
        }
        
        return textStorage;
    }
    
    /**
     * 计算文件MD5校验和
     */
    private String calculateChecksum(String filePath) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            byte[] fileBytes = Files.readAllBytes(Path.of(filePath));
            byte[] digest = md.digest(fileBytes);
            
            StringBuilder sb = new StringBuilder();
            for (byte b : digest) {
                sb.append(String.format("%02x", b));
            }
            return sb.toString();
        } catch (Exception e) {
            log.warn("计算文件校验和失败: {}", filePath, e);
            return null;
        }
    }
}
