package controllers

import (
	"encoding/json"
	"oppmg/common/log"
	"oppmg/protocol"
	servecompany "oppmg/services/company"
	"oppmg/utils"
	"strings"
)

type CompanyController struct {
	BaseController
}

//URLMapping 实现ControllerInterface中的URLMapping
func (c *CompanyController) URLMapping() {
	//c.Mapping("AccessToken", c.AccessToken)
}

// DepartmentList 部门列表
// @router /department [post]
func (c *CompanyController) DepartmentList() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	companyid := c.GetCompanyId()
	if companyid <= 0 {
		msg = protocol.BadRequestParam("1")
		return
	}
	departs, err := servecompany.DepartmentListAll(companyid)
	listdata := protocol.ResponseDepartmentList{
		List: departs,
	}
	msg = protocol.NewReturnResponse(listdata, err)
}

// DepartmentAdd 添加部门
// @router /department/add [post]
func (c *CompanyController) DepartmentAdd() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestDepartmentAdd
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	s := []rune(param.Name)
	if len(s) == 0 {
		msg = protocol.BadRequestParam("10044")
		return
	}
	if len(s) > 20 {
		msg = protocol.BadRequestParam("10043")
		return
	}
	if param.ParentID <= 0 {
		msg = protocol.BadRequestParam("10042")
		return
	}
	if param.ParentID == 0 {
		//部门必定有上级部门 ,至少是公司一级
		msg = protocol.BadRequestParam("10042")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.CompanyID <= 0 {
		log.Error("param.CompanyID <= 0")
		msg = protocol.BadRequestParam("1")
		return
	}
	result, err := servecompany.DepartmentAdd(param)
	msg = protocol.NewReturnResponse(result, err)
	return
}

// DepartmentUpdate 更新部门
// @router /department/edit [post]
func (c *CompanyController) DepartmentUpdate() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()

	var param protocol.RequestDepartmentEdit
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	s := []rune(param.Name)
	if len(s) == 0 {
		msg = protocol.BadRequestParam("10044")
		return
	}
	if len(s) > 20 {
		msg = protocol.BadRequestParam("10043")
		return
	}
	param.CompanyID = c.GetCompanyId()
	if param.ID <= 0 {
		log.Error(" param.ID <= 0 ")
		msg = protocol.BadRequestParam("1")
		return
	}
	err := servecompany.DepartmentEdit(param)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

// DepartmentUpdate 删除部门
// @router /department/delete [post]
func (c *CompanyController) DepartmentDelete() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()

	var param protocol.RequestDepartmentDelete
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	err := servecompany.DepartmentDelete(param)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

// PositionAdd 添加职位
// @router /position/add [post]
func (c *CompanyController) PositionAdd() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestPositionAdd
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}

	param.CompanyID = c.GetCompanyId()
	positioninfo, err := servecompany.PositionAdd(param)
	msg = protocol.NewReturnResponse(positioninfo, err)
	return
}

// PositionEdit 编辑职位
// @router /position/edit [post]
func (c *CompanyController) PositionEdit() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestPositionEdit
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	positioninfo, err := servecompany.PositionEdit(param)
	msg = protocol.NewReturnResponse(positioninfo, err)
	return
}

// PositionDelete 删除职位
// @router /position/delete [post]
func (c *CompanyController) PositionDelete() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestPositionDelete
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	param.CompanyID = c.GetCompanyId()
	err := servecompany.PositionDelete(param)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

// PositionList 获取职位列表
// @router /position [get]
func (c *CompanyController) PositionList() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()

	var (
		err       error
		list      []protocol.ResponsePositionInfo
		companyid int64
	)

	companyid = c.GetCompanyId()
	if companyid <= 0 {
		msg = protocol.BadRequestParam("1")
		return
	}
	list, err = servecompany.GetPositionList(companyid)
	msg = protocol.NewReturnResponse(list, err)
	return
}

//UserAdd 添加用户
//@Router /user/add [post]
func (c *CompanyController) UserAdd() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestUserAdd
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	name := []rune(strings.TrimSpace(param.Name))
	if len(name) == 0 {
		msg = protocol.BadRequestParam("10035")
		return
	}
	if len(name) > 10 {
		msg = protocol.BadRequestParam("10034")
		return
	}
	ok := utils.PhoneMatch.MatchString(param.Phone)
	if !ok {
		msg = protocol.BadRequestParam("10036")
		return
	}
	if len(param.Departments) == 0 {
		msg = protocol.BadRequestParam("10037")
		return
	}
	if len(param.Roles) == 0 {
		msg = protocol.BadRequestParam("10038")
		return
	}
	param.CompanyId = c.GetCompanyId()
	err := servecompany.UserAdd(param)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

