1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
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)
}
|
