From 309d17b921f75b16298cfe88ada9ba89bf646ac5 Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Mon, 5 Apr 2021 20:58:28 +0200
Subject: Added basic webui to show results

---
 internal/webui/app.go | 74 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 internal/webui/app.go

(limited to 'internal/webui/app.go')

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))
+}
-- 
cgit v1.2.3