swiftadmin/app/AdminController.php

<?php
// +----------------------------------------------------------------------
// | swiftAdmin 极速开发框架 [基于WebMan开发]
// +----------------------------------------------------------------------
// | Copyright (c) 2020-2030 http://www.swiftadmin.net
// +----------------------------------------------------------------------
// | swiftAdmin.net High Speed Development Framework
// +----------------------------------------------------------------------
// | Author: meystack <coolsec@foxmail.com> Apache 2.0 License
// +----------------------------------------------------------------------
namespace app;

use app\admin\enums\AdminEnum;
use app\admin\service\AuthService;
use support\Log;
use support\Response;
use think\helper\Str;

class AdminController extends BaseController
{
    /**
     * 数据库实例
     * @var object
     */
    public object $model;

    /**
     * 数据表名称
     * @var string
     */
    public string $tableName;

    /**
     * 操作状态
     * @var mixed
     */
    public mixed $status;

    /**
     * 获取模板
     * @var      string
     */
    public string $template = '';

    /**
     * 权限验证类
     * @var object
     */
    public object $authService;

    /**
     * 当前表字段
     * @var array
     */
    protected array $tableFields = [];

    /**
     * 默认开关
     * @var string
     */
    protected string $keepField = 'status';

    /**
     * 开启数据限制
     * @var boolean
     */
    protected bool $dataLimit = false;

    /**
     * 是否开启部门限制
     * @var bool
     */
    protected bool $departmentLimit = false;

    /**
     * 数据限制字段
     * @var string
     */
    protected string $dataLimitField = 'admin_id';

    /**
     * 需要排除的字段
     * @var mixed
     */
    protected mixed $ruleOutFields = '';

    /**
     * 查询过滤字段
     * @var array
     */
    protected array $filterWhere = ['page', 'limit'];

    /**
     * 查询转换字段
     * @var array
     */
    protected array $converTime = ['create_time', 'update_time', 'delete_time'];

    /**
     * 定义关联模型
     * @var array
     */
    protected array $relationModel = [];

    /**
     * 跳转URL地址
     * @var string
     */
    protected string $JumpUrl = '/';

    /**
     * 构造函数
     */
    public function __construct()
    {
        parent::__construct();
        $this->authService = AuthService::instance();
    }

    /**
     * 获取资源列表
     * @return Response
     */
    public function index()
    {
        if (request()->isAjax()) {
            $page = (int)input('page', 1);
            $limit = (int)input('limit', 18);
            $where = $this->buildSelectParams();
            $count = $this->model->where($where)->count();
            $page = $count <= $limit ? 1 : $page;
            $fieldList = $this->model->getFields();
            $order = !array_key_exists('sort', $fieldList) ? 'id' : 'sort';
            $subQuery = $this->model->field('id')->where($where)->order($order, 'desc')->limit($limit)->page($page)->buildSql();
            $subQuery = '( SELECT object.id FROM ' . $subQuery . ' AS object )';
            $list = $this->model->with($this->relationModel)->where('id in' . $subQuery)->order($order, 'desc')->select()->toArray();

            foreach ($list as $key => $value) {
                if (isset($value['user_id'])) {
                    $list[$key]['user_id'] = $value['user']['nickname'] ?? $value['user_id'];
                }
                if (isset($value['admin_id'])) {
                    $list[$key]['admin_id'] = $value['admin']['nickname'] ?? $value['admin_id'];
                }
            }

            return $this->success('查询成功', '/', $list, $count);
        }

        return $this->view();
    }

    /**
     * 添加资源
     * @return Response|void
     */
    public function add()
    {
        if (request()->isPost()) {

            $post = $this->preRuleOutFields(\request()->post());
            if ($this->dataLimit) {
                $post[$this->dataLimitField] = get_admin_id();
            }

            $validate = $this->isValidate ? get_class($this->model) : $this->isValidate;
            $post = request_validate_rules($post, $validate, $this->scene);
            if (empty($post) || !is_array($post)) {
                return $this->error($post);
            }

            $this->status = $this->model->create($post);
            return $this->status ? $this->success() : $this->error();
        }

        return $this->view('', ['data' => $this->getTableFields()]);
    }

