XT_New/controllers/doctor_schedule_api_control...

package controllers

import (
	"XT_New/enums"
	"XT_New/models"
	"XT_New/service"
	"fmt"
	"github.com/astaxie/beego"
	"github.com/jinzhu/gorm"
	"time"
)

type DoctorScheduleApiController struct {
	BaseAuthAPIController
}

func DoctorScheduleRouters() {

	beego.Router("/api/schedules/savedoctorschedules", &DoctorScheduleApiController{}, "Get:SaveDoctorSchedules")
	beego.Router("/api/schedules/getdoctorschedulelist", &DoctorScheduleApiController{}, "Get:GetDoctorScheduleList")
	beego.Router("/api/schedules/getscheduledetail", &DoctorScheduleApiController{}, "Get:GetScheduleDetail")
	beego.Router("/api/schedules/updateschedule", &DoctorScheduleApiController{}, "Get:UpdateSchedule")
	beego.Router("/api/schedule/deleteschedule", &DoctorScheduleApiController{}, "Get:DeleteSchedule")
	beego.Router("/api/schedule/getdoctorlist", &DoctorScheduleApiController{}, "Get:GetDoctorList")
	beego.Router("/api/schedule/getschedulelist", &DoctorScheduleApiController{}, "Get:GetScheduleList")
	beego.Router("/api/schedule/addschedule", &DoctorScheduleApiController{}, "Get:AddSchedule")
	beego.Router("/api/schedule/getstaffschedulelist", &DoctorScheduleApiController{}, "Get:GetStaffScheduleList")
	beego.Router("/api/schedule/getnextweeklist", &DoctorScheduleApiController{}, "Get:GetNextWeekList")
	beego.Router("/api/schedule/getschedulebydoctorid", &DoctorScheduleApiController{}, "Get:GetScheduleByDoctorId")
	beego.Router("/api/schedule/tosearchsechedulelist", &DoctorScheduleApiController{}, "Get:ToSearchScheduleList")
	beego.Router("api/schedule/deletestaffschedule", &DoctorScheduleApiController{}, "Get:DeleteStaffSchedule")
	beego.Router("/api/schedule/copystaffschedule", &DoctorScheduleApiController{}, "Get:CopyStaffSchedule")
	beego.Router("/api/scheudle/updatecontinusschedule", &DoctorScheduleApiController{}, "Get:UpdateContinusSchedule")
}

func (this *DoctorScheduleApiController) SaveDoctorSchedules() {
	adminUser := this.GetAdminUserInfo()
	orgId := adminUser.CurrentOrgId
	class_name := this.GetString("class_name")
	class_attributes, _ := this.GetInt64("class_attributes")
	timeone_start := this.GetString("timeone_start")
	timeone_type, _ := this.GetInt64("timeone_type")
	timeone_end := this.GetString("timeone_end")
	timetwo_start := this.GetString("timetwo_start")
	timetwo_type, _ := this.GetInt64("timetwo_type")
	timetwo_end := this.GetString("timetwo_end")
	work_time := this.GetString("work_time")
	remarks := this.GetString("remarks")

	schedules := models.DoctorSchedules{
		ClassName:       class_name,
		ClassAttributes: class_attributes,
		TimeoneStart:    timeone_start,
		TimeoneType:     timeone_type,
		TimeoneEnd:      timeone_end,
		TimetwoStart:    timetwo_start,
		TimetwoEnd:      timetwo_end,
		TimetwoType:     timetwo_type,
		WorkTime:        work_time,
		Remarks:         remarks,
		UserOrgId:       orgId,
		Status:          1,
		Ctime:           time.Now().Unix(),
	}

	//查询班种名称是否存在
	_, errcode := service.GetClassName(orgId, class_name)
	//如果不存在
	if errcode == gorm.ErrRecordNotFound {
		err := service.CreateDotorSchedule(&schedules)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}

		this.ServeSuccessJSON(map[string]interface{}{
			"schedules": schedules,
		})
	} else if errcode == nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "班种名称已存在")
		return
	}

}

