From 5e6c05ea592e10f40be81691d2b561b000fe28bf Mon Sep 17 00:00:00 2001
From: devlike <devlike@163.com>
Date: Thu, 8 Feb 2018 11:06:22 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=B0=E5=BD=95=E5=BD=93=E5=89=8D=E7=BD=91?=
 =?UTF-8?q?=E7=AB=99=E7=9A=84ID?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 extend/seven/behavior/GetSiteId.php | 35 +++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 extend/seven/behavior/GetSiteId.php

diff --git a/extend/seven/behavior/GetSiteId.php b/extend/seven/behavior/GetSiteId.php
new file mode 100644
index 00000000..1f7e4e6b
--- /dev/null
+++ b/extend/seven/behavior/GetSiteId.php
@@ -0,0 +1,35 @@
+<?php
+/**
+ * @desc: 获取管理员能管理的站点
+ * @Author: Dr.Xing <ezfwork@foxmail.com>
+ * @Since: 2018/2/1 21:43
+ */
+
+namespace seven\behavior;
+
+use think\Session;
+
+class GetSiteId
+{
+    public function run()
+    {
+        $siteId = request()->param('user_site_id');
+        if(!is_null($siteId)) {
+            Session::set("user_site_id", $siteId);
+            return;
+        }
+        $auth =\app\admin\library\Auth::instance();
+        if($auth->isSuperAdmin()){
+            Session::delete("user_site_id");
+            return;
+        }
+        $user = $auth->getUserInfo();
+        $site = \app\common\model\Sites::get(['user_id'=>$user['id']]); //\think\Db::name('sites')->where('user_id', $user['id'])->find();
+        if($site!=false) {
+            Session::set("user_site_id", $site['id']);
+        }else{
+            Session::set("user_site_id", -1);
+        }
+    }
+
+}
\ No newline at end of file