nilopaim
/
apifiber
1
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases Wiki Activity

starting channel time limit verification

main
nilo 2025-03-13 17:36:49 -03:00
parent 46235de720
commit 9b2936c810
4 changed files with 70 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
controllers/webhookController.go
View File

@ -24,12 +24,10 @@ func KickSession(stream_name, session_id string) {
if err != nil { if err != nil {
log.Printf("Error Marshall KickSession: %s\n", err) log.Printf("Error Marshall KickSession: %s\n", err)
} }
resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonValue)) _, err = http.Post(url, "application/json", bytes.NewBuffer(jsonValue))
if err != nil { if err != nil {
log.Printf("Error Post KickSession: %s\n", err) log.Printf("Error Post KickSession: %s\n", err)
} }
log.Printf("KickSession status code: %d\n", resp.StatusCode)
} }
func ServerStart(c *fiber.Ctx) error { func ServerStart(c *fiber.Ctx) error {
@ -49,55 +47,56 @@ func ServerStart(c *fiber.Ctx) error {
return c.SendString("Server started: " + string(c.Body())) return c.SendString("Server started: " + string(c.Body()))
} }
func OnRtmpConnect(c *fiber.Ctx) error { // func OnRtmpConnect(c *fiber.Ctx) error {
body := new(models.Connect) // body := new(models.Connect)
if err := c.BodyParser(body); err != nil { // if err := c.BodyParser(body); err != nil {
log.Printf("Error Start: %s\n", err) // log.Printf("Error Start: %s\n", err)
return err // return err
} // }
utils.PrettyPrintJson(body) // utils.PrettyPrintJson(body)
return c.SendString("On_Rtmp_Connect: " + string(c.Body())) // return c.SendString("On_Rtmp_Connect: " + string(c.Body()))
} // }
func OnUpdate(c *fiber.Ctx) error { // func OnUpdate(c *fiber.Ctx) error {
p := new(models.Update) // p := new(models.Update)
if err := c.BodyParser(p); err != nil { // if err := c.BodyParser(p); err != nil {
log.Printf("Error Update: %s\n", err) // log.Printf("Error Update: %s\n", err)
return err // return err
} // }
log.Printf("======================== Update") // log.Printf("======================== Update")
utils.PrettyPrintJson(p) // utils.PrettyPrintJson(p)
if len(p.Groups) > 0 { // if len(p.Groups) > 0 {
for _, g := range p.Groups { // for _, g := range p.Groups {
log.Printf("======================== Update %s %s [(%dx%d) %d]\n", g.Channel, g.UpdPub.StartTime, g.VideoWidth, g.VideoHeight, g.UpdPub.ReadBytesSum) // log.Printf("======================== Update %s %s [(%dx%d) %d]\n", g.Channel, g.UpdPub.StartTime, g.VideoWidth, g.VideoHeight, g.UpdPub.ReadBytesSum)
} // }
} // }
return c.SendString("On_Update: " + string(c.Body())) // return c.SendString("On_Update: " + string(c.Body()))
} // }
func OnSubStart(c *fiber.Ctx) error {
p := new(models.Update)
if err := c.BodyParser(p); err != nil {
log.Printf("Error SubStart: %s\n", err)
return err
}
log.Printf("SubStart") // func OnSubStart(c *fiber.Ctx) error {
utils.PrettyPrintJson(p) // p := new(models.Update)
// if err := c.BodyParser(p); err != nil {
// log.Printf("Error SubStart: %s\n", err)
// return err
// }
// if len(p.Groups) > 0 { // log.Printf("SubStart")
// for _, g := range p.Groups { // utils.PrettyPrintJson(p)
// log.Printf("Update %s %s [(%dx%d) %d]\n", g.Channel, g.UpdPub.StartTime, g.VideoWidth, g.VideoHeight, g.UpdPub.ReadBytesSum)
// }
// }
return c.SendString("On_Substart: " + string(c.Body())) // // if len(p.Groups) > 0 {
} // // for _, g := range p.Groups {
// // log.Printf("Update %s %s [(%dx%d) %d]\n", g.Channel, g.UpdPub.StartTime, g.VideoWidth, g.VideoHeight, g.UpdPub.ReadBytesSum)
// // }
// // }
// return c.SendString("On_Substart: " + string(c.Body()))
// }
func OnPubStart(c *fiber.Ctx) error { func OnPubStart(c *fiber.Ctx) error {
// ================================================================ // ================================================================
@ -115,8 +114,9 @@ func OnPubStart(c *fiber.Ctx) error {
log.Printf("======================== Start StreamName %s, Transmission Key %s, PlayerKey %s, SessionId %s\n", p.StreamName, transmissionkey, playerkey, p.SessionId) log.Printf("======================== Start StreamName %s, Transmission Key %s, PlayerKey %s, SessionId %s\n", p.StreamName, transmissionkey, playerkey, p.SessionId)
// Get the channel from the database // Get the channel from the database
channel := services.GetChannelByName(p.StreamName) channel, currentTransmission := services.GetChannelByName(p.StreamName)
log.Printf("======================== %v\n", channel)
log.Printf("%v\n", currentTransmission)
// If the channel is not found, kick the session // If the channel is not found, kick the session
if channel.ID == 0 { if channel.ID == 0 {
@ -141,9 +141,6 @@ func OnPubStart(c *fiber.Ctx) error {
// Check if the channel has authorized time to transmit // Check if the channel has authorized time to transmit
// KickSession(p.StreamName, p.SessionId)
// return fiber.ErrForbidden
services.AddTransmissionlog(p.StreamName, "Transmissão iniciada") services.AddTransmissionlog(p.StreamName, "Transmissão iniciada")
return c.SendString("On_Pub_Start: " + string(c.Body())) return c.SendString("On_Pub_Start: " + string(c.Body()))
} }
@ -156,7 +153,8 @@ func OnPubStop(c *fiber.Ctx) error {
if err := c.BodyParser(p); err != nil { if err := c.BodyParser(p); err != nil {
return err return err
} }
log.Printf("======================== Stop Channel %s\n", p.StreamName)
services.AddTransmissionlog(p.StreamName, "Transmissão encerrada")
return c.SendString("On_Pub_Stop: " + string(c.Body())) return c.SendString("On_Pub_Stop: " + string(c.Body()))
} }

6
models/models.go
View File

@ -228,3 +228,9 @@ type TransmissionLog struct {
func (TransmissionLog) TableName() string { func (TransmissionLog) TableName() string {
return "transmissionlog" return "transmissionlog"
} }
type CurrentTransmission struct {
Channel string
StartTime time.Time
Limit time.Time
}

6
routes/routes.go
View File

@ -23,9 +23,9 @@ func Setup(app *fiber.App) {
// Webhooks // Webhooks
app.Post("/on_server_start", controllers.ServerStart) app.Post("/on_server_start", controllers.ServerStart)
app.Post("/on_rtmp_connect", controllers.OnRtmpConnect) // app.Post("/on_rtmp_connect", controllers.OnRtmpConnect)
app.Post("/on_update", controllers.OnUpdate) // app.Post("/on_update", controllers.OnUpdate)
app.Post("/on_pub_start", controllers.OnPubStart) app.Post("/on_pub_start", controllers.OnPubStart)
app.Post("/on_pub_stop", controllers.OnPubStop) app.Post("/on_pub_stop", controllers.OnPubStop)
app.Post("/on_sub_start", controllers.OnSubStart) // app.Post("/on_sub_start", controllers.OnSubStart)
} }

19
services/channelservices.go
View File

@ -3,12 +3,23 @@ package services
import ( import (
"api/globals" "api/globals"
"api/models" "api/models"
"time"
) )
func GetChannelByName(channelname string) models.Channel { func GetChannelByName(channelname string) (models.Channel, models.CurrentTransmission) {
var channel models.Channel var (
channel models.Channel
plan models.Plan
globals.DB.Debug().Where("name = ?", channelname).First(&channel) currentTransmission models.CurrentTransmission
)
return channel globals.DB.Where("name = ?", channelname).First(&channel)
globals.DB.First(&plan, channel.PlanID)
currentTransmission.Channel = channel.Name
currentTransmission.StartTime = time.Now()
currentTransmission.Limit = currentTransmission.StartTime.Add(time.Duration(plan.DailyLimitTransmission) * time.Minute)
return channel, currentTransmission
} }