"""PinMAP ↔ PinList bidirectional converter

Usage:
  python main.py              # Interactive — choose direction + file
  python main.py input.xls    # MAP→List mode (legacy, specify file directly)
"""

import sys
import os


# ── Banner ──────────────────────────────────────────────────────────

def show_banner():
    """显示程序启动说明"""
    print("=" * 60)
    print("  PinMAP ↔ PinList 双向转换器")
    print("  支持 PinMAP→PinList 与 PinList→PinMAP 互转")
    print("  支持.xls和.xlsx格式，输出.xlsx格式")
    print("=" * 60)
    print()


def wait_for_exit():
    """等待用户按键后退出（Windows任意键，其他平台Enter键）"""
    try:
        import msvcrt
        print("按任意键退出...")
        msvcrt.getch()
    except ImportError:
        input("按Enter键退出...")


# ── Path helpers ────────────────────────────────────────────────────

def _build_output_path_map_to_list(input_path: str) -> str:
    """Generate output path: {original_filename}_PinList.xlsx"""
    base, _ = os.path.splitext(input_path)
    return f"{base}_PinList.xlsx"


def _build_output_path_list_to_map(input_path: str) -> str:
    """Generate output path: {original_filename}_PinMAP.xlsx"""
    base, _ = os.path.splitext(input_path)
    return f"{base}_PinMAP.xlsx"


# ── Direction 1: MAP → List ────────────────────────────────────────

def run_map_to_list(filepath: str):
    """执行 PinMAP → PinList 转换流程。"""
    from file_selector import select_file
    from xls_reader import read_excel_cells
    from xlsx_reader import read_excel_cells as read_xlsx_cells
    from pinmap_parser import parse_pinmap
    from validator import validate_pinmap
    from pinlist_generator import generate_pinlist
    from xlsx_writer import write_xlsx
    from models import FileFormatError, StructureError

    # ── 1. File selection ───────────────────────────────────────────
    if not filepath:
        filepath = select_file(mode="map_to_list")

    if not filepath:
        print("未选择文件，退出。")
        wait_for_exit()
        return

    # ── 2. Read Excel ───────────────────────────────────────────────
    print(f"[INFO] 正在读取文件: {filepath}")
    try:
        if filepath.lower().endswith('.xlsx'):
            cells = read_xlsx_cells(filepath)
        else:
            cells = read_excel_cells(filepath)
    except Exception as e:
        print(f"[FATAL] 文件读取失败: {e}")
        wait_for_exit()
        return

    print(f"[INFO] 文件读取完成，共 {len(cells)} 个非空单元格")

    # ── 3. Parse PinMAP ─────────────────────────────────────────────
    print("[INFO] 正在解析 PinMAP 结构...")
    try:
        pinmap = parse_pinmap(cells)
        print(f"[INFO] 解析完成: {pinmap.width}x{pinmap.height} 方形，共 {len(pinmap.pins)} 个Pin")
        print(f"[INFO] 封装信息: {pinmap.package_info}")
    except (FileFormatError, StructureError) as e:
        print(f"[FATAL] 结构错误: {e}")
        wait_for_exit()
        return

    # ── 4. Validate ─────────────────────────────────────────────────
    print("[INFO] 正在验证数据...")
    validation = validate_pinmap(pinmap)

    if validation.errors:
        print(f"[ERROR] 验证未通过，发现 {len(validation.errors)} 个错误:")
        for err in validation.errors:
            print(f"  - {err.message}: {err.details}")
        print("\n转换终止，请修正PinMAP文件后重试。")
        wait_for_exit()
        return

    if validation.warnings:
        print(f"[WARN] 发现 {len(validation.warnings)} 个警告:")
        for warn in validation.warnings:
            print(f"  - {warn.message}: {warn.details}")
    else:
        print("[INFO] 验证通过")

    # ── 5. Generate PinList ─────────────────────────────────────────
    print("[INFO] 正在生成 PinList...")
    pinlist = generate_pinlist(pinmap, validation)

    # ── 6. Write XLSX ───────────────────────────────────────────────
    output_path = _build_output_path_map_to_list(filepath)
    print(f"[INFO] 正在写入输出文件: {output_path}")
    try:
        data = {}
        data['A1'] = pinlist.package_info
        for i, (pin_name, pin_num) in enumerate(pinlist.rows):
            row = i + 2
            data[f'A{row}'] = pin_name
            data[f'B{row}'] = str(pin_num)

        write_xlsx(data, output_path)
    except Exception as e:
        print(f"[FATAL] 输出失败: {e}")
        wait_for_exit()
        return

    # ── 7. Result summary ───────────────────────────────────────────
    print()
    print("[SUCCESS] 转换完成！")
    print(f"  输出文件: {output_path}")
    print(f"  封装信息: {pinlist.package_info}")
    print(f"  Pin数量: {len(pinlist.rows)}")

    wait_for_exit()


