import argparse
import os
from cyber_py3 import record
from cyber_py3.record import RecordReader, RecordWriter, RecordMessage
 
def parse_record_file(file_path):
    """
    解析單個 Cyber Record 文件
    """
    try:
        reader = RecordReader(file_path)
        print(f"\n===== 正在解析文件: {os.path.basename(file_path)} =====")
        print(f"總消息數(shù): {reader.get_messagenumber()}")
        print(f"開始時間: {reader.get_starttime()}")
        print(f"結(jié)束時間: {reader.get_endtime()}")
        print(f"通道列表: {reader.get_channellist()}\n")
 
        # 遍歷所有消息
        for channel_name, msg, datatype, timestamp in reader.read_messages():
            print(f"[通道] {channel_name}")
            print(f"  時間戳: {timestamp}")
            print(f"  數(shù)據(jù)類型: {datatype}")
            print(f"  消息長度: {len(msg)} bytes")
            print("-" * 60)
 
    except Exception as e:
        print(f"解析文件 {file_path} 失敗: {str(e)}")
 
def parse_record_directory(directory):
    """
    解析目錄下的所有 Cyber Record 文件
    """
    if not os.path.isdir(directory):
        print(f"錯誤: {directory} 不是有效目錄")
        return
 
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)
        
        # 通過文件頭驗證是否為合法 Record 文件
        if record.is_valid_record_file(file_path):
            parse_record_file(file_path)
        else:
            print(f"跳過非 Record 文件: {filename}")
 
if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Cyber Record 文件解析工具')
    parser.add_argument('path', type=str, help='文件路徑或目錄路徑')
    args = parser.parse_args()
 
    target_path = args.path
 
    if os.path.isfile(target_path):
        if record.is_valid_record_file(target_path):
            parse_record_file(target_path)
        else:
            print(f"錯誤: {target_path} 不是有效的 Cyber Record 文件")
    elif os.path.isdir(target_path):
        parse_record_directory(target_path)
    else:
        print(f"錯誤: {target_path} 不存在")

# 解析單個文件
python parse_record.py /path/to/your.record
 
# 解析目錄
python parse_record.py /path/to/record_dir/