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

namespace app\service\admin\sys;

use app\dict\sys\MenuDict;
use app\dict\sys\MenuTypeDict;
use app\model\sys\SysMenu;
use core\base\BaseAdminService;
use core\dict\DictLoader;
use core\exception\AdminException;
use think\db\exception\DataNotFoundException;
use think\db\exception\DbException;
use think\db\exception\ModelNotFoundException;
use think\facade\Cache;
use think\Model;

/**
 * 用户服务层
 * Class BaseService
 * @package app\service
 */
class MenuService extends BaseAdminService
{

    public static $cache_tag_name = 'menu_cache';

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

    /**
     * 新增菜单接口
     * @param array $data
     * @return SysMenu|Model
     */
    public function add(array $data)
    {
        $menu = $this->find($data[ 'menu_key' ]);
        if (!$menu->isEmpty()) throw new AdminException('validate_menu.exit_menu_key');//创建失败
        $data[ 'source' ] = MenuDict::CREATE;
        $res = ( new SysMenu() )->create($data);
        if (!$res) throw new AdminException('ADD_FAIL');//创建失败

        Cache::tag(self::$cache_tag_name)->clear();
        return $res;
    }


    /**
     * 更新菜单
     * @param string $menu_key
     * @param array $data
     * @return SysMenu
     */
    public function edit(string $menu_key, array $data)
    {
        $where = array(
            [ 'menu_key', '=', $menu_key ]
        );

        //校验菜单是否可以修改
        $res = ( new SysMenu() )->update($data, $where);
        Cache::tag(self::$cache_tag_name)->clear();
        return $res;
    }

    /**
     * 获取信息
     * @param string $menu_key
     * @return array
     */
    public function get(string $menu_key)
    {
        return ( new SysMenu() )->where([ [ 'menu_key', '=', $menu_key ] ])->findOrEmpty()->toArray();
    }

    /**
     * @param string $menu_key
     * @return SysMenu|array|mixed|Model
     */
    public function find(string $menu_key)
    {
        $where = array(
            [ 'menu_key', '=', $menu_key ]
        );
        $menu = ( new SysMenu() )->where($where)->findOrEmpty();
        return $menu;
    }

    /**
     * 菜单删除
     * @param string $menu_key
     * @return bool
     * @throws DbException
     */
    public function del(string $menu_key)
    {
        //查询是否有下级菜单或按钮
        $menu = $this->find($menu_key);
        if (( new SysMenu() )->where([ [ 'parent_key', '=', $menu_key ] ])->count() > 0)
            throw new AdminException('MENU_NOT_ALLOW_DELETE');

        $res = $menu->delete();
        Cache::tag(self::$cache_tag_name)->clear();
        return $res;
    }

