LJH_Shop/niucloud/app/service/admin/site/SiteService.php

<?php
// +----------------------------------------------------------------------
// | Niucloud-admin 企业快速开发的saas管理平台
// +----------------------------------------------------------------------
// | 官方网址：https://www.niucloud.com
// +----------------------------------------------------------------------
// | niucloud团队 版权所有 开源版本可自由商用
// +----------------------------------------------------------------------
// | Author: Niucloud Team
// +----------------------------------------------------------------------

namespace app\service\admin\site;

use app\dict\addon\AddonDict;
use app\dict\site\SiteDict;
use app\dict\sys\AppTypeDict;
use app\model\addon\Addon;
use app\model\site\Site;
use app\model\site\SiteGroup;
use app\model\sys\SysUserRole;
use app\service\admin\addon\AddonService;
use app\service\admin\auth\AuthService;
use app\service\admin\generator\GenerateService;
use app\service\admin\sys\MenuService;
use app\service\admin\sys\RoleService;
use app\service\admin\user\UserRoleService;
use app\service\admin\user\UserService;
use app\service\core\site\CoreSiteService;
use core\base\BaseAdminService;
use core\exception\AdminException;
use core\exception\CommonException;
use Exception;
use think\db\exception\DataNotFoundException;
use think\db\exception\DbException;
use think\db\exception\ModelNotFoundException;
use think\facade\Cache;
use think\facade\Db;

/**
 * 站点服务层
 * Class Site
 * @package app\service\admin\site
 */
class SiteService extends BaseAdminService
{
    public static $cache_tag_name = 'site_cash';

    public function __construct()
    {
        parent::__construct();
        $this->model = new Site();
    }

    /**
     * 获取站点列表
     * @param array $where
     * @return array
     * @throws DbException
     */
    public function getPage(array $where = [])
    {

        $field = 'site_id, site_name, front_end_name, front_end_logo, app_type, keywords, logo, icon, `desc`, status, latitude, longitude, province_id, city_id, 
        district_id, address, full_address, phone, business_hours, create_time, expire_time, group_id, app, addons, site_domain';
        $condition = [
            [ 'app_type', '<>', 'admin' ]
        ];
        $search_model = $this->model->where($condition)->withSearch([ 'create_time', 'expire_time', 'keyword', 'status', 'group_id', 'app', 'site_domain' ], $where)->with(['groupName'])->field($field)->append([ 'status_name' ])->order('create_time desc');
        return $this->pageQuery($search_model, function ($item){
            $item['admin'] = (new SysUserRole())->where([ ['site_id', '=', $item['site_id'] ], ['is_admin', '=', 1] ])
                ->field('uid')
                ->with(['userinfo'])
                ->find()->toArray();
        });
    }

    /**
     * 站点信息
     * @param int $site_id
     * @return array
     */
    public function getInfo(int $site_id)
    {
        $field = 'site_id, site_name, front_end_name, front_end_logo, app_type, keywords, logo, icon, `desc`, status, latitude, longitude, province_id, city_id, 
        district_id, address, full_address, phone, business_hours, create_time, expire_time, group_id, app, addons, site_domain';
        $info = $this->model->where([ [ 'site_id', '=', $site_id ] ])->with([ 'groupName' ])->field($field)->append([ 'status_name' ])->findOrEmpty()->toArray();
        if (!empty($info)) {
            $site_addons = (new CoreSiteService())->getAddonKeysBySiteId($site_id);
            $info['site_addons'] = (new Addon())->where([ ['key', 'in', $site_addons]])->field('key,title,desc,icon,type')->select()->toArray();
        }
        return $info;
    }

