代码更新

app/middleware/AdminAuth.php

@@ -8,21 +8,81 @@
 // +----------------------------------------------------------------------
 namespace app\middleware;
 
-use think\facade\View;
+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{
+		} 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);
+	}
 }