diff options
Diffstat (limited to 'internal/webui/app.go')
-rw-r--r-- | internal/webui/app.go | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/internal/webui/app.go b/internal/webui/app.go new file mode 100644 index 0000000..f510074 --- /dev/null +++ b/internal/webui/app.go @@ -0,0 +1,74 @@ +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 + +//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(); 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) { + 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)) +} |