    /**
     * 编辑资源
     * @return Response|void
     */
    public function edit()
    {
        $id = input('id');
        $data = $this->model->where('id', $id)->findOrEmpty()->toArray();

        // 限制数据调用
        if (!$this->authService->SuperAdmin() && $this->dataLimit
            && in_array($this->dataLimitField, $this->model->getFields())) {
            if ($data[$this->dataLimitField] != get_admin_id()) {
                return $this->error('没有权限');
            }
        }

        if (request()->isPost()) {
            $post = $this->preRuleOutFields(\request()->post());
            $validate = $this->isValidate ? get_class($this->model) : $this->isValidate;
            $post = request_validate_rules($post, $validate, $this->scene);
            if (empty($post) || !is_array($post)) {
                return $this->error($post);
            }

            $this->status = $this->model->update($post);
            return $this->status ? $this->success() : $this->error();
        }

        /**
         * 默认共享模板
         */
        $template = str_replace('/_', '/', Str::snake(request()->getController()));
        return $this->view($template . '/add', [
            'data' => $data
        ]);
    }

    /**
     * 删除资源
     * @return Response
     */
    public function del()
    {
        $id = input('id');
        if (!is_array($id)) {
            $id = [$id];
        }

        try {
            $list = $this->model->whereIn('id', $id)->select();
            foreach ($list as $item) {
                if (!$this->authService->SuperAdmin() && $this->dataLimit
                    && in_array($this->dataLimitField, $this->model->getFields())) {
                    if ($item[$this->dataLimitField] != get_admin_id()) {
                        continue;
                    }
                }
                if (isset($item->isSystem) && $item->isSystem) {
                    throw new \Exception('禁止删除系统级数据');
                }

                $item->delete();
                $this->status = true;
            }
        } catch (\Throwable $th) {
            $this->status = false;
            return $this->error($th->getMessage());
        }

        return $this->status ? $this->success() : $this->error();
    }

    /**
     * 修改资源状态
     * @return Response|void
     */
    public function status()
    {
        if (request()->isAjax()) {

            $where[] = ['id', '=', input('id')];
            if (!$this->authService->SuperAdmin() && $this->dataLimit
                && in_array($this->dataLimitField, $this->model->getFields())) {
                $where[] = [$this->dataLimitField, '=', get_admin_id()];
            }

            try {
                $this->status = $this->model->where($where)->update(['status' => input('status')]);
            } catch (\Throwable $th) {
                return $this->error($th->getMessage());
            }

            if ($this->status) {
                return $this->success();
            }
        }

        return $this->error();
    }

    /**
     * 数据表排序
     * @return Response
     */
    public function sort()
    {
        if (request()->isPost()) {

            if (array_search('sort', $this->model->getTableFields())) {
                try {

                    $ids = request()->post('ids');
                    $list = $this->model->where('id', 'in', $ids)->orderRaw('field(id,' . implode(',', $ids) . ')')->select()->toArray();
                    $newSort = array_column($list, 'sort');
                    rsort($newSort);
                    $array = [];

                    // 循环处理排序字段
                    foreach ($list as $key => $value) {
                        $array[] = [
                            'id'   => $value['id'],
                            'sort' => $newSort[$key],
                        ];
                    }

                    $this->model->saveAll($array);
                } catch (\Throwable $th) {
                    return $this->error($th->getMessage());
                }

            } else {
                return $this->error('数据表未包含排序字段');
            }
        }

        return $this->success();
    }

    /**
     * 自动获取view模板
     * @param string $template
     * @param array $vars
     * @param null $app
     * @return Response
     */
    public function view(string $template = '', array $vars = [], $app = null): Response
    {
        $request = explode('/', \request()->getController());
        if (empty($template)) {
            $parseArr = array_map(function ($item) {
                return Str::snake($item);
            }, $request);
            $template = implode('/', $parseArr) . '/' . Str::snake(\request()->getAction());
        }

        return view($template, $vars, $app);
    }

    /**
     * 排除特定字段
     *
     * @param [type] $params
     * @return array
     */
    protected function preRuleOutFields($params): array
    {
        if (is_array($this->ruleOutFields)) {
            foreach ($this->ruleOutFields as $field) {
                if (key_exists($field, $params)) {
                    unset($params[$field]);
                }
            }
        } else {
            if (key_exists($this->ruleOutFields, $params)) {
                unset($params[$this->ruleOutFields]);
            }
        }

        return $params;
    }

