84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
package feishu_util
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
larkcore "github.com/larksuite/oapi-sdk-go/v3/core"
|
||
larkwiki "github.com/larksuite/oapi-sdk-go/v3/service/wiki/v2"
|
||
)
|
||
|
||
// WikiNode 知识库节点解析结果。
|
||
type WikiNode struct {
|
||
SpaceID string
|
||
NodeToken string
|
||
ObjToken string
|
||
ObjType string
|
||
Title string
|
||
}
|
||
|
||
// ResolveWikiNode 通过知识库 node_token 获取实际云文档 obj_token。
|
||
// 知识库 URL 中的 token 不是 spreadsheet_token，读写表格内容前必须先转换。
|
||
func (c *Client) ResolveWikiNode(ctx context.Context, wikiToken string) (*WikiNode, error) {
|
||
if wikiToken == "" {
|
||
return nil, fmt.Errorf("wikiToken 为空")
|
||
}
|
||
req := larkwiki.NewGetNodeSpaceReqBuilder().
|
||
Token(wikiToken).
|
||
Build()
|
||
resp, err := c.Lark.Wiki.V2.Space.GetNode(ctx, req)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("获取知识库节点失败: %w", err)
|
||
}
|
||
if !resp.Success() {
|
||
return nil, fmt.Errorf("获取知识库节点失败: code=%d msg=%s log_id=%s detail=%s",
|
||
resp.Code, resp.Msg, resp.RequestId(), larkcore.Prettify(resp.CodeError))
|
||
}
|
||
if resp.Data == nil || resp.Data.Node == nil {
|
||
return nil, fmt.Errorf("获取知识库节点返回空数据")
|
||
}
|
||
n := resp.Data.Node
|
||
node := &WikiNode{}
|
||
if n.SpaceId != nil {
|
||
node.SpaceID = *n.SpaceId
|
||
}
|
||
if n.NodeToken != nil {
|
||
node.NodeToken = *n.NodeToken
|
||
}
|
||
if n.ObjToken != nil {
|
||
node.ObjToken = *n.ObjToken
|
||
}
|
||
if n.ObjType != nil {
|
||
node.ObjType = *n.ObjType
|
||
}
|
||
if n.Title != nil {
|
||
node.Title = *n.Title
|
||
}
|
||
if node.ObjToken == "" {
|
||
return nil, fmt.Errorf("知识库节点未返回 obj_token")
|
||
}
|
||
return node, nil
|
||
}
|
||
|
||
// ResolveSpreadsheetToken 将 SpreadsheetRef 解析为可调用 Sheets API 的 spreadsheet_token。
|
||
func (c *Client) ResolveSpreadsheetToken(ctx context.Context, ref *SpreadsheetRef) (string, error) {
|
||
if ref == nil {
|
||
return "", fmt.Errorf("SpreadsheetRef 为空")
|
||
}
|
||
if ref.SpreadsheetToken != "" {
|
||
return ref.SpreadsheetToken, nil
|
||
}
|
||
if ref.WikiToken == "" {
|
||
return "", fmt.Errorf("既无 SpreadsheetToken 也无 WikiToken")
|
||
}
|
||
node, err := c.ResolveWikiNode(ctx, ref.WikiToken)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
if node.ObjType != "" && node.ObjType != "sheet" {
|
||
return "", fmt.Errorf("知识库节点类型为 %s，不是电子表格(sheet)", node.ObjType)
|
||
}
|
||
ref.SpreadsheetToken = node.ObjToken
|
||
return node.ObjToken, nil
|
||
}
|