Main.java

/*
 * Клиент API AML-скоринга Atlorium — скрининг криптокошелька по адресу и сети.
 *
 * Запуск (работает сразу, без регистрации — на демо-ключе).
 * Начиная с Java 11 файл запускается напрямую, без компиляции и без зависимостей:
 *
 *     java Main.java
 *     java Main.java 0x26a5919cd0392df875a53873b0323c42094bb216 ETH
 *
 * Боевой ключ: получить на https://atlorium.com и положить в переменную окружения
 * ATLORIUM_API_KEY. Код при этом не меняется.
 */

import java.io.IOException;
import java.net.URI;
import java.net.URLEncoder;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class Main {

    /**
     * Публичный демо-ключ. С ним API отвечает правдоподобными МОКАМИ (не реальными
     * данными) — чтобы можно было встроить и протестировать интеграцию до оплаты.
     * Ответы детерминированы: один и тот же запрос всегда даёт один и тот же результат,
     * поэтому на них можно писать стабильные тесты.
     */
    static final String SANDBOX_KEY = "ak_sandbox_demo_mockdata_v1";

    static final String API_KEY = envOr("ATLORIUM_API_KEY", SANDBOX_KEY);
    static final String BASE_URL = envOr("ATLORIUM_BASE_URL", "https://atlorium.com");

    static final HttpClient CLIENT = HttpClient.newBuilder()
            .connectTimeout(Duration.ofSeconds(30))
            .build();

    static String envOr(String key, String fallback) {
        String value = System.getenv(key);
        return (value == null || value.isBlank()) ? fallback : value;
    }

    /** Ошибка API: HTTP-код разложен в человекочитаемую причину. */
    static class AtloriumException extends RuntimeException {
        private static final Map<Integer, String> REASONS = Map.of(
                400, "Адрес или сеть не указаны либо имеют неверный формат (неподдерживаемая сеть, некорректный адрес)",
                401, "API-ключ отсутствует, просрочен или недействителен",
                402, "Недостаточно кредитов на балансе — пополните на https://atlorium.com",
                404, "Объект не найден",
                429, "Превышен лимит запросов — повторите позже",
                503, "Сервис AML-скоринга временно недоступен (за сбой на своей стороне мы не списываем деньги)");

        final int status;

        AtloriumException(int status, String body) {
            super("HTTP " + status + ": "
                    + REASONS.getOrDefault(status, "Неизвестная ошибка")
                    + ". Ответ сервера: " + body.substring(0, Math.min(200, body.length())));
            this.status = status;
        }
    }

    static String send(HttpRequest request) throws IOException, InterruptedException {
        HttpResponse<String> response = CLIENT.send(request, HttpResponse.BodyHandlers.ofString(StandardCharsets.UTF_8));
        if (response.statusCode() != 200) {
            throw new AtloriumException(response.statusCode(), response.body());
        }
        return response.body();
    }

    /**
     * AML-скрининг кошелька: балл риска 0–100, санкции и PEP, источники и назначение
     * средств, контрагенты.
     *
     * Это POST с JSON-телом, а не GET: адрес и сеть уходят в теле запроса.
     * Тарифицируется только успешный скрининг — неверный ввод и недоступность
     * источника денег не списывают.
     */
    static String screenWallet(String walletAddress, String blockchain)
            throws IOException, InterruptedException {
        String payload = "{\"walletAddress\":\"" + escape(walletAddress)
                + "\",\"blockchain\":\"" + escape(blockchain) + "\"}";

        HttpRequest request = HttpRequest.newBuilder(URI.create(BASE_URL + "/api/aml/screen"))
                .header("Authorization", "Bearer " + API_KEY)
                .header("Accept", "application/json")
                .header("Content-Type", "application/json")
                .timeout(Duration.ofSeconds(60))
                .POST(HttpRequest.BodyPublishers.ofString(payload, StandardCharsets.UTF_8))
                .build();

        return send(request);
    }

    /**
     * Проверка ФОРМАТА адреса и поддержки сети — без обращения к блокчейну.
     *
     * Отвечает на вопрос «можно ли вообще скринить такую пару адрес+сеть», а не
     * «существует ли этот адрес». Учитывается так же, как и скрининг: это
     * обращение к сервису.
     */
    static String validateWallet(String walletAddress, String blockchain)
            throws IOException, InterruptedException {
        String query = "walletAddress=" + URLEncoder.encode(walletAddress, StandardCharsets.UTF_8)
                + "&blockchain=" + URLEncoder.encode(blockchain, StandardCharsets.UTF_8);

        HttpRequest request = HttpRequest.newBuilder(URI.create(BASE_URL + "/api/aml/validate?" + query))
                .header("Authorization", "Bearer " + API_KEY)
                .header("Accept", "application/json")
                .timeout(Duration.ofSeconds(60))
                .GET()
                .build();

        return send(request);
    }

    static String escape(String value) {
        return value.replace("\\", "\\\\").replace("\"", "\\\"");
    }

    // ── Разбор JSON ──────────────────────────────────────────────────────────
    // Пример намеренно оставлен без внешних зависимостей, чтобы запускаться одной
    // командой `java Main.java`. В рабочем проекте берите Jackson или Gson и
    // маппьте ответ в полноценную запись — эти регулярки и ручной обход скобок
    // существуют только ради отсутствия pom.xml.

    static String str(String json, String field) {
        Matcher matcher = Pattern.compile("\"" + field + "\"\\s*:\\s*\"((?:[^\"\\\\]|\\\\.)*)\"").matcher(json);
        return matcher.find() ? matcher.group(1).replace("\\\"", "\"") : null;
    }

    static boolean bool(String json, String field) {
        Matcher matcher = Pattern.compile("\"" + field + "\"\\s*:\\s*(true|false)").matcher(json);
        return matcher.find() && "true".equals(matcher.group(1));
    }

    static Double number(String json, String field) {
        Matcher matcher = Pattern.compile("\"" + field + "\"\\s*:\\s*(-?\\d+(?:\\.\\d+)?)").matcher(json);
        return matcher.find() ? Double.valueOf(matcher.group(1)) : null;
    }

    static int integer(String json, String field, int fallback) {
        Double value = number(json, field);
        return value == null ? fallback : value.intValue();
    }

    /** Возвращает объекты массива `"field": [ {...}, {...} ]` как список JSON-строк. */
    static List<String> objects(String json, String field) {
        List<String> items = new ArrayList<>();
        int start = json.indexOf("\"" + field + "\"");
        if (start < 0) {
            return items;
        }
        int open = json.indexOf('[', start);
        if (open < 0) {
            return items;
        }

        int depth = 0;
        int itemStart = -1;
        for (int i = open; i < json.length(); i++) {
            char symbol = json.charAt(i);
            if (symbol == ']' && depth == 0) {
                break;
            }
            if (symbol == '{') {
                if (depth == 0) {
                    itemStart = i;
                }
                depth++;
            } else if (symbol == '}') {
                depth--;
                if (depth == 0 && itemStart >= 0) {
                    items.add(json.substring(itemStart, i + 1));
                }
            }
        }
        return items;
    }

    // ── Применение данных: комплаенс-гейт перед выводом или зачислением средств ──
    // Карточка скрининга сама по себе — просто JSON. Ценность появляется, когда по ней
    // принимают решение. Ниже — гейт из трёх исходов: ALLOW / REVIEW / BLOCK.

    // ВНИМАНИЕ: пороги ниже — отправная точка, а не универсальная истина.
    // Настраивайте их под свою риск-политику, юрисдикцию и аппетит к риску.
    static final int SCORE_BLOCK = 75;              // балл >= этого — блокируем
    static final int SCORE_REVIEW = 40;             // балл >= этого — на ручную проверку
    static final double EXPOSURE_ALERT_PCT = 5.0;   // доля средств из «плохой» категории — уже сигнал

    /**
     * Категории источников/назначения средств, которые сами по себе поднимают флаг.
     * Сравнение по подстроке — коды категорий у провайдеров различаются в написании.
     */
    static final List<String> HIGH_RISK_CATEGORY_KEYWORDS = List.of(
            "sanction", "mixer", "tumbler", "darknet", "ransomware",
            "stolen", "scam", "fraud", "terror", "illicit", "hack", "phishing");

    /** Исход комплаенс-гейта. Порядок важен: чем дальше, тем строже. */
    enum Decision { ALLOW, REVIEW, BLOCK }

    /** Решение комплаенс-гейта и факторы, которые на него повлияли. */
    static final class GateResult {
        Decision decision = Decision.ALLOW;
        final List<String> factors = new ArrayList<>();

        /** Решение может только ужесточаться: ALLOW → REVIEW → BLOCK. */
        void escalate(Decision next) {
            if (next.ordinal() > decision.ordinal()) {
                decision = next;
            }
        }
    }

    static boolean isHighRiskCategory(String category) {
        String lowered = category == null ? "" : category.toLowerCase(Locale.ROOT);
        return HIGH_RISK_CATEGORY_KEYWORDS.stream().anyMatch(lowered::contains);
    }

    /** Комплаенс-гейт: пропустить операцию, отправить на ручную проверку или заблокировать. */
    static GateResult complianceGate(String result) {
        GateResult gate = new GateResult();

        // 1. Санкции — безусловный стоп. Никакой низкий балл это не перевешивает.
        if (bool(result, "sanctionsHit")) {
            gate.escalate(Decision.BLOCK);
            gate.factors.add("[!] Санкционное совпадение — блокировка безусловна");
        } else {
            gate.factors.add("[+] Санкционных совпадений нет");
        }

        // 2. Публичное должностное лицо (PEP) среди контрагентов — тоже безусловный стоп:
        //    такие операции требуют отдельного решения комплаенс-офицера, а не автомата.
        if (bool(result, "pepCounterparty")) {
            gate.escalate(Decision.BLOCK);
            gate.factors.add("[!] Контрагент из списка PEP — блокировка безусловна");
        } else {
            gate.factors.add("[+] PEP-контрагентов нет");
        }

        // 3. Балл риска 0–100.
        int score = integer(result, "riskScore", 0);
        if (score >= SCORE_BLOCK) {
            gate.escalate(Decision.BLOCK);
            gate.factors.add("[!] Риск-балл " + score + " >= порога блокировки " + SCORE_BLOCK
                    + " (severity " + str(result, "severity") + ")");
        } else if (score >= SCORE_REVIEW) {
            gate.escalate(Decision.REVIEW);
            gate.factors.add("[?] Риск-балл " + score + " в диапазоне ручной проверки "
                    + SCORE_REVIEW + "–" + (SCORE_BLOCK - 1));
        } else {
            gate.factors.add("[+] Риск-балл " + score + " ниже порога ручной проверки (" + SCORE_REVIEW + ")");
        }

        // 4. Экспозиция на токсичные категории: миксеры, даркнет, вымогатели, краденое.
        //    Балл может быть умеренным, а вот прямая связь с миксером — нет.
        List<String> exposures = new ArrayList<>(objects(result, "sourceOfFunds"));
        exposures.addAll(objects(result, "destinationOfFunds"));

        for (String exposure : exposures) {
            String category = str(exposure, "category");
            Double percentage = number(exposure, "percentage");
            if (isHighRiskCategory(category) && percentage != null && percentage >= EXPOSURE_ALERT_PCT) {
                gate.escalate(Decision.REVIEW);
                String name = str(exposure, "entityName");
                gate.factors.add(String.format(Locale.ROOT,
                        "[?] Экспозиция %.1f%% на категорию «%s» (%s)",
                        percentage, category, name == null ? "контрагент не назван" : name));
            }
        }

        String dominant = str(result, "dominantRiskCategory");
        if (dominant != null) {
            gate.factors.add("[i] Доминирующая категория риска: " + dominant);
        }

        return gate;
    }

    // ── Печать отчёта ────────────────────────────────────────────────────────

    static String money(Double value) {
        if (value == null) {
            return "н/д";
        }
        String formatted = String.format(Locale.ROOT, "%,.2f", value).replace(',', ' ');
        return "$" + formatted;
    }

    static String orDash(String value) {
        return value == null || value.isEmpty() ? "—" : value;
    }

    static String yesNo(boolean value) {
        return value ? "ДА" : "нет";
    }

    static void printExposures(String title, List<String> exposures) {
        if (exposures.isEmpty()) {
            return;
        }
        System.out.println("  " + title + ":");
        for (String exposure : exposures) {
            Double percentage = number(exposure, "percentage");
            StringBuilder line = new StringBuilder("    ")
                    .append(String.format(Locale.ROOT, "%.1f%%", percentage == null ? 0.0 : percentage))
                    .append(" · ").append(orDash(str(exposure, "category")))
                    .append(" · ").append(orDash(str(exposure, "entityName")));

            String description = str(exposure, "description");
            if (description != null) {
                line.append(" · ").append(description);
            }
            String country = str(exposure, "country");
            if (country != null) {
                line.append(" · ").append(country);
            }
            System.out.println(line);
        }
    }

    public static void main(String[] args) throws Exception {
        if (API_KEY.equals(SANDBOX_KEY)) {
            System.out.println("Демо-ключ: ответы сгенерированы (моки), не реальные данные.\n");
        }

        String wallet = args.length > 0 ? args[0] : "1A1zP1eP5QGefi2DMPTfTL5SLmv7DivfNa";
        String blockchain = args.length > 1 ? args[1] : "BTC";

        String result;
        try {
            result = screenWallet(wallet, blockchain);
        } catch (AtloriumException error) {
            System.err.println("Ошибка: " + error.getMessage());
            System.exit(1);
            return;
        }

        System.out.println(str(result, "blockchain") + " " + str(result, "walletAddress"));
        System.out.println("  Риск-балл: " + integer(result, "riskScore", 0) + "/100"
                + " · severity: " + str(result, "severity")
                + " · статус: " + str(result, "status"));

        String summary = str(result, "summary");
        if (summary != null) {
            System.out.println("  Резюме: " + summary);
        }
        System.out.println("  Санкции: " + yesNo(bool(result, "sanctionsHit"))
                + " · PEP: " + yesNo(bool(result, "pepCounterparty")));
        System.out.println("  Оборот: " + money(number(result, "totalValueUsd"))
                + " · комиссии: " + money(number(result, "feeUsd")));

        printExposures("Источники средств", objects(result, "sourceOfFunds"));
        printExposures("Назначение средств", objects(result, "destinationOfFunds"));

        List<String> counterparties = objects(result, "counterpartyConnections");
        if (!counterparties.isEmpty()) {
            System.out.println("  Контрагенты:");
            for (String counterparty : counterparties) {
                Double percentage = number(counterparty, "percentage");
                System.out.println("    " + orDash(str(counterparty, "entityName"))
                        + " · " + orDash(str(counterparty, "entityType"))
                        + " · риск " + orDash(str(counterparty, "riskLevel"))
                        + " · " + String.format(Locale.ROOT, "%.1f%%", percentage == null ? 0.0 : percentage));
            }
        }

        GateResult gate = complianceGate(result);
        System.out.println("\nКОМПЛАЕНС-ГЕЙТ: " + gate.decision);
        gate.factors.forEach(factor -> System.out.println("  " + factor));
    }
}