mirror of
https://github.com/langhuihui/monibuca.git
synced 2025-10-05 18:06:51 +08:00
107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
package gateway
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"log"
|
|
"mime"
|
|
"net/http"
|
|
"path"
|
|
"runtime"
|
|
"time"
|
|
|
|
. "github.com/langhuihui/monibuca/monica"
|
|
. "github.com/langhuihui/monibuca/monica/util"
|
|
)
|
|
|
|
var (
|
|
config = new(ListenerConfig)
|
|
startTime = time.Now()
|
|
dashboardPath string
|
|
)
|
|
|
|
func init() {
|
|
_, currentFilePath, _, _ := runtime.Caller(0)
|
|
dashboardPath = path.Join(path.Dir(currentFilePath), "../../dashboard/dist")
|
|
log.Println(dashboardPath)
|
|
InstallPlugin(&PluginConfig{
|
|
Name: "GateWay",
|
|
Type: PLUGIN_HOOK,
|
|
Config: config,
|
|
Run: run,
|
|
})
|
|
}
|
|
func run() {
|
|
http.HandleFunc("/api/sysInfo", sysInfo)
|
|
http.HandleFunc("/api/stop", stopPublish)
|
|
http.HandleFunc("/api/summary", summary)
|
|
http.HandleFunc("/api/logs", watchLogs)
|
|
http.HandleFunc("/api/config", getConfig)
|
|
http.HandleFunc("/", website)
|
|
log.Printf("server gateway start at %s", config.ListenAddr)
|
|
log.Fatal(http.ListenAndServe(config.ListenAddr, nil))
|
|
}
|
|
func getConfig(w http.ResponseWriter, r *http.Request) {
|
|
w.Write(ConfigRaw)
|
|
}
|
|
func watchLogs(w http.ResponseWriter, r *http.Request) {
|
|
AddWriter(NewSSE(w, r.Context()))
|
|
<-r.Context().Done()
|
|
}
|
|
func stopPublish(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
if streamPath := r.URL.Query().Get("stream"); streamPath != "" {
|
|
if b, ok := AllRoom.Load(streamPath); ok {
|
|
b.(*Room).Cancel()
|
|
w.Write([]byte("success"))
|
|
} else {
|
|
w.Write([]byte("no query stream"))
|
|
}
|
|
} else {
|
|
w.Write([]byte("no such stream"))
|
|
}
|
|
}
|
|
func website(w http.ResponseWriter, r *http.Request) {
|
|
filePath := r.URL.Path
|
|
if filePath == "/" {
|
|
filePath = "/index.html"
|
|
} else if filePath == "/docs" {
|
|
filePath = "/docs/index.html"
|
|
}
|
|
if mime := mime.TypeByExtension(path.Ext(filePath)); mime != "" {
|
|
w.Header().Set("Content-Type", mime)
|
|
}
|
|
if f, err := ioutil.ReadFile(dashboardPath + filePath); err == nil {
|
|
if _, err = w.Write(f); err != nil {
|
|
w.WriteHeader(505)
|
|
}
|
|
} else {
|
|
w.Header().Set("Location", "/")
|
|
w.WriteHeader(302)
|
|
}
|
|
}
|
|
func summary(w http.ResponseWriter, r *http.Request) {
|
|
sse := NewSSE(w, r.Context())
|
|
Summary.Add()
|
|
defer Summary.Done()
|
|
sse.WriteJSON(&Summary)
|
|
ticker := time.NewTicker(time.Second)
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
if sse.WriteJSON(&Summary) != nil {
|
|
return
|
|
}
|
|
case <-r.Context().Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
func sysInfo(w http.ResponseWriter, r *http.Request) {
|
|
bytes, err := json.Marshal(EngineInfo)
|
|
if err == nil {
|
|
_, err = w.Write(bytes)
|
|
}
|
|
}
|