func (this *DoctorScheduleApiController) GetDoctorScheduleList() {
	adminUserInfo := this.GetAdminUserInfo()
	orgId := adminUserInfo.CurrentOrgId
	page, _ := this.GetInt64("page")
	limit, _ := this.GetInt64("limit")
	schedules, total, err := service.GetDoctorScheduleList(orgId, page, limit)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"schedules": schedules,
		"total":     total,
	})
	//查询该机构是否有班种，如果没有去读系统默认的
	//_, errcode := service.GetDoctorScheduleByOrgId(orgId)
	//if errcode == gorm.ErrRecordNotFound{
	//  schedules, total, err := service.GetDoctorScheduleList(0,page, limit)
	//  if err != nil {
	//    this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
	//    return
	//  }
	//
	//  this.ServeSuccessJSON(map[string]interface{}{
	//    "schedules": schedules,
	//    "total":total,
	//  })
	//}else if errcode == nil{
	//  schedules, total, err := service.GetDoctorScheduleList(orgId, page, limit)
	//  if err != nil {
	//    this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
	//    return
	//  }
	//
	//  this.ServeSuccessJSON(map[string]interface{}{
	//    "schedules": schedules,
	//    "total":total,
	//  })
	//}
}

func (this *DoctorScheduleApiController) GetScheduleDetail() {

	id, _ := this.GetInt64("id")
	scheduleDetail, err := service.GetScheduleDetail(id)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"scheduleDetail": scheduleDetail,
	})
}

func (this *DoctorScheduleApiController) UpdateSchedule() {

	id, _ := this.GetInt64("id")
	class_name := this.GetString("class_name")
	class_attributes, _ := this.GetInt64("class_attributes")
	timeone_start := this.GetString("timeone_start")
	timeone_type, _ := this.GetInt64("timeone_type")
	timeone_end := this.GetString("timeone_end")
	timetwo_start := this.GetString("timetwo_start")
	timetwo_type, _ := this.GetInt64("timetwo_type")
	timetwo_end := this.GetString("timetwo_end")
	work_time := this.GetString("work_time")
	remarks := this.GetString("remarks")
	user_org_id, _ := this.GetInt64("user_org_id")

	schedules := models.DoctorSchedules{
		ClassName:       class_name,
		ClassAttributes: class_attributes,
		TimeoneStart:    timeone_start,
		TimeoneType:     timeone_type,
		TimeoneEnd:      timeone_end,
		TimetwoStart:    timetwo_start,
		TimetwoEnd:      timetwo_end,
		TimetwoType:     timetwo_type,
		WorkTime:        work_time,
		Remarks:         remarks,
	}

	if user_org_id == 0 {
		err := service.CreateDotorSchedule(&schedules)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}
	} else {
		//查询是否班种名称已存在
		scheduledata, _ := service.GetClassNameIsExsit(class_name, user_org_id, id)

		if scheduledata.ID > 0 && scheduledata.ID != id {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "班种名称已存在")
			return
		}

		err := service.UpdateScheduleList(id, &schedules)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}
	}
	this.ServeSuccessJSON(map[string]interface{}{
		"schedules": schedules,
	})
}

func (this *DoctorScheduleApiController) DeleteSchedule() {
	adminUserInfo := this.GetAdminUserInfo()
	orgId := adminUserInfo.CurrentOrgId
	id, _ := this.GetInt64("id")
	//产寻该班种是否存在排班，如果存在不能删除
	_, errcode := service.GetStaffScheduleByScheduleType(id, orgId)
	if errcode == gorm.ErrRecordNotFound {
		err := service.DeleteScheduleById(id)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}
		returnData := make(map[string]interface{}, 0)
		returnData["msg"] = "ok"
		this.ServeSuccessJSON(returnData)
		return
	} else if errcode == nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "该班种已存在排班,不能删除")
		return
	}

}

func (this *DoctorScheduleApiController) GetDoctorList() {

	orgId := this.GetAdminUserInfo().CurrentOrgId
	appId := this.GetAdminUserInfo().CurrentAppId
	//admin_user_id := this.GetAdminUserInfo().AdminUser.Id
	list, err := service.GetDoctorList(orgId, appId)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"list": list,
	})
}