    /**
     * 通过菜单menu_key获取
     * @param array $menu_keys
     * @param int $is_tree
     * @return mixed
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public function getMenuListByMenuKeys(array $menu_keys, int $is_tree = 0, $addon = 'all', $is_button = 1)
    {
        sort($menu_keys);
        $cache_name = 'menu' . md5(implode("_", $menu_keys)) . $is_tree . $addon;
        $menu_list = cache_remember(
            $cache_name,
            function() use ($menu_keys, $is_tree, $addon) {
                $where = [
                    [ 'menu_key', 'in', $menu_keys ],
                ];
                $addons = get_site_addons();
                $addons[] = '';

                $delete_menu_addon = [];
                $addon_loader = new DictLoader("Menu");

                if ($addon != 'all') {
                    $where[] = [ 'addon', '=', $addon ];

                    $delete_menu = $addon_loader->load([ "addon" => $addon, "app_type" => 'admin' ])[ 'delete' ] ?? [];
                    if (!empty($delete_menu) && is_array($delete_menu)) $delete_menu_addon[] = $delete_menu;
                } else {
                    $where[] = [ 'addon', 'in', $addons ];

                    foreach ($addons as $addon) {
                        $delete_menu = $addon_loader->load([ "addon" => $addon, "app_type" => 'admin' ])[ 'delete' ] ?? [];
                        if (!empty($delete_menu) && is_array($delete_menu)) $delete_menu_addon[] = $delete_menu;
                    }
                }

                // 排除插件中delete的菜单
                if (!empty($delete_menu_addon)) {
                    $delete_intersect = array_intersect(...$delete_menu_addon);
                    if (!empty($delete_intersect)) {
                        $where[] = [ 'menu_key', 'not in', $delete_intersect ];
                    }
                }
                if (!empty($app_type)) {
                    $where[] = [ 'app_type', '=', $app_type ];
                }
                return ( new SysMenu() )->where($where)->order('sort', 'desc')->select()->toArray();
            },
            self::$cache_tag_name
        );
        foreach ($menu_list as &$v) {
            $lang_menu_key = "dict_menu_" . $v[ 'app_type' ] . '.' . $v[ 'menu_key' ];
            $lang_menu_name = get_lang("dict_menu_" . $v[ 'app_type' ] . '.' . $v[ 'menu_key' ]);
            //语言已定义
            if ($lang_menu_key != $lang_menu_name) {
                $v[ 'menu_name' ] = $lang_menu_name;
            }
        }

        return $is_tree ? $this->menuToTree($menu_list, 'menu_key', 'parent_key', 'children', 'auth', '', $is_button) : $menu_list;

    }

    /**
     * 获取所有接口菜单
     */
    public function getAllMenuList($status = 'all', $is_tree = 0, $is_button = 0)
    {
        $cache_name = 'menu_api_' . $status . '_' . $is_tree . '_' . $is_button;
        $menu_list = cache_remember(
            $cache_name,
            function() use ($status, $is_tree, $is_button) {
                $where = [
                    [ 'addon', 'in', array_merge([ '' ], get_site_addons()) ]
                ];
                if ($status != 'all') {
                    $where[] = [ 'status', '=', $status ];
                }
                // 排除菜单
                $delete_menu_addon = [];
                $addon_loader = new DictLoader("Menu");
                foreach (get_site_addons() as $addon) {
                    $delete_menu = $addon_loader->load([ "addon" => $addon, "app_type" => 'admin' ])[ 'delete' ] ?? [];
                    if (!empty($delete_menu) && is_array($delete_menu)) $delete_menu_addon[] = $delete_menu;
                }
                if (!empty($delete_menu_addon)) {
                    $delete_intersect = array_intersect(...$delete_menu_addon);
                    if (!empty($delete_intersect)) {
                        $where[] = [ 'menu_key', 'not in', $delete_intersect ];
                    }
                }
                return ( new SysMenu() )->where($where)->order('sort desc')->select()->toArray();
            },
            self::$cache_tag_name
        );
        foreach ($menu_list as &$v) {
            $lang_menu_key = 'dict_menu_admin' . '.' . $v[ 'menu_key' ];
            $lang_menu_name = get_lang($lang_menu_key);
            //语言已定义
            if ($lang_menu_key != $lang_menu_name) {
                $v[ 'menu_name' ] = $lang_menu_name;
            }
        }

        return $is_tree ? $this->menuToTree($menu_list, 'menu_key', 'parent_key', 'children', 'auth', '', $is_button) : $menu_list;

    }


    /**
     * 通过菜单menu_key组获取接口数组
     * @param array $menu_keys
     * @return mixed|string
     */
    public function getApiListByMenuKeys(array $menu_keys)
    {
        sort($menu_keys);
        $cache_name = 'api' . md5(implode('_', $menu_keys));
        return cache_remember(
            $cache_name,
            function() use ($menu_keys) {
                $where = [
                    [ 'menu_key', 'in', $menu_keys ]
                ];
                $menu_list = ( new SysMenu() )->where($where)->order('sort', 'desc')->column('api_url,methods');
                foreach ($menu_list as $v) {
                    $auth_menu_list[ $v[ 'methods' ] ][] = $v[ 'api_url' ];
                }
                return $auth_menu_list ?? [];
            },
            self::$cache_tag_name
        );
    }


