mirror of
https://git.nadeko.net/Fijxu/invidious.git
synced 2025-12-29 02:51:25 +00:00
c124c4854e
Some checks are pending
Build and release container directly from master / release (push) Waiting to run
Invidious CI / build - crystal: ${{ matrix.crystal }}, stable: ${{ matrix.stable }} (1.12.1, true) (push) Waiting to run
Invidious CI / build - crystal: ${{ matrix.crystal }}, stable: ${{ matrix.stable }} (1.13.2, true) (push) Waiting to run
Invidious CI / build - crystal: ${{ matrix.crystal }}, stable: ${{ matrix.stable }} (1.14.0, true) (push) Waiting to run
Invidious CI / build - crystal: ${{ matrix.crystal }}, stable: ${{ matrix.stable }} (1.15.0, true) (push) Waiting to run
Invidious CI / build - crystal: ${{ matrix.crystal }}, stable: ${{ matrix.stable }} (nightly, false) (push) Waiting to run
Invidious CI / build-docker (push) Waiting to run
Invidious CI / build-docker-arm64 (push) Waiting to run
Invidious CI / lint (push) Waiting to run
86 lines
2.3 KiB
Crystal
86 lines
2.3 KiB
Crystal
module BackendInfo
|
|
extend self
|
|
@@exvpp_url : Array(String) = Array.new(CONFIG.invidious_companion.size, "")
|
|
@@status : Array(Int32) = Array.new(CONFIG.invidious_companion.size, 0)
|
|
@@csp : Array(String) = Array.new(CONFIG.invidious_companion.size, "")
|
|
@@mutex : Mutex = Mutex.new
|
|
|
|
def check_backends
|
|
check_companion()
|
|
end
|
|
|
|
private def check_companion
|
|
CONFIG.invidious_companion.each_with_index do |companion, index|
|
|
spawn do
|
|
begin
|
|
response = HTTP::Client.get "#{companion.private_url}/healthz"
|
|
if response.status_code == 200
|
|
check_videoplayback_proxy(companion, index)
|
|
generate_csp([companion.public_url.to_s, companion.i2p_public_url.to_s], @@exvpp_url[index], index)
|
|
else
|
|
@@status[index] = 0
|
|
end
|
|
rescue
|
|
@@status[index] = 0
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
private def check_videoplayback_proxy(companion : Config::CompanionConfig, index : Int32)
|
|
begin
|
|
info = HTTP::Client.get "#{companion.private_url}/info"
|
|
exvpp_url = JSON.parse(info.body)["external_videoplayback_proxy"]?.try &.to_s
|
|
rescue JSON::ParseException
|
|
@@status[index] = 2
|
|
return
|
|
end
|
|
|
|
exvpp_url = "" if exvpp_url.nil?
|
|
@@exvpp_url[index] = exvpp_url
|
|
if exvpp_url.empty?
|
|
@@status[index] = 2
|
|
return
|
|
else
|
|
begin
|
|
exvpp_health = HTTP::Client.get "#{exvpp_url}/health"
|
|
if exvpp_health.status_code == 200
|
|
@@status[index] = 2
|
|
return exvpp_url
|
|
else
|
|
@@status[index] = 1
|
|
end
|
|
rescue
|
|
@@status[index] = 1
|
|
end
|
|
end
|
|
end
|
|
|
|
private def generate_csp(companion_url : Array(String), exvpp_url : String? = nil, index : Int32? = nil)
|
|
@@mutex.synchronize do
|
|
@@csp[index] = ""
|
|
companion_url.each do |url|
|
|
@@csp[index] += " #{url}"
|
|
end
|
|
@@csp[index] += " #{exvpp_url}"
|
|
end
|
|
end
|
|
|
|
def get_status
|
|
return @@status
|
|
end
|
|
|
|
def get_exvpp
|
|
return @@exvpp_url
|
|
end
|
|
|
|
def get_csp(index : Int32)
|
|
# A little mutex to prevent sending a partial CSP header
|
|
# Not sure if this is necessary. But if the @@csp[index] is being assigned
|
|
# at the same time when it's being accessed, a data race will appear
|
|
@@mutex.synchronize do
|
|
return @@csp[index], @@csp[index]
|
|
end
|
|
end
|
|
end
|