func (this *DoctorScheduleApiController) GetScheduleList() {

	orgId := this.GetAdminUserInfo().CurrentOrgId
	_, errcode := service.GetDoctorScheduleByOrgId(orgId)
	if errcode == gorm.ErrRecordNotFound {
		scheduleList, err := service.GetScheduleList(0)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}

		this.ServeSuccessJSON(map[string]interface{}{
			"scheduleList": scheduleList,
		})
	} else {
		scheduleList, err := service.GetScheduleList(orgId)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}

		this.ServeSuccessJSON(map[string]interface{}{
			"scheduleList": scheduleList,
		})
	}
}

func (this *DoctorScheduleApiController) AddSchedule() {

	orgId := this.GetAdminUserInfo().CurrentOrgId
	doctor_id, _ := this.GetInt64("doctor_id")
	fmt.Println("doctor_id", doctor_id)
	doctor_type, _ := this.GetInt64("doctor_type")
	fmt.Println("doctor_type", doctor_type)
	start_time, _ := this.GetInt64("start_time")
	fmt.Println("start_time", start_time)
	end_time, _ := this.GetInt64("end_time")
	fmt.Println("end_time", end_time)
	schedule_type, _ := this.GetInt64("schedule_type")
	fmt.Println("schedule_type", schedule_type)
	schedule_week, _ := this.GetInt64("schedule_week")
	fmt.Println("schedule_week", schedule_week)
	schedule_date, _ := this.GetInt64("schedule_date")
	schedule := models.StaffSchedule{
		DoctorId:     doctor_id,
		DoctorType:   doctor_type,
		ScheduleType: schedule_type,
		ScheduleWeek: schedule_week,
		UserOrgId:    orgId,
		StartTime:    start_time,
		EndTime:      end_time,
		Status:       1,
		Ctime:        time.Now().Unix(),
		ScheduleDate: schedule_date,
	}

	_, errcode := service.GetScheduleListDetail(doctor_id, orgId, schedule_date)
	fmt.Println("errcode-----", errcode)
	if errcode == gorm.ErrRecordNotFound {
		err := service.AddSchedule(&schedule)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}

		this.ServeSuccessJSON(map[string]interface{}{
			"schedule": schedule,
		})
	} else if errcode == nil {
		schedules, _ := service.GetScheduleListDetail(doctor_id, orgId, schedule_date)
		err := service.UpdateStaffList(&schedule, schedules.ID)
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}

		this.ServeSuccessJSON(map[string]interface{}{
			"schedule": schedule,
		})
	}

}

func (this *DoctorScheduleApiController) GetStaffScheduleList() {

	start_time, _ := this.GetInt64("start_time")
	fmt.Println("start_time", start_time)
	end_time, _ := this.GetInt64("end_time")
	fmt.Println("end_time", end_time)
	orgId := this.GetAdminUserInfo().CurrentOrgId
	staffList, err := service.GetStaffScheduleList(orgId, start_time, end_time)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"staffList": staffList,
	})
}

func (this *DoctorScheduleApiController) GetNextWeekList() {
	start_time, _ := this.GetInt64("start_time")
	fmt.Println("start_time", start_time)
	end_time, _ := this.GetInt64("end_time")
	fmt.Println("end_time", end_time)
	orgId := this.GetAdminUserInfo().CurrentOrgId
	staffList, err := service.GetStaffScheduleList(orgId, start_time, end_time)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"staffList": staffList,
	})
}

func (this *DoctorScheduleApiController) GetScheduleByDoctorId() {

	doctor_id, _ := this.GetInt64("doctor_id")
	start_time, _ := this.GetInt64("start_time")
	end_time, _ := this.GetInt64("end_time")
	orgId := this.GetAdminUserInfo().CurrentOrgId
	staffList, err := service.GetScheduleByDoctorId(doctor_id, start_time, end_time, orgId)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"staffList": staffList,
	})
}

