{
  "log_id": "20250622150000XXXX",
  "error_code": 0,
  "error_message": "success",
  "data": {}
}

curl --location 'https://aip.baidubce.com/rpc/2.0/speech/publiccloudspeech/v1/echo/footage/upload' \
--header 'authorization: Bearer xxxxxxx' \
--header 'Content-Type: application/json' \
--data '{
    "file_name": "video_sample.mp4"
}'

{
  "log_id": "20250622150000XXXX",
  "error_code": 0,
  "error_message": "success",
  "data": {
    "access_key_id": "STSxxxxxxxxxxxx",
    "secret_access_key": "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
    "session_token": "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
    "expiration": "2025-06-22T16:00:00Z",
    "resource": "prefix/projects/100001/550e8400-e29b-41d4-a716-446655440000/video_sample.mp4",
    "bucket_name": "echo-prod",
    "endpoint": "https://bos.bj.bcebos.com"
  }
}

// 创建素材
{
  "action_type": "create",
  "create": {
    "footage_name": "角色A参考视频",
    "footage_url": "https://bucket.bj.bcebos.com/path/to/footage.mp4",
    "footage_desc": "角色A 的参考素材"
  }
}

// 更新素材
{
  "action_type": "update",
  "update": { "footage_id": 123456, "footage_name": "角色A参考视频-新版", "footage_desc": "更新后的描述" }
}

// 查询列表
{
  "action_type": "list",
  "list": { "page": 1, "page_size": 20, "sort": "desc" }
}

// 查询详情
{
  "action_type": "detail",
  "detail": { "footage_id": 123456 }
}

// 删除
{
  "action_type": "delete",
  "delete": { "footage_ids": [123456, 123457] }
}

// list
{
  "log_id": "20250622150000XXXX",
  "error_code": 0,
  "error_message": "success",
  "data": {
    "list": {
      "page": 1,
      "page_size": 20,
      "total": 35,
      "list": [
        {
          "footage_id": 123456,
          "footage_name": "角色A参考视频",
          "footage_url": "https://bucket.bj.bcebos.com/path/...?authorization=...",
          "footage_desc": "角色A 的参考素材",
          "create_time": "2025-06-22 15:00:00"
        }
      ]
    }
  }
}

// detail
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "detail": { "footage_id": 123456, "footage_name": "角色A参考视频",
    "footage_url": "https://...", "footage_desc": "...", "create_time": "2025-06-22 15:00:00" } } }

// create / update
{ "log_id": "...", "error_code": 0, "error_message": "success", "data": { "create": { "footage_id": 123456 } } }

// delete
{ "log_id": "...", "error_code": 0, "error_message": "success", "data": { "delete": { "footage_ids": [123456, 123457] } } }

// 1) single_video
{
  "action_type": "single_video",
  "single_video": {
    "series_id": 0,
    "series_name": "我的剧集",
    "series_desc": "可选",
    "video_url": "https://bucket.bj.bcebos.com/path/input.mp4",
    "video_name": "第一集",
    "video_desc": "第一集",
    "source_language": "zh",
    "target_language": "en",
    "analysis_config": { "face_type": "normal" },
    "subtitle_config": { "extractor_type": "advance", "subtitle_remove_mode": "none" },
    "translate_config": {},
    "dubbing_config": { "with_original_vocal": false }
  }
}

// 2) batch_video
{
  "action_type": "batch_video",
  "batch_video": {
    "series_name": "我的剧集",
    "source_language": "zh",
    "target_language": "en",
    "video_list": [
      { "video_url": "https://.../01.mp4", "video_name": "第一集" },
      { "video_url": "https://.../02.mp4", "video_name": "第二集" }
    ],
    "analysis_config": {},
    "subtitle_config": { "extractor_type": "advance", "subtitle_remove_mode": "none" },
    "translate_config": {},
    "dubbing_config": {}
  }
}

// 3) single_task
{
  "action_type": "single_task",
  "single_task": {
    "video_id": 100001,
    "task_type": "subtitle",
    "subtitle_config": { "extractor_type": "advance", "subtitle_remove_mode": "none" }
  }
}

// single_video
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "all_task": { "series_id": 1001, "video_id": 200001, "task_id": 3000001 } } }

// batch_video
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "batch_task": { "series_id": 1001, "list": [
      { "series_id": 1001, "video_id": 200001, "task_id": 3000001 },
      { "series_id": 1001, "video_id": 200002, "task_id": 3000002 }
  ] } } }

// single_task
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "single_task": { "video_id": 0, "task_id": 3000010 } } }

// 查询视频列表
{ "action_type": "list_video", "list_video": { "page": 1, "page_size": 20, "sort": "desc" } }

// 查询任务详情（按 video_id）
{ "action_type": "detail_task", "detail_task": { "video_id": 200001 } }

// 更新任务结果
{
  "action_type": "update_task_result",
  "update_task_result": {
    "task_id": 3000001,
    "task_status": "checked",
    "subtitle_result": {
      "subtitles": [
        { "id": "473e1c93-1b8c-46", "text": "直到死去5年的丈夫站在我面前的时候",
          "start_time": 0.3, "end_time": 2.7,
          "character_id": "1000261_ebee37e3-d9fe-48_cluster_0", "language": "zh" },
        { "id": "a238d69d-3d9e-43", "text": "我才知道原来他没死",
          "start_time": 2.9, "end_time": 4.83,
          "character_id": "1000261_ebee37e3-d9fe-48_cluster_0", "language": "zh" }
      ]
    }
  }
}

