from house_service import add_house, delete_house, update_house, query_house
from house_service import push_operation_stack, push_browse_stack
from house_service import pop_operation_stack, pop_browse_stack
from house_service import get_recent_records, clear_stack, get_next_house_id
# V3.0 导入新增业务
from house_service import query_by_area, add_community_distance, show_adj_matrix


def print_menu():
    print("\n===== 房屋出租系统 V3.0 =====")
    print("1. 新增房源")
    print("2. 删除房源")
    print("3. 修改房源")
    print("4. 查询房源")
    print("5. 查看/管理操作记录")
    print("6. 查看/管理浏览历史")
    # V3.0 新增菜单
    print("7. 地址解析与区域查询")
    print("8. 小区邻接矩阵管理")
    print("9. 退出系统")
    print("==============================")


def main():
    while True:
        print_menu()
        choice = input("请输入操作编号（1-9）：")

        # 1-6 功能完全保留 V2.0 代码
        if choice == "1":
            print("\n----- 新增房源 -----")
            address = input("请输入房源地址：")
            try:
                price = float(input("请输入租金（元/月）："))
                area = float(input("请输入面积（㎡）："))
            except ValueError:
                print("错误：租金/面积必须是数字！")
                continue
            house_type = input("请输入户型（如：一居室、两居室）：")
            if add_house(address, price, area, house_type):
                new_id = get_next_house_id() - 1
                push_operation_stack(f"新增房源ID:{new_id}，地址:{address}")

        elif choice == "2":
            print("\n----- 删除房源 -----")
            try:
                house_id = int(input("请输入要删除的房源ID："))
                house = query_house("id", house_id)
                house_addr = house[0]["address"] if house else "未知地址"
                if delete_house(house_id):
                    push_operation_stack(f"删除房源ID:{house_id}，地址:{house_addr}")
            except ValueError:
                print("错误：房源ID必须是整数！")

        elif choice == "3":
            print("\n----- 修改房源 -----")
            try:
                house_id = int(input("请输入要修改的房源ID："))
                new_info = {}
                address = input("请输入新地址（不修改按回车）：")
                if address:
                    new_info["address"] = address
                price_input = input("请输入新租金（不修改按回车）：")
                if price_input:
                    try:
                        new_info["price"] = float(price_input)
                    except ValueError:
                        print("错误：租金必须是数字！")
                        continue
                area_input = input("请输入新面积（不修改按回车）：")
                if area_input:
                    try:
                        new_info["area"] = float(area_input)
                    except ValueError:
                        print("错误：面积必须是数字！")
                        continue
                house_type = input("请输入新户型（不修改按回车）：")
                if house_type:
                    new_info["house_type"] = house_type
                if new_info:
                    if update_house(house_id, new_info):
                        push_operation_stack(f"修改房源ID:{house_id}，修改内容:{new_info}")
                else:
                    print("未输入任何修改内容！")
            except ValueError:
                print("错误：房源ID必须是整数！")

        elif choice == "4":
            print("\n----- 查询房源 -----")
            print("查询条件类型：1-ID  2-地址  3-租金  4-户型")
            cond_choice = input("请输入条件类型编号（1-4）：")
            cond_type_map = {"1": "id", "2": "address", "3": "price", "4": "house_type"}
            if cond_choice not in cond_type_map:
                print("错误：无效的条件类型！")
                continue
            condition_type = cond_type_map[cond_choice]
            condition_value = input(f"请输入{condition_type}查询值：")
            result = query_house(condition_type, condition_value)
            if result:
                print("\n查询结果：")
                for house in result:
                    print(
                        f"ID：{house['id']} | 地址：{house['address']} | 租金：{house['price']}元/月 | 面积：{house['area']}㎡ | 户型：{house['house_type']}")
                    if condition_type == "id":
                        push_browse_stack(house["id"])
            else:
                print("未找到符合条件的房源！")

        elif choice == "5":
            while True:
                print("\n----- 操作记录管理 -----")
                print("1. 查看最新10条操作记录")
                print("2. 获取最新1条操作记录（并删除）")
                print("3. 清空所有操作记录")
                print("4. 返回主菜单")
                c = input("请输入操作编号：")
                if c == "1":
                    records = get_recent_records("operation", 10)
                    if records:
                        print("\n最新10条操作记录（从新到旧）：")
                        for i, r in enumerate(records, 1):
                            print(f"{i}. {r}")
                    else:
                        print("暂无操作记录！")
                elif c == "2":
                    latest = pop_operation_stack()
                    print(latest if latest else "暂无操作记录！")
                elif c == "3":
                    clear_stack("operation")
                elif c == "4":
                    break
                else:
                    print("无效编号！")

        elif choice == "6":
            while True:
                print("\n----- 浏览历史管理 -----")
                print("1. 查看最新10条浏览历史")
                print("2. 获取最新1条浏览房源（并删除）")
                print("3. 清空所有浏览历史")
                print("4. 返回主菜单")
                c = input("请输入操作编号：")
                if c == "1":
                    ids = get_recent_records("browse", 10)
                    if ids:
                        print("\n最新10条浏览历史（从新到旧）：")
                        for i, hid in enumerate(ids, 1):
                            h = query_house("id", hid)
                            if h:
                                h = h[0]
                                print(f"{i}. ID:{hid} | 地址:{h['address']} | 租金:{h['price']}")
                            else:
                                print(f"{i}. ID:{hid}（房源已删除）")
                    else:
                        print("暂无浏览历史！")
                elif c == "2":
                    hid = pop_browse_stack()
                    if hid:
                        h = query_house("id", hid)
                        if h:
                            h = h[0]
                            print(f"最新浏览：ID:{hid} | 地址:{h['address']}")
                        else:
                            print(f"最新浏览ID:{hid}（已删除）")
                    else:
                        print("暂无浏览历史！")
                elif c == "3":
                    clear_stack("browse")
                elif c == "4":
                    break
                else:
                    print("无效编号！")

        # ==================== V3.0 新增：地址解析与区域查询 ====================
        elif choice == "7":
            while True:
                print("\n----- 地址解析与区域查询 -----")
                print("1. 按区域关键词查询房源")
                print("2. 返回主菜单")
                c = input("请输入操作编号：")
                if c == "1":
                    keyword = input("请输入区域关键词（如：朝阳、花园）：")
                    res = query_by_area(keyword)
                    if res:
                        print("\n查询结果：")
                        for h in res:
                            print(f"ID:{h['id']} | 地址:{h['address']} | 租金:{h['price']}")
                    else:
                        print("未找到匹配房源！")
                elif c == "2":
                    break
                else:
                    print("无效编号！")

        # ==================== V3.0 新增：邻接矩阵管理 ====================
        elif choice == "8":
            while True:
                print("\n----- 小区邻接矩阵管理 -----")
                print("1. 设置小区间距离")
                print("2. 查看邻接矩阵")
                print("3. 返回主菜单")
                c = input("请输入操作编号：")
                if c == "1":
                    c1 = input("请输入小区1名称：")
                    c2 = input("请输入小区2名称：")
                    try:
                        dis = int(input("请输入距离（米）："))
                        if add_community_distance(c1, c2, dis):
                            print("距离设置成功！")
                        else:
                            print("小区不存在，请先添加房源！")
                    except ValueError:
                        print("错误：距离必须是数字！")
                elif c == "2":
                    show_adj_matrix()
                elif c == "3":
                    break
                else:
                    print("无效编号！")

        elif choice == "9":
            print("感谢使用房屋出租系统 V3.0，再见！")
            break
        else:
            print("错误：请输入1-9之间的有效编号！")


if __name__ == "__main__":
    main()