Initial commit of libmkwebpage

1 files changed, 101 insertions, 0 deletions

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..1032943
--- /dev/null
+++ b/main.go
@@ -0,0 +1,101 @@
+package libmkwebpage
+
+import (
+	"fmt"
+	"net/http"
+	"net/http/cgi"
+	"os"
+)
+
+type Webpage struct {
+	muxStatic        *http.ServeMux
+	muxDynamic       *http.ServeMux
+	muxInternal      *http.ServeMux
+	templates        templates
+	staticFilePrefix string
+}
+
+type handler struct {
+	w *Webpage
+}
+
+func (h handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	if _, pattern := h.w.muxInternal.Handler(r); pattern != "" {
+		h.w.muxInternal.ServeHTTP(w, r)
+		return
+	}
+	h.w.getHandler().ServeHTTP(w, r)
+}
+
+func New() *Webpage {
+	webpage := Webpage{muxStatic: http.NewServeMux(), muxDynamic: http.NewServeMux(), muxInternal: http.NewServeMux()}
+	webpage.muxInternal.HandleFunc("/.libmkpage/", func(w http.ResponseWriter, r *http.Request) { fmt.Fprintln(w, "/.libmkpage/") })
+	return &webpage
+}
+
+// See http.HandleFunc
+func (w *Webpage) Handle(pattern string, handler func(http.ResponseWriter, *http.Request)) *Webpage {
+	return w.HandleDynamic(pattern, handler).HandleStatic(pattern, handler)
+}
+
+// See http.HandleFunc, only executed in dynamic requests (i.e. in a dev setup)
+func (w *Webpage) HandleDynamic(pattern string, handler func(http.ResponseWriter, *http.Request)) *Webpage {
+	w.muxDynamic.HandleFunc(pattern, handler)
+	return w
+}
+
+// See http.HandleFunc, only executed in static requests (i.e. in a prod setup)
+func (w *Webpage) HandleStatic(pattern string, handler func(http.ResponseWriter, *http.Request)) *Webpage {
+	w.muxStatic.HandleFunc(pattern, handler)
+	return w
+}
+
+// See http.Handle
+func (w *Webpage) Handler(pattern string, handler http.Handler) *Webpage {
+	return w.HandlerDynamic(pattern, handler).HandlerStatic(pattern, handler)
+}
+
+// See http.Handle, only executed in dynamic requests (i.e. in a dev setup)
+func (w *Webpage) HandlerDynamic(pattern string, handler http.Handler) *Webpage {
+	w.muxDynamic.Handle(pattern, handler)
+	return w
+}
+
+// See http.Handle, only executed in static requests (i.e. in a prod setup)
+func (w *Webpage) HandlerStatic(pattern string, handler http.Handler) *Webpage {
+	w.muxStatic.Handle(pattern, handler)
+	return w
+}
+
+// Returns the proper Mux for the current Request based on the request type without the magic internal mux
+func (w *Webpage) getHandler() http.Handler {
+	if w.isDynamic() {
+		return w.muxDynamic
+	} else {
+		return w.muxStatic
+	}
+}
+
+// Returns a http.Handler capable of hosting the application. Useful for combining more than one webpage in one binary
+func (w *Webpage) GetHandler() http.Handler {
+	return handler{w}
+}
+
+func (w *Webpage) isDynamic() bool {
+	return os.Getenv("PATH_INFO") == ""
+}
+func (w *Webpage) isStatic() bool {
+	return !w.isDynamic()
+}
+
+func (w *Webpage) Serve() {
+	if w.isDynamic() {
+		fmt.Println("Listening on [::]:8080")
+		panic(http.ListenAndServe(":8080", w.GetHandler()))
+	}
+	err := cgi.Serve(w.GetHandler())
+	if err != nil {
+		panic(err)
+	}
+	os.Exit(0)
+}