103 lines
4.0 KiB
Go
103 lines
4.0 KiB
Go
// Copyright 2026 Alibaba Group
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package helpers
|
|
|
|
import "github.com/spf13/cobra"
|
|
|
|
const sheetImportLong = `将本地表格文件导入为一个新的钉钉在线电子表格，与 dws sheet export（导出）对称。
|
|
|
|
支持的文件格式 (按扩展名):
|
|
xlsx, xls Microsoft Excel 表格
|
|
|
|
文件大小限制: 20MB
|
|
固定导入为电子表格类型；只新建文档，不操作已有表格。
|
|
|
|
CLI 内部自动完成全部流程:
|
|
1. 创建导入会话（获取 OSS 上传凭证）
|
|
2. 上传文件到 OSS
|
|
3. 确认导入（触发格式转换）
|
|
4. 渐进式退避轮询等待完成（最多约 5 分钟）
|
|
|
|
如果轮询超时仍未完成，会输出 taskId 供后续手动查询:
|
|
dws sheet import get --task-id <taskId>`
|
|
|
|
func newSheetImportCmd() *cobra.Command {
|
|
return newSheetImportCmdWithConfig(sheetImportFlowConfig())
|
|
}
|
|
|
|
func newSheetImportCmdWithConfig(cfg importFlowConfig) *cobra.Command {
|
|
importCmd := &cobra.Command{
|
|
Use: "import",
|
|
Short: "导入本地表格文件为在线电子表格 (xlsx / xls)",
|
|
Long: sheetImportLong,
|
|
Example: ` # 导入 xlsx 为在线电子表格（默认表格名取文件名）
|
|
dws sheet import --file ./quote.xlsx --folder-token <FOLDER_TOKEN>
|
|
|
|
# 指定目标文件夹与导入后表格名称
|
|
dws sheet import --file ./report.xls --folder-token <FOLDER_TOKEN> --name "月度报表"
|
|
|
|
# 导入到指定知识库
|
|
dws sheet import --file ./data.xls --workspace <WORKSPACE_ID>`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return runImportCommand(cmd, args, cfg)
|
|
},
|
|
}
|
|
addSheetImportFlags(importCmd)
|
|
|
|
// Schema deliberately binds only runnable leaves. Keep the historical
|
|
// runnable parent for CLI compatibility, and expose the same action through
|
|
// a leaf so agents can discover and invoke the import operation.
|
|
importCreateCmd := &cobra.Command{
|
|
Use: "create",
|
|
Short: "导入本地表格文件为在线电子表格 (xlsx / xls)",
|
|
Long: sheetImportLong,
|
|
Example: ` dws sheet import create --file ./quote.xlsx --folder-token <FOLDER_TOKEN>
|
|
dws sheet import create --file ./data.xls --workspace <WORKSPACE_ID> --name "月度报表"`,
|
|
Args: cobra.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return runImportCommand(cmd, args, cfg)
|
|
},
|
|
}
|
|
addSheetImportFlags(importCreateCmd)
|
|
|
|
importGetCmd := &cobra.Command{
|
|
Use: "get",
|
|
Short: "查询表格导入任务结果（手动兜底）",
|
|
Long: `根据 taskId 查询表格导入任务的执行结果。
|
|
通常不需要手动调用，dws sheet import 会自动完成轮询。
|
|
仅在导入命令超时或中断后，用于手动查询任务状态。
|
|
|
|
任务状态:
|
|
processing 转换中
|
|
completed 导入成功，返回 documentUrl
|
|
failed 导入失败`,
|
|
Example: ` dws sheet import get --task-id <TASK_ID>`,
|
|
RunE: func(cmd *cobra.Command, _ []string) error {
|
|
return runImportGetCommand(cmd, cfg)
|
|
},
|
|
}
|
|
importGetCmd.Flags().String("task-id", "", "导入任务 ID (必填)")
|
|
importCmd.AddCommand(importCreateCmd, importGetCmd)
|
|
return importCmd
|
|
}
|
|
|
|
func addSheetImportFlags(cmd *cobra.Command) {
|
|
cmd.Flags().String("file", "", "本地表格文件路径 (必填，支持 xlsx/xls)")
|
|
cmd.Flags().String("folder-token", "", "目标文件夹 ID 或 URL (与 --workspace 至少传一个)")
|
|
cmd.Flags().String("workspace", "", "目标知识库 ID 或 URL (与 --folder-token 至少传一个)")
|
|
cmd.Flags().StringP("name", "n", "", "导入后表格名称 (可选，默认取文件名)")
|
|
cmd.Flags().String("folder", "", "")
|
|
_ = cmd.Flags().MarkHidden("folder")
|
|
}
|