Редагування Модуль:FetchData2

local p = {}

-- Цільова дата для розрахунку днів (25.10.2024)
local TARGET_DATE_STR = "25.10.2024"
local TARGET_TIMESTAMP = os.time({year=2024, month=10, day=25, hour=0, min=0, sec=0}) -- Припускаємо однаковий часовий пояс для обох дат

-- Допоміжна функція для вилучення значення з таблиці на вікі-сторінці
-- page_title: Назва сторінки, звідки брати дані (наприклад, "Гравці")
-- player_name: Нік гравця (для пошуку в першій колонці)
-- column_index: Індекс колонки, значення якої потрібно повернути (починаючи з 2)
local function fetch_data_from_table(page_title, player_name, column_index)
    local title = mw.title.new(page_title)
    local content = title and title:getContent()

    if not content then
        return nil
    end

    -- Регулярний вираз для пошуку таблиці (wikitable або будь-яка таблиця)
    local table_content = mw.ustring.match(content, "{|%s*class%s*=%s*\"wikitable\"(.-)|}")
    if not table_content then
        table_content = mw.ustring.match(content, "{|(.+?)|}")
    end

    if not table_content then
        return nil
    end

    -- Екранування спеціальних символів у імені гравця для regex
    local escaped_player = mw.ustring.gsub(player_name, "([%(%)%.%-%+%[%]])", "%%%1")
    
    -- Шаблони пошуку: шукає рядок, що починається з першої колонки, яка містить [[гравець]] або гравець
    local pattern = "|%s*%[%[" .. escaped_player .. "%]%]" .. "((.-)|)"
    local direct_pattern = "|%s*" .. escaped_player .. "%s*" .. "((.-)|)"
    
    local row_match = mw.ustring.match(table_content, pattern) or mw.ustring.match(table_content, direct_pattern)

    if not row_match then
        return nil
    end

    -- Розділяємо захоплену частину (всі колонки після першої) на окремі значення
    -- Використовуємо функцію string.split, але оскільки її немає в базовому mw, 
    -- використаємо gmatch і тимчасову таблицю.
    local columns = {}
    for value in mw.ustring.gmatch(row_match, "[^|]+") do
        table.insert(columns, mw.text.trim(value))
    end

    -- Індекс колонки 2 у вікі-таблиці відповідає індексу 1 у масиві 'columns' (тобто column_index - 1)
    local result = columns[column_index - 1] 

    if result then
        -- Видаляємо зайві [[ та ]]
        result = mw.text.trim(mw.ustring.gsub(result, "%[?%[?(.+)%]%]?", "%1"))
        return result
    else
        return nil
    end
end

-- 1. Дата приєднання (Гравці, колонка 2)
-- Формат: DD.MM.YYYY (XXX д)
function p.date_added(frame)
    local player = frame.args.player
    local date_str = fetch_data_from_table("Гравці", player, 2) 

    if not date_str or date_str == "" then
        return "''Невідомо''"
    end

    -- Припускаємо формат DD.MM.YYYY
    local day, month, year = mw.ustring.match(date_str, "(%d%d)%.(%d%d)%.(%d%d%d%d)")
    
    if not day then
        return date_str .. " (Помилка формату)"
    end

    local join_timestamp = os.time({year=tonumber(year), month=tonumber(month), day=tonumber(day), hour=0, min=0, sec=0})
    
    -- Розрахунок різниці в секундах
    local diff_seconds = TARGET_TIMESTAMP - join_timestamp
    
    if diff_seconds < 0 then
        return date_str -- Якщо дата в майбутньому (хоча не повинна)
    end

    -- Переведення секунд у дні (86400 сек/день)
    local diff_days = math.floor(diff_seconds / 86400)
    
    return string.format("%s (%d д)", date_str, diff_days)
end

-- 2. Провідник (Гравці, колонка 3)
-- Формат: [[Демон|Демон]]
function p.recruiter(frame)
    local player = frame.args.player
    local recruiter_name = fetch_data_from_table("Гравці", player, 3) 

    if not recruiter_name or recruiter_name == "" or recruiter_name == "Не вказано" then
        return "Не вказано"
    end
    
    -- Якщо ім'я вже містить посилання (наприклад, [[Демон|Демон]]), залишаємо як є.
    if mw.ustring.match(recruiter_name, "%[?%[?") then
        return recruiter_name
    else
        return string.format("[[%s|%s]]", recruiter_name, recruiter_name)
    end
end

-- 3. Призовий фонд (Призовий_фонд, колонка 2)
-- Формат: XXX грн
function p.prize_pool(frame)
    local player = frame.args.player
    local amount = fetch_data_from_table("Призовий_фонд", player, 2) 

    if not amount or amount == "" then
        return "0 ₴"
    end

    -- Видаляємо всі символи, крім цифр та пробілів, для чистоти
    amount = mw.ustring.gsub(amount, "[^%d%s]", "")
    amount = mw.text.trim(amount)
    
    return amount .. " ₴"
end

-- 4. Фундація (Фундація, колонка 2)
-- Формат: XXX грн
function p.foundation(frame)
    local player = frame.args.player
    local amount = fetch_data_from_table("Фундація", player, 2) 

    if not amount or amount == "" then
        return "0 ₴"
    end

    -- Видаляємо всі символи, крім цифр та пробілів, для чистоти
    amount = mw.ustring.gsub(amount, "[^%d%s]", "")
    amount = mw.text.trim(amount)
    
    return amount .. " ₴"
end

-- 5. Фіналіст (Фіналіст, колонка 2)
-- Формат: X/9
function p.final(frame)
    local player = frame.args.player
    local count = fetch_data_from_table("Фіналіст", player, 2) 

    if not count or count == "" then
        return "0/9"
    end

    -- Переконуємось, що це лише цифра
    count = mw.ustring.match(count, "(%d+)")
    
    if not count then
         return "0/9"
    end
    
    return count .. "/9"
end

return p