优化插件，完善TreeListView

alielie

2025-07-04 0940f31b7eba1cf6b6a8c1fca72d97cf686e79f2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
using HMI.Core.Entity;
using HMI.Core.Util;
using HMI.MenuManage.Entity;
using HMI.MenuManage.Util;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Data;
using System.IO;
 
namespace HMI.MenuManage.Model
{
    public class RoleManageModel : NotifyBase
    {
        public SqliteHelper sqliteHelper;
 
        private ObservableCollection<RoleData> _DataSource;
 
        public ObservableCollection<RoleData> DataSource
        {
            get { return _DataSource; }
            set { Set(ref _DataSource, value); }
        }
 
        private RoleData _SelectedValue;
        public RoleData SelectedValue
        {
            get => _SelectedValue;
            set
            {
                Set(ref _SelectedValue, value);
                LoadRelData(value);
            }
        }
 
        private ObservableCollection<RoleMenuRel> _TreeDataSource;
 
        public ObservableCollection<RoleMenuRel> TreeDataSource
        {
            get { return _TreeDataSource; }
            set { Set(ref _TreeDataSource, value); }
        }
 
        public RoleManageModel()
        {
            if (!File.Exists(PathHelper.DatabasePath))
            {
                SqliteHelper.CreateDatabase(PathHelper.DatabasePath);
            }
            sqliteHelper = new SqliteHelper(PathHelper.DatabasePath);
            sqliteHelper.ExecuteSql(@"CREATE TABLE IF NOT EXISTS role (
                    id INTEGER PRIMARY KEY AUTOINCREMENT,
                    code TEXT NOT NULL,
                    name TEXT,
                    UNIQUE(code)
                    );
CREATE TABLE IF NOT EXISTS role_menu_rel (
                    id INTEGER PRIMARY KEY AUTOINCREMENT,
                    roleCode TEXT NOT NULL,
                    menuCode TEXT NOT NULL,
                    UNIQUE(roleCode,menuCode)
                    );
");
        }
 
        public void LoadData()
        {
            ObservableCollection<RoleData> data = new ObservableCollection<RoleData>();
            DataTable dt = sqliteHelper.GetDataTable("SELECT * FROM role Order By id;");
            if (dt != null)
            {
                foreach (DataRow dr in dt.Rows)
                {
                    var menu = new RoleData
                    {
                        Id = dr["id"].ToString(),
                        Code = dr["code"].ToString(),
                        Name = dr["name"].ToString()
                    };
                    data.Add(menu);
                }
            }
            DataSource = data;
        }
 
        public void LoadRelData(RoleData data)
        {
            if (data == null)
            {
                TreeDataSource.Clear();
                return;
            }
            TreeDataSource = RoleMenuHelper.SelectRoleMenuTreeData(sqliteHelper, data.Code);
        }
 
        public void DelSingleData(RoleData data)
        {
            sqliteHelper.ExecuteBatchSql(new List<string>{
            $"DELETE FROM role WHERE id = '{data.Id}'",
            $"DELETE FROM role_menu_rel WHERE roleCode = '{data.Code}'",
            $"DELETE FROM user_role_rel WHERE roleCode = '{data.Code}'"
            });
        }
 
        public void DeleteRoleByCodes(IEnumerable<string> codes)
        {
            List<string> sqls = new List<string>();
            foreach (string code in codes)
            {
                sqls.Add($"DELETE FROM role WHERE code = '{code}';");
                sqls.Add($"DELETE FROM role_menu_rel WHERE roleCode = '{code}';");
                sqls.Add($"DELETE FROM user_role_rel WHERE roleCode = '{code}';");
            }
            sqliteHelper.ExecuteBatchSql(sqls);
        }
 
        public void SaveRelData(RoleData role)
        {
            if (role == null) return;
            RoleMenuHelper.SaveRelData(sqliteHelper, TreeDataSource, role.Code, null, null);
        }
    }
}