import sys
from pathlib import Path

# 导入技能模块
sys.path.append(str(Path(__file__).parent))
from main_send_file import find_account_json, send_weixin_file

# 查询账号信息
account_id = "6e2f83e62b99-im-bot"
account_info = find_account_json(account_id)
if "error" in account_info:
    print(f"查询失败: {account_info['error']}")
else:
    print(f"成功查询到账号: {account_info['account_id']}")
    
    # 获取发送文件所需的配置
    result = account_info["data"]
    BASE_URL = result[f"{account_id}"]["baseUrl"]
    BOT_TOKEN = result[f"{account_id}"]["token"]
    TARGET_USER_ID = result[f"{account_id}"]["userId"]
    CONTEXT_TOKEN = result[f"{account_id}.context-tokens"][TARGET_USER_ID]
    FILE_PATH = "/path/to/your/file.jpg"
    
    # 发送文件
    send_weixin_file(BOT_TOKEN, TARGET_USER_ID, CONTEXT_TOKEN, FILE_PATH)

def send_weixin_file(BOT_TOKEN, TARGET_USER_ID, CONTEXT_TOKEN, IMAGE_PATH):
    """
    根据文件类型发送微信文件
    
    参数说明：
    BOT_TOKEN: 机器人令牌
    TARGET_USER_ID: 目标用户ID
    CONTEXT_TOKEN: 上下文令牌
    FILE_PATH: 文件路径（支持图片、视频、文件等多种类型）
    """

def aes_encrypt_file(file_path: str, aes_key_hex: str) -> bytes:
    """
    使用AES-128-ECB模式加密文件，使用PKCS7 padding
    """

prepare_result = prepare_image_upload(IMAGE_PATH)
# 返回: filekey, aeskey_hex, rawsize, rawfilemd5, filesize, file_data

upload_params = get_upload_params(
    use_token=BOT_TOKEN,
    filekey=filekey,
    media_type=media_type,  # 根据文件类型自动判断
    to_user_id=TARGET_USER_ID,
    rawsize=rawsize,
    rawfilemd5=rawfilemd5,
    filesize=filesize,
    aeskey_hex=aeskey_hex,
    no_need_thumb=True
)

encrypted_data = aes_encrypt_file(IMAGE_PATH, aeskey_hex)
# 使用AES-128-ECB + PKCS7填充

upload_result = upload_to_cdn(
    upload_params["upload_param"],
    filekey,
    encrypted_data
)