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;

@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) {
        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 void clear() {
        enterpriseMap.clear();
        reverseMap.clear();
        counter.set(1);
    }
}