    /**
     * 添加站点(平台端添加站点，同时添加用户以及密码)
     * @param array $data
     * ['site_name' => '', 'username' => '', 'head_img' => '', 'real_name' => '', 'password' => '']
     * @return mixed
     * @throws DbException
     */
    public function add(array $data)
    {
        $user_service = new UserService();
        if ($user_service->checkUsername($data[ 'username' ])) throw new AdminException('USERNAME_REPEAT');

        $site_group = (new SiteGroup())->where([ ['group_id', '=', $data[ 'group_id' ] ] ])->field('app,addon')->findOrEmpty();

        $data[ 'app_type' ] = 'site';
        //添加站点
        $data_site = [
            'site_name' => $data[ 'site_name' ],
            'app_type' => $data[ 'app_type' ],
            'group_id' => $data[ 'group_id' ],
            'create_time' => time(),
            'expire_time' => $data[ 'expire_time' ],
            'app' => $site_group['app'],
            'addons' => ''
        ];
        Db::startTrans();
        try {
            $site = $this->model->create($data_site);
            $site_id = $site->site_id;

            if ($data['uid']) {
                (new UserRoleService())->add($data['uid'], ['role_ids' => '', 'is_admin' => 1], $site_id);
            } else {
                //添加用户
                $data_user = [
                    'username' => $data[ 'username' ],
                    'head_img' => $data[ 'head_img' ] ?? '',
                    'status' => $data[ 'status' ] ?? 1,
                    'real_name' => $data[ 'real_name' ] ?? '',
                    'password' => $data[ 'password' ],
                    'role_ids' => '',
                    'is_admin' => 1
                ];
                $data['uid'] = ( new UserService() )->addSiteUser($data_user, $site_id);
            }

            //添加站点成功事件
            event("AddSiteAfter", [ 'site_id' => $site_id, 'main_app' => $site_group['app'], 'site_addons' => $site_group['addon'] ]);

            Cache::delete('user_role_list_' . $data['uid']);

            Db::commit();
            return $site_id;
        } catch ( Exception $e) {
            Db::rollback();
            throw new AdminException($e->getMessage().$e->getFile().$e->getLine());
        }
    }

    /**
     * 修改站点
     * @param int $site_id
     * @param array $data
     * @return bool
     */
    public function edit(int $site_id, array $data)
    {
        $site = $this->model->where([ [ 'site_id', '=', $site_id ] ])->with(['site_group'])->findOrEmpty();
        if ($site->isEmpty()) throw new AdminException('SITE_NOT_EXIST');

        Db::startTrans();
        try {
            if (isset($data[ 'group_id' ]) && $site['group_id'] != $data[ 'group_id' ]) {
                $old_site_group = $site['site_group'];

                $site_group = (new SiteGroup())->where([ ['group_id', '=', $data[ 'group_id' ] ] ])->field('app,addon')->findOrEmpty();
                $data['app'] = $site_group['app'];

                if (empty($site->initalled_addon)) {
                    $site->initalled_addon = array_merge($old_site_group['app'], $old_site_group['addon']);
                }
                //添加站点成功事件
                event("AddSiteAfter", [ 'site_id' => $site_id, 'main_app' => array_diff($site_group['app'], $site->initalled_addon) , 'site_addons' => array_diff($site_group['addon'], $site->initalled_addon) ]);

                $data['initalled_addon'] = array_values(array_unique(array_merge($site->initalled_addon, $site_group['app'], $site_group['addon'])));
            }

            if (isset($data['expire_time']) && $site['status'] != SiteDict::CLOSE) {
                $data['status'] = strtotime($data['expire_time']) > time() ? SiteDict::ON : SiteDict::EXPIRE;
            }
            $site->save($data);

            Cache::tag(self::$cache_tag_name . $site_id)->clear();
            Db::commit();
            return true;
        } catch ( Exception $e) {
            Db::rollback();
            throw new AdminException($e->getMessage().$e->getFile().$e->getLine());
        }
    }

    /**
     * 删除站点
     * @param int $site_id
     */
    public function del(int $site_id) {
        Db::startTrans();
        try {
            $site = $this->model->where([ [ 'site_id', '=', $site_id ] ])->findOrEmpty()->toArray();

            // 删除站点相关数据
            $sys_models = (new GenerateService())->getModels(['addon' => 'system']);
            $addon_models = [];
            $addons = (new CoreSiteService())->getAddonKeysBySiteId($site_id);
            foreach($addons as $addon) {
                $addon_models[] = (new GenerateService())->getModels(['addon' => $addon ]);
            }
            $models = array_merge($sys_models, ...$addon_models);

            foreach ($models as $model) {
                $name = "\\$model";
                $class = new $name();

                if (in_array('site_id', $class->getTableFields())) {
                    $class->where([ ['site_id', '=', $site['site_id'] ] ])->delete();
                }
            }

            Cache::tag(self::$cache_tag_name . $site_id)->clear();
            Db::commit();
            return true;
        } catch (\Exception $e) {
            Db::rollback();
            throw new CommonException($e->getMessage());
        }
    }

