95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"io/fs"
|
|
"net/http"
|
|
"os"
|
|
|
|
userctx "git.kealoha.me/lks/lenslocked/context"
|
|
ctrlrs "git.kealoha.me/lks/lenslocked/controllers"
|
|
"git.kealoha.me/lks/lenslocked/migrations"
|
|
"git.kealoha.me/lks/lenslocked/models"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/gorilla/csrf"
|
|
"github.com/pressly/goose/v3"
|
|
|
|
"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 MigrateDB(db *sql.DB, subfs fs.FS) error {
|
|
goose.SetBaseFS(subfs)
|
|
defer func() { goose.SetBaseFS(nil) }()
|
|
err := goose.SetDialect("postgres")
|
|
if err != nil {
|
|
return fmt.Errorf("Migrate: %w", err)
|
|
}
|
|
err = goose.Up(db, ".")
|
|
if err != nil {
|
|
return fmt.Errorf("Migrate: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
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()
|
|
err := MigrateDB(db, migrations.FS)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
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.Post("/signout", usersCtrlr.GetSignout)
|
|
|
|
r.Get("/user", usersCtrlr.CurrentUser)
|
|
|
|
r.NotFound(notFoundHandler)
|
|
|
|
umw := userctx.UserMiddleware{SS: &sessionService}
|
|
|
|
fmt.Println("Starting the server on :3000...")
|
|
http.ListenAndServe(":3000", csrf.Protect(csrfKey, csrf.Secure(!DEBUG))(umw.SetUser(r)))
|
|
}
|