udemy-go-web-1/cmd/web/main.go

133 lines
3.2 KiB
Go
Raw Normal View History

2024-06-26 15:30:39 +00:00
package main
import (
"encoding/gob"
2024-07-29 12:20:51 +00:00
"flag"
2024-06-26 15:30:39 +00:00
"fmt"
2024-06-30 14:41:46 +00:00
"go-udemy-web-1/internal/config"
"go-udemy-web-1/internal/driver"
2024-06-30 14:41:46 +00:00
"go-udemy-web-1/internal/handlers"
"go-udemy-web-1/internal/helpers"
"go-udemy-web-1/internal/models"
2024-06-30 14:41:46 +00:00
"go-udemy-web-1/internal/render"
"log"
2024-06-26 15:30:39 +00:00
"net/http"
"os"
"time"
2024-07-29 19:54:37 +00:00
"github.com/alexedwards/scs/redisstore"
"github.com/alexedwards/scs/v2"
2024-07-29 19:54:37 +00:00
"github.com/gomodule/redigo/redis"
2024-06-26 15:30:39 +00:00
)
2024-06-26 15:41:50 +00:00
const portNumber = ":8080"
var (
app config.AppConfig
session *scs.SessionManager
infoLog *log.Logger
errorLog *log.Logger
)
2024-06-26 15:41:50 +00:00
// main is the main application function
2024-06-26 15:30:39 +00:00
func main() {
db, err := run()
if err != nil {
log.Fatal(err)
}
defer db.SQL.Close()
defer close(app.MailChan)
fmt.Println("Starting mail listener...")
fmt.Printf("Starting application on port %s\n", portNumber)
srv := &http.Server{
Addr: portNumber,
Handler: routes(&app),
}
err = srv.ListenAndServe()
log.Fatal(err)
}
func run() (*driver.DB, error) {
// what am I going to put in the session
gob.Register(models.Reservation{})
2024-07-07 20:37:48 +00:00
gob.Register(models.User{})
gob.Register(models.Room{})
gob.Register(models.Restriction{})
2024-07-27 20:39:23 +00:00
gob.Register(map[string]int{})
2024-07-29 12:20:51 +00:00
// read flags
inProduction := flag.Bool("production", true, "Application is in production")
useCache := flag.Bool("cache", true, "Use template cache")
dbHost := flag.String("dbhost", "localhost", "Database host")
dbName := flag.String("dbname", "bookings", "Database name")
dbUser := flag.String("dbuser", os.Getenv("PGUSER"), "Database user")
dbPass := flag.String("dbpass", os.Getenv("PGPWD"), "Database password")
dbPort := flag.String("dbport", "5432", "Database port")
dbSSL := flag.String("dbssl", "disable", "Database ssl settings (disable, prefer, require)")
flag.Parse()
if *dbUser == "" {
fmt.Println("Missing required flags")
os.Exit(1)
}
mailChan := make(chan models.MailData)
app.MailChan = mailChan
listenForMail()
// change this to true when in production
2024-07-29 12:20:51 +00:00
app.InProduction = *inProduction
2024-07-29 19:54:37 +00:00
// Establish connection pool to Redis.
pool := &redis.Pool{
MaxIdle: 10,
Dial: func() (redis.Conn, error) {
return redis.Dial("tcp", "localhost:6379")
},
}
session = scs.New()
2024-07-29 19:54:37 +00:00
session.Store = redisstore.New(pool)
session.Lifetime = 24 * time.Hour
session.Cookie.Persist = true
session.Cookie.SameSite = http.SameSiteLaxMode
session.Cookie.Secure = app.InProduction
app.Session = session
// connect to database
log.Println("Connecting to database...")
2024-07-29 12:20:51 +00:00
dsn := fmt.Sprintf("host=%s port=%s dbname=%s user=%s password=%s sslmode=%s",
*dbHost, *dbPort, *dbName, *dbUser, *dbPass, *dbSSL)
db, err := driver.ConnectSQL(dsn)
if err != nil {
log.Fatal("Cannot connect to database! Dying...")
}
log.Println("Connected to database")
tc, err := render.CreateTemplateCache()
if err != nil {
2024-06-29 19:30:06 +00:00
log.Fatalf("cannot create template cache: %s", err)
return nil, err
}
app.TemplateCahce = tc
2024-07-29 12:20:51 +00:00
app.UseCache = *useCache
infoLog = log.New(os.Stdout, "INFO\t", log.Ldate|log.Ltime)
app.InfoLog = infoLog
errorLog = log.New(os.Stdout, "ERROR\t", log.Ldate|log.Ltime|log.Lshortfile)
app.ErrorLog = errorLog
repo := handlers.NewRepo(&app, db)
handlers.NewHandlers(repo)
helpers.NewHelpers(&app)
2024-07-07 20:37:48 +00:00
render.NewRenderer(&app)
return db, nil
2024-06-26 15:30:39 +00:00
}