    /**
     * 通过菜单menu_key组获取按钮数组
     * @param array $menu_keys
     * @return mixed
     */
    public function getButtonListBuMenuKeys(array $menu_keys)
    {
        sort($menu_keys);
        $cache_name = 'button' . md5(implode('_', $menu_keys));
        return cache_remember(
            $cache_name,
            function() use ($menu_keys) {
                $where = [
                    [ 'menu_key', 'in', $menu_keys ],
                    [ 'menu_type', '=', MenuTypeDict::BUTTON ]
                ];
                return ( new SysMenu() )->where($where)->order('sort', 'desc')->column('menu_key');
            },
            self::$cache_tag_name
        );
    }

    /**
     * 获取所有接口菜单权限
     * @param $status
     * @return mixed
     */
    public function getAllApiList($status = 'all')
    {
        $cache_name = 'all_api' . '_' . $status;
        return cache_remember(
            $cache_name,
            function() use ($status) {
                $where = [
                    [ 'api_url', '<>', '' ],
                ];
                if ($status != 'all') {
                    $where[] = [ 'status', '=', $status ];
                }
                $menu_list = ( new SysMenu() )->where($where)->order('sort', 'desc')->column('methods, api_url');
                $auth_menu_list = [];
                foreach ($menu_list as $v) {
                    $auth_menu_list[ $v[ 'methods' ] ][] = $v[ 'api_url' ];
                }
                return $auth_menu_list;
            },
            self::$cache_tag_name
        );
    }

    /**
     * 通过站点端口获取菜单id
     * @param $status
     * @return mixed|string
     */
    public function getAllMenuIdsByAppType($status = 'all')
    {
        $cache_name = 'menu_id_cache';
        return cache_remember(
            $cache_name,
            function() use ($status) {
                $where = [
                ];
                if ($status != 'all') {
                    $where[] = [ 'status', '=', $status ];
                }
                return ( new SysMenu() )->where($where)->order('sort desc')->column('menu_key');
            },
            self::$cache_tag_name
        );
    }


    /**
     * 获取所有按钮菜单
     */
    public function getAllButtonList($status = 'all', $is_tree = 0)
    {
        $cache_name = 'menu_api_' . $status . '_' . $is_tree;
        return cache_remember(
            $cache_name,
            function() use ($status, $is_tree) {
                $where = [
                    [ 'menu_type', '=', MenuTypeDict::BUTTON ],
                ];
                if ($status != 'all') {
                    $where[] = [ 'status', '=', $status ];
                }
                return ( new SysMenu() )->where($where)->order('sort', 'desc')->column('menu_key');
            },
            self::$cache_tag_name
        );
    }

    /**
     * 把返回的数据集转换成Tree(专属于)
     * @param $list 要转换的数据集
     * @param string $pk
     * @param string $pid
     * @param string $child
     * @param int $root
     * @return array
     */
    public function menuToTree($list, $pk = 'id', $pid = 'pid', $child = 'child', $button_name = 'auth', $root = '', $is_button = 0)
    {
        // 创建Tree
        $tree = array();
        if (is_array($list)) {
            // 创建基于主键的数组引用
            $refer = array();
            foreach ($list as $key => $data) {
                $refer[ $data[ $pk ] ] =& $list[ $key ];
            }
            foreach ($list as $key => $data) {
                // 判断是否存在parent
                $parent_id = $data[ $pid ];
                if ($root == $parent_id) {
                    $tree[] =& $list[ $key ];
                } else {
                    if (isset($refer[ $parent_id ])) {
                        $parent =& $refer[ $parent_id ];
                        if ($list[ $key ][ 'menu_type' ] == 2 && $is_button == 1) {
                            $parent[ $button_name ][] =& $list[ $key ][ 'menu_key' ];
                        } else {
                            $parent[ $child ][] =& $list[ $key ];
                        }

                    }
                }
            }
        }
        return $tree;

    }

