package com.trafficaudit.llmintegration.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 企业名称脱敏：把企业名称替换为 Company1、Company2… 后发给大模型，回答返回前再还原。
 * 仅保存在内存中，不落库。
 */
@Slf4j
@Service
public class DesensitizeService {

    private final Map<String, String> enterpriseMap = new ConcurrentHashMap<>();
    private final Map<String, String> reverseMap = new ConcurrentHashMap<>();
    private final AtomicInteger counter = new AtomicInteger(1);

    public String maskEnterprise(String name) {
        if (name == null) {
            return null;
        }
        return enterpriseMap.computeIfAbsent(name, k -> {
            String alias = "Company" + counter.getAndIncrement();
            reverseMap.put(alias, name);
            return alias;
        });
    }

    public String unmaskEnterprise(String alias) {
        return reverseMap.getOrDefault(alias, alias);
    }

    /** 把整段文本里出现过的脱敏别名还原为真实企业名称（回答返回给前端前调用） */
    public String unmaskAll(String text) {
        if (text == null || text.isEmpty() || reverseMap.isEmpty()) {
            return text;
        }
        String out = text;
        for (Map.Entry<String, String> e : reverseMap.entrySet()) {
            if (out.contains(e.getKey())) {
                out = out.replace(e.getKey(), e.getValue());
            }
        }
        return out;
    }

    public void clear() {
        enterpriseMap.clear();
        reverseMap.clear();
        counter.set(1);
    }
}