package controller

import (
	"time"

	"ppgo_job/consts"
	"ppgo_job/dao"
	"ppgo_job/model/entity"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
)

// ==================== 服务器管理 ====================

func ServerAjaxSave(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	groupId := r.Get("group_id", 0).Int()
	serverName := r.Get("server_name").String()
	serverIp := r.Get("server_ip").String()
	port := r.Get("port", 22).Int()
	serverAccount := r.Get("server_account").String()
	password := r.Get("password").String()
	connType := r.Get("connection_type", 0).Int()
	authType := r.Get("type", 0).Int()
	privateKeySrc := r.Get("private_key_src").String()
	detail := r.Get("detail").String()

	if serverName == "" || serverIp == "" || serverAccount == "" {
		ajaxMsg(r, "服务器名称、IP、账号不能为空", consts.MsgErr)
		return
	}

	now := time.Now().Unix()

	if id > 0 {
		srv, _ := dao.TaskServer.GetById(ctx, id)
		if srv == nil {
			ajaxMsg(r, "服务器不存在", consts.MsgErr)
			return
		}
		if password != "" {
			srv.Password = password
		}
		if privateKeySrc != "" {
			srv.PrivateKeySrc = privateKeySrc
		}
		srv.GroupId = groupId
		srv.ServerName = serverName
		srv.ServerIp = serverIp
		srv.Port = port
		srv.ServerAccount = serverAccount
		srv.ConnectionType = connType
		srv.Type = authType
		srv.Detail = detail
		srv.UpdateTime = now
		_ = dao.TaskServer.Update(ctx, srv)
		ajaxMsg(r, "更新成功", consts.MsgOK)
	} else {
		_, err := dao.TaskServer.Insert(ctx, &entity.TaskServer{
			GroupId: groupId, ServerName: serverName, ServerIp: serverIp,
			Port: port, ServerAccount: serverAccount, Password: password,
			ConnectionType: connType, Type: authType,
			PrivateKeySrc: privateKeySrc, Detail: detail,
			CreateTime: now, UpdateTime: now, Status: 0,
		})
		if err != nil {
			ajaxMsg(r, "添加失败: "+err.Error(), consts.MsgErr)
			return
		}
		ajaxMsg(r, "添加成功", consts.MsgOK)
	}
}

// ServerAjaxDel 删除服务器
func ServerAjaxDel(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	if id <= 0 {
		ajaxMsg(r, "参数错误", consts.MsgErr)
		return
	}
	_ = dao.TaskServer.Delete(ctx, id)
	ajaxMsg(r, "删除成功", consts.MsgOK)
}

// ServerAjaxTest 测试服务器连接
func ServerAjaxTest(r *ghttp.Request) {
	// 暂不实现实际测试，返回成功
	ajaxMsg(r, "连接测试功能暂未实现", consts.MsgOK)
}

// ServerGetByGroupId 按分组获取服务器列表
func ServerGetByGroupId(r *ghttp.Request) {
	ctx := r.GetCtx()
	groupId := r.Get("groupId", 0).Int()
	list, _, _ := dao.TaskServer.GetList(ctx, 1, 1000, "group_id", groupId)
	r.Response.WriteJson(list)
}

// ServerApiSave Agent注册/更新API
func ServerApiSave(r *ghttp.Request) {
	ajaxMsg(r, "Agent API 待完善", consts.MsgOK)
}

// ServerApiStatus Agent更新状态API
func ServerApiStatus(r *ghttp.Request) {
	ajaxMsg(r, "Agent API 待完善", consts.MsgOK)
}

// ServerApiGet Agent获取信息API
func ServerApiGet(r *ghttp.Request) {
	r.Response.WriteJson(g.Map{"status": 0, "message": "ok"})
}

// ==================== 任务分组 ====================