//UserEdit 编辑用户
//@Router /user/edit [post]
func (c *CompanyController) UserEdit() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestUserEdit
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	name := []rune(strings.TrimSpace(param.Name))
	if len(name) == 0 {
		msg = protocol.BadRequestParam("10035")
		return
	}
	if len(name) > 10 {
		msg = protocol.BadRequestParam("10034")
		return
	}

	if len(param.Departments) == 0 {
		msg = protocol.BadRequestParam("10037")
		return
	}
	if len(param.Roles) == 0 {
		msg = protocol.BadRequestParam("10038")
		return
	}
	param.CompanyId = c.GetCompanyId()
	err := servecompany.UserEdit(param)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

//UserDelete 删除用户
//@Router /user/delete [post]
func (c *CompanyController) UserDelete() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Parameter struct {
		UserCompanyId []int64 `json:"ids"`
	}
	var param Parameter
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	companyId := c.GetCompanyId()
	err := servecompany.UserDelete(param.UserCompanyId, companyId)

	msg = protocol.NewReturnResponse(nil, err)
	return
}

//UserEdit 禁用、启用用户 TODO
//@Router /user/allow_forbid [post]
func (c *CompanyController) UserAllowForbid() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Parameter struct {
		UserCompanyIds []int64 `json:"ids"`
		Status         int     `json:"status"` // [1:执行禁用操作][2:执行启用操作]
	}
	var param Parameter
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	var (
		err error
	)
	companyid := c.GetCompanyId()
	switch param.Status {
	case 1:
		err = servecompany.UserForbid(param.UserCompanyIds, companyid)
	case 2:
		err = servecompany.UserAllow(param.UserCompanyIds, companyid)
	default:
		msg = protocol.BadRequestParam("1")
		return
	}

	msg = protocol.NewReturnResponse(nil, err)
	return
}

//UserEdit用户列表
//@Router /user/list [post]
func (c *CompanyController) UserList() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	var param protocol.RequestUserList
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}

	param.Companyid = c.GetCompanyId()
	result, err := servecompany.UserList(param)
	msg = protocol.NewPageDataResponse(result, err)
	return
}

//InitCompany 初始化企业
//@router /ucenter/company [post]
func (c *CompanyController) InitCompany() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()

	var param protocol.CenterCompanyInfo
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	var (
		err error
	)
	if ok := param.IsEnable(); ok {
		err = servecompany.InitCompanyInfo(param)
	} else if ok := param.IsForbid(); ok {
		err = servecompany.ForbidCompany(param.CompanyId)
	} else {
		err = protocol.NewErrWithMessage("1")
	}

	msg = protocol.NewReturnResponse(nil, err)
	return
}

//@router /ucenter/company/info [post]
func (c *CompanyController) GetCompanyForUCenter() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Paremeter struct {
		Companyid int64 `json:"company_id"`
	}
	var param protocol.CenterCompanyInfo
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}

	msg = protocol.NewReturnResponse(nil, nil)
	return
}

//DepartmentUser 获取部门下成员
//@router /department/user [post]
func (c *CompanyController) DepartmentUser() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Parameter struct {
		DepartmentId int64 `json:"id"`
	}
	var param Parameter
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	companyid := c.GetCompanyId()
	resultdata, err := servecompany.GetDepartmentUser(companyid, param.DepartmentId)
	msg = protocol.NewReturnResponse(resultdata, err)
	return
}

//CurrentCompanyEdit  编辑当前公司
func (c *CompanyController) CurrentCompanyEdit() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	type Parameter struct {
		Logo string `json:"string"`
	}
	var param Parameter
	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
		log.Error("json 解析失败 err:%s", err)
		msg = protocol.BadRequestParam("1")
		return
	}
	companyid := c.GetCompanyId()
	err := servecompany.CompanyInfoEdit(companyid, param.Logo)
	msg = protocol.NewReturnResponse(nil, err)
	return
}

//CurrentCompanyInfo 获取当前公司数据
func (c *CompanyController) CurrentCompanyInfo() {
	var msg *protocol.ResponseMessage
	defer func() {
		c.ResposeJson(msg)
	}()
	// type Parameter struct {
	// 	Logo string `json:"string"`
	// }
	// var param Parameter
	// if err := json.Unmarshal(c.Ctx.Input.RequestBody, &param); err != nil {
	// 	log.Error("json 解析失败 err:%s", err)
	// 	msg = protocol.BadRequestParam("1")
	// 	return
	// }
	companyid := c.GetCompanyId()
	companybase, err := servecompany.CompanyBaseInfo(companyid)
	msg = protocol.NewReturnResponse(companybase, err)
	return
}