Zoran: Pywikibot 6.4.0

2021-07-16T05:55:54Z

Pywikibot 6.4.0

New page

-- This module creates a warning about sensitive IP addresses that is intended
-- to be placed in [[Special:Block]] via the [[Mediawiki:Blockiptext]] message.

local querySensitiveIPs = require('Module:Sensitive IP addresses').query
local mIP = require('Module:IP')
local IPAddress = mIP.IPAddress
local Subnet = mIP.Subnet

local function normalizeIPOrSubnet(ipOrSubnet)
-- Normalize an IP address or subnet.
-- If ipOrSubnet is not a valid IP address or subnet, returns nil.
local ipSuccess, ipObj = pcall(IPAddress.new, ipOrSubnet)
if ipSuccess and ipObj then
return tostring(ipObj)
end
local subnetSuccess, subnetObj = pcall(Subnet.new, ipOrSubnet)
if subnetSuccess and subnetObj then
return tostring(subnetObj)
end
return nil
end

local function parseTitle(title)
-- Parse a title. If the subpage(s) of the root page are a valid IP
-- address or subnet, then return the normalized IP address or subnet.
local ipOrSubnet = title.text:gsub('^.-/', '')
return normalizeIPOrSubnet(ipOrSubnet)
end

local function fetchSensitivityData(ipOrSubnet)
return querySensitiveIPs{test = {ipOrSubnet}}
end

local function parseSensitivityData(sensitivityData)
-- Parse the data from the sensitive IP API call, and return a table with
-- information about the first match that we can use as arguments to a
-- template call.
local function parse(sensitivityData)
sensitivityData = sensitivityData.sensitiveips
if #sensitivityData.matches == 0 then
return nil
end
local ret = {}
ret.type = sensitivityData.matches[1].type
if ret.type == 'ip' then
ret.ip = sensitivityData.matches[1].ip
elseif ret.type == 'range' then
ret.range = sensitivityData.matches[1].range
else
return nil
end
ret['entity-id'] = sensitivityData.matches[1]['entity-id']
ret['matches-range'] = sensitivityData.matches[1]['matches-range']
ret['entity-name'] = sensitivityData.entities[ret['entity-id']].name
ret['entity-description'] = sensitivityData.entities[ret['entity-id']].description
ret.reason = sensitivityData.entities[ret['entity-id']].reason
return ret
end

local success, ret = pcall(parse, sensitivityData)
if success and ret then
return ret
else
return nil
end
end

local function renderMessage(rawMessage, messageArgs)
local message
if rawMessage then
message = mw.message.newRawMessage(rawMessage)
else
message = mw.message.new('sensitive-ip-block-text')
end
local wikitext = message:params(
messageArgs.type,
messageArgs.type == 'ip' and messageArgs.ip or messageArgs.range,
messageArgs['matches-range'],
messageArgs['entity-id'],
messageArgs['entity-name'],
messageArgs['entity-description'],
messageArgs.reason
):plain()
local frame = mw.getCurrentFrame()
return frame:preprocess(wikitext)
end

local p = {}

function p._main(args, title)
args = args or {}
title = title or mw.title.getCurrentTitle()
local ipOrSubnet = parseTitle(title)
if not ipOrSubnet then
return nil
end
local sensitivityData = fetchSensitivityData(ipOrSubnet)
if not sensitivityData then
return nil
end
local messageArgs = parseSensitivityData(sensitivityData)
return renderMessage(args.message, messageArgs)
end

function p.main(frame)
local args = require('Module:Arguments').getArgs(frame)
return p._main(args)
end

function p._exportFunctions()
return {
parseTitle = parseTitle,
fetchSensitivityData = fetchSensitivityData,
parseSensitivityData = parseSensitivityData,
renderMessage = renderMessage,
}
end

return p

Module:Sensitive IP addresses/blocktext - Revision history

Zoran: Pywikibot 6.4.0