88 lines
2.7 KiB
Lua
88 lines
2.7 KiB
Lua
local cjson = require('cjson')
|
|
local http = require('socket.http')
|
|
local stdlib = require('posix.stdlib')
|
|
local redis = require('redis')
|
|
|
|
local client = redis.connect('redis', 6379)
|
|
|
|
local function get_mirrors(source)
|
|
local body, _, _, _ = http.request(source)
|
|
return cjson.decode(body)
|
|
end
|
|
|
|
local function prepare_pool_map(mirrors)
|
|
local pool_map = {}
|
|
for _, site in pairs(mirrors['sites']) do
|
|
for _, alternative in pairs(site['available_alternatives']) do
|
|
if alternative['type'] == 'mirror' then
|
|
local canonical_host = site['main_domain']
|
|
if select(2, canonical_host:gsub("%.", "")) == 1 then
|
|
canonical_host = "www." .. canonical_host
|
|
end
|
|
pool_map[canonical_host] = alternative['url']:sub(9)
|
|
break
|
|
end
|
|
end
|
|
end
|
|
return pool_map
|
|
end
|
|
|
|
local function redis_set(client, key, data)
|
|
local res, err = client:set(key, data)
|
|
if not res then
|
|
print("Error setting key in Redis:", err)
|
|
else
|
|
print("Data successfully stored in Redis under '" .. key .. "'")
|
|
end
|
|
end
|
|
|
|
local function get_config(host)
|
|
local function read_file(file_path)
|
|
local file = io.open(file_path, "r")
|
|
if file then
|
|
local content = file:read("*a")
|
|
file:close()
|
|
return content
|
|
else
|
|
return nil
|
|
end
|
|
end
|
|
local file_paths = {
|
|
"/configs/" .. host .. ".json", -- Check host-specific file
|
|
"/configs/default.json" -- Check default file
|
|
}
|
|
for _, file_path in ipairs(file_paths) do
|
|
local content = read_file(file_path)
|
|
if content then
|
|
print("Using " .. file_path .. " for " .. host .. " configuration.")
|
|
return content
|
|
end
|
|
end
|
|
return "{}"
|
|
end
|
|
|
|
local function update_pool_data(pool_name, pool_source)
|
|
local pool_map = prepare_pool_map(get_mirrors(pool_source))
|
|
redis_set(client, 'jasima:poolmap:' .. pool_name, cjson.encode(pool_map))
|
|
for host, _ in pairs(pool_map) do
|
|
local redis_key = 'jasima:config:' .. host
|
|
local config = get_config(host)
|
|
redis_set(client, redis_key, config)
|
|
end
|
|
end
|
|
|
|
local interval = 5 * 60 -- 5 minutes in seconds
|
|
local last_update_time = os.clock() - interval
|
|
|
|
while true do
|
|
local current_time = os.clock()
|
|
if current_time - last_update_time >= interval then
|
|
for pool_name, pool_source in pairs(stdlib.getenv()) do
|
|
if pool_name:match("^JASIMA_MIRRORS_POOL_") then
|
|
local pool_name = pool_name:sub(21)
|
|
update_pool_data(pool_name, pool_source)
|
|
end
|
|
end
|
|
last_update_time = current_time
|
|
end
|
|
end
|