// GroupAjaxSave 保存任务分组
func GroupAjaxSave(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	groupName := r.Get("group_name").String()
	description := r.Get("description").String()
	if groupName == "" {
		ajaxMsg(r, "分组名称不能为空", consts.MsgErr)
		return
	}

	now := time.Now().Unix()
	if id > 0 {
		g, _ := dao.Group.GetById(ctx, id)
		if g == nil {
			ajaxMsg(r, "分组不存在", consts.MsgErr)
			return
		}
		g.GroupName = groupName
		g.Description = description
		g.UpdateTime = now
		_ = dao.Group.Update(ctx, g)
		ajaxMsg(r, "更新成功", consts.MsgOK)
	} else {
		_, err := dao.Group.Insert(ctx, &entity.Group{
			GroupName: groupName, Description: description,
			CreateTime: now, UpdateTime: now, Status: 1,
		})
		if err != nil {
			ajaxMsg(r, "添加失败: "+err.Error(), consts.MsgErr)
			return
		}
		ajaxMsg(r, "添加成功", consts.MsgOK)
	}
}

// GroupAjaxDel 删除任务分组
func GroupAjaxDel(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	if id <= 0 {
		ajaxMsg(r, "参数错误", consts.MsgErr)
		return
	}
	_ = dao.Task.ResetGroupId(ctx, id)
	_ = dao.Group.Delete(ctx, id)
	ajaxMsg(r, "删除成功", consts.MsgOK)
}

// ==================== 资源分组 ====================

func ServerGroupAjaxSave(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	groupName := r.Get("group_name").String()
	description := r.Get("description").String()
	if groupName == "" {
		ajaxMsg(r, "分组名称不能为空", consts.MsgErr)
		return
	}

	now := time.Now().Unix()
	if id > 0 {
		sg, _ := dao.ServerGroup.GetById(ctx, id)
		if sg == nil {
			ajaxMsg(r, "分组不存在", consts.MsgErr)
			return
		}
		sg.GroupName = groupName
		sg.Description = description
		sg.UpdateTime = now
		_ = dao.ServerGroup.Update(ctx, sg)
		ajaxMsg(r, "更新成功", consts.MsgOK)
	} else {
		_, err := dao.ServerGroup.Insert(ctx, &entity.ServerGroup{
			GroupName: groupName, Description: description,
			CreateTime: now, UpdateTime: now, Status: 1,
		})
		if err != nil {
			ajaxMsg(r, "添加失败: "+err.Error(), consts.MsgErr)
			return
		}
		ajaxMsg(r, "添加成功", consts.MsgOK)
	}
}

func ServerGroupAjaxDel(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	if id <= 0 {
		ajaxMsg(r, "参数错误", consts.MsgErr)
		return
	}
	_ = dao.ServerGroup.Delete(ctx, id)
	ajaxMsg(r, "删除成功", consts.MsgOK)
}

// ==================== 禁用命令 ====================

func BanAjaxSave(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	code := r.Get("code").String()
	if code == "" {
		ajaxMsg(r, "命令关键字不能为空", consts.MsgErr)
		return
	}

	now := time.Now().Unix()
	if id > 0 {
		b, _ := dao.Ban.GetById(ctx, id)
		if b == nil {
			ajaxMsg(r, "记录不存在", consts.MsgErr)
			return
		}
		b.Code = code
		b.UpdateTime = now
		_ = dao.Ban.Update(ctx, b)
		ajaxMsg(r, "更新成功", consts.MsgOK)
	} else {
		_, err := dao.Ban.Insert(ctx, &entity.Ban{Code: code, CreateTime: now, UpdateTime: now, Status: 1})
		if err != nil {
			ajaxMsg(r, "添加失败: "+err.Error(), consts.MsgErr)
			return
		}
		ajaxMsg(r, "添加成功", consts.MsgOK)
	}
}

func BanAjaxDel(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	if id <= 0 {
		ajaxMsg(r, "参数错误", consts.MsgErr)
		return
	}
	_ = dao.Ban.Delete(ctx, id)
	ajaxMsg(r, "删除成功", consts.MsgOK)
}

// ==================== 通知模板 ====================

