package tasklog

// 任务日志

import (
	"fmt"
	"html/template"

	"github.com/Unknwon/paginater"
	"github.com/ouqiang/gocron/models"
	"github.com/ouqiang/gocron/modules/logger"
	"github.com/ouqiang/gocron/modules/utils"
	"github.com/ouqiang/gocron/routers/base"
	"github.com/ouqiang/gocron/service"
	"gopkg.in/macaron.v1"
)

func Index(ctx *macaron.Context) {
	logModel := new(models.TaskLog)
	queryParams := parseQueryParams(ctx)
	total, err := logModel.Total(queryParams)
	if err != nil {
		logger.Error(err)
	}
	logs, err := logModel.List(queryParams)
	if err != nil {
		logger.Error(err)
	}
	PageParams := fmt.Sprintf("task_id=%d&protocol=%d&status=%d&page_size=%d",
		queryParams["TaskId"], queryParams["Protocol"], queryParams["Status"],
		queryParams["PageSize"])
	queryParams["PageParams"] = template.URL(PageParams)
	p := paginater.New(int(total), queryParams["PageSize"].(int), queryParams["Page"].(int), 5)
	ctx.Data["Pagination"] = p
	ctx.Data["Title"] = "任务日志"
	ctx.Data["Logs"] = logs
	ctx.Data["Params"] = queryParams
	ctx.HTML(200, "task/log")
}

// 清空日志
func Clear(ctx *macaron.Context) string {
	taskLogModel := new(models.TaskLog)
	_, err := taskLogModel.Clear()
	json := utils.JsonResponse{}
	if err != nil {
		return json.CommonFailure(utils.FailureContent)
	}

	return json.Success(utils.SuccessContent, nil)
}

// 停止运行中的任务
func Stop(ctx *macaron.Context) string {
	id := ctx.QueryInt64("id")
	taskId := ctx.QueryInt("task_id")
	taskModel := new(models.Task)
	task, err := taskModel.Detail(taskId)
	json := utils.JsonResponse{}
	if err != nil {
		return json.CommonFailure("获取任务信息失败#"+err.Error(), err)
	}
	if task.Protocol != models.TaskRPC {
		return json.CommonFailure("仅支持SHELL任务手动停止")
	}
	if len(task.Hosts) == 0 {
		return json.CommonFailure("任务节点列表为空")
	}
	for _, host := range task.Hosts {
		service.ServiceTask.Stop(host.Name, host.Port, id)

	}

	return json.Success("已执行停止操作, 请等待任务退出", nil)
}

// 删除N个月前的日志
func Remove(ctx *macaron.Context) string {
	month := ctx.ParamsInt(":id")
	json := utils.JsonResponse{}
	if month < 1 || month > 12 {
		return json.CommonFailure("参数取值范围1-12")
	}
	taskLogModel := new(models.TaskLog)
	_, err := taskLogModel.Remove(month)
	if err != nil {
		return json.CommonFailure("删除失败", err)
	}

	return json.Success("删除成功", nil)
}

// 解析查询参数
func parseQueryParams(ctx *macaron.Context) models.CommonMap {
	var params models.CommonMap = models.CommonMap{}
	params["TaskId"] = ctx.QueryInt("task_id")
	params["Protocol"] = ctx.QueryInt("protocol")
	status := ctx.QueryInt("status")
	if status >= 0 {
		status -= 1
	}
	params["Status"] = status
	base.ParsePageAndPageSize(ctx, params)

	return params
}