<?php

namespace app\admin\controller\oa;

use app\common\controller\Backend;
use think\Db;
use think\exception\DbException;
use think\exception\PDOException;
use think\exception\ValidateException;

/**
 * 排班
 *
 * @icon fa fa-circle-o
 */
class CustomSchedule extends Backend
{

    /**
     * Schedule模型对象
     * @var \app\admin\model\oa\Schedule
     */
    protected $model = null;

    protected $noNeedRight = ['editable'];


    public function _initialize()
    {
        parent::_initialize();
        $this->model = new \app\admin\model\oa\Schedule;
        $this->view->assign("typeList", $this->model->getTypeList());
    }


    /**
     * 默认生成的控制器所继承的父类中有index/add/edit/del/multi五个基础方法、destroy/restore/recyclebin三个回收站方法
     * 因此在当前控制器中可不用编写增删改查的代码,除非需要自己控制这部分逻辑
     * 需要将application/admin/library/traits/Backend.php中对应的方法复制到当前控制器,然后进行修改
     */


    public function index()
    {
        //当前是否为关联查询
        $this->relationSearch = true;
        //设置过滤方法
        $this->request->filter(['strip_tags', 'trim']);
        if ($this->request->isAjax()) {


            $filter = $this->request->param('filter');
            $filter = json_decode($filter, true);

            $startDate = date('Y-m-01');
            $endDate = date('Y-m-t');

            if (isset($filter['timetype']) && $filter['timetype'] == 2) {
                $startDate = date('Y-m-d', strtotime('monday this week'));
                $endDate = date('Y-m-d', strtotime('sunday this week'));
            }

            $res = [];

            if (!$this->auth->check('oa/custom_schedule/edit')) {
                $admins = Db::name('admin')->where('id', $this->auth->id)->field('id,nickname')->select();
            } else{
                $admins = Db::name('admin')->field('id,nickname')->select();
            }

            $admins = array_column($admins, NULL, 'id');
            $adminIds = array_keys($admins);

            $adminNames = array_column($admins, 'nickname', 'id');

            $tmpDate = $startDate;
            $dates = [];
            while(true) {

                if ($tmpDate > $endDate) {
                    break;
                }

                foreach ($adminIds as $adminId) {
                    $res[$adminId][$tmpDate] = '无';
                    $res[$adminId]['admin_id'] = $adminId;
                    $res[$adminId]['name'] = $adminNames[$adminId];
                    $res[$adminId]['editable'] = $this->auth->check('oa/custom_schedule/edit') ? 1 : 0;
                }

                $dates[] = $tmpDate;
                $tmpDate = date('Y-m-d', strtotime($tmpDate) + 86400);
            }

            $builder = $this->model
                ->with([
                    'admin'
                ])
                ->where('date', '>=', $startDate)
                ->where('date', '<=', $endDate);

            if (!$this->auth->check('oa/custom_schedule/edit')) {
                $builder = $builder->where('exec_admin_id', $this->auth->id);
            }

            $queryData = $builder->select();
            foreach ($queryData as $queryDatum) {
                $queryDatum = $queryDatum->toArray();

                if (empty($queryDatum['admin']['id'])) {
                    continue;
                }

                $adminId = $queryDatum['admin']['id'];
                $date = $queryDatum['date'];
                $type = $queryDatum['type'];

                if (isset($res[$adminId][$date])) {
                    $res[$adminId][$date] = $this->model->getTypeList()[$type];
                }
            }


            $res = array_values($res);

            $result = array("total" => count($res), "rows" => $res);

            return json($result);
        }

        $timeType = [
            1 => '本月排班',
            2 => '本周排班',
        ];
        $this->view->assign("timetype", $timeType);
        return $this->view->fetch();
    }

