/
main.go
98 lines (90 loc) · 3.06 KB
/
main.go
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
package main
import (
"flag"
"fmt"
"io/ioutil"
"net/http"
"path/filepath"
"sync"
"text/template"
"github.com/a-know/yukizuri/trace"
stats_api "github.com/fukata/golang-stats-api-handler"
"github.com/stretchr/objx"
)
type templateHandler struct {
once sync.Once
filename string
templ *template.Template
}
// Handling HTTP Request
func (t *templateHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
t.once.Do(func() {
f, err := Assets.Open(filepath.Join("/templates", t.filename))
if err != nil {
tracer := trace.New()
logContent := tracer.LogContent("system", "-", "-", "TemplateHandler")
tracer.TraceError(logContent, err)
}
defer f.Close()
data, err := ioutil.ReadAll(f)
if err != nil {
tracer := trace.New()
logContent := tracer.LogContent("system", "-", "-", "TemplateHandler")
tracer.TraceError(logContent, err)
}
var ns = template.New("template")
t.templ, _ = ns.Parse(string(data))
})
data := map[string]interface{}{
"Host": r.Host,
}
if cookie, err := r.Cookie("yukizuri"); err == nil {
data["UserData"] = objx.MustFromBase64(cookie.Value)
}
t.templ.Execute(w, data)
}
func heartbeatHandler(w http.ResponseWriter, r *http.Request) {
tracer := trace.New()
logContent := tracer.LogContent("system", "-", "-", "heartbeat")
tracer.TraceInfo(logContent)
fmt.Fprintf(w, "OK")
}
func main() {
var addr = flag.String("addr", ":8080", "port number")
var openaitoken = flag.String("openaitoken", "openaitoken", "OpenAI API Token String")
var logging = flag.Bool("logging", true, "logging with stdout")
flag.Parse()
r := newRoom(*logging, *openaitoken)
http.Handle("/chat", MustJoin(&templateHandler{filename: "yukizuri.html"}))
http.Handle("/join", &templateHandler{filename: "join.html"})
http.Handle("/", &templateHandler{filename: "join.html"})
http.HandleFunc("/leave", func(w http.ResponseWriter, r *http.Request) {
http.SetCookie(w, &http.Cookie{
Name: "yukizuri",
Value: "",
Path: "/",
MaxAge: -1,
})
w.Header()["Location"] = []string{"/chat"}
w.WriteHeader(http.StatusTemporaryRedirect)
})
http.HandleFunc("/heartbeat", heartbeatHandler)
http.HandleFunc("/join/", joinHandler)
http.HandleFunc("/api/stats", stats_api.Handler)
http.Handle("/room", r) // for WebSocket connection endpoint
http.Handle("/css/", http.StripPrefix("/css/", http.FileServer(http.Dir("./css"))))
http.Handle("/js/", http.StripPrefix("/js/", http.FileServer(http.Dir("./js"))))
http.Handle("/fonts/", http.StripPrefix("/fonts/", http.FileServer(http.Dir("./fonts"))))
http.Handle("/images/", http.StripPrefix("/images/", http.FileServer(http.Dir("./images"))))
http.Handle("/plugins/", http.StripPrefix("/plugins/", http.FileServer(http.Dir("./plugins"))))
// Starting chatroom
go r.run()
// Starting web server
tracer := trace.New()
logContent := tracer.LogContent("system", "-", "-", "Starting Web server...")
tracer.TraceInfo(logContent)
if err := http.ListenAndServe(*addr, nil); err != nil {
logContent := tracer.LogContent("system", "-", "-", "ListenAndServe")
tracer.TraceError(logContent, err)
}
}