gocron/routers/routers.go

184 lines
5.3 KiB
Go
Raw Normal View History

2017-03-24 05:06:53 +00:00
package routers
2017-03-24 09:55:44 +00:00
import (
2017-04-02 02:38:49 +00:00
"github.com/go-macaron/binding"
2017-04-07 01:22:00 +00:00
"github.com/ouqiang/gocron/routers/install"
2017-04-02 02:38:49 +00:00
"gopkg.in/macaron.v1"
2017-04-07 01:22:00 +00:00
"github.com/ouqiang/gocron/routers/task"
"github.com/ouqiang/gocron/routers/host"
"github.com/ouqiang/gocron/routers/tasklog"
2017-04-13 09:35:59 +00:00
"github.com/ouqiang/gocron/modules/utils"
"github.com/go-macaron/session"
"github.com/go-macaron/csrf"
"github.com/go-macaron/toolbox"
"strings"
"github.com/ouqiang/gocron/modules/app"
2017-04-23 06:11:53 +00:00
"github.com/ouqiang/gocron/modules/logger"
2017-04-23 09:00:47 +00:00
"github.com/ouqiang/gocron/routers/user"
"github.com/go-macaron/gzip"
2017-03-24 09:55:44 +00:00
)
2017-03-24 05:06:53 +00:00
2017-04-13 09:35:59 +00:00
// 静态文件目录
const StaticDir = "public"
2017-03-24 05:06:53 +00:00
// 路由注册
func Register(m *macaron.Macaron) {
2017-04-02 02:38:49 +00:00
// 所有GET方法自动注册HEAD方法
m.SetAutoHead(true)
// 首页
2017-04-13 09:35:59 +00:00
m.Get("/", Home)
2017-04-02 02:38:49 +00:00
// 系统安装
m.Group("/install", func() {
2017-04-07 01:13:36 +00:00
m.Get("", install.Create)
m.Post("/store", binding.Bind(install.InstallForm{}), install.Store)
})
// 用户
m.Group("/user", func() {
2017-04-23 09:00:47 +00:00
m.Get("/login", user.Login)
m.Post("/login", user.ValidateLogin)
m.Get("/logout", user.Logout)
2017-04-07 01:13:36 +00:00
})
// 任务
m.Group("/task", func() {
m.Get("/create", task.Create)
2017-04-07 09:26:46 +00:00
m.Post("/store", binding.Bind(task.TaskForm{}), task.Store)
2017-04-17 10:04:30 +00:00
m.Get("/edit/:id", task.Edit)
m.Get("", task.Index)
m.Get("/log", tasklog.Index)
2017-04-13 09:35:59 +00:00
m.Post("/log/clear", tasklog.Clear)
m.Post("/remove/:id", task.Remove)
m.Post("/enable/:id", task.Enable)
m.Post("/disable/:id", task.Disable)
2017-04-21 05:36:45 +00:00
m.Get("/run/:id", task.Run)
2017-04-07 01:13:36 +00:00
})
// 主机
m.Group("/host", func() {
m.Get("/create", host.Create)
2017-04-20 01:36:42 +00:00
m.Get("/edit/:id", host.Edit)
m.Get("/ping/:id", host.Ping)
2017-04-07 09:26:46 +00:00
m.Post("/store", binding.Bind(host.HostForm{}), host.Store)
m.Get("", host.Index)
2017-04-13 09:35:59 +00:00
m.Post("/remove/:id", host.Remove)
2017-04-07 01:13:36 +00:00
})
2017-04-23 09:00:47 +00:00
// 404错误
m.NotFound(func(ctx *macaron.Context) {
if isGetRequest(ctx) && !isAjaxRequest(ctx) {
ctx.Data["Title"] = "404 - NOT FOUND"
ctx.HTML(404, "error/404")
} else {
json := utils.JsonResponse{}
ctx.Resp.Write([]byte(json.Failure(utils.NotFound, "您访问的地址不存在")))
}
})
// 50x错误
m.InternalServerError(func(ctx *macaron.Context) {
logger.Debug("500错误")
if isGetRequest(ctx) && !isAjaxRequest(ctx) {
ctx.Data["Title"] = "500 - INTERNAL SERVER ERROR"
ctx.HTML(500, "error/500")
} else {
json := utils.JsonResponse{}
ctx.Resp.Write([]byte(json.Failure(utils.ServerError, "网站暂时无法访问,请稍后再试")))
}
})
2017-04-13 09:35:59 +00:00
}
// 中间件注册
func RegisterMiddleware(m *macaron.Macaron) {
m.Use(macaron.Logger())
m.Use(macaron.Recovery())
2017-04-23 09:00:47 +00:00
m.Use(gzip.Gziper())
2017-04-13 09:35:59 +00:00
m.Use(macaron.Static(StaticDir))
m.Use(macaron.Renderer(macaron.RenderOptions{
Directory: "templates",
Extensions: []string{".html"},
// 模板语法分隔符,默认为 ["{{", "}}"]
Delims: macaron.Delims{"{{{", "}}}"},
// 追加的 Content-Type 头信息,默认为 "UTF-8"
Charset: "UTF-8",
// 渲染具有缩进格式的 JSON默认为不缩进
IndentJSON: true,
// 渲染具有缩进格式的 XML默认为不缩进
IndentXML: true,
}))
m.Use(session.Sessioner(session.Options{
Provider: "file",
ProviderConfig: app.DataDir + "/sessions",
}))
2017-04-13 09:35:59 +00:00
m.Use(csrf.Csrfer())
m.Use(toolbox.Toolboxer(m))
2017-04-23 09:00:47 +00:00
checkAppInstall(m)
userAuth(m)
setShareData(m)
}
2017-04-07 01:13:36 +00:00
2017-04-23 09:00:47 +00:00
// 系统未安装,重定向到安装页面
func checkAppInstall(m *macaron.Macaron) {
2017-04-13 09:35:59 +00:00
m.Use(func(ctx *macaron.Context) {
installUrl := "/install"
if strings.HasPrefix(ctx.Req.URL.Path, installUrl) {
2017-04-13 09:35:59 +00:00
return
}
if !app.Installed {
ctx.Redirect(installUrl)
}
})
2017-04-23 09:00:47 +00:00
}
// 用户认证
func userAuth(m *macaron.Macaron) {
m.Use(func(ctx *macaron.Context, sess session.Store) {
if user.IsLogin(sess) {
return
}
uri := ctx.Req.URL.Path
found := false
excludePaths := []string{"/install", "/user/login"}
for _, path := range excludePaths {
if strings.HasPrefix(uri, path) {
found = true
break
}
}
if !found {
ctx.Redirect("/user/login")
}
})
}
// 设置共享数据
func setShareData(m *macaron.Macaron) {
m.Use(func(ctx *macaron.Context, sess session.Store) {
2017-04-21 09:41:59 +00:00
ctx.Data["URI"] = ctx.Req.URL.Path
urlPath := strings.TrimPrefix(ctx.Req.URL.Path, "/")
paths := strings.Split(urlPath, "/")
ctx.Data["Controller"] = ""
ctx.Data["Action"] = ""
if len(paths) > 0 {
ctx.Data["Controller"] = paths[0]
}
if len(paths) > 1 {
ctx.Data["Action"] = paths[1]
}
2017-04-23 09:00:47 +00:00
ctx.Data["LoginUsername"] = user.Username(sess)
2017-04-23 11:17:29 +00:00
ctx.Data["LoginUid"] = user.Uid(sess)
2017-04-02 02:38:49 +00:00
})
2017-04-02 02:19:52 +00:00
}
func isAjaxRequest(ctx *macaron.Context) bool {
req := ctx.Req.Header.Get("X-Requested-With")
if req == "XMLHttpRequest" {
return true
}
return false
}
func isGetRequest(ctx *macaron.Context) bool {
return ctx.Req.Method == "GET"
}