package main import ( "net/http" "gitlab.quimbo.fr/odwrtw/canape-sql/auth" "gitlab.quimbo.fr/odwrtw/canape-sql/config" "gitlab.quimbo.fr/odwrtw/canape-sql/movies" "gitlab.quimbo.fr/odwrtw/canape-sql/users" "gitlab.quimbo.fr/odwrtw/canape-sql/web" "github.com/Sirupsen/logrus" "github.com/codegangsta/negroni" "github.com/gorilla/sessions" "github.com/jmoiron/sqlx" _ "github.com/lib/pq" ) type UserBackend struct { Database *sqlx.DB } func (b *UserBackend) Get(username string) (auth.User, error) { return users.Get(b.Database, username) } func main() { log := logrus.NewEntry(logrus.New()) cf, err := config.Load("config.yml") if err != nil { log.Panic(err) } db, err := sqlx.Connect("postgres", cf.PGDSN) if err != nil { log.Panic(err) } uBackend := &UserBackend{Database: db} authParams := auth.Params{ Backend: uBackend, Pepper: cf.Authorizer.Pepper, CookieName: cf.Authorizer.CookieName, Cookiejar: sessions.NewCookieStore([]byte(cf.Authorizer.Key)), Cost: cf.Authorizer.Cost, } authorizer := auth.New(authParams) env := web.NewEnv(web.EnvParams{ Database: db, Auth: authorizer, Log: log, Config: cf, TemplatesDir: "./templates", }) authMiddleware := auth.NewMiddleware(env.Auth) env.Handle("/users/login", users.LoginGETHandler).Name("users.login").Methods("GET") env.Handle("/users/login", users.LoginPOSTHandler).Name("users.login").Methods("POST") env.Handle("/users/logout", users.LogoutHandler).Name("users.logout") env.Handle("/users/details", users.DetailsHandler).Name("users.details").WithRole(users.UserRole) env.Handle("/users/edit", users.EditHandler).Name("users.edit").WithRole(users.UserRole) env.Handle("/movies/polochon", movies.FromPolochon).Name("movies.polochon").WithRole(users.UserRole) env.Handle("/movies/explore/popular", movies.ExplorePopular).Name("movies.explore.popular").WithRole(users.UserRole) err = env.SetLoginRoute("users.login") if err != nil { log.Panic(err) } n := negroni.Classic() n.Use(authMiddleware) n.Use(negroni.NewStatic(http.Dir("./static"))) n.UseHandler(env.Router) n.Run(":" + cf.Port) }