    /**
     * 站点数量
     * @return int
     * @throws DbException
     */
    public function getCount(array $where = [])
    {
        $condition = [];
        if(!empty($where['group_id'])){
            $condition[] = ['group_id', '=', $where['group_id']];
        }
        if(!empty($where['create_time'])){
            $condition[] = ['create_time', 'between', $where['create_time']];
        }
        return $this->model->where($condition)->count();
    }


    /**
     * 获取授权当前站点信息(用做缓存)
     * @return mixed
     */
    public function getSiteCache(int $site_id)
    {
        return (new CoreSiteService())->getSiteCache($site_id);
    }


    /**
     * 通过站点id获取菜单列表
     * @param int $site_id
     * @param $is_tree
     * @param $status
     * @param $addon   所以应用名一般不建议叫all
     * @return mixed
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function getMenuList(int $site_id, $is_tree, $status, $addon = 'all', int $is_button = 1)
    {
        $site_info = $this->getSiteCache($site_id);
        if (empty($site_info))
            return [];
        $app_type = $site_info[ 'app_type' ];
//        if ($app_type == AppTypeDict::ADMIN) {
//            return ( new MenuService() )->getAllMenuList($app_type, $status, $is_tree, $is_button);
//        } else {
//            $addons = ( new AddonService() )->getAddonKeysBySiteId($site_id);
//            $addons[] = '';
//            if($addon != 'all'){
//                $addons = [$addon];
//            }
//            return ( new MenuService() )->getMenuListBySystem($this->app_type, $addons, $is_tree, $is_button, $status);
//        }

        if (AuthService::isSuperAdmin()) {
            $is_admin = 1;
        } else {
            $user_role_info = (new AuthService())->getAuthRole($this->site_id);
            if(empty($user_role_info))
                return [];
            $is_admin = $user_role_info['is_admin'];//是否是超级管理员组
        }

        if ($is_admin) {
            return ( new MenuService() )->getAllMenuList($app_type, $status, $is_tree, $is_button);
        } else {
            $user_role_ids = $user_role_info['role_ids'];
            $role_service = new RoleService();
            $menu_keys = $role_service->getMenuIdsByRoleIds($this->site_id, $user_role_ids);
            return ( new MenuService() )->getMenuListByMenuKeys($this->site_id, $menu_keys, $this->app_type, $is_tree, $addon, $is_button);
        }
    }

    /**
     * 通过站点id获取站点菜单极限
     * @param int $site_id
     * @param $status
     * @return array|mixed|string|null
     */
    public function getMenuIdsBySiteId(int $site_id, $status)
    {
        $site_info = $this->getSiteCache($site_id);
        if (empty($site_info))
            return [];
        $app_type = $site_info[ 'app_type' ];
        if ($app_type == AppTypeDict::ADMIN) {
            return ( new MenuService() )->getAllMenuIdsByAppType($app_type, $status);
        } else {

            $addons = ( new AddonService() )->getAddonKeysBySiteId($site_id);
            return ( new MenuService() )->getMenuKeysBySystem($app_type, $addons);

        }
    }

    /**
     * 通过站点id获取菜单列表
     * @param int $site_id
     * @param $status
     * @return mixed
     */
    public function getApiList(int $site_id, $status)
    {
        $site_info = $this->getSiteCache($site_id);
        if (empty($site_info))
            return [];
        $app_type = $site_info[ 'app_type' ];
        if ($app_type == AppTypeDict::ADMIN) {
            return ( new MenuService() )->getAllApiList($app_type, $status);
        } else {
            $addons = ( new AddonService() )->getAddonKeysBySiteId($site_id);
            return ( new MenuService() )->getApiListBySystem($app_type, $addons);
        }
    }

    /**
     * 站点过期时间
     * @param int $site_id
     * @return array
     */
    public function getExpireTime(int $site_id)
    {
        $field = 'expire_time';
        return $this->model->where([ [ 'site_id', '=', $site_id ] ])->field($field)->findOrEmpty()->toArray();

    }

    /**
     * 获取站点的插件
     * @return array
     */
    public function getSiteAddons(array $where) {
        $site_addon = (new CoreSiteService())->getAddonKeysBySiteId($this->site_id);
        return (new Addon())->where([['type', '=', AddonDict::ADDON], ['status', '=', AddonDict::ON], ['key', 'in', $site_addon ]])->withSearch(['title'], $where)->append(['status_name'])->field('title, icon, key, desc, status, type, support_app')->select()->toArray();
    }
}