137 lines
4.7 KiB
Bash
137 lines
4.7 KiB
Bash
#!/bin/bash
|
||
#
|
||
# 腾讯文档 MCP 文件导入辅助脚本
|
||
#
|
||
# 功能：
|
||
# 完成文件导入的前两步操作：
|
||
# 1. 计算文件的 MD5 和大小
|
||
# 2. 调用 manage.pre_import 获取 COS 上传链接和 file_key
|
||
# 3. 使用 curl 将文件 PUT 上传到 COS
|
||
# 4. 输出 file_key、file_name、file_md5、file_size、task_id 供后续调用 manage.async_import
|
||
#
|
||
# 用法：
|
||
# bash import_file.sh <file_path>
|
||
#
|
||
# 依赖：
|
||
# - mcporter（已配置 tencent-docs 服务）
|
||
# - curl
|
||
# - md5sum 或 md5（macOS）
|
||
#
|
||
# 输出（成功时）：
|
||
# IMPORT_READY
|
||
# FILE_KEY:<file_key>
|
||
# FILE_NAME:<file_name>
|
||
# FILE_MD5:<file_md5>
|
||
#
|
||
# 输出（失败时）：
|
||
# ERROR:<error_message>
|
||
#
|
||
|
||
set -euo pipefail
|
||
|
||
# ── 参数校验 ──────────────────────────────────────────────────────────────────
|
||
if [[ $# -lt 1 ]]; then
|
||
echo "ERROR:missing_argument - 用法: bash import_file.sh <file_path>"
|
||
exit 1
|
||
fi
|
||
|
||
FILE_PATH="$1"
|
||
|
||
if [[ ! -f "$FILE_PATH" ]]; then
|
||
echo "ERROR:file_not_found - 文件不存在: $FILE_PATH"
|
||
exit 1
|
||
fi
|
||
|
||
# ── 提取文件名（格式支持性由后端 manage.pre_import 判定）────────────────────
|
||
FILE_NAME=$(basename "$FILE_PATH")
|
||
|
||
# ── 计算文件大小 ──────────────────────────────────────────────────────────────
|
||
if [[ "$(uname)" == "Darwin" ]]; then
|
||
FILE_SIZE=$(stat -f%z "$FILE_PATH")
|
||
else
|
||
FILE_SIZE=$(stat -c%s "$FILE_PATH")
|
||
fi
|
||
|
||
if [[ "$FILE_SIZE" -le 0 ]]; then
|
||
echo "ERROR:empty_file - 文件为空: $FILE_PATH"
|
||
exit 1
|
||
fi
|
||
|
||
# ── 计算文件 MD5 ─────────────────────────────────────────────────────────────
|
||
if command -v md5sum &>/dev/null; then
|
||
FILE_MD5=$(md5sum "$FILE_PATH" | awk '{print $1}')
|
||
elif command -v md5 &>/dev/null; then
|
||
FILE_MD5=$(md5 -q "$FILE_PATH")
|
||
else
|
||
echo "ERROR:no_md5_tool - 未找到 md5sum 或 md5 命令"
|
||
exit 1
|
||
fi
|
||
|
||
echo "📄 文件: $FILE_NAME"
|
||
echo "📏 大小: $FILE_SIZE bytes"
|
||
echo "🔑 MD5: $FILE_MD5"
|
||
echo ""
|
||
|
||
# ── Step 1: 调用 manage.pre_import 获取 COS 上传链接 ─────────────────────────
|
||
echo "⏳ 正在获取上传链接..."
|
||
|
||
PRE_IMPORT_ARGS=$(cat <<EOF
|
||
{"file_name": "$FILE_NAME", "file_size": $FILE_SIZE, "file_md5": "$FILE_MD5"}
|
||
EOF
|
||
)
|
||
|
||
PRE_IMPORT_RESULT=$(mcporter call "tencent-docs" "manage.pre_import" --args "$PRE_IMPORT_ARGS" 2>&1) || {
|
||
echo "ERROR:pre_import_failed - manage.pre_import 调用失败: $PRE_IMPORT_RESULT"
|
||
exit 1
|
||
}
|
||
|
||
# 解析返回的 upload_url 和 file_key
|
||
UPLOAD_URL=$(echo "$PRE_IMPORT_RESULT" | jq -r '.upload_url // empty' 2>/dev/null || echo "")
|
||
FILE_KEY=$(echo "$PRE_IMPORT_RESULT" | jq -r '.file_key // empty' 2>/dev/null || echo "")
|
||
TASK_ID=$(echo "$PRE_IMPORT_RESULT" | jq -r '.task_id // empty' 2>/dev/null || echo "")
|
||
|
||
if [[ -z "$UPLOAD_URL" ]]; then
|
||
echo "ERROR:no_upload_url - 未获取到上传链接，pre_import 返回: $PRE_IMPORT_RESULT"
|
||
exit 1
|
||
fi
|
||
|
||
if [[ -z "$FILE_KEY" ]]; then
|
||
echo "ERROR:no_file_key - 未获取到 file_key，pre_import 返回: $PRE_IMPORT_RESULT"
|
||
exit 1
|
||
fi
|
||
|
||
echo "✅ 获取上传链接成功"
|
||
echo ""
|
||
|
||
# ── Step 2: 使用 curl PUT 上传文件到 COS ─────────────────────────────────────
|
||
echo "⏳ 正在上传文件到 COS..."
|
||
|
||
HTTP_STATUS=$(curl -s -o /dev/null -w "%{http_code}" \
|
||
-X PUT \
|
||
-H "Content-Type: application/octet-stream" \
|
||
--data-binary "@$FILE_PATH" \
|
||
"$UPLOAD_URL") || {
|
||
echo "ERROR:upload_failed - curl 上传文件失败"
|
||
exit 1
|
||
}
|
||
|
||
if [[ "$HTTP_STATUS" -ge 200 && "$HTTP_STATUS" -lt 300 ]]; then
|
||
echo "✅ 文件上传成功 (HTTP $HTTP_STATUS)"
|
||
else
|
||
echo "ERROR:upload_http_error - COS 上传返回 HTTP $HTTP_STATUS"
|
||
exit 1
|
||
fi
|
||
|
||
echo ""
|
||
|
||
# ── 输出结果 ──────────────────────────────────────────────────────────────────
|
||
echo "IMPORT_READY"
|
||
echo "FILE_KEY:$FILE_KEY"
|
||
echo "FILE_NAME:$FILE_NAME"
|
||
echo "FILE_MD5:$FILE_MD5"
|
||
echo "TASK_ID:$TASK_ID"
|
||
echo "FILE_SIZE:$FILE_SIZE"
|
||
echo ""
|
||
echo "📋 下一步：调用 manage.async_import 触发导入"
|
||
echo " mcporter call \"tencent-docs\" \"manage.async_import\" --args '{\"task_id\": \"$TASK_ID\", \"file_size\": \"$FILE_SIZE\", \"file_key\": \"$FILE_KEY\", \"file_name\": \"$FILE_NAME\", \"file_md5\": \"$FILE_MD5\"}'"
|