67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
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
|
|
}
|
|
|
|
func NewIndexController(l *logrus.Logger) *IndexController {
|
|
return &IndexController{
|
|
l: l,
|
|
}
|
|
}
|
|
|
|
func (ic IndexController) serveUI(w http.ResponseWriter, r *http.Request) (int, int, error) {
|
|
lp := filepath.Join("templates", "index.html")
|
|
tmpl, err := template.ParseFiles(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.Errorf("Unhandled route %q", r.RequestURI)
|
|
helpers.HandleResponse(w, r, http.StatusNotFound, nil, ic.l)
|
|
return
|
|
}
|
|
returncode, contentLength, 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)
|
|
} else {
|
|
helpers.AddToContext(r, returncode, contentLength)
|
|
}
|
|
}
|