// 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 (
	"time"

	"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"
)

// ThisMonthAttendance: show MY punch-in (打卡流水) records for THIS MONTH in one
// step. It is the month-scoped sibling of +my-attendance (which covers today).
//
// Steps:
//  1. resolve the current logged-in user's userId via the contact server's
//     zero-arg get_current_user_profile (no --name needed — it's always "me");
//  2. compute this month's window [1st 00:00, next-month 1st 00:00) in the local
//     timezone and format both bounds as "yyyy-MM-dd HH:mm:ss" (the format the
//     query_check_record helper feeds the tool);
//  3. call query_check_record on the attendance-wukong server with the exact
//     nested QueryCheckRecordRequest shape used by
//     `dws attendance check record`, then print via rt.Output.
//
// The month window spans at most 31 days, within query_check_record's span cap.
// Read-only; it never modifies any attendance data.
//
//	dws attendance +this-month
var ThisMonthAttendance = shortcut.Shortcut{
	Service:     "attendance",
	Command:     "+this-month",
	Product:     "attendance",
	Description: "查我本月的考勤打卡记录（打卡流水，自动解析当前用户）",
	Intent: "当你想快速看自己本月的打卡流水（几点上下班打卡、打卡地址/定位方式）、又不想先查自己的 userId " +
		"再手动填写本月的起止时间时使用；内部先取当前登录用户的 userId，再按本地时区算出本月 1 号 00:00 到下月 1 号 00:00 的时间窗，" +
		"最后查询你本月的打卡流水记录。只读操作，不会修改任何考勤数据；本月若还没有任何打卡则返回空结果。",
	Risk: shortcut.RiskRead,
	Safety: contract.SafetySpec{
		Effect: "read", Risk: "low",
		Confirmation: "not_required", Idempotency: "idempotent",
	},
	Contract: corecmd.ContractDecl{
		Identity: contract.ToolIdentitySpec{
			ProductID:      "attendance",
			Name:           "shortcut_this_month",
			CanonicalPath:  "attendance.shortcut_this_month",
			CLIPath:        "attendance +this-month",
			PrimaryCLIPath: "attendance +this-month",
		},
		Description: "查我本月的考勤打卡记录（打卡流水，自动解析当前用户）",
		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: "查我本月的考勤打卡记录（打卡流水，自动解析当前用户）",
			UseWhen:      []string{"当你想快速看自己本月的打卡流水（几点上下班打卡、打卡地址/定位方式）、又不想先查自己的 userId 再手动填写本月的起止时间时使用；内部先取当前登录用户的 userId，再按本地时区算出本月 1 号 00:00 到下月 1 号 00:00 的时间窗，最后查询你本月的打卡流水记录。只读操作，不会修改任何考勤数据；本月若还没有任何打卡则返回空结果。"},
			AvoidWhen:    []string{"需要该 Shortcut 未公开的底层参数、原始响应或不同执行语义时，改用对应原子命令"},
			Examples:     []string{"dws attendance +this-month"},
		},
	},
	Flags: []shortcut.Flag{},
	Tips: []string{
		`dws attendance +this-month`,
	},
	Execute: func(rt *shortcut.RuntimeContext) error {
		// Step 1 — resolve the current user's own userId (zero-arg "我的" API).
		profile, err := rt.CallMCPData("contact", "get_current_user_profile", nil)
		if err != nil {
			return err
		}
		userID := myAttendanceCurrentUserID(profile)
		if userID == "" {
			return apperrors.NewValidation(
				"没能解析出当前登录用户的 userId，无法查询你的打卡记录；请确认已登录后重试。")
		}

		// Step 2 — this month's window [1st 00:00, next-month 1st 00:00) in local
		// time, formatted as the tool expects (yyyy-MM-dd HH:mm:ss).
		now := time.Now()
		start := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
		end := start.AddDate(0, 1, 0)
		const layout = "2006-01-02 15:04:05"

		// Step 3 — query my punch-in records for this month. The nested
		// QueryCheckRecordRequest shape + field names mirror the
		// `dws attendance check record` helper exactly.
		data, err := rt.CallMCPData("attendance-wukong", "query_check_record", map[string]any{
			"QueryCheckRecordRequest": map[string]any{
				"userIds":       []string{userID},
				"checkDateFrom": start.Format(layout),
				"checkDateTo":   end.Format(layout),
			},
		})
		if err != nil {
			return err
		}
		return rt.Output(data)
	},
}

func init() {
	shortcut.Register(ThisMonthAttendance)
}