97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
|
package webserver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gorilla/handlers"
|
||
|
|
"github.com/gorilla/mux"
|
||
|
|
"github.com/gorilla/sessions"
|
||
|
|
|
||
|
|
"fmt"
|
||
|
|
"html/template"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
//var store = sessions.NewCookieStore(os.Getenv("SESSION_KEY"))
|
||
|
|
|
||
|
|
var store = sessions.NewCookieStore([]byte("GO_SESS"))
|
||
|
|
|
||
|
|
func HealthHandler(response http.ResponseWriter, request *http.Request) {
|
||
|
|
response.Header().Add("Content-type", "text/plain")
|
||
|
|
fmt.Fprint(response, "I'm okay jack!")
|
||
|
|
}
|
||
|
|
|
||
|
|
func NotFoundHandler(response http.ResponseWriter, request *http.Request) {
|
||
|
|
response.Header().Add("X-Template-File", "html"+request.URL.Path)
|
||
|
|
tmpl := template.Must(template.ParseFiles("web/404.html"))
|
||
|
|
tmpl.Execute(response, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func CSSHandler(response http.ResponseWriter, request *http.Request) {
|
||
|
|
response.Header().Add("Content-type", "text/css")
|
||
|
|
tmpl := template.Must(template.ParseFiles("web/cover.css"))
|
||
|
|
tmpl.Execute(response, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RootHandler(response http.ResponseWriter, request *http.Request) {
|
||
|
|
request.URL.Path = "/index.html"
|
||
|
|
TemplateHandler(response, request)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TemplateHandler(response http.ResponseWriter, request *http.Request) {
|
||
|
|
response.Header().Add("X-Template-File", "web"+request.URL.Path)
|
||
|
|
type TemplateData struct {
|
||
|
|
Prompt string
|
||
|
|
AvailCount int
|
||
|
|
}
|
||
|
|
// tmpl, err := template.New("html"+request.URL.Path).Funcs(template.FuncMap{
|
||
|
|
// "ToUpper": strings.ToUpper,
|
||
|
|
// "ToLower": strings.ToLower,
|
||
|
|
// }).ParseFiles("html"+request.URL.Path)
|
||
|
|
_ = strings.ToLower("Hello")
|
||
|
|
if strings.Index(request.URL.Path, "/") < 0 {
|
||
|
|
http.Error(response, "No slashes wat - "+request.URL.Path, http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
basenameSlice := strings.Split(request.URL.Path, "/")
|
||
|
|
basename := basenameSlice[len(basenameSlice)-1]
|
||
|
|
//fmt.Fprintf(response, "%q", basenameSlice)
|
||
|
|
tmpl, err := template.New(basename).Funcs(template.FuncMap{
|
||
|
|
"ToUpper": strings.ToUpper,
|
||
|
|
"ToLower": strings.ToLower,
|
||
|
|
}).ParseFiles("web" + request.URL.Path)
|
||
|
|
if err != nil {
|
||
|
|
http.Error(response, err.Error(), http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
// NotFoundHandler(response, request)
|
||
|
|
// return
|
||
|
|
}
|
||
|
|
var td = TemplateData{"Hello", 1}
|
||
|
|
err = tmpl.Execute(response, td)
|
||
|
|
if err != nil {
|
||
|
|
http.Error(response, err.Error(), http.StatusInternalServerError)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func HandleHTTP() {
|
||
|
|
r := mux.NewRouter()
|
||
|
|
loggedRouter := handlers.LoggingHandler(os.Stdout, r)
|
||
|
|
r.NotFoundHandler = http.HandlerFunc(NotFoundHandler)
|
||
|
|
r.HandleFunc("/", RootHandler)
|
||
|
|
r.HandleFunc("/healthz", HealthHandler)
|
||
|
|
r.HandleFunc("/example/{.*}", TemplateHandler)
|
||
|
|
r.HandleFunc("/cover.css", CSSHandler)
|
||
|
|
http.Handle("/", r)
|
||
|
|
srv := &http.Server{
|
||
|
|
Handler: loggedRouter,
|
||
|
|
Addr: "0.0.0.0:5353",
|
||
|
|
WriteTimeout: 15 * time.Second,
|
||
|
|
ReadTimeout: 15 * time.Second,
|
||
|
|
}
|
||
|
|
fmt.Println("Listening on 0.0.0.0:5353")
|
||
|
|
srv.ListenAndServe()
|
||
|
|
}
|