func (this *DoctorScheduleApiController) ToSearchScheduleList() {

	//doctor_id, _ := this.GetInt64("doctor_id")
	user_name := this.GetString("user_name")
	start_time, _ := this.GetInt64("start_time")
	end_time, _ := this.GetInt64("end_time")
	orgId := this.GetAdminUserInfo().CurrentOrgId
	staffList, err := service.ToSearchSeacheduleList(user_name, start_time, end_time, orgId)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"staffList": staffList,
	})
}

func (this *DoctorScheduleApiController) DeleteStaffSchedule() {
	adminUserInfo := this.GetAdminUserInfo()
	orgId := adminUserInfo.CurrentOrgId
	start_time, _ := this.GetInt64("start_time")
	fmt.Println("start_time", start_time)
	end_time, _ := this.GetInt64("end_time")
	err := service.DeleteStaffSchedule(orgId, start_time, end_time)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}
	returnData := make(map[string]interface{}, 0)
	returnData["msg"] = "ok"
	this.ServeSuccessJSON(returnData)
	return
}

func (this *DoctorScheduleApiController) CopyStaffSchedule() {

	adminUserInfo := this.GetAdminUserInfo()
	orgId := adminUserInfo.CurrentOrgId
	start_time, _ := this.GetInt64("start_time")
	fmt.Println("start_time", start_time)
	end_time, _ := this.GetInt64("end_time")
	fmt.Println("end_time", end_time)
	copy_startime, _ := this.GetInt64("copy_startime")
	fmt.Println("copy_startime", copy_startime)
	copy_endtime, _ := this.GetInt64("copy_endtime")
	schedule, err := service.GetStaffScheduleListTwo(orgId, start_time, end_time)
	//查询选中日期的排班是否存在
	_, errcode := service.GetNextWeekSchedule(orgId, copy_startime, copy_endtime)
	if errcode == gorm.ErrRecordNotFound {
		for _, item := range schedule {
			//fmt.Println(item.StartTime+604800)
			//礼拜1
			if item.ScheduleWeek == 1 {
				item.ScheduleDate = copy_startime
			}
			if item.ScheduleWeek == 2 {
				item.ScheduleDate = copy_startime + 86400
			}
			if item.ScheduleWeek == 3 {
				item.ScheduleDate = copy_startime + 172800
			}
			if item.ScheduleWeek == 4 {
				item.ScheduleDate = copy_startime + 259200
			}
			if item.ScheduleWeek == 5 {
				item.ScheduleDate = copy_startime + 345600
			}

			if item.ScheduleWeek == 6 {
				item.ScheduleDate = copy_startime + 432000
			}
			//礼拜天
			if item.ScheduleWeek == 0 {
				item.ScheduleDate = copy_endtime
			}
			staffSchedule := models.StaffSchedule{
				DoctorId:     item.DoctorId,
				DoctorType:   item.DoctorType,
				ScheduleType: item.ScheduleType,
				ScheduleWeek: item.ScheduleWeek,
				UserOrgId:    item.UserOrgId,
				StartTime:    copy_startime,
				EndTime:      copy_endtime,
				Status:       1,
				Ctime:        time.Now().Unix(),
				ScheduleDate: item.ScheduleDate,
			}
			service.AddSchedule(&staffSchedule)
		}
		if err != nil {
			this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
			return
		}
		returnData := make(map[string]interface{}, 0)
		returnData["msg"] = "ok"
		this.ServeSuccessJSON(returnData)
		return
	} else if errcode == nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "排班数据已存在")
		fmt.Print("下周排班已存在")
		return
	}

}

func (this *DoctorScheduleApiController) UpdateContinusSchedule() {
	adminUserInfo := this.GetAdminUserInfo()
	orgId := adminUserInfo.CurrentOrgId
	is_status, _ := this.GetInt64("is_status")
	fmt.Println("is_status", is_status)
	schedule := models.ContinueSchedule{
		IsStatus: is_status,
	}
	err := service.UpdateContinusSchedule(&schedule, orgId)
	if err != nil {
		this.ServeFailJsonSend(enums.ErrorCodeDataException, "添加设备失败")
		return
	}

	this.ServeSuccessJSON(map[string]interface{}{
		"schedule": schedule,
	})
}