mirror of
https://github.com/lkmio/gb-cms.git
synced 2025-12-24 11:51:52 +08:00
346 lines
16 KiB
Go
346 lines
16 KiB
Go
package api
|
||
|
||
import (
|
||
"gb-cms/common"
|
||
"gb-cms/dao"
|
||
"github.com/gorilla/mux"
|
||
"github.com/gorilla/websocket"
|
||
"net/http"
|
||
"os"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type ApiServer struct {
|
||
router *mux.Router
|
||
upgrader *websocket.Upgrader
|
||
}
|
||
|
||
type InviteParams struct {
|
||
DeviceID string `json:"serial"`
|
||
ChannelID string `json:"code"`
|
||
StartTime string `json:"starttime"`
|
||
EndTime string `json:"endtime"`
|
||
Setup string `json:"setup"`
|
||
Speed string `json:"speed"`
|
||
Token string `json:"token"`
|
||
Download bool `json:"download"`
|
||
streamId common.StreamID
|
||
}
|
||
|
||
type StreamParams struct {
|
||
Stream common.StreamID `json:"stream"` // Source
|
||
Protocol int `json:"protocol"` // 推拉流协议
|
||
RemoteAddr string `json:"remote_addr"` // peer地址
|
||
}
|
||
|
||
type PlayDoneParams struct {
|
||
StreamParams
|
||
Sink string `json:"sink"`
|
||
}
|
||
|
||
type QueryRecordParams struct {
|
||
DeviceID string `json:"serial"`
|
||
ChannelID string `json:"code"`
|
||
Timeout int `json:"timeout"`
|
||
StartTime string `json:"starttime"`
|
||
EndTime string `json:"endtime"`
|
||
//Type_ string `json:"type"`
|
||
Command string `json:"command"` // 云台控制命令 left/up/right/down/zoomin/zoomout
|
||
}
|
||
|
||
type DeviceChannelID struct {
|
||
DeviceID string `json:"device_id"`
|
||
ChannelID string `json:"channel_id"`
|
||
}
|
||
|
||
type SeekParams struct {
|
||
StreamId common.StreamID `json:"stream_id"`
|
||
Seconds int `json:"seconds"`
|
||
}
|
||
|
||
type BroadcastParams struct {
|
||
DeviceID string `json:"device_id"`
|
||
ChannelID string `json:"channel_id"`
|
||
StreamId common.StreamID `json:"stream_id"`
|
||
Setup *common.SetupType `json:"setup"`
|
||
}
|
||
|
||
type RecordParams struct {
|
||
StreamParams
|
||
Path string `json:"path"`
|
||
}
|
||
|
||
type StreamIDParams struct {
|
||
StreamID common.StreamID `json:"streamid"`
|
||
Command string `json:"command"`
|
||
Scale float64 `json:"scale"`
|
||
}
|
||
|
||
type PageQuery struct {
|
||
PageNumber *int `json:"page_number"` // 页数
|
||
PageSize *int `json:"page_size"` // 每页大小
|
||
TotalPages int `json:"total_pages"` // 总页数
|
||
TotalCount int `json:"total_count"` // 总记录数
|
||
Data interface{} `json:"data"`
|
||
}
|
||
|
||
type SetMediaTransportReq struct {
|
||
DeviceID string `json:"serial"`
|
||
MediaTransport string `json:"media_transport"`
|
||
MediaTransportMode string `json:"media_transport_mode"`
|
||
}
|
||
|
||
// QueryDeviceChannel 查询设备和通道的参数
|
||
type QueryDeviceChannel struct {
|
||
DeviceID string `json:"serial"`
|
||
GroupID string `json:"dir_serial"`
|
||
PCode string `json:"pcode"`
|
||
Start int `json:"start"`
|
||
Limit int `json:"limit"`
|
||
Keyword string `json:"q"`
|
||
Online string `json:"online"`
|
||
Enable string `json:"enable"`
|
||
ChannelType string `json:"channel_type"` // dir-查询子目录
|
||
Order string `json:"order"` // asc/desc
|
||
Sort string `json:"sort"` // Channel-根据数据库ID排序/iD-根据通道ID排序
|
||
SMS string `json:"sms"`
|
||
Filter string `json:"filter"`
|
||
|
||
Priority int `json:"priority"` // 报警参数
|
||
Method int `json:"method"`
|
||
StartTime string `json:"starttime"`
|
||
EndTime string `json:"endtime"`
|
||
}
|
||
|
||
type DeleteDevice struct {
|
||
DeviceID string `json:"serial"`
|
||
IP string `json:"ip"`
|
||
Forbid bool `json:"forbid"`
|
||
UA string `json:"ua"`
|
||
}
|
||
|
||
type SetEnable struct {
|
||
ID int `json:"id"`
|
||
Enable bool `json:"enable"`
|
||
ShareAllChannel bool `json:"shareallchannel"`
|
||
}
|
||
|
||
type QueryCascadeChannelList struct {
|
||
QueryDeviceChannel
|
||
ID string `json:"id"`
|
||
Related bool `json:"related"` // 只看已选择
|
||
Reverse bool `json:"reverse"`
|
||
}
|
||
|
||
type ChannelListResult struct {
|
||
ChannelCount int `json:"ChannelCount"`
|
||
ChannelList []*LiveGBSChannel `json:"ChannelList"`
|
||
}
|
||
|
||
type CascadeChannel struct {
|
||
CascadeID string
|
||
*LiveGBSChannel
|
||
}
|
||
|
||
type CustomChannel struct {
|
||
DeviceID string `json:"serial"`
|
||
ChannelID string `json:"code"`
|
||
CustomID string `json:"id"`
|
||
}
|
||
|
||
type DeviceInfo struct {
|
||
DeviceID string `json:"serial"`
|
||
CustomName string `json:"custom_name"`
|
||
MediaTransport string `json:"media_transport"`
|
||
MediaTransportMode string `json:"media_transport_mode"`
|
||
StreamMode string `json:"stream_mode"`
|
||
SMSID string `json:"sms_id"`
|
||
SMSGroupID string `json:"sms_group_id"`
|
||
RecvStreamIP string `json:"recv_stream_ip"`
|
||
ContactIP string `json:"contact_ip"`
|
||
Charset string `json:"charset"`
|
||
CatalogInterval int `json:"catalog_interval"`
|
||
SubscribeInterval int `json:"subscribe_interval"`
|
||
CatalogSubscribe bool `json:"catalog_subscribe"`
|
||
AlarmSubscribe bool `json:"alarm_subscribe"`
|
||
PositionSubscribe bool `json:"position_subscribe"`
|
||
PTZSubscribe bool `json:"ptz_subscribe"`
|
||
RecordCenter bool `json:"record_center"`
|
||
RecordIndistinct bool `json:"record_indistinct"`
|
||
CivilCodeFirst bool `json:"civil_code_first"`
|
||
KeepOriginalTree bool `json:"keep_original_tree"`
|
||
Password string `json:"password"`
|
||
DropChannelType string `json:"drop_channel_type"`
|
||
Longitude float64 `json:"longitude"`
|
||
Latitude float64 `json:"latitude"`
|
||
}
|
||
|
||
type Empty struct {
|
||
}
|
||
|
||
var apiServer *ApiServer
|
||
|
||
func init() {
|
||
apiServer = &ApiServer{
|
||
upgrader: &websocket.Upgrader{
|
||
CheckOrigin: func(r *http.Request) bool {
|
||
return true
|
||
},
|
||
},
|
||
|
||
router: mux.NewRouter(),
|
||
}
|
||
}
|
||
|
||
// 验证和刷新token
|
||
func withVerify(f func(w http.ResponseWriter, req *http.Request)) func(http.ResponseWriter, *http.Request) {
|
||
return func(w http.ResponseWriter, req *http.Request) {
|
||
cookie, err := req.Cookie("token")
|
||
if err != nil {
|
||
w.WriteHeader(http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
ok := TokenManager.Refresh(cookie.Value, time.Now())
|
||
if !ok {
|
||
w.WriteHeader(http.StatusUnauthorized)
|
||
return
|
||
} else if AdminMD5 == PwdMD5 && req.URL.Path != "/api/v1/modifypassword" && req.URL.Path != "/api/v1/userinfo" {
|
||
// 如果没有修改默认密码, 只允许放行这2个接口
|
||
return
|
||
}
|
||
|
||
f(w, req)
|
||
}
|
||
}
|
||
|
||
func withVerify2(onSuccess func(w http.ResponseWriter, req *http.Request), onFailure func(w http.ResponseWriter, req *http.Request)) func(http.ResponseWriter, *http.Request) {
|
||
return func(w http.ResponseWriter, req *http.Request) {
|
||
cookie, err := req.Cookie("token")
|
||
if err == nil && TokenManager.Refresh(cookie.Value, time.Now()) {
|
||
onSuccess(w, req)
|
||
} else {
|
||
onFailure(w, req)
|
||
}
|
||
}
|
||
}
|
||
|
||
func StartApiServer(addr string) {
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_play", common.WithJsonParams(apiServer.OnPlay, &PlayDoneParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_play_done", common.WithJsonParams(apiServer.OnPlayDone, &PlayDoneParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_publish", common.WithJsonParams(apiServer.OnPublish, &StreamParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_publish_done", common.WithJsonParams(apiServer.OnPublishDone, &StreamParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_idle_timeout", common.WithJsonParams(apiServer.OnIdleTimeout, &StreamParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_receive_timeout", common.WithJsonParams(apiServer.OnReceiveTimeout, &StreamParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_record", common.WithJsonParams(apiServer.OnRecord, &RecordParams{}))
|
||
apiServer.router.HandleFunc("/api/v1/hook/on_started", apiServer.OnStarted)
|
||
|
||
apiServer.router.HandleFunc("/api/v1/stream/start", withVerify(common.WithFormDataParams(apiServer.OnStreamStart, InviteParams{}))) // 实时预览
|
||
apiServer.router.HandleFunc("/api/v1/stream/stop", withVerify(common.WithFormDataParams(apiServer.OnCloseLiveStream, InviteParams{}))) // 关闭实时预览
|
||
apiServer.router.HandleFunc("/api/v1/playback/start", withVerify(common.WithFormDataParams(apiServer.OnPlaybackStart, InviteParams{}))) // 回放/下载
|
||
apiServer.router.HandleFunc("/api/v1/playback/stop", withVerify(common.WithFormDataParams(apiServer.OnCloseStream, StreamIDParams{}))) // 关闭回放/下载
|
||
apiServer.router.HandleFunc("/api/v1/playback/control", withVerify(common.WithFormDataParams(apiServer.OnPlaybackControl, StreamIDParams{}))) // 回放控制
|
||
|
||
apiServer.router.HandleFunc("/api/v1/device/list", withVerify(common.WithQueryStringParams(apiServer.OnDeviceList, QueryDeviceChannel{}))) // 查询设备列表
|
||
apiServer.router.HandleFunc("/api/v1/device/channeltree", withVerify(common.WithQueryStringParams(apiServer.OnDeviceTree, QueryDeviceChannel{}))) // 设备树
|
||
apiServer.router.HandleFunc("/api/v1/device/channellist", withVerify(common.WithQueryStringParams(apiServer.OnChannelList, QueryDeviceChannel{}))) // 查询通道列表
|
||
apiServer.router.HandleFunc("/api/v1/device/fetchcatalog", withVerify(common.WithQueryStringParams(apiServer.OnCatalogQuery, QueryDeviceChannel{}))) // 更新通道
|
||
apiServer.router.HandleFunc("/api/v1/device/remove", withVerify(common.WithFormDataParams(apiServer.OnDeviceRemove, DeleteDevice{}))) // 删除设备
|
||
apiServer.router.HandleFunc("/api/v1/device/setmediatransport", withVerify(common.WithFormDataParams(apiServer.OnDeviceMediaTransportSet, SetMediaTransportReq{}))) // 设置设备媒体传输模式
|
||
|
||
apiServer.router.HandleFunc("/api/v1/playback/recordlist", withVerify(common.WithQueryStringParams(apiServer.OnRecordList, QueryRecordParams{}))) // 查询录像列表
|
||
apiServer.router.HandleFunc("/api/v1/stream/info", withVerify(apiServer.OnStreamInfo))
|
||
apiServer.router.HandleFunc("/api/v1/playback/streaminfo", withVerify(apiServer.OnStreamInfo))
|
||
apiServer.router.HandleFunc("/api/v1/device/session/list", withVerify(common.WithQueryStringParams(apiServer.OnSessionList, QueryDeviceChannel{}))) // 推流列表
|
||
apiServer.router.HandleFunc("/api/v1/device/session/stop", withVerify(common.WithFormDataParams(apiServer.OnSessionStop, StreamIDParams{}))) // 关闭流
|
||
apiServer.router.HandleFunc("/api/v1/device/setchannelid", withVerify(common.WithFormDataParams(apiServer.OnCustomChannelSet, CustomChannel{}))) // 自定义通道ID
|
||
|
||
apiServer.router.HandleFunc("/api/v1/playback/seek", common.WithJsonResponse(apiServer.OnSeekPlayback, &SeekParams{})) // 回放seek
|
||
apiServer.router.HandleFunc("/api/v1/control/ptz", withVerify(common.WithFormDataParams(apiServer.OnPTZControl, QueryRecordParams{}))) // 云台控制
|
||
|
||
apiServer.router.HandleFunc("/api/v1/cascade/list", withVerify(common.WithQueryStringParams(apiServer.OnPlatformList, QueryDeviceChannel{}))) // 级联设备列表
|
||
apiServer.router.HandleFunc("/api/v1/cascade/save", withVerify(common.WithFormDataParams(apiServer.OnPlatformAdd, LiveGBSCascade{}))) // 添加级联设备
|
||
apiServer.router.HandleFunc("/api/v1/cascade/setenable", withVerify(common.WithFormDataParams(apiServer.OnEnableSet, SetEnable{}))) // 添加级联设备
|
||
apiServer.router.HandleFunc("/api/v1/cascade/remove", withVerify(common.WithFormDataParams(apiServer.OnPlatformRemove, SetEnable{}))) // 删除级联设备
|
||
apiServer.router.HandleFunc("/api/v1/cascade/channellist", withVerify(common.WithQueryStringParams(apiServer.OnPlatformChannelList, QueryCascadeChannelList{}))) // 级联设备通道列表
|
||
|
||
apiServer.router.HandleFunc("/api/v1/cascade/savechannels", withVerify(apiServer.OnPlatformChannelBind)) // 级联绑定通道
|
||
apiServer.router.HandleFunc("/api/v1/cascade/removechannels", withVerify(apiServer.OnPlatformChannelUnbind)) // 级联解绑通道
|
||
apiServer.router.HandleFunc("/api/v1/cascade/setshareallchannel", withVerify(common.WithFormDataParams(apiServer.OnShareAllChannel, SetEnable{}))) // 开启或取消级联所有通道
|
||
apiServer.router.HandleFunc("/api/v1/cascade/pushcatalog", withVerify(common.WithFormDataParams(apiServer.OnCatalogPush, SetEnable{}))) // 推送目录
|
||
apiServer.router.HandleFunc("/api/v1/device/setinfo", withVerify(common.WithFormDataParams(apiServer.OnDeviceInfoSet, DeviceInfo{}))) // 编辑设备信息
|
||
apiServer.router.HandleFunc("/api/v1/alarm/list", withVerify(common.WithQueryStringParams(apiServer.OnAlarmList, QueryDeviceChannel{}))) // 报警查询
|
||
apiServer.router.HandleFunc("/api/v1/alarm/remove", withVerify(common.WithFormDataParams(apiServer.OnAlarmRemove, SetEnable{}))) // 删除报警
|
||
apiServer.router.HandleFunc("/api/v1/alarm/clear", withVerify(common.WithFormDataParams(apiServer.OnAlarmClear, Empty{}))) // 清空报警
|
||
|
||
// 暂未开发
|
||
apiServer.router.HandleFunc("/api/v1/sms/list", withVerify(func(w http.ResponseWriter, req *http.Request) {})) // 流媒体服务器列表
|
||
apiServer.router.HandleFunc("/api/v1/cloudrecord/querychannels", withVerify(func(w http.ResponseWriter, req *http.Request) {})) // 云端录像
|
||
apiServer.router.HandleFunc("/api/v1/user/list", withVerify(func(w http.ResponseWriter, req *http.Request) {})) // 用户管理
|
||
apiServer.router.HandleFunc("/api/v1/log/list", withVerify(func(w http.ResponseWriter, req *http.Request) {})) // 操作日志
|
||
apiServer.router.HandleFunc("/api/v1/getbaseconfig", withVerify(func(w http.ResponseWriter, req *http.Request) {}))
|
||
apiServer.router.HandleFunc("/api/v1/gm/cert/list", withVerify(func(w http.ResponseWriter, req *http.Request) {}))
|
||
apiServer.router.HandleFunc("/api/v1/getrequestkey", withVerify(func(w http.ResponseWriter, req *http.Request) {}))
|
||
|
||
apiServer.router.HandleFunc("/api/v1/record/start", withVerify(apiServer.OnRecordStart)) // 开启录制
|
||
apiServer.router.HandleFunc("/api/v1/record/stop", withVerify(apiServer.OnRecordStop)) // 关闭录制
|
||
|
||
apiServer.router.HandleFunc("/api/v1/broadcast/invite", common.WithJsonResponse(apiServer.OnBroadcast, &BroadcastParams{Setup: &common.DefaultSetupType})) // 发起语音广播
|
||
apiServer.router.HandleFunc("/api/v1/broadcast/hangup", common.WithJsonResponse(apiServer.OnHangup, &BroadcastParams{})) // 挂断广播会话
|
||
apiServer.router.HandleFunc("/api/v1/control/ws-talk/{device}/{channel}", withVerify(apiServer.OnTalk)) // 一对一语音对讲
|
||
|
||
apiServer.router.HandleFunc("/api/v1/jt/device/add", common.WithJsonResponse(apiServer.OnVirtualDeviceAdd, &dao.JTDeviceModel{}))
|
||
apiServer.router.HandleFunc("/api/v1/jt/device/edit", common.WithJsonResponse(apiServer.OnVirtualDeviceEdit, &dao.JTDeviceModel{}))
|
||
apiServer.router.HandleFunc("/api/v1/jt/device/remove", common.WithJsonResponse(apiServer.OnVirtualDeviceRemove, &dao.JTDeviceModel{}))
|
||
apiServer.router.HandleFunc("/api/v1/jt/device/list", common.WithJsonResponse(apiServer.OnVirtualDeviceList, &PageQuery{}))
|
||
|
||
apiServer.router.HandleFunc("/api/v1/jt/channel/add", common.WithJsonResponse(apiServer.OnVirtualChannelAdd, &dao.ChannelModel{}))
|
||
apiServer.router.HandleFunc("/api/v1/jt/channel/edit", common.WithJsonResponse(apiServer.OnVirtualChannelEdit, &dao.ChannelModel{}))
|
||
apiServer.router.HandleFunc("/api/v1/jt/channel/remove", common.WithJsonResponse(apiServer.OnVirtualChannelRemove, &dao.ChannelModel{}))
|
||
apiServer.router.HandleFunc("/logout", func(writer http.ResponseWriter, req *http.Request) {
|
||
cookie, err := req.Cookie("token")
|
||
if err == nil {
|
||
TokenManager.Remove(cookie.Value)
|
||
writer.Header().Set("Location", "/login.html")
|
||
writer.WriteHeader(http.StatusFound)
|
||
return
|
||
}
|
||
})
|
||
|
||
registerLiveGBSApi()
|
||
|
||
// 前端路由
|
||
htmlRoot := "./html/"
|
||
fileServer := http.FileServer(http.Dir(htmlRoot))
|
||
apiServer.router.PathPrefix("/").HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
|
||
// 处理无扩展名的路径，自动添加.html扩展名
|
||
path := request.URL.Path
|
||
if !strings.Contains(path, ".") {
|
||
// 检查是否存在对应的.html文件
|
||
htmlPath := htmlRoot + path + ".html"
|
||
if _, err := os.Stat(htmlPath); err == nil {
|
||
// 如果存在对应的.html文件，则直接返回该文件
|
||
http.ServeFile(writer, request, htmlPath)
|
||
return
|
||
}
|
||
}
|
||
|
||
fileServer.ServeHTTP(writer, request)
|
||
})
|
||
|
||
srv := &http.Server{
|
||
Handler: apiServer.router,
|
||
Addr: addr,
|
||
// Good practice: enforce timeouts for servers you create!
|
||
WriteTimeout: 30 * time.Second,
|
||
ReadTimeout: 30 * time.Second,
|
||
}
|
||
|
||
err := srv.ListenAndServe()
|
||
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
}
|