package ui import ( "bytes" "fmt" "html/template" "io" "net/http" "path/filepath" "git.faercol.me/faercol/polyculeconnect/polyculeconnect/helpers" "github.com/sirupsen/logrus" ) const StaticRoute = "/static/" type StaticController struct { } func (sc *StaticController) ServeHTTP(w http.ResponseWriter, r *http.Request) { fs := http.FileServer(http.Dir("./static")) http.StripPrefix(StaticRoute, fs).ServeHTTP(w, r) } type IndexController struct { l *logrus.Logger downstreamConstroller http.Handler } func NewIndexController(l *logrus.Logger, downstream http.Handler) *IndexController { return &IndexController{ l: l, downstreamConstroller: downstream, } } func (ic IndexController) serveUI(w http.ResponseWriter, r *http.Request) (int, int, error) { funcs := template.FuncMap{ "issuer": func() string { return "toto" }, } lp := filepath.Join("templates", "index.html") hdrTpl := filepath.Join("templates", "header.html") footTpl := filepath.Join("templates", "footer.html") tmpl, err := template.New("index.html").Funcs(funcs).ParseFiles(hdrTpl, footTpl, lp) if err != nil { return http.StatusInternalServerError, -1, fmt.Errorf("failed to init template: %w", err) } buf := new(bytes.Buffer) if err := tmpl.Execute(buf, nil); err != nil { return http.StatusInternalServerError, -1, fmt.Errorf("failed to execute template: %w", err) } n, err := io.Copy(w, buf) if err != nil { return http.StatusInternalServerError, int(n), fmt.Errorf("failed to write response; %w", err) } return http.StatusOK, int(n), nil } func (ic *IndexController) ServeHTTP(w http.ResponseWriter, r *http.Request) { if r.RequestURI != "/" { ic.l.Debugf("Serving URI %q to dex handler", r.RequestURI) ic.downstreamConstroller.ServeHTTP(w, r) return } returncode, _, err := ic.serveUI(w, r) if err != nil { ic.l.Errorf("Error serving UI: %s", err.Error()) helpers.HandleResponse(w, r, returncode, nil, ic.l) } }