Модуль:FetchData2

Документацію для цього модуля можна створити у Модуль:FetchData2/документація

local p = {}

-- [[25.10.2024]] для розрахунку днів
local TARGET_TIMESTAMP = os.time({year=2024, month=10, day=25, hour=0, min=0, sec=0}) 

--- Функція для виведення помилки на сторінку
local function error_output(context, error_details)
    -- !!! УВАГА: ЦЕ ВІДОБРАЖАЄТЬСЯ НА СТОРІНЦІ !!!
    return string.format("<span style='color:red; font-weight:bold;'>[FD2 Error: %s. Player: %s]</span>", context, error_details)
end


----------------------------------------------------------------------
-- ГОЛОВНИЙ ПОШУК (Використовується для Гравці, Призовий фонд, Фундація, Фіналіст)
-- Індексація: 1=№, 2=Ім'я, 3=Значення, 4=Додатково
----------------------------------------------------------------------

local function fetch_from_table(page_title, player_name, column_index)
    if not player_name or player_name == "" then
        return error_output("No Player Name", "N/A")
    end
    
    local title = mw.title.new(page_title)
    if not title or not title.exists then 
        return error_output("Page Missing", page_title)
    end

    local content = title:getContent()
    if not content then return error_output("Content Read Fail", page_title) end

    local escaped = mw.ustring.gsub(player_name, "([%^%$%(%)%%%.%[%]%*%+%-%?])", "%%%1")
    local player_pattern = "%[%[" .. escaped .. "%]%]"

    -- 1. Вилучаємо вміст таблиці wikitable
    local table_content = mw.ustring.match(content, "{|.-([|].-.-[|]})")
    if not table_content then 
        return error_output("Table Missing", page_title)
    end

    -- 2. Знаходимо весь блок даних гравця (4 комірки: №, Ім'я, Значення, Додатково)
    local full_row_pattern = "([|]%s*.-%s*[|]%s*" .. player_pattern .. "%s*[|]%s*.-%s*[|]%s*.-%s*)"
    local full_row_match = mw.ustring.match(table_content, full_row_pattern)
    
    if not full_row_match then 
        -- Повертаємо nil, щоб функції-обгортки (recruiter, date_added) могли встановити значення за замовчуванням
        return nil 
    end

    -- 3. Розділяємо захоплений блок на комірки
    local cells = {}
    
    for cell in mw.ustring.gmatch(full_row_match, "[|](.-)") do
        local trimmed_cell = mw.text.trim(cell)
        if trimmed_cell ~= "" then
             table.insert(cells, trimmed_cell)
        end
    end
    
    local result = cells[column_index]
    if not result then 
        -- Якщо рядок знайдено, але колонка відсутня (наприклад, 4-та), повертаємо nil.
        return nil
    end

    -- 4. Прибираємо вікі-посилання [[link|text]] і [[link]]
    result = mw.ustring.gsub(result, "%[%[([^%]|]+)|([^%]]+)%]%]", "%2")
    result = mw.ustring.gsub(result, "%[%[([^%]]+)%]%]", "%1")

    return mw.text.trim(result)
end


----------------------------------------------------------------------
-- Рахує різницю в днях між датою та 25.10.2024
----------------------------------------------------------------------

local function date_with_days(dateString)
    if not dateString or mw.ustring.find(dateString, "Error") then return "Невідомо" end

    local d, m, y = dateString:match("(%d%d)%.(%d%d)%.(%d%d%d%d)")
    if not d then return dateString end

    local start = os.time({day = tonumber(d), month = tonumber(m), year = tonumber(y), hour = 0, min = 0, sec = 0})
    local finish = TARGET_TIMESTAMP
    
    local diff = math.floor((finish - start) / 86400)
    if diff < 0 then diff = 0 end 

    return string.format("%s (%d д)", dateString, diff)
end


----------------------------------------------------------------------
-- ФУНКЦІЇ ДЛЯ {{MCC Player New}}
----------------------------------------------------------------------

-- 1. Провідник (Колонка 4)
function p.recruiter(frame)
    local name = frame.args.player
    local raw = fetch_from_table("Гравці", name, 4) 

    if type(raw) == "string" and mw.ustring.find(raw, "Error") then return raw end
    
    if raw == "Відсутній" then return "Не вказано" end
    return raw or "Не вказано"
end

-- 2. Дата приєднання (Колонка 3)
function p.date_added(frame)
    local name = frame.args.player
    local raw = fetch_from_table("Гравці", name, 3) 
    
    if type(raw) == "string" and mw.ustring.find(raw, "Error") then return raw end

    return raw and date_with_days(raw) or "Невідомо"
end

-- 3. Призовий фонд (Колонка 3)
function p.prize_pool(frame)
    local name = frame.args.player
    local raw = fetch_from_table("Призовий_фонд", name, 3) 

    if type(raw) == "string" and mw.ustring.find(raw, "Error") then return raw end

    if raw then 
        raw = mw.ustring.gsub(raw, "[^%d%s]", "")
        raw = mw.text.trim(raw)
    end
    return (raw or "0") .. " ₴"
end

-- 4. Фундація (Колонка 3)
function p.foundation(frame)
    local name = frame.args.player
    local raw = fetch_from_table("Фундація", name, 3)

    if type(raw) == "string" and mw.ustring.find(raw, "Error") then return raw end
    
    if raw then 
        raw = mw.ustring.gsub(raw, "[^%d%s]", "")
        raw = mw.text.trim(raw)
    end
    return (raw or "0") .. " ₴"
end

-- 5. Фіналіст (Колонка 3)
function p.final(frame)
    local name = frame.args.player
    local raw = fetch_from_table("Фіналіст", name, 3)
    
    if type(raw) == "string" and mw.ustring.find(raw, "Error") then return raw end
    
    local count = raw and mw.ustring.match(raw, "(%d+)")
    
    if count then
        return count .. "/9"
    else
        return "0/9"
    end
end

-- 6. Результат сезону (Додано функціонал season_result)
function p.season_result(frame)
    local season = frame.args.season
    local player = frame.args.player

    if not player or player == "" then return error_output("No Player Name", "Season") end
    
    local season_titles = {
        "Перший_сезон", "Другий_сезон", "Третій_сезон",
        "Четвертий_сезон", "П'ятий_сезон", "Шостий_сезон", 
        "Сьомий_сезон", "Восьмий_сезон", "Дев'ятий_сезон"
    }
    
    local season_title = season_titles[tonumber(season)]

    if not season_title then return error_output("Invalid Season Index", season) end
    
    local title = mw.title.new(season_title)
    if not title or not title.exists then return error_output("Season Page Missing", season_title) end
    
    local content = title:getContent()
    if not content then return error_output("Season Content Read Fail", season_title) end

    -- Шукаємо секцію "Рейтинг" та таблицю
    local rating_section = mw.ustring.match(content, "==%s*Рейтинг%s*==.-{|%s*class%s*=%s*\"wikitable sortable\"(.-)|}")
    if not rating_section then 
         return error_output("Rating Section Missing", season_title) 
    end

    -- Патерн для пошуку позиції (перша колонка) гравця (друга колонка)
    -- | <№> | [[Гравець]] 
    local escaped_player = mw.ustring.gsub(player, "([%^%$%(%)%%%.%[%]%*%+%-%?])", "%%%1")
    local pattern = "[|]%s*(%d+)%s*[|]%s*%[%[" .. escaped_player .. "%]%]"
    
    local rank = mw.ustring.match(rating_section, pattern)
    
    if not rank then
        return "--"
    end

    return rank
end

return p