mirror of
https://github.com/Monibuca/engine.git
synced 2025-10-04 16:22:41 +08:00
123 lines
3.5 KiB
Go
123 lines
3.5 KiB
Go
package engine
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"time"
|
|
|
|
"m7s.live/engine/v4/config"
|
|
"m7s.live/engine/v4/util"
|
|
)
|
|
|
|
const (
|
|
NO_SUCH_CONIFG = "no such config"
|
|
NO_SUCH_STREAM = "no such stream"
|
|
)
|
|
|
|
type GlobalConfig struct {
|
|
*config.Engine
|
|
}
|
|
|
|
func (conf *GlobalConfig) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
|
|
rw.Write([]byte("Monibuca API Server\n"))
|
|
for _, api := range apiList {
|
|
rw.Write([]byte(api + "\n"))
|
|
}
|
|
}
|
|
|
|
func (conf *GlobalConfig) API_summary(rw http.ResponseWriter, r *http.Request) {
|
|
util.ReturnJson(summary.collect, time.Second, rw, r)
|
|
}
|
|
|
|
func (conf *GlobalConfig) API_plugins(rw http.ResponseWriter, r *http.Request) {
|
|
if err := json.NewEncoder(rw).Encode(Plugins); err != nil {
|
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
}
|
|
|
|
func (conf *GlobalConfig) API_stream(rw http.ResponseWriter, r *http.Request) {
|
|
if streamPath := r.URL.Query().Get("streamPath"); streamPath != "" {
|
|
if s := Streams.Get(streamPath); s != nil {
|
|
if err := json.NewEncoder(rw).Encode(s); err != nil {
|
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
} else {
|
|
http.Error(rw, NO_SUCH_STREAM, http.StatusNotFound)
|
|
}
|
|
} else {
|
|
http.Error(rw, "no streamPath", http.StatusBadRequest)
|
|
}
|
|
}
|
|
|
|
func (conf *GlobalConfig) API_sysInfo(rw http.ResponseWriter, r *http.Request) {
|
|
if err := json.NewEncoder(rw).Encode(&SysInfo); err != nil {
|
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
}
|
|
|
|
func (conf *GlobalConfig) API_closeStream(w http.ResponseWriter, r *http.Request) {
|
|
if streamPath := r.URL.Query().Get("streamPath"); streamPath != "" {
|
|
if s := Streams.Get(streamPath); s != nil {
|
|
s.Close()
|
|
w.Write([]byte("ok"))
|
|
} else {
|
|
http.Error(w, NO_SUCH_STREAM, http.StatusNotFound)
|
|
}
|
|
} else {
|
|
http.Error(w, "no streamPath", http.StatusBadRequest)
|
|
}
|
|
}
|
|
|
|
// API_getConfig 获取指定的配置信息
|
|
func (conf *GlobalConfig) API_getConfig(w http.ResponseWriter, r *http.Request) {
|
|
if configName := r.URL.Query().Get("name"); configName != "" {
|
|
if c, ok := Plugins[configName]; ok {
|
|
if err := json.NewEncoder(w).Encode(c.RawConfig); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
} else {
|
|
http.Error(w, NO_SUCH_CONIFG, http.StatusNotFound)
|
|
}
|
|
} else if err := json.NewEncoder(w).Encode(Engine.RawConfig); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
}
|
|
|
|
// API_modifyConfig 修改并保存配置
|
|
func (conf *GlobalConfig) API_modifyConfig(w http.ResponseWriter, r *http.Request) {
|
|
if configName := r.URL.Query().Get("name"); configName != "" {
|
|
if c, ok := Plugins[configName]; ok {
|
|
if err := json.NewDecoder(r.Body).Decode(&c.Modified); err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
} else {
|
|
c.Save()
|
|
c.RawConfig.Assign(c.Modified)
|
|
w.Write([]byte("ok"))
|
|
}
|
|
} else {
|
|
http.Error(w, NO_SUCH_CONIFG, http.StatusNotFound)
|
|
}
|
|
} else if err := json.NewDecoder(r.Body).Decode(&Engine.Modified); err == nil {
|
|
Engine.Save()
|
|
Engine.RawConfig.Assign(Engine.Modified)
|
|
w.Write([]byte("ok"))
|
|
} else {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
}
|
|
}
|
|
|
|
// API_updateConfig 热更新配置
|
|
func (conf *GlobalConfig) API_updateConfig(w http.ResponseWriter, r *http.Request) {
|
|
if configName := r.URL.Query().Get("name"); configName != "" {
|
|
if c, ok := Plugins[configName]; ok {
|
|
c.Update(c.Modified)
|
|
w.Write([]byte("ok"))
|
|
} else {
|
|
http.Error(w, NO_SUCH_CONIFG, http.StatusNotFound)
|
|
}
|
|
} else {
|
|
Engine.Update(Engine.Modified)
|
|
w.Write([]byte("ok"))
|
|
}
|
|
}
|