package handlers import ( "encoding/json" "fmt" "go-udemy-web-1/internal/config" "go-udemy-web-1/internal/driver" "go-udemy-web-1/internal/forms" "go-udemy-web-1/internal/helpers" "go-udemy-web-1/internal/models" "go-udemy-web-1/internal/render" "go-udemy-web-1/internal/repository" "go-udemy-web-1/internal/repository/dbrepo" "net/http" "strconv" "time" ) // Repo the repository used by the handlers var Repo *Repository // Repository is the repository type type Repository struct { App *config.AppConfig DB repository.DatabaseRepo } // NewRepo creates a new repository func NewRepo(a *config.AppConfig, db *driver.DB) *Repository { return &Repository{ App: a, DB: dbrepo.NewPostgresRepo(db.SQL, a), } } // NewHandlers sets the repository for the handlers func NewHandlers(r *Repository) { Repo = r } // Home is the home page handler func (m *Repository) Home(w http.ResponseWriter, r *http.Request) { render.Template(w, r, "home.page.tmpl", &models.TemplateData{}) } // About is the about page handler func (m *Repository) About(w http.ResponseWriter, r *http.Request) { // send the data to the template render.Template(w, r, "about.page.tmpl", &models.TemplateData{}) } // Contact is the contact page handler func (m *Repository) Contact(w http.ResponseWriter, r *http.Request) { render.Template(w, r, "contact.page.tmpl", &models.TemplateData{}) } // Generals is the General's Quarters page handler func (m *Repository) Generals(w http.ResponseWriter, r *http.Request) { render.Template(w, r, "generals.page.tmpl", &models.TemplateData{}) } // Majors is the Major's Suite page handler func (m *Repository) Majors(w http.ResponseWriter, r *http.Request) { render.Template(w, r, "majors.page.tmpl", &models.TemplateData{}) } // MakeReservation is the make reservation page handler func (m *Repository) MakeReservation(w http.ResponseWriter, r *http.Request) { // For the first time render emptyReservation so that this object is // filled with the info when sent back. var emptyReservation models.Reservation data := make(map[string]interface{}) data["reservation"] = emptyReservation render.Template(w, r, "make-reservation.page.tmpl", &models.TemplateData{ Form: forms.New(nil), Data: data, }) } // PostMakeReservation is the make reservation page post handler func (m *Repository) PostMakeReservation(w http.ResponseWriter, r *http.Request) { err := r.ParseForm() if err != nil { helpers.ServerError(w, err) return } sd := r.Form.Get("start_date") ed := r.Form.Get("end_date") // 2024-07-08 -- 01/02 03:04:05PM '06 -0700 layout := "2006-01-02" startDate, err := time.Parse(layout, sd) if err != nil { helpers.ServerError(w, err) return } endDate, err := time.Parse(layout, ed) if err != nil { helpers.ServerError(w, err) return } roomId, err := strconv.Atoi(r.Form.Get("room_id")) if err != nil { helpers.ServerError(w, err) return } reservation := models.Reservation{ FirstName: r.Form.Get("first_name"), LastName: r.Form.Get("last_name"), Email: r.Form.Get("email"), Phone: r.Form.Get("phone"), StartDate: startDate, EndDate: endDate, RoomID: roomId, } form := forms.New(r.PostForm) form.Required("first_name", "last_name", "email") form.MinLength("first_name", 2) form.IsEmail("email") if !form.Valid() { data := make(map[string]interface{}) data["reservation"] = reservation render.Template(w, r, "make-reservation.page.tmpl", &models.TemplateData{ Data: data, Form: form, }) return } newReservationID, err := m.DB.InsertReservation(reservation) if err != nil { helpers.ServerError(w, err) return } restriction := models.RoomRestriction{ StartDate: startDate, EndDate: endDate, ID: 0, RoomID: roomId, ReservationID: newReservationID, RestrictionID: 1, } err = m.DB.InsertRoomRestriction(restriction) if err != nil { helpers.ServerError(w, err) return } m.App.Session.Put(r.Context(), "reservation", reservation) http.Redirect(w, r, "/reservation-summary", http.StatusSeeOther) } // ReservationSummary is the reservation summary page handler func (m *Repository) ReservationSummary(w http.ResponseWriter, r *http.Request) { reservation, ok := m.App.Session.Get(r.Context(), "reservation").(models.Reservation) if !ok { m.App.ErrorLog.Println("connot get item from session") m.App.Session.Put(r.Context(), "error", "Can't get reservation from session") http.Redirect(w, r, "/", http.StatusTemporaryRedirect) } m.App.Session.Remove(r.Context(), "reservation") data := make(map[string]interface{}) data["reservation"] = reservation render.Template(w, r, "reservation-summary.page.tmpl", &models.TemplateData{ Data: data, }) } // Availability is the search for availability page handler func (m *Repository) Availability(w http.ResponseWriter, r *http.Request) { render.Template(w, r, "availability.page.tmpl", &models.TemplateData{}) } // PostAvailability is the search for availability page handler func (m *Repository) PostAvailability(w http.ResponseWriter, r *http.Request) { start := r.Form.Get("start") end := r.Form.Get("end") fmt.Fprintf(w, "Posted to search availability from %s to %s", start, end) } type responseJSON struct { OK string `json:"ok"` Message string `json:"message"` } // AvailabilityJSON is the search for availability page handler func (m *Repository) AvailabilityJSON(w http.ResponseWriter, r *http.Request) { resp := responseJSON{ OK: "true", Message: "Available!", } out, err := json.MarshalIndent(resp, "", " ") if err != nil { helpers.ServerError(w, err) return } w.Header().Set("Content-Type", "application/json") w.Write(out) }