#!/usr/bin/env python
# -*- coding: utf-8 -*-

import requests
import json
import sys
import time

def get_emails_by_address(email_address, limit=10, unread_only=False, since=None):
    """
    通过邮箱地址获取最新邮件
    
    参数:
        email_address: 完整邮箱地址 (例如: user@example.com)
        limit: 返回的邮件数量 (默认: 10)
        unread_only: 是否只返回未读邮件 (默认: False)
        since: 从指定时间戳后获取邮件 (可选)
    
    返回:
        API响应的JSON数据
    """
    base_url = "http://localhost:5000/api/emails/by-address"
    
    # 构建参数
    params = {
        'email_address': email_address,
        'limit': limit
    }
    
    if unread_only:
        params['unread_only'] = 'true'
        
    if since:
        params['since'] = since
    
    # 发送请求
    try:
        print(f"请求 URL: {base_url}?{'&'.join([f'{k}={v}' for k, v in params.items()])}")
        response = requests.get(base_url, params=params)
        
        # 打印响应信息
        print(f"状态码: {response.status_code}")
        
        if response.status_code == 200:
            data = response.json()
            return data
        else:
            print(f"请求失败: {response.text}")
            return None
    except Exception as e:
        print(f"请求出错: {str(e)}")
        return None

def format_email_info(email):
    """格式化邮件信息显示"""
    result = []
    result.append(f"ID: {email.get('id')}")
    result.append(f"主题: {email.get('subject')}")
    result.append(f"发件人: {email.get('sender')}")
    result.append(f"收件人: {email.get('recipients')}")
    result.append(f"接收时间: {email.get('received_at')}")
    
    if 'verification_code' in email and email['verification_code']:
        result.append(f"验证码: {email.get('verification_code')}")
        
    if 'verification_link' in email and email['verification_link']:
        result.append(f"验证链接: {email.get('verification_link')}")
    
    return "\n".join(result)

def main():
    # 检查命令行参数
    if len(sys.argv) < 2:
        print("用法: python test_email_by_address.py <邮箱地址> [limit] [unread_only] [since]")
        print("例如: python test_email_by_address.py test@example.com 5 true")
        return
    
    # 解析参数
    email_address = sys.argv[1]
    limit = int(sys.argv[2]) if len(sys.argv) > 2 else 10
    unread_only = sys.argv[3].lower() == 'true' if len(sys.argv) > 3 else False
    since = float(sys.argv[4]) if len(sys.argv) > 4 else None
    
    print(f"获取邮箱 {email_address} 的最新邮件")
    print(f"参数: limit={limit}, unread_only={unread_only}, since={since}")
    
    # 获取邮件
    result = get_emails_by_address(email_address, limit, unread_only, since)
    
    # 显示结果
    if result and result.get('success'):
        print("\n===== 查询结果 =====")
        print(f"邮箱地址: {result.get('email_address')}")
        print(f"邮箱ID: {result.get('mailbox_id')}")
        print(f"总邮件数: {result.get('total')}")
        print(f"返回邮件数: {result.get('count')}")
        
        # 显示邮件详情
        emails = result.get('emails', [])
        if emails:
            print("\n----- 邮件列表 -----")
            for i, email in enumerate(emails, 1):
                print(f"\n邮件 {i}:")
                print(format_email_info(email))
                print("-" * 40)
        else:
            print("\n没有找到邮件")
            
        # 返回时间戳，可用于下次查询
        print(f"\n当前时间戳: {result.get('timestamp')}")
        print("在下次查询时可以使用此时间戳作为since参数，仅获取新邮件")
    else:
        print("查询失败或没有结果")

if __name__ == "__main__":
    main()