energylink/backend/api/api.go

package api

import (
	"html/template"

	log "github.com/alecthomas/log4go"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/sessions"
	"github.com/torkelo/grafana-pro/backend/components"
	"github.com/torkelo/grafana-pro/backend/stores"
)

type HttpServer struct {
	port     string
	shutdown chan bool
	store    stores.Store
	renderer *components.PhantomRenderer
	router   *gin.Engine
}

var sessionStore = sessions.NewCookieStore([]byte("something-very-secret"))

// var hashKey = []byte("very-secret")
// var blockKey = []byte("a-lot-secret")
// var s = securecookie.New(hashKey, blockKey)

func NewHttpServer(port string, store stores.Store) *HttpServer {
	self := &HttpServer{}
	self.port = port
	self.store = store
	self.renderer = &components.PhantomRenderer{ImagesDir: "data/png", PhantomDir: "_vendor/phantomjs"}

	return self
}

func (self *HttpServer) ListenAndServe() {
	log.Info("Starting Http Listener on port %v", self.port)
	defer func() { self.shutdown <- true }()

	self.router = gin.Default()
	self.router.Use(CacheHeadersMiddleware())

	self.router.Static("/public", "./public")
	self.router.Static("/app", "./public/app")
	self.router.Static("/img", "./public/img")

	// register & parse templates
	templates := template.New("templates")
	templates.Delims("[[", "]]")
	templates.ParseFiles("./views/index.html")
	self.router.SetHTMLTemplate(templates)

	for _, fn := range routeHandlers {
		fn(self)
	}

	// register default route
	self.router.GET("/", self.AuthMiddleware(), self.index)
	self.router.GET("/login/*_", self.index)
	self.router.GET("/dashboard/*_", self.index)

	self.router.Run(":" + self.port)
}

func (self *HttpServer) index(c *gin.Context) {
	c.HTML(200, "index.html", &indexViewModel{title: "hello from go"})
}

func (self *HttpServer) login(c *gin.Context) {
	c.HTML(200, "login.html", &indexViewModel{title: "hello from go"})
}

func (self *HttpServer) AuthMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		session, _ := sessionStore.Get(c.Request, "grafana-session")
		// if session.Values["login"] == nil {
		// 	c.Writer.Header().Set("Location", "/login/login#login")
		// 	c.Abort(302)
		// }
		session.Values["asd"] = 1
		session.Save(c.Request, c.Writer)
	}
}

func CacheHeadersMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Writer.Header().Add("Cache-Control", "max-age=0, public, must-revalidate, proxy-revalidate")
	}
}

// Api Handler Registration
var routeHandlers = make([]routeHandlerRegisterFn, 0)

type routeHandlerRegisterFn func(self *HttpServer)

func addRoutes(fn routeHandlerRegisterFn) {
	routeHandlers = append(routeHandlers, fn)
}