// 确认任务结果
{ "action_type": "confirm_task_result", "confirm_task_result": { "task_id": 3000001 } }

// 删除任务
{ "action_type": "delete_task", "delete_task": { "task_ids": [3000001, 3000002] } }

// 导出合成视频
{ "action_type": "export_task", "export_task": { "task_id": 3000100 } }

// list_video
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "list": { "page": 1, "page_size": 20, "total": 1, "list": [
    { "video_id": 200001, "name": "第一集",
      "video_url": "https://.../input.mp4?authorization=...",
      "source_language": "zh", "target_language": "en",
      "duration": 600, "description": "",
      "create_time": "2025-06-22 15:00", "update_time": "2025-06-22 15:05" }
  ] } } }

// detail_task
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "detail": {
    "video_id": 200001,
    "task_id": 0,
    "analysis": [
      { "video_id": 200001, "task_id": 3000001, "task_type": "analysis",
        "task_status": "done", "need_check": false,
        "created_time": "2025-06-22 15:00:00", "updated_time": "2025-06-22 15:03:00" }
    ],
    "subtitle": [], "translation": [], "dubbing": [], "compose": []
  } } }

// update_task_result / confirm_task_result
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "update_task_result": { "task_id": 3000001 } } }

// delete_task
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "delete": { "task_ids": [3000001, 3000002] } } }

// export_task
{ "log_id": "...", "error_code": 0, "error_message": "success",
  "data": { "export": { "task_id": 3000100, "final_url": "https://.../final.mp4?authorization=..." } } }

// 重试字幕任务，并调整提取器与擦除模式
{
  "task_id": 3000002,
  "subtitle_config": { "extractor_type": "advance", "subtitle_remove_mode": "advanced", "need_human_check": false }
}

// 重试失败的分析任务（failed 状态不计重试次数）
{
  "task_id": 3000001,
  "analysis_config": { "disable_face_detection": true, "need_human_check": false }
}

// 仅重试，不调整配置
{ "task_id": 3000003 }

{
  "log_id": "20250622150000XXXX",
  "error_code": 0,
  "error_message": "success",
  "data": {
    "task_id": 3000002
  }
}

"""通过 BOS 的 STS 临时凭证上传文件并返回文件链接"""

import os
import sys
import time
from io import BytesIO
from baidubce.bce_client_configuration import BceClientConfiguration
from baidubce.auth.bce_credentials import BceCredentials
from baidubce.services.bos.bos_client import BosClient

# ===== 配置区域：填入 /footage/upload 返回的 STS 临时凭证 =====
STS_ACCESS_KEY_ID = "<access_key_id>"
STS_SECRET_ACCESS_KEY = "<secret_access_key>"
STS_SESSION_TOKEN = "<session_token>"
BOS_HOST = "http://bj.bcebos.com"      # 对应响应中的 endpoint
BUCKET_NAME = "<bucket_name>"          # 对应响应中的 bucket_name


def upload_file(local_file_path, object_key=None):
    if not os.path.exists(local_file_path):
        print(f"文件不存在: {local_file_path}")
        return None

    if object_key is None:
        object_key = os.path.basename(local_file_path)

    bos_config = BceClientConfiguration(
        credentials=BceCredentials(STS_ACCESS_KEY_ID, STS_SECRET_ACCESS_KEY),
        endpoint=BOS_HOST,
        security_token=STS_SESSION_TOKEN
    )
    bos_client = BosClient(bos_config)

    # 大于 1GB 用分块上传，否则用简单上传
    file_size = os.path.getsize(local_file_path)
    max_simple_size = 1 * 1024 * 1024 * 1024  # 1GB

    if file_size >= max_simple_size:
        part_size = 100 * 1024 * 1024  # 每块 100MB
        total_parts = (file_size + part_size - 1) // part_size
        upload_id = bos_client.initiate_multipart_upload(BUCKET_NAME, object_key).upload_id

        part_list = []
        with open(local_file_path, 'rb') as f:
            part_number = 1
            while True:
                data = f.read(part_size)
                if not data:
                    break
                response = bos_client.upload_part(
                    BUCKET_NAME, object_key, upload_id, part_number, len(data), BytesIO(data))
                part_list.append({"partNumber": part_number, "eTag": response.metadata.etag})
                print(f"  [{part_number}/{total_parts}] 分块上传完成")
                part_number += 1

        bos_client.complete_multipart_upload(BUCKET_NAME, object_key, upload_id, part_list)
    else:
        bos_client.put_object_from_file(BUCKET_NAME, object_key, local_file_path)

    # 生成预签名 URL（1 小时有效）
    timestamp = int(time.time())
    signed_url = bos_client.generate_pre_signed_url(
        BUCKET_NAME, object_key, timestamp, expiration_in_seconds=3600).decode('utf-8')
    print(f"上传成功，签名链接（1 小时有效）：\n{signed_url}")
    return signed_url


if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("用法: python upload_by_bos_sts.py <文件路径> [对象路径]")
        sys.exit(1)