<?php
// +----------------------------------------------------------------------
// | SentCMS [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2013 http://www.tensent.cn All rights reserved.
// +----------------------------------------------------------------------
// | Author: molong <molong@tensent.cn> <http://www.tensent.cn>
// +----------------------------------------------------------------------
namespace app\middleware;

use think\facade\Cache;
use think\facade\Session;
use think\facade\View;

/**
 * @title 后台中间件
 */
class AdminAuth {

	protected $data = ['headerMenu' => [], 'asideMenu' => []];

	public function handle($request, \Closure $next) {
		$user = Session::get('user');

		if (Session::has('user') && $user['uid']) {
			$request->user = $user;
			if ($user['uid'] == 1) {
				$request->isAdmin = true;
			}

			$this->getMenu($request); //设置菜单
			return $next($request);
		} else {
			return redirect(url('admin.index/login'))->remember();
		}
	}

	/**
	 * @title 显示菜单
	 */
	protected function getMenu($request) {
		$current_controller = str_replace('.', '/', strtolower($request->controller()));
		$current_url        = $current_controller . '/' . strtolower($request->action());
		$menu               = Cache::get('menu');
		if (!$menu) {
			$dao  = new \app\model\Menu();
			$res  = $dao->where('is_menu', 1)->select();
			$menu = $res->toArray();
			Cache::set('menu', $menu);
		}
		$current_pid = 0;
		foreach ($menu as $key => $value) {
			if (strpos($value['url'], $current_controller) !== false) {
				if ($value['pid'] == 0) {
					$current_pid = $value['id'];
				} else {
					$current_pid = $value['pid'];
				}
			}
			if ($request->isAdmin || in_array($value['id'], array())) {
				$list[$value['id']] = $value;
			}
		}
		$headerMenu = list_to_tree($list);
		foreach ($headerMenu as $key => $value) {
			if ($value['id'] == $current_pid) {
				$value['active'] = true;
				$current_aside   = $value['_child'];
			} else {
				$value['active'] = false;
			}
			$headerMenu[$key] = $value;
		}

		foreach ($current_aside as $key => $value) {
			if ($current_url == $value['url']) {
				$value['active'] = true;
			} else {
				$value['active'] = false;
			}
			$asideMenu[$value['group']][] = $value;
		}

		$this->data['asideMenu']  = $asideMenu;
		$this->data['headerMenu'] = $headerMenu;
		View::assign($this->data);
	}
}