lsky-pro/application/index/controller/admin/Group.php

<?php
/**
 * User: Wisp X
 * Date: 2019-01-19
 * Time: 11:59
 * Link: https://github.com/wisp-x
 */

namespace app\index\controller\admin;

use app\common\model\Group as GroupModel;
use think\Db;
use think\Exception;
use think\facade\Config;

class Group extends Base
{
    protected $strategyList;

    protected function initialize()
    {
        parent::initialize();
        $this->strategyList = Config::pull('strategy');
    }

    public function index()
    {
        $groups = GroupModel::select()->order('id', 'desc')->each(function ($item) {
            $item->strategy_str = isset($this->strategyList[$item->strategy]) ? $this->strategyList[$item->strategy]['name'] : '未知';
            return $item;
        });
        $this->assign([
            'groups' => $groups,
            'strategy_list' => $this->strategyList
        ]);

        return $this->fetch();
    }

    public function add()
    {
        if ($this->request->isPost()) {
            try {
                $data = $this->request->post();
                $validate = $this->validate($data, 'Group');
                if (true !== $validate) {
                    throw new Exception($validate);
                }
                if (!GroupModel::create($data)) {
                    throw new Exception('添加失败');
                }
            } catch (Exception $e) {
                $this->error($e->getMessage());
            }

            $this->success('添加成功');
        }
    }

    public function edit()
    {
        if ($this->request->isPost()) {
            try {
                $data = $this->request->post();
                $validate = $this->validate($data, 'Group');
                if (true !== $validate) {
                    throw new Exception($validate);
                }
                $data['default'] = array_key_exists('default', $data) ? 1 : 0;
                if ($data['default'] === 0) {
                    if (!GroupModel::where('default', 1)->where('id', 'neq', $data['id'])->count()) {
                        throw new Exception('至少保留一个默认分组');
                    }
                }
                if (!GroupModel::update($data)) {
                    throw new Exception('编辑失败');
                }
            } catch (Exception $e) {
                $this->error($e->getMessage());
            }
            $this->success('编辑成功');
        }
    }

    public function del()
    {
        if ($this->request->isPost()) {
            Db::startTrans();
            try {
                $id = $this->request->post('id');
                if (1 == $id) {
                    throw new Exception('默认组不可删除');
                }
                $group = GroupModel::find($id);
                // 至少保留一个默认分组
                $defaultId = GroupModel::where('default', 1)->where('id', 'neq', $id)->value('id');
                if (!$defaultId) {
                    throw new Exception('至少保留一个默认分组');
                }
                // 转移该组下的用户到默认分组
                \app\common\model\Users::where('group_id', $group->id)->setField('group_id', $defaultId);
                $group->delete();
                Db::commit();
            } catch (Exception $e) {
                Db::rollback();
                $this->error($e->getMessage());
            }
            $this->success('删除成功');
        }
    }

    public function getGroup()
    {
        if ($this->request->isPost()) {
            $id = $this->request->post('id');
            $this->success('success', null, GroupModel::find($id));
        }
    }

    public function setDefault()
    {
        if ($this->request->isPost()) {
            $id = $this->request->post('id');
            $value = $this->request->post('value');
            if (1 != $value) {
                if (!GroupModel::where('default', 1)->where('id', 'neq', $id)->count()) {
                    $this->error('至少保留一个默认分组');
                }
            }
            if (!GroupModel::update([
                'id' => $id,
                'default' => $value
            ])) {
                $this->error('设置失败');
            }
            $this->success('设置成功');
        }
    }

    public function setStrategy()
    {
        if ($this->request->isPost()) {
            $id = $this->request->post('id');
            $strategy = $this->request->post('strategy');
            if (!array_key_exists($strategy, $this->strategyList)) {
                $this->error('储存策略不存在');
            }
            if (!GroupModel::update([
                'id' => $id,
                'strategy' => $strategy
            ])) {
                $this->error('设置失败');
            }
            $this->success('设置成功');
        }
    }
}