mirror of
https://github.com/cuigh/swirl
synced 2024-12-30 15:53:24 +00:00
127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/cuigh/auxo/errors"
|
|
"github.com/cuigh/auxo/net/web"
|
|
"github.com/cuigh/swirl/biz"
|
|
"github.com/cuigh/swirl/biz/docker"
|
|
"github.com/cuigh/swirl/model"
|
|
)
|
|
|
|
// HomeController is a basic controller of site
|
|
type HomeController struct {
|
|
Index web.HandlerFunc `path:"/" name:"index" authorize:"?" desc:"index page"`
|
|
Login web.HandlerFunc `path:"/login" name:"login.view" authorize:"*" desc:"sign in page"`
|
|
InitGet web.HandlerFunc `path:"/init" name:"init.view" authorize:"*" desc:"initialize page"`
|
|
InitPost web.HandlerFunc `path:"/init" name:"init" method:"post" authorize:"*" desc:"initialize system"`
|
|
Error403 web.HandlerFunc `path:"/403" name:"403" authorize:"?" desc:"403 page"`
|
|
Error404 web.HandlerFunc `path:"/404" name:"404" authorize:"*" desc:"404 page"`
|
|
}
|
|
|
|
// Home creates an instance of HomeController
|
|
func Home() (c *HomeController) {
|
|
return &HomeController{
|
|
Index: homeIndex,
|
|
Login: homeLogin,
|
|
InitGet: homeInitGet,
|
|
InitPost: homeInitPost,
|
|
Error403: homeError403,
|
|
Error404: homeError404,
|
|
}
|
|
}
|
|
|
|
func homeIndex(ctx web.Context) (err error) {
|
|
var (
|
|
count int
|
|
dashboard *model.ChartDashboard
|
|
charts []*model.Chart
|
|
m = newModel(ctx)
|
|
)
|
|
|
|
if count, err = docker.NodeCount(); err != nil {
|
|
return
|
|
}
|
|
m.Set("NodeCount", count)
|
|
|
|
if count, err = docker.NetworkCount(); err != nil {
|
|
return
|
|
}
|
|
m.Set("NetworkCount", count)
|
|
|
|
if count, err = docker.ServiceCount(); err != nil {
|
|
return
|
|
}
|
|
m.Set("ServiceCount", count)
|
|
|
|
if count, err = docker.StackCount(); err != nil {
|
|
return
|
|
}
|
|
m.Set("StackCount", count)
|
|
|
|
if dashboard, err = biz.Chart.GetDashboard("home", ""); err != nil {
|
|
return
|
|
}
|
|
charts, err = biz.Chart.GetDashboardCharts(dashboard)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
m.Set("Charts", charts)
|
|
|
|
return ctx.Render("index", m)
|
|
}
|
|
|
|
func homeLogin(ctx web.Context) error {
|
|
count, err := biz.User.Count()
|
|
if err != nil {
|
|
return err
|
|
} else if count == 0 {
|
|
return ctx.Redirect("init")
|
|
}
|
|
if ctx.User() != nil {
|
|
u := ctx.Q("from")
|
|
if u == "" {
|
|
u = "/"
|
|
}
|
|
return ctx.Redirect(u)
|
|
}
|
|
return ctx.Render("login", nil)
|
|
}
|
|
|
|
func homeInitGet(ctx web.Context) error {
|
|
count, err := biz.User.Count()
|
|
if err != nil {
|
|
return err
|
|
} else if count > 0 {
|
|
return ctx.Redirect("login")
|
|
}
|
|
return ctx.Render("init", nil)
|
|
}
|
|
|
|
func homeInitPost(ctx web.Context) error {
|
|
count, err := biz.User.Count()
|
|
if err != nil {
|
|
return err
|
|
} else if count > 0 {
|
|
return errors.New("Swirl was already initialized")
|
|
}
|
|
|
|
user := &model.User{}
|
|
err = ctx.Bind(user)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
user.Admin = true
|
|
user.Type = model.UserTypeInternal
|
|
err = biz.User.Create(user, nil)
|
|
return ajaxResult(ctx, err)
|
|
}
|
|
|
|
func homeError403(ctx web.Context) error {
|
|
return ctx.Render("403", nil)
|
|
}
|
|
|
|
func homeError404(ctx web.Context) error {
|
|
return ctx.Render("404", nil)
|
|
}
|