From 70c4b53498956c247d34f25dd40d2ea6433f8c91 Mon Sep 17 00:00:00 2001
From: niu <xyniu623@hotmail.com>
Date: Tue, 3 Dec 2024 18:32:16 +0800
Subject: [PATCH] =?UTF-8?q?improvement:=E6=94=B9=E8=BF=9B=E4=BA=86?=
 =?UTF-8?q?=E5=8D=95=E4=B8=80=E7=A7=9F=E6=88=B7=E7=9A=84=E4=BD=BF=E7=94=A8?=
 =?UTF-8?q?=EF=BC=8C=E4=BF=AE=E5=A4=8D=E4=BA=86=E5=A4=9A=E7=A7=9F=E6=88=B7?=
 =?UTF-8?q?=E7=94=A8=E6=88=B7=E7=95=8C=E9=9D=A2=E5=90=8D=E7=A7=B0=E4=B8=8D?=
 =?UTF-8?q?=E5=AF=B9=E9=BD=90=E7=9A=84=E6=A0=B7=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Areas/Admin/Controllers/UserController.cs | 67 +++++++++++++++++--
 NewLife.CubeNC/Views/ACE/_Navbar.cshtml       |  2 +-
 NewLife.CubeNC/Views/Shared/_Form_Item.cshtml | 17 +++++
 3 files changed, 80 insertions(+), 6 deletions(-)

diff --git a/NewLife.CubeNC/Areas/Admin/Controllers/UserController.cs b/NewLife.CubeNC/Areas/Admin/Controllers/UserController.cs
index 7b43423f..c9b47aae 100644
--- a/NewLife.CubeNC/Areas/Admin/Controllers/UserController.cs
+++ b/NewLife.CubeNC/Areas/Admin/Controllers/UserController.cs
@@ -26,7 +26,7 @@ namespace NewLife.Cube.Areas.Admin.Controllers;
 [DisplayName("用户")]
 [Description("系统基于角色授权，每个角色对不同的功能模块具备添删改查以及自定义权限等多种权限设定。")]
 [AdminArea]
