215 lines
9.0 KiB
Go
215 lines
9.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 smart
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/corecmd"
|
||
|
||
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/corecmd/contract"
|
||
apperrors "github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/errors"
|
||
"github.com/DingTalk-Real-AI/dingtalk-workspace-cli/internal/shortcut"
|
||
)
|
||
|
||
// Book: create a calendar event AND (optionally) invite participants BY NAME,
|
||
// in one command, with automatic rollback if inviting fails.
|
||
//
|
||
// Steps: create the event (summary + start/end) → if --with is given, resolve
|
||
// each name to a unique userId and add them all as participants; if that add
|
||
// fails, delete the just-created event so we never leave a half-built event
|
||
// behind. Replaces `calendar event create` (copy eventId) →
|
||
// `contact +search-user` (copy each userId) → `calendar attendee add`.
|
||
//
|
||
// dws calendar +book --title "Q1 复盘会" \
|
||
// --start "2026-03-10T14:00:00+08:00" --end "2026-03-10T15:00:00+08:00" \
|
||
// --with 张三,李四
|
||
var Book = shortcut.Shortcut{
|
||
Service: "calendar",
|
||
Command: "+book",
|
||
Product: "calendar",
|
||
Description: "创建日程，并可按姓名邀请参会人（自动解析 userId，失败自动回滚删除日程）",
|
||
Intent: "当你想快速排一个会/日程、并顺手把几位同事按姓名拉进来时使用；" +
|
||
"内部先建日程拿到 eventId，再把每个姓名解析成唯一 userId 批量加为参会人。" +
|
||
"如果加参会人失败，会自动删除刚建好的日程回滚，避免留下一个没人的空日程。" +
|
||
"会真实创建日程并发出参会邀请。",
|
||
Risk: shortcut.RiskWrite,
|
||
Safety: contract.SafetySpec{
|
||
Effect: "write", Risk: "medium",
|
||
Confirmation: "user_required", Idempotency: "unknown",
|
||
},
|
||
Contract: corecmd.ContractDecl{
|
||
Identity: contract.ToolIdentitySpec{
|
||
ProductID: "calendar",
|
||
Name: "shortcut_book",
|
||
CanonicalPath: "calendar.shortcut_book",
|
||
CLIPath: "calendar +book",
|
||
PrimaryCLIPath: "calendar +book",
|
||
},
|
||
Description: "创建日程，并可按姓名邀请参会人（自动解析 userId，失败自动回滚删除日程）",
|
||
Interface: &contract.InterfaceSpec{
|
||
Mode: "composite",
|
||
Availability: "available",
|
||
Reason: "Reviewed built-in shortcut adapter: the executable CLI owns validation, optional multi-step orchestration, output projection, and confirmation; the complete command contract is not represented by one pinned MCP interface_ref.",
|
||
},
|
||
Selection: contract.SelectionSpec{
|
||
AgentSummary: "创建日程，并可按姓名邀请参会人（自动解析 userId，失败自动回滚删除日程）",
|
||
UseWhen: []string{"当你想快速排一个会/日程、并顺手把几位同事按姓名拉进来时使用；内部先建日程拿到 eventId，再把每个姓名解析成唯一 userId 批量加为参会人。如果加参会人失败，会自动删除刚建好的日程回滚，避免留下一个没人的空日程。会真实创建日程并发出参会邀请。"},
|
||
AvoidWhen: []string{"需要该 Shortcut 未公开的底层参数、原始响应或不同执行语义时，改用对应原子命令"},
|
||
Examples: []string{
|
||
"dws calendar +book --title \"周会\" --start \"2026-03-10T14:00:00+08:00\" --end \"2026-03-10T15:00:00+08:00\"",
|
||
"dws calendar +book --title \"Q1 复盘会\" --start \"2026-03-10T14:00:00+08:00\" --end \"2026-03-10T15:00:00+08:00\" --with 张三,李四",
|
||
},
|
||
},
|
||
},
|
||
Flags: []shortcut.Flag{
|
||
{Name: "title", Type: shortcut.FlagString, Desc: "日程标题", Required: true},
|
||
{Name: "start", Type: shortcut.FlagString, Desc: "开始时间（ISO8601，如 2026-03-10T14:00:00+08:00）", Required: true},
|
||
{Name: "end", Type: shortcut.FlagString, Desc: "结束时间（ISO8601，如 2026-03-10T15:00:00+08:00）", Required: true},
|
||
{Name: "with", Type: shortcut.FlagString, Desc: "参会人姓名，逗号分隔（可选）"},
|
||
},
|
||
Tips: []string{
|
||
`dws calendar +book --title "周会" --start "2026-03-10T14:00:00+08:00" --end "2026-03-10T15:00:00+08:00"`,
|
||
`dws calendar +book --title "Q1 复盘会" --start "2026-03-10T14:00:00+08:00" --end "2026-03-10T15:00:00+08:00" --with 张三,李四`,
|
||
},
|
||
Execute: func(rt *shortcut.RuntimeContext) error {
|
||
start := rt.Str("start")
|
||
end := rt.Str("end")
|
||
if err := calendarSmartValidateRange(start, end); err != nil {
|
||
return err
|
||
}
|
||
// create_calendar_event params copied verbatim from the helper's create
|
||
// call site: summary + startDateTime/endDateTime (ISO strings, not millis).
|
||
createArgs := map[string]any{
|
||
"summary": rt.Str("title"),
|
||
"startDateTime": start,
|
||
"endDateTime": end,
|
||
}
|
||
|
||
// Step 1 — resolve every participant name to a unique userId BEFORE
|
||
// creating the event, so an unknown/ambiguous name fails cheaply without
|
||
// leaving a dangling event behind.
|
||
var userIDs []string
|
||
var userNames []string
|
||
withPeople := rt.Changed("with") && strings.TrimSpace(rt.Str("with")) != ""
|
||
if withPeople {
|
||
for _, name := range strings.Split(rt.Str("with"), ",") {
|
||
name = strings.TrimSpace(name)
|
||
if name == "" {
|
||
continue
|
||
}
|
||
user, err := resolveUser(rt, name)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
userIDs = append(userIDs, user.userID)
|
||
userNames = append(userNames, user.name)
|
||
}
|
||
if len(userIDs) == 0 {
|
||
return apperrors.NewValidation("--with 需要至少一个有效的参会人姓名")
|
||
}
|
||
}
|
||
|
||
// Under --dry-run we resolved names (reads) to validate them, but must not
|
||
// create the event or send invites. Preview what would happen and return.
|
||
if rt.DryRun() {
|
||
return rt.Output(map[string]any{
|
||
"success": true,
|
||
"dryRun": true,
|
||
"executed": false,
|
||
"wouldCreate": createArgs,
|
||
"inviteeCount": len(userIDs),
|
||
})
|
||
}
|
||
|
||
// Step 2 — create the event and require an explicit terminal receipt plus
|
||
// a stable id. Empty/unknown write acknowledgements are never success.
|
||
created, err := rt.CallMCPWriteDataStrict("calendar", "create_calendar_event", createArgs)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err := calendarSmartWriteReceipt(created, "calendar/create_calendar_event"); err != nil {
|
||
return err
|
||
}
|
||
eventID := calendarSmartEventID(created)
|
||
if eventID == "" {
|
||
return calendarSmartError("calendar/create_calendar_event", "missing_event_id", "创建回执缺少稳定日程 id，远端效果未知")
|
||
}
|
||
|
||
// Step 3 — add all participants in one batch. attendeesToAdd copied
|
||
// verbatim from the helper's `attendee add` call site.
|
||
if len(userIDs) > 0 {
|
||
added, addErr := rt.CallMCPWriteDataStrict("calendar", "add_calendar_participant", map[string]any{
|
||
"eventId": eventID,
|
||
"attendeesToAdd": userIDs,
|
||
})
|
||
if addErr == nil {
|
||
addErr = calendarSmartWriteReceipt(added, "calendar/add_calendar_participant")
|
||
}
|
||
if addErr != nil {
|
||
// Rollback itself is not enough: verify the newly-created event is
|
||
// absent before saying rollback succeeded.
|
||
if rollbackErr := calendarSmartDeleteAndVerify(rt, eventID); rollbackErr != nil {
|
||
return apperrors.NewValidation(fmt.Sprintf(
|
||
"添加参会人失败：%v；回滚删除或删除后验证失败：%v，请人工核查日程状态",
|
||
addErr, rollbackErr))
|
||
}
|
||
return apperrors.NewValidation(fmt.Sprintf("添加参会人失败：%v；新建日程已回滚并验证不存在", addErr))
|
||
}
|
||
}
|
||
|
||
// Step 4 — prove the final state before returning one composed result.
|
||
readback, err := rt.CallMCPData("calendar", "get_calendar_detail", map[string]any{"eventId": eventID})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
event, err := calendarSmartRequireEvent(readback, "calendar/get_calendar_detail", eventID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err := calendarSmartVerifyCreatedEvent(event, eventID, rt.Str("title"), start, end); err != nil {
|
||
return err
|
||
}
|
||
if len(userIDs) > 0 {
|
||
participants, err := rt.CallMCPData("calendar", "get_calendar_participants", map[string]any{"eventId": eventID})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
present, err := calendarSmartAttendees(participants)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
currentUserID, err := calendarSmartCurrentUserID(rt, present)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err := calendarSmartVerifyAttendees(present, userIDs, userNames, currentUserID); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return rt.Output(map[string]any{
|
||
"success": true,
|
||
"eventId": eventID,
|
||
"verified": true,
|
||
"event": event,
|
||
})
|
||
},
|
||
}
|
||
|
||
func init() {
|
||
finalizeCalendarSmart(&Book, "已创建并通过精确读回验证的日程")
|
||
shortcut.Register(Book)
|
||
}
|