    /**
     * 获取查询参数
     * @return array
     */
    protected function buildSelectParams(): array
    {
        $where = [];
        $params = request()->all();
        if (!empty($params) && is_array($params)) {

            $this->tableFields = $this->model->getFields();
            foreach ($params as $field => $value) {

                // 过滤字段
                if (in_array($field, $this->filterWhere)) {
                    continue;
                }

                // 非表内字段
                if (!array_key_exists($field, $this->tableFields)) {
                    continue;
                }

                // 默认状态字段
                if ($field == $this->keepField && $value) {
                    $where[] = [$field, '=', intval($value - 1)];
                    continue;
                }

                // 获取类型
                $type = $this->tableFields[$field]['type'];
                $type = explode('(', $type)[0];
                $value = str_replace('/\s+/', '', $value);
                switch ($type) {
                    case 'char':
                    case 'text':
                    case 'varchar':
                    case 'tinytext':
                    case 'longtext':
                        $where[] = [$field, 'like', '%' . $value . '%'];
                        break;
                    case 'int':
                    case 'bigint':
                    case 'integer':
                    case 'tinyint':
                    case 'smallint':
                    case 'mediumint':
                    case 'float':
                    case 'double':
                    case 'timestamp':
                    case 'year':
                        $value = str_replace(',', '-', $value);
                        if (strpos($value, '-')) {

                            $arr = explode(' - ', $value);
                            if (empty($arr)) {
                                continue 2;
                            }
                            if (in_array($field, $this->converTime)) {
                                if (isset($arr[0])) {
                                    $arr[0] = strtotime($arr[0]);
                                }
                                if (isset($arr[1])) {
                                    $arr[1] = strtotime($arr[1]);
                                }
                            }
                            $exp = 'between';
                            if ($arr[0] === '') {
                                $exp = '<=';
                                $arr = $arr[1];
                            } elseif ($arr[1] === '') {
                                $exp = '>=';
                                $arr = $arr[0];
                            }
                            $where[] = [$field, $exp, $arr];
                        } else {
                            $where[] = [$field, '=', $value];
                        }
                        break;
                    case 'set';
                        $where[] = [$field, 'find in set', $value];
                        break;
                    case 'enum';
                        $where[] = [$field, '=', $value];
                        break;
                    case 'date';
                    case 'time';
                    case 'datetime';
                        $value = str_replace(',', '-', $value);
                        if (strpos($value, '-')) {
                            $arr = explode(' - ', $value);
                            if (!array_filter($arr)) {
                                continue 2;
                            }
                            $exp = 'between';
                            if ($arr[0] === '') {
                                $exp = '<=';
                                $arr = $arr[1];
                            } elseif ($arr[1] === '') {
                                $exp = '>=';
                                $arr = $arr[0];
                            }
                            $where[] = [$field, $exp, $arr];
                        } else {
                            $where[] = [$field, '=', $value];
                        }
                        break;
                    case 'blob';
                        break;
                    default:
                        // 默认值
                        break;
                }
            }

            // 限制个人数据权限
            $superAdmin = $this->authService->SuperAdmin();
            if (!$superAdmin && $this->dataLimit) {
                if (in_array($this->dataLimitField, $this->tableFields)) {
                    $where[] = [$this->dataLimitField, '=', get_admin_id()];
                }
            } // 限制部门数据权限
            else if (!$superAdmin && $this->departmentLimit
                && in_array('department_id', $this->tableFields)) {
                $where[] = ['department_id', 'in', get_admin_info('AdminLogin.department_id')];
            }
        }

        return $where;
    }

    /**
     * 递归查询父节点
     * @access public
     * @param int $pid 查询条件
     * @param array $array 返回数组
     * @return array
     */
    public function parentNode(int $pid, array &$array = []): array
    {
        $result = $this->model->where('id', $pid)->find()->toArray();
        if (!empty($result)) {
            /**
             * 多语言字段
             */
            if (isset($result['title'])) {
                $result['title'] = __($result['title']);
            }

            $array[] = $result;
            if ($result['pid'] !== 0) {
                $this->parentNode($result['pid'], $array);
            }
        }

        return $array;
    }

    /**
     * 管理员退出
     * @return Response
     */
    public function logout(): Response
    {
        request()->session()->set(AdminEnum::ADMIN_SESSION, null);
        return $this->success('退出成功！', '/');
    }

    /**
     * 错误页面
     * @param int $code
     * @param string $msg
     * @return Response
     */
    public function abortPage(string $msg = '', int $code = 404): Response
    {
        $exception = config('app.exception_template');
        if (isset($exception[$code])) {
            $template = @file_get_contents($exception[$code]);
        } else {
            $template = $msg;
        }

        return \response($template, $code);
    }
}