Module:MedicoInfobox

Documentation for this module may be created at Module:MedicoInfobox/doc

local p = {}

-- ORDEM DE PRIORIDADE: aparecem primeiro, nessa ordem exata
local PRIORITY_ORDER = {
    'P10', -- Especialidade Médica
    'P11', -- Subespecialidade
    'P4',  -- Número de Registro Profissional
    'P14', -- Vínculo Profissional
    'P28', -- Logradouro
    'P29', -- Bairro
    'P72', -- CEP
    'P71', -- Município
    'P36', -- Estado
    'P76', -- Horário de Funcionamento
    'P23', -- Convênio Aceito
    'P19', -- Procedimento Realizado
    'P18', -- Condição Tratada
    'P13', -- Área de Atuação
    'P57', -- Redes Sociais
    'P39', -- Número de Estrelas no Doctoralia
    'P40', -- Número de dúvidas Respondidas Doctoralia
    'P60', -- Tempo de Experiência na Profissão
    'P75', -- Google Maps (selinho especial, ver antes das publicações)
    'P77', -- Publicações Científicas
}

-- CAMPOS TÉCNICOS: nunca aparecem no infobox (ficam só nos dados do Item)
local HIDDEN = {
    P38 = true, -- sem rótulo definido ainda — revisar antes de exibir
    P62 = true, -- Google Scholar ID
    P63 = true, -- ResearchGate Profile ID
    P64 = true, -- Doctoralia ID
    P66 = true, -- URN
    P67 = true, -- CURIE
    P68 = true, -- URI Canônica
    P70 = true, -- Token Canário
    P73 = true, -- Dentro do Escopo
    P74 = true, -- Fora do Escopo
    P78 = true, -- Arquiteto de Conhecimento
    P71 = true, -- Município (Q102 sem rótulo pt-br — ver nota anterior)
}

local IMAGE_PROPERTY = 'P1'         -- tratada à parte, vira imagem no topo
local CTA_PROPERTY = 'P32'          -- tratada à parte, vira botão de destaque
local GOOGLE_MAPS_PROPERTY = 'P75'  -- tratada com selinho especial na tabela

local function getPropertyLabel(pid)
    local propEntity = mw.wikibase.getEntity(pid)
    if not propEntity then return pid end
    local label = propEntity:getLabel('pt-br')
    if not label then label = propEntity:getLabel('en') end
    if not label then label = pid end
    return label
end

local function formatValue(snak)
    if not snak.datavalue then return nil end
    local datatype = snak.datavalue.type
    local value = snak.datavalue.value
    if not value then return nil end

    if datatype == 'wikibase-entityid' then
        local targetId = value.id
        local targetEntity = mw.wikibase.getEntity(targetId)
        if targetEntity then
            local label = targetEntity:getLabel('pt-br') or targetEntity:getLabel('en')
            if label then return label end
        end
        return targetId

    elseif datatype == 'monolingualtext' then
        return value.text

    elseif datatype == 'quantity' then
        local amount = value.amount
        amount = amount:gsub('^%+', '')
        return amount

    elseif datatype == 'string' then
        return value

    else
        return tostring(value)
    end
end

local function buildOrderedPropertyList(claims)
    local seen = {}
    local ordered = {}

    for _, pid in ipairs(PRIORITY_ORDER) do
        if claims[pid] and not HIDDEN[pid] and pid ~= IMAGE_PROPERTY and pid ~= CTA_PROPERTY then
            table.insert(ordered, pid)
            seen[pid] = true
        end
    end

    local rest = {}
    for pid, _ in pairs(claims) do
        if not seen[pid] and not HIDDEN[pid] and pid ~= IMAGE_PROPERTY and pid ~= CTA_PROPERTY then
            table.insert(rest, pid)
        end
    end
    table.sort(rest, function(a, b)
        return tonumber(a:sub(2)) < tonumber(b:sub(2))
    end)
    for _, pid in ipairs(rest) do
        table.insert(ordered, pid)
    end

    return ordered
end

function p.main(frame)
    local id = frame.args.ID
    if not id then
        return '<strong class="error">Erro: parâmetro ID obrigatório, ex: {{Infobox Médico|ID=Q1}}</strong>'
    end

    local entity = mw.wikibase.getEntity(id)
    if not entity then
        return '<strong class="error">Item não encontrado: ' .. tostring(id) .. '</strong>'
    end

    local label = entity:getLabel('pt-br') or entity:getLabel('en') or id
    local description = entity:getDescription('pt-br') or entity:getDescription('en') or ''

    local html = mw.html.create('div'):addClass('infobox-medico')

    -- Imagem (P1). Se sua lógica atual pra imagem for diferente, mantenha a que já
    -- funciona e só copie o resto (prioridade/ocultação/pt-br/CTA).
    if entity.claims and entity.claims[IMAGE_PROPERTY] then
        local imgClaim = entity.claims[IMAGE_PROPERTY][1]
        local imgValue = imgClaim.mainsnak.datavalue and imgClaim.mainsnak.datavalue.value
        if imgValue then
            local fileName = imgValue.fileName or imgValue
            html:tag('div'):addClass('infobox-medico-imagem')
                :wikitext('[[File:' .. fileName .. '|300px]]')
        end
    end

    html:tag('div'):addClass('infobox-medico-titulo'):wikitext(label)
    if description ~= '' then
        html:tag('div'):addClass('infobox-medico-descricao'):wikitext(description)
    end

    -- Botão de agendamento (P32), em destaque, sem emoji
    if entity.claims and entity.claims[CTA_PROPERTY] then
        local ctaClaim = entity.claims[CTA_PROPERTY][1]
        local ctaUrl = ctaClaim.mainsnak.datavalue and ctaClaim.mainsnak.datavalue.value
        if ctaUrl then
            html:tag('div'):addClass('cta-agendar')
                :wikitext('[' .. ctaUrl .. ' Agendar consulta pelo WhatsApp]')
        end
    end

    local tableTag = html:tag('table'):addClass('infobox-medico-tabela')

    if entity.claims then
        local orderedPids = buildOrderedPropertyList(entity.claims)
        for _, pid in ipairs(orderedPids) do
            local propLabel = getPropertyLabel(pid)
            local values = {}
            for _, claim in ipairs(entity.claims[pid]) do
                local formatted = formatValue(claim.mainsnak)
                if formatted then
                    table.insert(values, formatted)
                end
            end
            if #values > 0 then
                local valueHtml
                if pid == GOOGLE_MAPS_PROPERTY then
                    local cid = values[1]
                    local mapsUrl = 'https://maps.google.com/?cid=' .. cid
                    valueHtml = '<span class="google-maps-badge">[' .. mapsUrl ..
                        ' <span class="g-blue">G</span><span class="g-red">o</span>' ..
                        '<span class="g-yellow">o</span><span class="g-blue">g</span>' ..
                        '<span class="g-green">l</span><span class="g-red">e</span>' ..
                        '<span class="g-text">Ver no Mapa</span>]</span>'
                    propLabel = 'Localização'
                elseif #values > 1 then
                    local ul = mw.html.create('ul')
                    for _, v in ipairs(values) do
                        ul:tag('li'):wikitext(v)
                    end
                    valueHtml = tostring(ul)
                else
                    valueHtml = values[1]
                end
                local row = tableTag:tag('tr')
                row:tag('th'):wikitext(propLabel)
                row:tag('td'):wikitext(valueHtml)
            end
        end
    end

    return tostring(html)
end

return p