#!/usr/bin/env python3
"""简易文件上传客户端，用于通过 API 上传文件到临时文件传输服务"""

import requests
import sys
import os

BASE_URL = "http://localhost:5000"

def upload_file(filepath, expiry='24h'):
    """
    上传文件到临时文件传输服务

    Args:
        filepath: 要上传的文件路径
        expiry: 过期时间，可选值: '1h', '24h', '7d'

    Returns:
        dict: 包含文件信息的字典
    """
    if not os.path.exists(filepath):
        raise FileNotFoundError(f"文件不存在: {filepath}")

    url = f"{BASE_URL}/api/upload"
    with open(filepath, 'rb') as f:
        files = {'file': (os.path.basename(filepath), f)}
        data = {'expiry': expiry}
        response = requests.post(url, files=files, data=data)

    if response.status_code == 200:
        return response.json()
    else:
        raise Exception(f"上传失败: {response.json().get('error', '未知错误')}")

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("用法: python upload_client.py <文件路径> [过期时间]")
        print("过期时间可选: 1h, 24h, 7d (默认: 24h)")
        sys.exit(1)

    filepath = sys.argv[1]
    expiry = sys.argv[2] if len(sys.argv) > 2 else '24h'

    try:
        result = upload_file(filepath, expiry)
        print(f"上传成功!")
        print(f"文件ID: {result['id']}")
        print(f"文件名: {result['filename']}")
        print(f"分享链接: {result['share_url']}")
        if 'filesize' in result:
            print(f"文件大小: {result['filesize']} 字节")
    except Exception as e:
        print(f"错误: {e}")
        sys.exit(1)