gganli/application/admin/controller/UserInvite.php

<?php

namespace app\admin\controller;

use app\common\controller\Backend;
use think\Db;

/**
 * 用户邀请管理
 *
 * @icon fa fa-circle-o
 */
class UserInvite extends Backend
{
    
    /**
     * UserInvite模型对象
     * @var \app\admin\model\UserInvite
     */
    protected $model = null;

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

    }

    public function import()
    {
        parent::import();
    }

    /**
     * 默认生成的控制器所继承的父类中有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()) {
            //如果发送的来源是Selectpage，则转发到Selectpage
            if ($this->request->request('keyField')) {
                return $this->selectpage();
            }
            list($where, $sort, $order, $offset, $limit) = $this->buildparams();

            $list = $this->model
                    ->with(['user','inviteuser'])
                    ->where($where)
                    ->order($sort, $order)
                    ->paginate($limit);

            foreach ($list as $row) {
                
                $row->getRelation('user')->visible(['nickname']);
                $row->getRelation('inviteuser')->visible(['nickname']);
            }

            $result = array("total" => $list->total(), "rows" => $list->items());

            return json($result);
        }
        return $this->view->fetch();
    }

    /**
     * 添加
     */
    public function add()
    {
        if ($this->request->isPost()) {
            $params = $this->request->post("row/a");
            $params = $this->preExcludeFields($params);
            if (!$params) {
                $this->error(__('Parameter %s can not be empty', ''));
            }
            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);
                }
                if ($params['invite_user_id'] == $params['user_id']) {
                    throw new Exception('不能邀请自己');
                }
                if (!empty($params['invite_user_id'])) {
                    $userWhere['id'] = $params['invite_user_id'];
                    $user = Db::name('user')->where($userWhere)->find();
                    if ($user['pre_userid'] != 0) {
                        throw new Exception('该用户已被邀请');
                    }
                    $userRes = Db::name('user_id')->where($userWhere)->update(['pre_userid'=>$params['user_id']]);
                    if (!$userRes) {
                        throw new Exception('更新用户邀请信息失败');
                    }
                }
                $result = $this->model->allowField(true)->save($params);
                if ($result == false) {
                    throw new Exception(__('No rows were inserted'));
                }
                Db::commit();
                $this->success();
            } catch (ValidateException|PDOException|Exception $e) {
                Db::rollback();
                $this->error($e->getMessage());
            }
        }
        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) {
                $this->error(__('Parameter %s can not be empty', ''));
            }
            $params = $this->preExcludeFields($params);
            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);
                }
                if ($params['invite_user_id'] == $params['user_id']) {
                    throw new Exception('不能邀请自己');
                }
                if (!empty($params['invite_user_id'])) {
                    $userWhere['id'] = $params['invite_user_id'];
                    $user = Db::name('user')->where($userWhere)->find();
                    if ($user['pre_userid'] != $params['user_id']) {
                        $userRes = Db::name('user_id')->where($userWhere)->update(['pre_userid'=>$params['user_id']]);
                        if (!$userRes) {
                            throw new Exception('更新用户邀请信息失败');
                        }
                    }
                }
                $result = $row->allowField(true)->save($params);
                if ($result == false) {
                    throw new Exception(__('No rows were inserted'));
                }
                Db::commit();
                $this->success();
            } catch (ValidateException|PDOException|Exception $e) {
                Db::rollback();
                $this->error($e->getMessage());
            }
        }
        $this->view->assign("row", $row);
        return $this->view->fetch();
    }
}