import os
import re
import subprocess
import time
import xml.etree.ElementTree as ET

def run_adb_command(command):
    """执行ADB命令并返回输出"""
    result = subprocess.run(command, shell=True, capture_output=True, text=True)
    if result.returncode != 0:
        return None
    return result.stdout.strip()

def pull_ui_dump(file_path):
    """拉取UI Dump文件，确保文件生成成功"""
    device_path = f"/sdcard/{os.path.basename(file_path)}"
    
    # 生成UI转储并等待文件创建
    run_adb_command(f"adb shell uiautomator dump {device_path}")
    time.sleep(1.2)  # 延长等待时间，确保文件生成
    run_adb_command(f"adb pull {device_path} {file_path}")
    run_adb_command(f"adb shell rm {device_path}")
    
    return os.path.exists(file_path)

def extract_list_items(xml_file):
    """提取列表项，确保元素识别稳定"""
    items = []
    try:
        tree = ET.parse(xml_file)
        root = tree.getroot()
        
        # 元素标识常量（与UI Dump匹配）
        LIST_CONTAINER_ID = "com.alicom.smartdail:id/xiaohao_conversation_list"
        NAME_ID = "com.alicom.smartdail:id/xiaohao_item_conversation_name"
        TIME_ID = "com.alicom.smartdail:id/xiaohao_item_conversation_time"
        CONTENT_ID = "com.alicom.smartdail:id/xiaohao_item_conversation_body"
        
        # 查找列表容器
        list_container = root.find(f".//node[@resource-id='{LIST_CONTAINER_ID}']")
        if list_container is None:
            return items
            
        # 遍历所有可点击的列表项
        for node in list_container.iter('node'):
            if node.get('class') == 'android.widget.LinearLayout' and node.get('clickable') == 'true':
                # 查找关键子元素
                name_node = node.find(f".//node[@resource-id='{NAME_ID}']")
                time_node = node.find(f".//node[@resource-id='{TIME_ID}']")
                content_node = node.find(f".//node[@resource-id='{CONTENT_ID}']")
                
                # 确保元素存在且有内容
                if (name_node is not None and time_node is not None and content_node is not None and
                    name_node.get('text') and time_node.get('text') and content_node.get('text')):
                    items.append({
                        'name': name_node.get('text'),
                        'time': time_node.get('text'),
                        'content': content_node.get('text')
                    })
        
    except Exception as e:
        print(f"解析错误: {e}")
    return items

def get_screen_size():
    """获取屏幕尺寸（缓存结果）"""
    if not hasattr(get_screen_size, 'cached_size'):
        output = run_adb_command("adb shell wm size")
        if output:
            size_match = re.search(r"(\d+)x(\d+)", output)
            if size_match:
                get_screen_size.cached_size = (int(size_match.group(1)), int(size_match.group(2)))
                return get_screen_size.cached_size
        get_screen_size.cached_size = (1080, 2210)  # 默认值
    return get_screen_size.cached_size

def scroll_down():
    """优化滚动逻辑，确保滚动距离足够"""
    width, height = get_screen_size()
    # 增加滚动距离（从屏幕2/3处滑到1/4处）
    start_x, start_y = width//2, height*2//3
    end_x, end_y = width//2, height//4
    run_adb_command(f"adb shell input swipe {start_x} {start_y} {end_x} {end_y} 250")  # 延长滑动时间

def main():
    os.makedirs("temp_dumps", exist_ok=True)
    all_items = []
    prev_items = []  # 存储上一页完整内容，用于更准确的底部判断
    page = 1
    consecutive_same = 0  # 连续相同页数计数（避免误判）
    max_consecutive = 2   # 连续2页相同才判断为底部
    max_pages = 1000       # 增加最大页数限制
    
    print("开始收集列表数据（完整模式）...")
    
    while consecutive_same < max_consecutive and page <= max_pages:
        dump_file = f"temp_dumps/page_{page}.xml"
        
        if not pull_ui_dump(dump_file):
            print(f"页面 {page} 获取失败，重试...")
            page += 1
            continue
        
        current_items = extract_list_items(dump_file)
        if not current_items:
            print(f"页面 {page} 未找到数据，继续...")
            page += 1
            continue
        
        # 计算新增数据量
        seen_items = set((i['name'], i['time'], i['content']) for i in all_items)
        new_count = 0
        for item in current_items:
            item_key = (item['name'], item['time'], item['content'])
            if item_key not in seen_items:
                seen_items.add(item_key)
                all_items.append(item)
                new_count += 1
        
        # 判断是否与上一页内容相同（优化底部判断）
        if prev_items and len(current_items) == len(prev_items):
            # 比较所有项而非仅最后一项，避免误判
            if all(current_items[i] == prev_items[i] for i in range(len(current_items))):
                consecutive_same += 1
            else:
                consecutive_same = 0
        else:
            consecutive_same = 0
        
        prev_items = current_items.copy()
        print(f"页面 {page} 新增 {new_count} 项，总计 {len(all_items)} 项")
        
        # 继续滚动
        if consecutive_same < max_consecutive:
            scroll_down()
            page += 1
            time.sleep(1.5)  # 延长滚动后等待时间，确保内容加载
        else:
            print(f"连续 {max_consecutive} 页内容相同，判断为已到达底部")
    
    # 导出结果
    if all_items:
        csv_file = "conversation_list.csv"
        with open(csv_file, "w", encoding="utf-8") as f:
            f.write("号码,时间,内容\n")
            for item in all_items:
                name = item['name'].replace('"', '""')
                time_val = item['time'].replace('"', '""')
                content = item['content'].replace('"', '""')
                f.write(f'"{name}","{time_val}","{content}"\n')
        
        print(f"\n完成! 共收集 {len(all_items)} 条记录")
        print(f"结果已导出到: {csv_file}")
    else:
        print("未找到任何列表项")
    
    # 清理临时文件
    try:
        for file in os.listdir("temp_dumps"):
            os.remove(os.path.join("temp_dumps", file))
        os.rmdir("temp_dumps")
    except:
        pass

if __name__ == "__main__":
    main()