    /**
     * 获取完整的路由地址
     * @param $menu_key
     * @return string
     */
    public function getFullRouterPath($menu_key)
    {
        $menu = ( new SysMenu() )->where([ [ 'menu_key', '=', $menu_key ] ])->findOrEmpty($menu_key);
        if ($menu->isEmpty()) return '';
        $parents = [];
        $this->getParentDirectory($menu, $parents);
        $parents = array_reverse($parents);
        $router_path = implode('/', $parents);
        if (!empty($router_path)) {
            $router_path .= '/' . $menu[ 'router_path' ];
        } else {
            $router_path = $menu[ 'router_path' ];
        }
        return $router_path;
    }

    /**
     * 递归查询模板集合
     * @param SysMenu $menu
     * @param $parents
     * @return void
     */
    public function getParentDirectory(SysMenu $menu, &$parents)
    {
        if (!$menu->isEmpty() && !empty($menu[ 'parent_key' ])) {
            $parent_menu = ( new SysMenu() )->where([ [ 'menu_key', '=', $menu[ 'parent_key' ] ] ])->findOrEmpty();
            if (!empty($parent_menu)) {
                if (!empty($parent_menu[ 'router_path' ])) $parents[] = $parent_menu[ 'router_path' ];
                $this->getParentDirectory($parent_menu, $parents);
            }
        }

    }

    public function getSystemMenu($status = 'all', $is_tree = 0, $is_button = 0)
    {

        if ($is_button == 0) {
            $where = [
                [ 'menu_type', 'in', [ 0, 1 ] ]
            ];
        }

        if ($status != 'all') {
            $where[] = [ 'status', '=', $status ];
        }
        $where[] = [ 'addon', '=', '' ];
        $menu_list = ( new SysMenu() )->where($where)->order('sort desc')->select()->toArray();
        foreach ($menu_list as &$v) {
            $lang_menu_key = 'dict_menu_admin' . '.' . $v[ 'menu_key' ];
            $lang_menu_name = get_lang($lang_menu_key);
            //语言已定义
            if ($lang_menu_key != $lang_menu_name) {
                $v[ 'menu_name' ] = $lang_menu_name;
            }
        }
        return $is_tree ? $this->menuToTree($menu_list, 'menu_key', 'parent_key', 'children', 'auth', '', $is_button) : $menu_list;
    }

    public function getAddonMenu($app_key, $status = 'all', $is_tree = 0, $is_button = 0)
    {

        if ($is_button == 0) {
            $where = [
                [ 'menu_type', 'in', [ 0, 1 ] ]
            ];
        }

        if ($status != 'all') {
            $where[] = [ 'status', '=', $status ];
        }
        $where[] = [ 'addon', '=', $app_key ];
        $menu_list = ( new SysMenu() )->where($where)->select()->toArray();
        return $is_tree ? $this->menuToTree($menu_list, 'menu_key', 'parent_key', 'children', 'auth', '', $is_button) : $menu_list;
    }

    /**
     * 查询菜单类型为目录的菜单
     * @param string $addon
     * @return array
     */
    public function getMenuByTypeDir(string $addon = 'system')
    {
        $cache_name = 'menu_api_by_type_dir' . $addon;
        $menu_list = cache_remember(
            $cache_name,
            function() use ($addon) {
                $where = [
                    [ 'menu_type', '=', 0 ]
                ];
                //查询应用
                $where[] = [ 'addon', '=', $addon == 'system' ? '' : $addon ];
                return ( new SysMenu() )->where($where)->order('sort desc')->select()->toArray();
            },
            self::$cache_tag_name
        );
        foreach ($menu_list as &$v) {
            $lang_menu_key = 'dict_menu_admin' . '.' . $v[ 'menu_key' ];
            $lang_menu_name = get_lang($lang_menu_key);
            //语言已定义
            if ($lang_menu_key != $lang_menu_name) {
                $v[ 'menu_name' ] = $lang_menu_name;
            }
        }

        return $this->menuToTree($menu_list, 'menu_key', 'parent_key', 'children', 'auth', '', 0);
    }
}