package controllers

import (
	"github.com/kataras/iris/v12/mvc"
	"xps/datamodels"
	"xps/service"
)

type ModelController struct {
	Base
	Service service.ModelService
}

func (c *ModelController) BeforeActivation(b mvc.BeforeActivation) {
	b.Handle("GET",         "/",                    "GetList"       )
	b.Handle("GET",         "/data",                "GetTreeData"   )
	b.Handle("GET",         "/page",                "GetPage"       )
	b.Handle("GET",         "/{modelId:int64}",     "GetById"       )
	b.Handle("POST",        "/add",                 "Create"        )
	b.Handle("PUT",         "/update",              "Update"        )
	b.Handle("DELETE",      "/{modelId:int64}",     "DeleteByID"    )
}

func (c *ModelController) GetList() *JsonResult {
	params    := c.buildParams()
	data, err := c.Service.GetList(params)
	if err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("获取成功", data)
	}
}

func (c *ModelController) GetTreeData() *JsonResult{
	data, err := c.Service.GetTreeData()
	if err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("获取成功", data)
	}
}

func (c *ModelController) GetPage() *JsonResult {
	params    := c.buildParams()
	data, err := c.Service.GetPage(params)
	if err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("获取成功", data)
	}
}

func (c *ModelController) GetById(modelId int64) *JsonResult {
	data, err := c.Service.GetById(modelId)
	if err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("获取成功", data)
	}
}

func (c *ModelController) Create() *JsonResult {
	userId := c.GetCurUserId()
	
	mod := datamodels.Model{}
	mod.CreatedBy = userId
	if err := c.Ctx.ReadJSON(&mod); err != nil {
		return ResultErr("读取数据失败", nil)
	}
	
	if err := c.Service.Create(&mod); err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("新增成功", nil)
	}
}

func (c *ModelController) Update() *JsonResult {
	userId := c.GetCurUserId()
	
	mod  := datamodels.Model{}
	if err := c.Ctx.ReadJSON(&mod); err != nil {
		return ResultErr("读取数据失败", nil)
	}
	
	mod.UpdatedBy = userId
	if err := c.Service.Update(&mod); err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("更新成功", nil)
	}
}

func (c *ModelController) DeleteByID(modelId int64) *JsonResult {
	userId := c.GetCurUserId()
	
	m := make(map[string]interface{})
	m["modelId"]   = modelId
	m["deletedBy"] = userId
	if err := c.Service.Delete(m); err != nil {
		return ResultErr(err.Error(), nil)
	} else {
		return ResultOk("删除成功", nil)
	}
}