feat: initial commit

legacy/updater.lua

@@ -0,0 +1,88 @@
+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