// Copyright © 2023 OpenIM open source community. All rights reserved.
//
// 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 admin

import (
	"context"

	chatdb "git.imall.cloud/openim/chat/pkg/common/db/table/chat"
	"git.imall.cloud/openim/chat/pkg/common/mctx"
	"git.imall.cloud/openim/chat/pkg/protocol/admin"
	"github.com/openimsdk/tools/errs"
)

// ==================== 定时任务管理相关 RPC ====================

// GetScheduledTasks 获取定时任务列表（管理员接口，可查看所有任务）
func (o *adminServer) GetScheduledTasks(ctx context.Context, req *admin.GetScheduledTasksReq) (*admin.GetScheduledTasksResp, error) {
	// 验证管理员权限
	_, err := mctx.CheckAdmin(ctx)
	if err != nil {
		return nil, err
	}

	// 获取所有定时任务列表
	total, tasks, err := o.ChatDatabase.GetAllScheduledTasks(ctx, req.Pagination)
	if err != nil {
		return nil, err
	}

	// 转换为响应格式
	taskInfos := make([]*admin.ScheduledTaskInfo, 0, len(tasks))
	for _, task := range tasks {
		taskInfos = append(taskInfos, convertScheduledTaskToAdminProto(task))
	}

	return &admin.GetScheduledTasksResp{
		Total: total,
		Tasks: taskInfos,
	}, nil
}

// DeleteScheduledTask 删除定时任务（管理员接口，可删除任何任务）
func (o *adminServer) DeleteScheduledTask(ctx context.Context, req *admin.DeleteScheduledTaskReq) (*admin.DeleteScheduledTaskResp, error) {
	// 验证管理员权限
	_, err := mctx.CheckAdmin(ctx)
	if err != nil {
		return nil, err
	}

	// 验证请求参数
	if len(req.TaskIDs) == 0 {
		return nil, errs.ErrArgs.WrapMsg("taskIDs is required")
	}

	// 删除任务
	if err := o.ChatDatabase.DeleteScheduledTask(ctx, req.TaskIDs); err != nil {
		return nil, err
	}

	return &admin.DeleteScheduledTaskResp{}, nil
}

// convertScheduledTaskToAdminProto 将数据库模型转换为 admin protobuf 消息
func convertScheduledTaskToAdminProto(task *chatdb.ScheduledTask) *admin.ScheduledTaskInfo {
	messages := make([]*admin.ScheduledTaskMessage, 0, len(task.Messages))
	for _, msg := range task.Messages {
		messages = append(messages, &admin.ScheduledTaskMessage{
			Type:      msg.Type,
			Content:   msg.Content,
			Thumbnail: msg.Thumbnail,
			Duration:  msg.Duration,
			FileSize:  msg.FileSize,
			Width:     msg.Width,
			Height:    msg.Height,
		})
	}

	return &admin.ScheduledTaskInfo{
		Id:             task.ID,
		UserID:         task.UserID,
		Name:           task.Name,
		CronExpression: task.CronExpression,
		Messages:       messages,
		RecvIDs:        task.RecvIDs,
		GroupIDs:       task.GroupIDs,
		Status:         task.Status,
		CreateTime:     task.CreateTime.UnixMilli(),
		UpdateTime:     task.UpdateTime.UnixMilli(),
	}
}