aboutsummaryrefslogtreecommitdiff
path: root/internal/webui/root.go
blob: c3a0ec514599fab3d9b2800cfd52ead97b1869f9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package webui

import (
	"fmt"
	"html/template"
	"log"
	"net/http"
	"time"

	"git.adyxax.org/adyxax/trains/pkg/model"
)

var rootTemplate = template.Must(template.ParseFS(templatesFS, "html/base.html", "html/root.html"))

// The page template variable
type Page struct {
	User       *model.User
	Departures []Departure
	Title      string
}
type Departure struct {
	DisplayName string
	Arrival     string
	Odd         bool
}

// The root handler of the webui
func rootHandler(e *env, w http.ResponseWriter, r *http.Request) error {
	if r.URL.Path == "/" {
		var p Page
		user, err := tryAndResumeSession(e, r)
		if err != nil {
			http.Redirect(w, r, "/login", http.StatusFound)
			return nil
		}
		p.User = user
		if d, err := e.navitia.GetDepartures(e.conf.TrainStop); err != nil {
			log.Printf("%+v\n%s\n", d, err)
		} else {
			for i := 0; i < len(d.Departures); i++ {
				t, err := time.Parse("20060102T150405", d.Departures[i].StopDateTime.ArrivalDateTime)
				if err != nil {
					panic(err)
				}
				p.Departures = append(p.Departures, Departure{d.Departures[i].DisplayInformations.Direction, t.Format("Mon, 02 Jan 2006 15:04:05"), i%2 == 1})
			}
			w.Header().Set("Cache-Control", "no-store, no-cache")
		}
		p.Title = "Horaires des prochains trains à Crépieux la Pape"
		err = rootTemplate.ExecuteTemplate(w, "root.html", p)
		if err != nil {
			return newStatusError(http.StatusInternalServerError, err)
		}
		return nil
	} else {
		return newStatusError(http.StatusNotFound, fmt.Errorf("Invalid path in rootHandler"))
	}
}