package main import ( "database/sql" "fmt" "net/http" "os" ctrlrs "git.kealoha.me/lks/lenslocked/controllers" "git.kealoha.me/lks/lenslocked/models" "github.com/go-chi/chi/v5" "github.com/gorilla/csrf" "github.com/go-chi/chi/v5/middleware" _ "github.com/jackc/pgx/v4/stdlib" ) const DEBUG bool = true func notFoundHandler(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "text/html; charset=utf8") w.WriteHeader(http.StatusNotFound) fmt.Fprint(w, "404 page not found") } func ConnectDB() *sql.DB { db, err := sql.Open("pgx", os.Getenv("LENSLOCKED_DB_STRING")) if err != nil { panic(fmt.Sprint("Error connecting to database: %w", err)) } err = db.Ping() if err != nil { panic(fmt.Sprint("Error connecting to database: %w", err)) } return db } func main() { csrfKey := []byte(os.Getenv("LENSLOCKED_CSRF_KEY")) if len(csrfKey) < 32 { panic("Error: bad csrf protection key\nPlease set a key with the LENSLOCKED_CSRF_KEY env var.") } db := ConnectDB() defer db.Close() userService := models.UserService{DB: db} sessionService := models.SessionService{DB: db} var usersCtrlr ctrlrs.Users = ctrlrs.Default(&userService, &sessionService) r := chi.NewRouter() r.Use(middleware.Logger) r.Get("/", ctrlrs.StaticController("home.gohtml", "tailwind.gohtml")) r.Get("/contact", ctrlrs.StaticController("contact.gohtml", "tailwind.gohtml")) r.Get("/faq", ctrlrs.FAQ("faq.gohtml", "tailwind.gohtml")) r.Get("/signup", usersCtrlr.GetSignup) r.Post("/signup", usersCtrlr.PostSignup) r.Get("/signin", usersCtrlr.GetSignin) r.Post("/signin", usersCtrlr.PostSignin) r.Get("/user", usersCtrlr.CurrentUser) r.NotFound(notFoundHandler) fmt.Println("Starting the server on :3000...") http.ListenAndServe(":3000", csrf.Protect(csrfKey, csrf.Secure(!DEBUG))(r)) }