aboutsummaryrefslogtreecommitdiff
path: root/internal/webui/app.go
blob: f6082dcb58186e9e5aff4bd6b97777d07c52e81e (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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package webui

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

	"git.adyxax.org/adyxax/trains/pkg/config"
	"git.adyxax.org/adyxax/trains/pkg/navitia_api_client"
)

// the api client object
var client *navitia_api_client.Client

// the webui configuration
var conf *config.Config

//go:embed html/*
var templatesFS embed.FS

//go:embed static/*
var staticFS embed.FS

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

// The root handler of the webui
func rootHandler(w http.ResponseWriter, r *http.Request) {
	if r.URL.Path == "/" {
		var p Page
		if d, err := client.GetDepartures(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"
		renderTemplate(w, "index", p)
	} else {
		http.Error(w, "404 Not Found", http.StatusNotFound)
	}
}

var templates = template.Must(template.ParseFS(templatesFS, "html/index.html"))

func renderTemplate(w http.ResponseWriter, tmpl string, p Page) {
	err := templates.ExecuteTemplate(w, tmpl+".html", p)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
	}
}

func Run(c *config.Config) {
	conf = c
	client = navitia_api_client.NewClient(c.Token)
	http.Handle("/static/", http.FileServer(http.FS(staticFS)))
	http.HandleFunc("/", rootHandler)

	listenStr := c.Address + ":" + c.Port
	log.Printf("Starting webui on %s", listenStr)
	log.Fatal(http.ListenAndServe(listenStr, nil))
}