// add middleware to server
func ServeMiddleware(ctx *cli.Context) {
vars.Server.Use(base.LoggingHandler())
vars.Server.Use(base.Recovery(true))
for prefix, path := range vars.StaticDirectory {
vars.Server.Use(tango.Static(tango.StaticOptions{
RootPath: path,
Prefix: prefix,
}))
}
vars.Server.Use(xsrf.New(time.Second * 3600))
vars.Server.Use(binding.Bind())
vars.Server.Use(base.AuthHandler())
vars.Server.Use(base.SettingHandler())
vars.Server.Use(base.I18nHandler())
vars.Server.Use(renders.New(renders.Options{
Reload: true,
Directory: "static",
Extensions: []string{".tmpl"},
Funcs: template.FuncMap{
"Str2HTML": utils.Str2HTML,
"Md2HTML": utils.Md2Html,
"Pager2HTML": utils.Pager2HTML,
"Pager2Simple": utils.Pager2HTMLSimple,
"TimeUnixFormat": utils.TimeUnixFormat,
"TimeUnixFormatFriend": utils.FriendTimeUnixFormat,
"FriendBytesSize": utils.FriendBytesSize,
"Nl2Br": utils.Nl2Br,
},
}))
}
func initTango(isprod bool) *tango.Tango {
middlewares.Init()
tg := tango.NewWithLog(setting.Log)
if false {
//if !isprod {
tg.Use(debug.Debug(debug.Options{
IgnorePrefix: "/static",
HideResponseBody: true,
HideRequestBody: true,
}))
}
tg.Use(tango.ClassicHandlers...)
sess := session.New(session.Options{
MaxAge: time.Duration(setting.SessionCookieLifeTime),
})
tg.Use(
tango.Static(tango.StaticOptions{
RootPath: "./static",
Prefix: "static",
}),
tango.Static(tango.StaticOptions{
RootPath: "./static_source",
Prefix: "static_source",
}),
sess,
middlewares.Renders,
setting.Captcha,
)
tg.Get("/favicon.ico", func(ctx *tango.Context) {
ctx.ServeFile("./static/favicon.ico")
})
if setting.EnableXSRF {
tg.Use(xsrf.New(time.Duration(setting.SessionCookieLifeTime)))
}
tg.Use(flash.Flashes(sess), events.Events())
return tg
}
func Web(listen, static, templates, admin, pass string, tls bool, certFile, keyFile string) {
_, err := DB.GetUser(admin)
if err != nil {
if err == leveldb.ErrNotFound {
err = DB.AddUser(admin, pass)
}
}
if err != nil {
panic(err)
}
adminUser = admin
t := tango.Classic()
sess := session.New(session.Options{
MaxAge: timeout,
})
t.Use(
tango.Static(tango.StaticOptions{
RootPath: static,
}),
renders.New(renders.Options{
Reload: true, // if reload when template is changed
Directory: templates,
}),
sess,
auth(),
binding.Bind(),
xsrf.New(timeout),
flash.Flashes(sess),
)
t.Get("/", new(MainAction))
t.Any("/login", new(LoginAction))
t.Get("/logout", new(LogoutAction))
t.Get("/down", new(DownAction))
t.Group("/user", func(g *tango.Group) {
g.Get("/", new(UserAction))
g.Any("/chgpass", new(ChgPassAction))
g.Any("/add", new(UserAddAction))
g.Any("/edit", new(UserEditAction))
g.Any("/del", new(UserDelAction))
})
t.Group("/group", func(g *tango.Group) {
g.Get("/", new(GroupAction))
g.Get("/add", new(GroupAddAction))
g.Get("/edit", new(GroupEditAction))
g.Get("/del", new(GroupDelAction))
})
t.Group("/perm", func(g *tango.Group) {
g.Get("/", new(PermAction))
g.Any("/add", new(PermAddAction))
g.Any("/edit", new(PermEditAction))
g.Any("/del", new(PermDelAction))
g.Any("/updateOwner", new(PermUpdateOwner))
g.Any("/updateGroup", new(PermUpdateGroup))
g.Any("/updatePerm", new(PermUpdatePerm))
})
if tls {
t.RunTLS(certFile, keyFile, listen)
return
}
t.Run(listen)
}
func (is *BootstrapService) Init(v interface{}) (*Result, error) {
opt, ok := v.(BootstrapInitOption)
if !ok {
return nil, ErrServiceFuncNeedType(is.Init, opt)
}
var err error
if opt.Config {
core.Cfg = core.NewConfig()
if err = core.Cfg.Sync(core.ConfigFile); err != nil {
return nil, err
}
}
if core.Cfg != nil && opt.Database { // database depends on config
log.SetLevelByString("error")
tidb.Debug = false // close tidb debug
core.Db, err = xorm.NewEngine(core.Cfg.Db.Driver, core.Cfg.Db.DSN)
if err != nil {
return nil, err
}
core.Db.SetLogger(nil)
// core.Db.ShowDebug = true
// core.Db.ShowSQL = true
}
if core.Cfg != nil && opt.Server { // server depends on config
core.Server = tango.New([]tango.Handler{
tango.Return(),
tango.Param(),
tango.Contexts(),
}...)
core.Server.Use(tango.Static(tango.StaticOptions{
RootPath: core.StaticDirectory,
Prefix: core.StaticPrefix,
}))
core.Server.Use(tango.Static(tango.StaticOptions{
RootPath: core.ThemeDirectory,
Prefix: core.ThemePrefix,
}))
core.Server.Use(renders.New(renders.Options{
Reload: true,
Directory: core.ThemeDirectory,
Extensions: []string{".tmpl"},
Funcs: template.FuncMap{
"TimeUnixFormat": utils.TimeUnixFormat,
"TimeUnixFriend": utils.TimeUnixFriend,
"Mardown2Str": utils.Markdown2String,
"Markdown2HTML": utils.Markdown2HTML,
"Nl2BrHTML": utils.Nl2Br,
"Nl2BrString": utils.Nl2BrString,
"BytesSizeFriend": utils.FriendBytesSize,
},
}))
sessions := session.New(session.Options{
SessionIdName: core.SessionName,
})
core.Server.Use(xsrf.New(time.Hour))
core.Server.Use(binding.Bind())
core.Server.Use(sessions)
core.Server.Use(flash.Flashes(sessions))
}
return nil, nil
}