Set middlewars

Author: mrLSD

.gitignore

@@ -1,2 +1,3 @@
 .idea
-*.iml
+*.iml
+echo-cms

config/config.go

@@ -18,21 +18,21 @@ func LoadConfig() {
 }
 
 type SiteHeader struct {
-	Title		string
-	Keywords	string
-	Description	string
+	Title       string
+	Keywords    string
+	Description string
 }
 
 func GetSiteHeader(moduleName string) SiteHeader {
 	var header SiteHeader
 	path := "site." + moduleName + ".head"
 	err := viper.UnmarshalKey(path, &header)
 	if err != nil {
-			panic(fmt.Errorf("Fatal error config file: %s \n", err))
+		panic(fmt.Errorf("Fatal error config file: %s \n", err))
 	}
 	return header
 }
 
 func (h *SiteHeader) SetTitle(title string) {
 	h.Title = title + h.Title
-}
+}

server.go

@@ -2,20 +2,30 @@ package main
 
 import (
 	"github.com/labstack/echo"
+	"github.com/labstack/gommon/log"
 
+	"github.com/labstack/echo/middleware"
 	"github.com/mrlsd/echo-cms/config"
 	"github.com/mrlsd/echo-cms/modules/backend"
-	"github.com/gpmgo/gopm/modules/log"
 )
 
 func main() {
 	config.LoadConfig()
 
 	e := echo.New()
+
+	// Routers
+	backend.UrlRules(e)
+	e.Static("/", "static")
+
 	e.Debug = true
 	e.Logger.SetLevel(log.DEBUG)
-	e.Static("/", "static")
+	e.Pre(middleware.NonWWWRedirect())
+	e.Use(middleware.Secure())
+	e.Use(middleware.CSRF())
+	//e.Use(middleware.Logger())
+	e.Use(middleware.Recover())
+	e.Use(middleware.BodyLimit("2M"))
 
-	backend.UrlRules(e)
-	e.Start(":3000")
+	e.Logger.Fatal(e.Start(":3000"))
 }