func NotifyTplTable(r *ghttp.Request) {
	ctx := r.GetCtx()
	page := r.Get("page", 1).Int()
	limit := r.Get("limit", 20).Int()
	var filters []interface{}
	if sortStr := parseSort(r); sortStr != "" {
		filters = append(filters, "@order", sortStr)
	}
	list, total, err := dao.NotifyTpl.GetList(ctx, page, limit, filters...)
	if err != nil {
		ajaxList(r, "查询失败", consts.MsgErr, 0, nil)
		return
	}
	rows := make([]g.Map, 0)
	for _, v := range list {
		tplTypeText := ""
		switch v.TplType {
		case consts.NotifyTypeEmail:
			tplTypeText = "邮件"
		case consts.NotifyTypeSMS:
			tplTypeText = "短信"
		case consts.NotifyTypeDingtalk:
			tplTypeText = "钉钉"
		case consts.NotifyTypeWechat:
			tplTypeText = "微信"
		}
		statusText := "正常"
		if v.Status == 0 {
			statusText = "禁用"
		}
		ct := ""
		ut := ""
		if v.CreateTime > 0 {
			ct = time.Unix(v.CreateTime, 0).Format("2006-01-02 15:04:05")
		}
		if v.UpdateTime > 0 {
			ut = time.Unix(v.UpdateTime, 0).Format("2006-01-02 15:04:05")
		}
		rows = append(rows, g.Map{"id": v.Id, "type": v.Type, "tpl_name": v.TplName, "tpl_type_text": tplTypeText, "create_time": ct, "update_time": ut, "status_text": statusText})
	}
	ajaxList(r, "成功", 0, total, rows)
}

func NotifyTplAjaxSave(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	tplName := r.Get("tpl_name").String()
	tplType := r.Get("tpl_type", 0).Int()
	title := r.Get("title").String()
	content := r.Get("content").String()
	tp := r.Get("type").String()
	if tp == "" {
		tp = "default"
	}

	if tplName == "" {
		ajaxMsg(r, "模板名称不能为空", consts.MsgErr)
		return
	}

	now := time.Now().Unix()
	if id > 0 {
		t, _ := dao.NotifyTpl.GetById(ctx, id)
		if t == nil {
			ajaxMsg(r, "模板不存在", consts.MsgErr)
			return
		}
		t.TplName = tplName
		t.TplType = tplType
		t.Title = title
		t.Content = content
		t.Type = tp
		t.UpdateTime = now
		_ = dao.NotifyTpl.Update(ctx, t)
		ajaxMsg(r, "更新成功", consts.MsgOK)
	} else {
		_, err := dao.NotifyTpl.Insert(ctx, &entity.NotifyTpl{
			TplName: tplName, TplType: tplType, Title: title,
			Content: content, Type: tp,
			CreateTime: now, UpdateTime: now, Status: 1,
		})
		if err != nil {
			ajaxMsg(r, "添加失败: "+err.Error(), consts.MsgErr)
			return
		}
		ajaxMsg(r, "添加成功", consts.MsgOK)
	}
}

func NotifyTplAjaxDel(r *ghttp.Request) {
	ctx := r.GetCtx()
	id := r.Get("id", 0).Int()
	if id <= 0 {
		ajaxMsg(r, "参数错误", consts.MsgErr)
		return
	}
	_ = dao.NotifyTpl.Delete(ctx, id)
	ajaxMsg(r, "删除成功", consts.MsgOK)
}

func NotifyTplAjaxByType(r *ghttp.Request) {
	ctx := r.GetCtx()
	notifyType := r.Get("notify_type", 0).Int()
	tpl, _ := dao.NotifyTpl.GetByTplType(ctx, notifyType, "system")
	if tpl == nil {
		tpl, _ = dao.NotifyTpl.GetByTplType(ctx, notifyType, "default")
	}
	if tpl == nil {
		r.Response.WriteJson(g.Map{
			"code": 0,
			"data": make([]g.Map, 0),
		})
		return
	}
	r.Response.WriteJson(g.Map{
		"code": 0,
		"data": []g.Map{
			{
				"id":       tpl.Id,
				"tpl_name": tpl.TplName,
				"title":    tpl.Title,
				"content":  tpl.Content,
			},
		},
	})
}