mirror of
https://github.com/gowvp/gb28181.git
synced 2025-10-10 01:40:09 +08:00
109 lines
3.6 KiB
Go
109 lines
3.6 KiB
Go
package api
|
||
|
||
import (
|
||
"log/slog"
|
||
"net/url"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/gowvp/gb28181/internal/conf"
|
||
"github.com/gowvp/gb28181/internal/core/media"
|
||
"github.com/gowvp/gb28181/internal/core/sms"
|
||
"github.com/ixugo/goweb/pkg/web"
|
||
)
|
||
|
||
type WebHookAPI struct {
|
||
smsCore sms.Core
|
||
mediaCore media.Core
|
||
conf *conf.Bootstrap
|
||
}
|
||
|
||
func NewWebHookAPI(core sms.Core, mediaCore media.Core, conf *conf.Bootstrap) WebHookAPI {
|
||
return WebHookAPI{
|
||
smsCore: core,
|
||
mediaCore: mediaCore,
|
||
conf: conf,
|
||
}
|
||
}
|
||
|
||
func registerZLMWebhookAPI(r gin.IRouter, api WebHookAPI, handler ...gin.HandlerFunc) {
|
||
{
|
||
group := r.Group("/webhook", handler...)
|
||
group.POST("/on_server_keepalive", web.WarpH(api.onServerKeepalive))
|
||
group.POST("/on_stream_changed", web.WarpH(api.onStreamChanged))
|
||
group.POST("/on_publish", web.WarpH(api.onPublish))
|
||
group.POST("/on_play", web.WarpH(api.onPlay))
|
||
}
|
||
}
|
||
|
||
// onServerKeepalive 服务器定时上报时间，上报间隔可配置，默认 10s 上报一次
|
||
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_16%E3%80%81on-server-keepalive
|
||
func (w WebHookAPI) onServerKeepalive(_ *gin.Context, in *onServerKeepaliveInput) (gin.H, error) {
|
||
w.smsCore.Keepalive(in.MediaServerID)
|
||
return gin.H{}, nil
|
||
}
|
||
|
||
// onPublish rtsp/rtmp/rtp 推流鉴权事件。
|
||
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_7%E3%80%81on-publish
|
||
func (w WebHookAPI) onPublish(c *gin.Context, in *onPublishInput) (*onPublishOutput, error) {
|
||
if in.Schema == "rtmp" {
|
||
params, err := url.ParseQuery(in.Params)
|
||
if err != nil {
|
||
return &onPublishOutput{DefaultOutput: DefaultOutput{Code: 1, Msg: err.Error()}}, nil
|
||
}
|
||
sign := params.Get("sign")
|
||
if err := w.mediaCore.Publish(c.Request.Context(), media.PublishInput{
|
||
App: in.App,
|
||
Stream: in.Stream,
|
||
MediaServerID: in.MediaServerID,
|
||
Sign: sign,
|
||
Secret: w.conf.Server.RTMPSecret,
|
||
Session: params.Get("session"),
|
||
}); err != nil {
|
||
return &onPublishOutput{DefaultOutput: DefaultOutput{Code: 1, Msg: err.Error()}}, nil
|
||
}
|
||
}
|
||
return &onPublishOutput{
|
||
DefaultOutput: newDefaultOutputOK(),
|
||
}, nil
|
||
}
|
||
|
||
// onStreamChanged rtsp/rtmp 流注册或注销时触发此事件；此事件对回复不敏感。
|
||
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_12%E3%80%81on-stream-changed
|
||
func (w WebHookAPI) onStreamChanged(c *gin.Context, in *onStreamChangedInput) (DefaultOutput, error) {
|
||
switch in.Schema {
|
||
case "rtmp":
|
||
if !in.Regist {
|
||
if err := w.mediaCore.UnPublish(c.Request.Context(), in.App, in.Stream); err != nil {
|
||
slog.Error("UnPublish", "err", err)
|
||
}
|
||
}
|
||
case "rtsp":
|
||
}
|
||
return newDefaultOutputOK(), nil
|
||
}
|
||
|
||
// onPlay rtsp/rtmp/http-flv/ws-flv/hls 播放触发播放器身份验证事件。
|
||
// 播放流时会触发此事件。如果流不存在，则首先触发 on_play 事件，然后触发 on_stream_not_found 事件。
|
||
// 播放rtsp流时，如果该流开启了rtsp专用认证（on_rtsp_realm），则不会触发on_play事件。
|
||
// https://docs.zlmediakit.com/guide/media_server/web_hook_api.html#_6-on-play
|
||
func (w WebHookAPI) onPlay(c *gin.Context, in *onPublishInput) (DefaultOutput, error) {
|
||
switch in.Schema {
|
||
case "rtmp":
|
||
params, err := url.ParseQuery(in.Params)
|
||
if err != nil {
|
||
return DefaultOutput{Code: 1, Msg: err.Error()}, nil
|
||
}
|
||
session := params.Get("session")
|
||
if err := w.mediaCore.OnPlay(c.Request.Context(), media.OnPlayInput{
|
||
App: in.App,
|
||
Stream: in.Stream,
|
||
Session: session,
|
||
}); err != nil {
|
||
return DefaultOutput{Code: 1, Msg: err.Error()}, nil
|
||
}
|
||
case "rtsp":
|
||
}
|
||
|
||
return newDefaultOutputOK(), nil
|
||
}
|