forked from GoAdminGroup/go-admin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
admin.go
140 lines (120 loc) · 3.91 KB
/
admin.go
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package admin
import (
"github.com/wowucco/go-admin/modules/config"
"github.com/wowucco/go-admin/modules/service"
"github.com/wowucco/go-admin/plugins"
"github.com/wowucco/go-admin/plugins/admin/controller"
"github.com/wowucco/go-admin/plugins/admin/models"
"github.com/wowucco/go-admin/plugins/admin/modules/guard"
"github.com/wowucco/go-admin/plugins/admin/modules/table"
"github.com/wowucco/go-admin/template/types"
_ "github.com/wowucco/go-admin/template/types/display"
)
// Admin is a GoAdmin plugin.
type Admin struct {
*plugins.Base
tableList table.GeneratorList
guardian *guard.Guard
handler *controller.Handler
}
// InitPlugin implements Plugin.InitPlugin.
// TODO: find a better way to manage the dependencies
func (admin *Admin) InitPlugin(services service.List) {
// DO NOT DELETE
admin.InitBase(services)
c := config.GetService(services.Get("config"))
st := table.NewSystemTable(admin.Conn, c)
admin.tableList.Combine(table.GeneratorList{
"manager": st.GetManagerTable,
"permission": st.GetPermissionTable,
"roles": st.GetRolesTable,
"op": st.GetOpTable,
"menu": st.GetMenuTable,
"normal_manager": st.GetNormalManagerTable,
"site": st.GetSiteTable,
})
admin.guardian = guard.New(admin.Services, admin.Conn, admin.tableList)
handlerCfg := controller.Config{
Config: c,
Services: services,
Generators: admin.tableList,
Connection: admin.Conn,
}
admin.handler.UpdateCfg(handlerCfg)
admin.initRouter()
admin.handler.SetRoutes(admin.App.Routers)
admin.handler.AddNavButton(admin.UI.NavButtons...)
// init site setting
site := models.Site().SetConn(admin.Conn)
site.Init(c.ToMap())
_ = c.Update(site.AllToMap())
table.SetServices(services)
}
// NewAdmin return the global Admin plugin.
func NewAdmin(tableCfg ...table.GeneratorList) *Admin {
return &Admin{
tableList: make(table.GeneratorList).CombineAll(tableCfg),
Base: &plugins.Base{PlugName: "admin"},
handler: controller.New(),
}
}
// SetCaptcha set captcha driver.
func (admin *Admin) SetCaptcha(captcha map[string]string) *Admin {
admin.handler.SetCaptcha(captcha)
return admin
}
// AddGenerator add table model generator.
func (admin *Admin) AddGenerator(key string, g table.Generator) *Admin {
admin.tableList.Add(key, g)
return admin
}
// AddGenerators add table model generators.
func (admin *Admin) AddGenerators(gen ...table.GeneratorList) *Admin {
admin.tableList.CombineAll(gen)
return admin
}
// AddGlobalDisplayProcessFn call types.AddGlobalDisplayProcessFn
func (admin *Admin) AddGlobalDisplayProcessFn(f types.DisplayProcessFn) *Admin {
types.AddGlobalDisplayProcessFn(f)
return admin
}
// AddDisplayFilterLimit call types.AddDisplayFilterLimit
func (admin *Admin) AddDisplayFilterLimit(limit int) *Admin {
types.AddLimit(limit)
return admin
}
// AddDisplayFilterTrimSpace call types.AddDisplayFilterTrimSpace
func (admin *Admin) AddDisplayFilterTrimSpace() *Admin {
types.AddTrimSpace()
return admin
}
// AddDisplayFilterSubstr call types.AddDisplayFilterSubstr
func (admin *Admin) AddDisplayFilterSubstr(start int, end int) *Admin {
types.AddSubstr(start, end)
return admin
}
// AddDisplayFilterToTitle call types.AddDisplayFilterToTitle
func (admin *Admin) AddDisplayFilterToTitle() *Admin {
types.AddToTitle()
return admin
}
// AddDisplayFilterToUpper call types.AddDisplayFilterToUpper
func (admin *Admin) AddDisplayFilterToUpper() *Admin {
types.AddToUpper()
return admin
}
// AddDisplayFilterToLower call types.AddDisplayFilterToLower
func (admin *Admin) AddDisplayFilterToLower() *Admin {
types.AddToUpper()
return admin
}
// AddDisplayFilterXssFilter call types.AddDisplayFilterXssFilter
func (admin *Admin) AddDisplayFilterXssFilter() *Admin {
types.AddXssFilter()
return admin
}
// AddDisplayFilterXssJsFilter call types.AddDisplayFilterXssJsFilter
func (admin *Admin) AddDisplayFilterXssJsFilter() *Admin {
types.AddXssJsFilter()
return admin
}