Robot.php 6.7 KB
<?php

namespace app\admin\controller\dinghorn;

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

/**
 * 钉钉机器人管理
 *
 * @icon fa fa-circle-o
 */
class Robot extends Backend
{

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

    protected $multiFields = 'openswitch';

    public function _initialize()
    {
        parent::_initialize();
        $this->model = new \app\admin\model\DinghornRobot;
    }

    /*验证机器人配置*/
    public function msg_test()
    {
        $data         = $this->request->only(['access_token', 'sign', 'keywords']);
        $access_token = $this->parse_hook_url($data['access_token']);
        $keywords     = isset($data['keywords']) ? $data['keywords'] : '';
        $sign         = (isset($data['sign']) && $data['sign']) ? $data['sign'] : false;

        $data = array(
            'msgtype' => 'text',
            'text'    => ['content' => '这是一条测试消息!' . $keywords],
            'at'      => [
                'atMobiles' => [],
                'isAtAll'   => true
            ]
        );

        if (!$access_token) {
            $this->error('Hook地址解析失败,没有找到 access_token');
        }

        $dinghorn = new \addons\dinghorn\library\DinghornLib();
        $res      = $dinghorn->msgSend($access_token, $data, $sign);

        if ($res['errcode'] == 0) {
            $this->success('消息发送成功!');
        } elseif ($res['errcode'] == 310000) {
            if ($res['errmsg'] == 'keywords not in content') {
                $this->error('消息内容中不包含任何 [钉钉机器人安全-自定义关键词],请检查钉钉机器人安全设置!');
            } elseif ($res['errmsg'] == 'sign not match') {
                $this->error('密匙不匹配,请确认钉钉机器人安全设置!');
            } else {
                $this->error($res['errmsg'] . '(' . $res['errcode'] . ')');
            }
        } else {
            $this->error($res['errmsg'] . '(' . $res['errcode'] . ')');
        }
    }

    public function parse_hook_url($url)
    {
        $params = [];

        if (strstr($url, '?') === false) {
            return $url;
        } else {
            $url = explode('?', $url);
            $url = explode('&', $url[1]);

            foreach ($url as $key => $value) {
                $value             = explode('=', $value);
                $params[$value[0]] = isset($value[1]) ? $value[1] : '';
            }
        }

        return isset($params['access_token']) ? $params['access_token'] : false;
    }

    /**
     * 添加
     */
    public function add()
    {
        if ($this->request->isPost()) {
            $params = $this->request->post("row/a");
            if ($params) {
                $params = method_exists($this, 'preExcludeFields') ? $this->preExcludeFields($params) : $params;

                $params['access_token'] = $this->parse_hook_url($params['access_token']);

                if ($this->dataLimit && $this->dataLimitFieldAutoFill) {
                    $params[$this->dataLimitField] = $this->auth->id;
                }
                $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 . '.add' : $name) : $this->modelValidate;
                        $this->model->validateFailException(true)->validate($validate);
                    }
                    $result = $this->model->allowField(true)->save($params);
                    Db::commit();
                } catch (ValidateException $e) {
                    Db::rollback();
                    $this->error($e->getMessage());
                } catch (PDOException $e) {
                    Db::rollback();
                    $this->error($e->getMessage());
                } catch (Exception $e) {
                    Db::rollback();
                    $this->error($e->getMessage());
                }
                if ($result !== false) {
                    $this->success();
                } else {
                    $this->error(__('No rows were inserted'));
                }
            }
            $this->error(__('Parameter %s can not be empty', ''));
        }
        return $this->view->fetch();
    }

    /**
     * 编辑
     */
    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)) {
            if (!in_array($row[$this->dataLimitField], $adminIds)) {
                $this->error(__('You have no permission'));
            }
        }
        if ($this->request->isPost()) {
            $params = $this->request->post("row/a");
            if ($params) {
                $params                 = method_exists($this, 'preExcludeFields') ? $this->preExcludeFields($params) : $params;
                $params['access_token'] = $this->parse_hook_url($params['access_token']);
                $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(true)->validate($validate);
                    }
                    $result = $row->allowField(true)->save($params);
                    Db::commit();
                } catch (ValidateException $e) {
                    Db::rollback();
                    $this->error($e->getMessage());
                } catch (PDOException $e) {
                    Db::rollback();
                    $this->error($e->getMessage());
                } catch (Exception $e) {
                    Db::rollback();
                    $this->error($e->getMessage());
                }
                if ($result !== false) {
                    $this->success();
                } else {
                    $this->error(__('No rows were updated'));
                }
            }
            $this->error(__('Parameter %s can not be empty', ''));
        }
        $this->view->assign("row", $row);
        return $this->view->fetch();
    }
}