Replace master branch with local files

This commit is contained in:
doomtube 2025-08-03 21:53:15 -04:00
commit 875a53f499
60 changed files with 21637 additions and 0 deletions

View file

@ -0,0 +1,84 @@
local redis_helper = require "redis_helper"
local cjson = require "cjson"
-- Safely load the http module
local has_http, http = pcall(require, "resty.http")
local function disconnect_stream(stream_key)
if not has_http then
ngx.log(ngx.WARN, "resty.http module not available, skipping stream disconnection")
return
end
local httpc = http.new()
local ome_url = os.getenv("OME_URL") or "http://ovenmediaengine:8081"
local ome_token = os.getenv("OME_API_TOKEN") or "your-api-token"
-- Get active streams from OME
local res, err = httpc:request_uri(ome_url .. "/v1/vhosts/default/apps/app/streams", {
method = "GET",
headers = {
["Authorization"] = "Bearer " .. ome_token,
["Content-Type"] = "application/json"
}
})
if not res then
ngx.log(ngx.ERR, "Failed to get streams: ", err)
return
end
local ok, data = pcall(cjson.decode, res.body)
if ok and data and data.response and data.response.streams then
for _, stream in ipairs(data.response.streams) do
if stream.name == stream_key then
-- Disconnect the stream
local del_res, del_err = httpc:request_uri(
ome_url .. "/v1/vhosts/default/apps/app/streams/" .. stream.name,
{
method = "DELETE",
headers = {
["Authorization"] = "Bearer " .. ome_token
}
}
)
if del_res and del_res.status == 200 then
ngx.log(ngx.INFO, "Disconnected stream: ", stream_key)
else
ngx.log(ngx.ERR, "Failed to disconnect stream: ", stream_key)
end
break
end
end
end
end
local function monitor_streams()
-- Check if Redis is available
local ok, keys = pcall(redis_helper.get_streams_to_disconnect)
if not ok then
ngx.log(ngx.WARN, "Redis not available yet")
return
end
for _, key in ipairs(keys or {}) do
disconnect_stream(key)
redis_helper.remove_stream_from_disconnect(key)
end
end
-- Start monitoring in a timer with error handling
local function start_monitoring()
local ok, err = ngx.timer.every(1, monitor_streams)
if not ok then
ngx.log(ngx.ERR, "Failed to create timer: ", err)
else
ngx.log(ngx.INFO, "Stream monitor started")
end
end
-- Delay start to ensure services are ready
ngx.timer.at(5, start_monitoring)