# File: drop_position_manager.py

import configparser
import os
from typing import Optional
from Model.Position import Real_Position

class DropPositionManager:
    def __init__(self, config_path="drop.ini"):
        self.config_path = config_path
        self.config = configparser.ConfigParser()
        self._load_config()

    def _load_config(self):
        if not os.path.exists(self.config_path):
            raise FileNotFoundError(f"配置文件不存在: {self.config_path}")
        self.config.read(self.config_path, encoding='utf-8')

    def _save_config(self):
        with open(self.config_path, 'w', encoding='utf-8') as f:
            self.config.write(f)

    def get_next_drop_position(self, lineid: int = 1) -> Optional[Real_Position]:
        print(f"\n🔄 开始获取 lineid={lineid} 的下一个 drop 点...")

        line_section = f"DropLine{lineid}"
        print(f"🔍 查找路径 section: {line_section}")

        if not self.config.has_section(line_section):
            print(f"❌ 错误：配置文件中不存在 section [{line_section}]")
            print(f"📊 所有 sections: {list(self.config.sections())}")
            return None

        current_index = self.config.getint(line_section, "current_index", fallback=0)
        print(f"📌 当前索引: {current_index} (fallback=0 如果未设置)")

        points = []
        found_sections = []
        matched_sections = []

        for sec in self.config.sections():
            if sec.startswith("DropPoints"):
                found_sections.append(sec)
                try:
                    file_lineid = self.config.getint(sec, "lineid")
                    print(f"🔍 {sec}: lineid={file_lineid}")
                    if file_lineid == lineid:
                        matched_sections.append(sec)
                        pos = Real_Position()
                        pos.X = self.config.getfloat(sec, "x")
                        pos.Y = self.config.getfloat(sec, "y")
                        pos.Z = self.config.getfloat(sec, "z")
                        pos.U = self.config.getfloat(sec, "u")
                        pos.V = self.config.getfloat(sec, "v")
                        pos.W = self.config.getfloat(sec, "w")
                        order = self.config.getint(sec, "order")
                        points.append((order, pos))
                        print(f"✅ 匹配到点: {sec}, order={order}, 位置=({pos.X:.3f}, {pos.Y:.3f})")
                except Exception as e:
                    print(f"❌ 读取 {sec} 失败: {e}")

        print(f"📁 找到 DropPoints: {found_sections}")
        print(f"🟢 匹配 lineid={lineid} 的点: {matched_sections}")
        print(f"📊 共收集 {len(points)} 个有效点")

        if not points:
            print("❌ 没有可用的 drop 点！请检查：lineid 是否匹配，字段是否正确")
            return None

        # 按 order 排序
        points.sort(key=lambda x: x[0])
        sorted_positions = [pos for _, pos in points]
        count = len(sorted_positions)
        actual_index = current_index % count

        selected_pos = sorted_positions[actual_index]
        print(f"🎯 选择第 {actual_index} 个点: X={selected_pos.X:.3f}, Y={selected_pos.Y:.3f}, Z={selected_pos.Z:.3f}")

        # 索引 +1 并保存回 ini
        next_index = current_index + 1
        self.config.set(line_section, "current_index", str(next_index))
        self._save_config()
        print(f"💾 已保存 current_index = {next_index} 到 [{line_section}]")

        return selected_pos


# ✅ 主函数：用于测试
if __name__ == "__main__":
    # 创建管理器，加载 drop.ini
    manager = DropPositionManager("drop.ini")

    # 获取第一个 drop 点（lineid=1）
    pos = manager.get_next_drop_position(lineid=1)

    if pos is not None:
        print(f"\n🎉 成功返回点位: X={pos.X}, Y={pos.Y}, Z={pos.Z}, U={pos.U}, V={pos.V}, W={pos.W}")
    else:
        print("\n❌ get_next_drop_position 返回了 None，请根据上面的日志排查")
        print("💡 常见原因：")
        print("   1. drop.ini 缺少 current_index")
        print("   2. lineid 不匹配")
        print("   3. Real_Position 属性名错误（应为大写 X/Y/Z）")
        print("   4. 文件路径不对")