package com.bjtds.brichat.util;

import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class TrainUtils {

    /***
     *  列车号转车厢
     * @param trainNo
     * @return
     */
    public static Map<String, String> convertTrainNo(String trainNo) {
        // 1. 提取前缀后的数字部分
        Pattern pattern = Pattern.compile("(\\d+)$");
        Matcher matcher = pattern.matcher(trainNo);

        if (!matcher.find()) {
            throw new IllegalArgumentException("无法在前缀 '" + trainNo + "' 中找到数字");
        }
        int i = Integer.parseInt(matcher.group(1));

        // 2. 计算三个基础编号
        int code1 = 3 * (i - 1) + 1;
        int code2 = code1 + 1;
        int code3 = code1 + 2;

        // 格式化为三位数
        String code1Str = String.format("%03d", code1);
        String code2Str = String.format("%03d", code2);
        String code3Str = String.format("%03d", code3);

        // 3. 组装九个车厢号
        List<String> part1 = Arrays.asList("D" + code1Str, "P" + code1Str, "M" + code1Str);
        List<String> part2 = Arrays.asList("M" + code2Str, "P" + code2Str, "F" + code2Str);
        List<String> part3 = Arrays.asList("M" + code3Str, "P" + code3Str, "D" + code3Str);

        // 合并结果
        List<String> allParts = new ArrayList<>();
        allParts.addAll(part1);
        allParts.addAll(part2);
        allParts.addAll(part3);

        // 转换为逗号分隔字符串
        String result = String.join(", ", allParts);

        return Collections.singletonMap("result", result);
    }

    /*
     * 将车厢号转换为列车号
     * @param carriageNumber 车厢号（例如 "D109"）
            * @return 列车号（例如 "RT37"）
            */
    public static String convertToTrainNo(String carriageNumber) {
        // 1. 提取数字部分并验证格式
        Matcher matcher = Pattern.compile("\\d+").matcher(carriageNumber);
        if (!matcher.find()) {
            throw new IllegalArgumentException("无效的车厢号格式: " + carriageNumber);
        }
        int n = Integer.parseInt(matcher.group());

        // 2. 计算行号 i
        int i = (n + 2) / 3; // 等效于 (n - 1)/3 + 1

        // 3. 验证行号范围
        if (i < 1 || i > 37) {
            throw new IllegalArgumentException("行号超出有效范围 1-37");
        }

        // 4. 验证数字是否符合规则
        int expectedCode1 = 3 * (i - 1) + 1;
        int expectedCode2 = expectedCode1 + 1;
        int expectedCode3 = expectedCode1 + 2;
        if (n != expectedCode1 && n != expectedCode2 && n != expectedCode3) {
            throw new IllegalArgumentException("数字 " + n + " 不符合编码规则");
        }

        // 5. 组合列车号（假设前缀为 RT）
        return "TS" + i;
    }
}