-[Menu(100, true, Icon = "fa-user", HelpUrl = "https://newlifex.com/cube/cube_security")]
+[Menu(100, true, Icon = "fa-user", HelpUrl = "https://newlifex.com/cube/cube_security", Mode = MenuModes.Admin | MenuModes.Tenant)]
 public class UserController : EntityController<User, UserModel>
 {
     /// <summary>用于防爆破登录。即使内存缓存，也有一定用处，最糟糕就是每分钟重试次数等于集群节点数的倍数</summary>
@@ -198,7 +198,41 @@ protected override IEnumerable<User> Search(Pager p)
 
         //if (roleId > 0) roleIds.Add(roleId);
         //if (departmentId > 0) departmentIds.Add(departmentId);
-        var list2 = XCode.Membership.User.Search(roleIds, departmentIds, areaIds, enable, start, end, key, p);
+        IList<User> list2 = new List<User>();
+
+        var tencentId = ManagerProviderHelper.GetTenantId(HttpContext);
+
+        if (tencentId > 0)//只读取租户相关的用户
+        {
+            var exp = TenantUser._.TenantId == tencentId;
+            if (roleIds != null && roleIds.Length > 0)
+            {
+                //exp &= _.RoleID.In(roleIds) | _.RoleIds.Contains("," + roleIds.Join(",") + ",");
+                var exp2 = new WhereExpression();
+                exp2 |= _.RoleID.In(roleIds);
+                foreach (var rid in roleIds)
+                {
+                    exp2 |= _.RoleIds.Contains("," + rid + ",");
+                }
+                exp &= exp2;
+            }
+            if (departmentIds != null && departmentIds.Length > 0) exp &= _.DepartmentID.In(departmentIds);
+            if (areaIds != null && areaIds.Length > 0) exp &= _.AreaId.In(areaIds);
+            if (enable != null) exp &= _.Enable == enable.Value;
+            exp &= _.LastLogin.Between(start, end);
+            if (!key.IsNullOrEmpty()) exp &= _.Code.StartsWith(key) | _.Name.StartsWith(key) | _.DisplayName.StartsWith(key) | _.Mobile.StartsWith(key) | _.Mail.StartsWith(key);
+
+            //var where= exp.ToString();
+            var sql = $"SELECT User.* FROM User INNER JOIN TenantUser ON User.ID= TenantUser.UserId where {exp.ToString()} ";
+
+            list2 = TenantUser.Meta.Session.Dal.Query<User>(sql, null, p)?.ToList();
+            //var sb = new SelectBuilder { Table = table.Name, Where = "Age>18" };
+        }
+        else
+        {
+            list2 = XCode.Membership.User.Search(roleIds, departmentIds, areaIds, enable, start, end, key, p);
+        }
+
 
         foreach (var user in list2)
         {
@@ -226,10 +260,13 @@ protected override Boolean Valid(User entity, DataObjectMethodType type, Boolean
         {
             // 非系统管理员，禁止修改任何人的角色
             var user = ManageProvider.User;
-            if (!user.Roles.Any(e => e.IsSystem) && entity is IEntity entity2)
+            if (TenantContext.CurrentId == 0)//非租户验证
             {
-                if (entity2.Dirtys["RoleID"]) throw new Exception("禁止修改角色！");
-                if (entity2.Dirtys["RoleIds"]) throw new Exception("禁止修改角色！");
+                if (!user.Roles.Any(e => e.IsSystem) && entity is IEntity entity2)
+                {
+                    if (entity2.Dirtys["RoleID"]) throw new Exception("禁止修改角色！");
+                    if (entity2.Dirtys["RoleIds"]) throw new Exception("禁止修改角色！");
+                }
             }
         }
 
@@ -786,6 +823,26 @@ public ActionResult TenantSetting()
         return View(model);
     }
 
+    protected override Int32 OnInsert(User entity)
+    {
+        var ef = base.OnInsert(entity);
+
+        if (TenantContext.CurrentId > 0)//默认插入当前租户下的用户
+        {
+            var tu = new TenantUser
+            {
+                TenantId = TenantContext.CurrentId,
+                UserId = entity.ID,
+                CreateIP = entity.RegisterIP,
+                Enable = entity.Enable,
+
+            };
+            tu.InsertAsync();
+        }
+
+        return ef;
+    }
+
     /// <summary>租户设置</summary>
     /// <param name="model"></param>
     /// <returns></returns>
diff --git a/NewLife.CubeNC/Views/ACE/_Navbar.cshtml b/NewLife.CubeNC/Views/ACE/_Navbar.cshtml
index fefecd89..11d72e2b 100644
--- a/NewLife.CubeNC/Views/ACE/_Navbar.cshtml
+++ b/NewLife.CubeNC/Views/ACE/_Navbar.cshtml
@@ -69,7 +69,7 @@
                         {
                             <img class="nav-user-photo hidden-xs" src="@user.GetAvatarUrl()" alt="@user" />
                         }
-                        <span class="user-info hidden-xs">
+                        <span class="user-info hidden-xs" style="line-height: 30px;">
                             @user<br />
                         </span>
 
diff --git a/NewLife.CubeNC/Views/Shared/_Form_Item.cshtml b/NewLife.CubeNC/Views/Shared/_Form_Item.cshtml
index 74bbf357..4589e053 100644
--- a/NewLife.CubeNC/Views/Shared/_Form_Item.cshtml
+++ b/NewLife.CubeNC/Views/Shared/_Form_Item.cshtml
@@ -50,6 +50,23 @@
     @if (dataSource != null)
     {
         var ds = dataSource(entity);
+
+        if (TenantContext.CurrentId > 0)
+        {
+            var roles = Tenant.FindById(TenantContext.CurrentId)?.RoleIds.SplitAsInt();
+            if (roles == null && roles.Length == 0)//强制把系统权限清空了
+            {
+                ds.Clear();
+            }
+            else//按照已有的角色组进行分配
+            {
+                if (ds is IDictionary<Int32, String> odic)
+                {
+                    ds = odic.Where(kvp => roles.Contains(kvp.Key)).ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
+                }
+            }
+        }
+
         if (ds is IDictionary<Int32, String> dic)
         {
             if (!ds.Contains(-1))