# ── Direction 2: List → MAP ────────────────────────────────────────

def run_list_to_map(filepath: str):
    """执行 PinList → PinMAP 转换流程。"""
    from file_selector import select_file
    from pinlist_parser import parse_pinlist
    from pinlist_validator import validate_pinlist
    from pinmap_generator import generate_pinmap, generate_output_path
    from template_reader import read_template_styles
    from models import StructureError, LayoutError

    # ── 1. File selection ───────────────────────────────────────────
    if not filepath:
        filepath = select_file(mode="list_to_map")

    if not filepath:
        print("未选择文件，退出。")
        wait_for_exit()
        return

    # ── 2. Input PinMAP dimensions ──────────────────────────────────
    while True:
        try:
            rows_input = input("请输入 PinMAP 行数: ").strip()
            rows = int(rows_input)
            if rows < 2:
                print("[ERROR] 行数至少为 2")
                continue
            break
        except ValueError:
            print("[ERROR] 请输入有效的整数")

    while True:
        try:
            cols_input = input("请输入 PinMAP 列数: ").strip()
            cols = int(cols_input)
            if cols < 2:
                print("[ERROR] 列数至少为 2")
                continue
            break
        except ValueError:
            print("[ERROR] 请输入有效的整数")

    print(f"[INFO] PinMAP 尺寸: {rows} 行 × {cols} 列")

    # ── 3. Parse PinList ────────────────────────────────────────────
    print(f"[INFO] 正在解析 PinList 文件: {filepath}")
    try:
        package_info, entries = parse_pinlist(filepath)
        print(f"[INFO] 解析完成: 封装信息 '{package_info}', 共 {len(entries)} 个引脚")
    except StructureError as e:
        print(f"[FATAL] 解析失败: {e}")
        wait_for_exit()
        return

    # ── 4. Validate ─────────────────────────────────────────────────
    print("[INFO] 正在验证数据...")
    validation = validate_pinlist(entries, rows, cols)

    if validation.errors:
        print(f"[ERROR] 验证未通过，发现 {len(validation.errors)} 个错误:")
        for err in validation.errors:
            print(f"  - {err.message}: {err.details}")
        print("\n转换终止，请修正PinList文件或网格尺寸后重试。")
        wait_for_exit()
        return

    if validation.warnings:
        print(f"[WARN] 发现 {len(validation.warnings)} 个警告:")
        for warn in validation.warnings:
            print(f"  - {warn.message}: {warn.details}")
    else:
        print("[INFO] 验证通过")

    # ── 5. Generate PinMAP ──────────────────────────────────────────
    output_path = generate_output_path(filepath)
    print(f"[INFO] 正在生成 PinMAP 并写入: {output_path}")

    try:
        # 尝试读取模板样式（优雅降级）
        template_style = read_template_styles(filepath)

        generate_pinmap(
            entries=entries,
            rows=rows,
            cols=cols,
            package_info=package_info,
            template_style=template_style,
            output_path=output_path,
        )
    except LayoutError as e:
        print(f"[FATAL] 布局计算失败: {e}")
        wait_for_exit()
        return
    except Exception as e:
        print(f"[FATAL] 输出失败: {e}")
        wait_for_exit()
        return

    # ── 6. Result summary ───────────────────────────────────────────
    print()
    print("[SUCCESS] 转换完成！")
    print(f"  输出文件: {output_path}")
    print(f"  封装信息: {package_info}")
    print(f"  PinMAP 尺寸: {rows}×{cols}")
    print(f"  Pin数量: {len(entries)}")

    wait_for_exit()


# ── Main entry ──────────────────────────────────────────────────────

def main():
    show_banner()

    # ── Direction selection ─────────────────────────────────────────
    if len(sys.argv) > 1:
        # Legacy mode: direct file argument → MAP→List
        direction = 1
        filepath = sys.argv[1]
    else:
        print("请选择转换方向:")
        print("  1 — PinMAP → PinList")
        print("  2 — PinList → PinMAP")
        print()

        while True:
            choice = input("请输入选项 (1/2): ").strip()
            if choice in ('1', '2'):
                direction = int(choice)
                break
            print("[ERROR] 无效选项，请输入 1 或 2")

        filepath = None  # will be selected inside the flow

    # ── Dispatch ────────────────────────────────────────────────────
    if direction == 1:
        print()
        print("─" * 40)
        print("  方向: PinMAP → PinList")
        print("─" * 40)
        print()
        run_map_to_list(filepath)
    else:
        print()
        print("─" * 40)
        print("  方向: PinList → PinMAP")
        print("─" * 40)
        print()
        run_list_to_map(filepath)


if __name__ == '__main__':
    main()