86 lines
2.6 KiB
Go
86 lines
2.6 KiB
Go
// Copyright 2026 Alibaba Group
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
package helpers
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// runDocMediaUpload 上传绑定到文档 nodeId 的可复用媒体资源，但不插入正文块。
|
|
// 白板 Vector/SVG 使用返回的 resourceId 与 resourceUrl 引用同一文档下的资源。
|
|
func runDocMediaUpload(cmd *cobra.Command, _ []string) error {
|
|
nodeID, err := mustFlagOrFallback(cmd, "node", "url", "id", "node-id", "doc-id", "file-id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
filePath := mustGetFlag(cmd, "file")
|
|
if filePath == "" {
|
|
return fmt.Errorf("flag --file is required")
|
|
}
|
|
fileInfo, err := os.Stat(filePath)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot read file %s: %w", filePath, err)
|
|
}
|
|
if fileInfo.IsDir() {
|
|
return fmt.Errorf("%s is a directory, not a file", filePath)
|
|
}
|
|
|
|
fileName, _ := cmd.Flags().GetString("name")
|
|
if fileName == "" {
|
|
fileName = filepath.Base(filePath)
|
|
} else if filepath.Ext(fileName) == "" {
|
|
fileName += filepath.Ext(filePath)
|
|
}
|
|
mimeType, _ := cmd.Flags().GetString("mime-type")
|
|
if mimeType == "" {
|
|
mimeType = inferMimeType(fileName)
|
|
}
|
|
|
|
if deps.Caller.DryRun() {
|
|
return callMCPToolOnServer("doc", "get_doc_attachment_upload_info", map[string]any{
|
|
"nodeId": nodeID,
|
|
"fileName": fileName,
|
|
"fileSize": float64(fileInfo.Size()),
|
|
"mimeType": mimeType,
|
|
})
|
|
}
|
|
|
|
// 用户确认由 DeclareLeafMetadata(user_required) 的 ConfirmSafety 门控接管：
|
|
// 推迟到首次 deps.Caller.CallTool（下方 get_doc_attachment_upload_info），
|
|
// 避免与门控双读 stdin。--yes / --dry-run 经 confirmationBypass 跳过。
|
|
text, err := callMCPToolReturnTextOnServer(cmd.Context(), "doc", "get_doc_attachment_upload_info", map[string]any{
|
|
"nodeId": nodeID,
|
|
"fileName": fileName,
|
|
"fileSize": float64(fileInfo.Size()),
|
|
"mimeType": mimeType,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
uploadURL, resourceID, resourceURL, err := parseAttachmentUploadInfo(text)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if resourceURL == "" {
|
|
return fmt.Errorf("incomplete attachment upload info: missing resourceUrl")
|
|
}
|
|
if err := httpPutFile(cmd.Context(), uploadURL, map[string]string{"Content-Type": mimeType}, filePath, fileInfo.Size()); err != nil {
|
|
message := strings.ReplaceAll(err.Error(), uploadURL, "<redacted upload URL>")
|
|
return fmt.Errorf("document media upload failed: %s", message)
|
|
}
|
|
return deps.Out.PrintJSON(map[string]any{
|
|
"nodeId": nodeID,
|
|
"resourceId": resourceID,
|
|
"resourceUrl": resourceURL,
|
|
"fileName": fileName,
|
|
"mimeType": mimeType,
|
|
"size": fileInfo.Size(),
|
|
})
|
|
}
|