    /**
     * 添加
     *
     * @return string
     * @throws \think\Exception
     */
    public function add()
    {
        if (false === $this->request->isPost()) {
            return $this->view->fetch();
        }
        $params = $this->request->post('row/a');
        if (empty($params)) {
            $this->error(__('Parameter %s can not be empty', ''));
        }

        Db::startTrans();
        try {

            $dates = [];

            if ($params['period'] == 'month') {
                $daysInMonth = date('t'); // 本月天数
                $month = date('Y-m');     // 当前年月
                for ($i = 1; $i <= $daysInMonth; $i++) {
                    $dates[] = $month . '-' . str_pad($i, 2, '0', STR_PAD_LEFT);
                }
            } elseif ($params['period'] == 'week') {
                $start = strtotime('monday this week');
                for ($i = 0; $i < 7; $i++) {
                    $dates[] = date('Y-m-d', strtotime("+$i day", $start));
                }
            } else {
                $dates = [date('Y-m-d')];
            }
            $type = $params['type'];
            $execAdminIds = explode(',', $params['exec_admin_id']);

            $schedules = [];
            foreach ($execAdminIds as $execAdminId) {
                foreach ($dates as $date) {
                    $schedules[] = [
                        'exec_admin_id' => $execAdminId,
                        'date' => $date,
                        'type' => $type,
                    ];
                }
            }

            Db::name('schedule')
                ->whereIn('exec_admin_id', $execAdminIds)
                ->whereIn('date', $dates)
                ->delete();

            Db::name('schedule')
                ->insertAll($schedules);

            Db::commit();
        } catch (ValidateException|PDOException|Exception $e) {
            Db::rollback();
            $this->error($e->getMessage());
        }
        $this->success();
    }

    /**
     * 编辑
     *
     * @param $ids
     * @return string
     * @throws DbException
     * @throws \think\Exception
     */
    public function edit($ids = null)
    {
        $row = $this->model->get($ids);
        if (!$row) {
            $this->error(__('No Results were found'));
        }
        $adminIds = $this->getDataLimitAdminIds();
        if (is_array($adminIds) && !in_array($row[$this->dataLimitField], $adminIds)) {
            $this->error(__('You have no permission'));
        }
        if (false === $this->request->isPost()) {
            $this->view->assign('row', $row);
            return $this->view->fetch();
        }
        $params = $this->request->post('row/a');
        if (empty($params)) {
            $this->error(__('Parameter %s can not be empty', ''));
        }
        $params = $this->preExcludeFields($params);
        $result = false;
        Db::startTrans();
        try {
            //是否采用模型验证
            if ($this->modelValidate) {
                $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.edit' : $name) : $this->modelValidate;
                $row->validateFailException()->validate($validate);
            }


            $params['update_time'] = date('Y-m-d H:i:s');

            $result = $row->allowField(true)->save($params);
            Db::commit();
        } catch (ValidateException|PDOException|Exception $e) {
            Db::rollback();
            $this->error($e->getMessage());
        }
        if (false === $result) {
            $this->error(__('No rows were updated'));
        }
        $this->success();
    }


    public function editable()
    {

        $params = $this->request->post('row/a');
        if (empty($params)) {
            $this->error(__('Parameter %s can not be empty', ''));
        }


        $date = $params['field'];
        $adminId = $params['admin_id'];

        $value = $params[$date] ?? '无';
        $valueMap = [
            '无' => 0,
            '早班' => 1,
            '中班' => 2,
            '晚班' => 3,
            '行政班' => 4,
        ];

        if ($value == '无') {
            $this->model->where('exec_admin_id', $adminId)->delete();
            $this->success();
        }

        $dbValue = $valueMap[$value];

        $data = $this->model
            ->where('date', '=', $date)
            ->where('exec_admin_id', '=', $adminId)
            ->find();

        $nowTime = date('Y-m-d H:i:s');
        if (!empty($data)) {
            $data->save([
                "date" => $date,
                "exec_admin_id" => $adminId,
                "type" => $dbValue,
                "update_time" => $nowTime,
            ]);
        }

        $this->model->insert([
            "date" => $date,
            "exec_admin_id" => $adminId,
            "type" => $dbValue,
            "create_time" => $nowTime,
            "update_time" => $nowTime,
        ]);

        $this->success();
    }
}