299
edits
(username removed) m (1 revision imported) |
Innocentuzo (talk | contribs) m (1 revision imported) |
||
(One intermediate revision by one other user not shown) | |||
Line 6: | Line 6: | ||
-- -- | -- -- | ||
-------------------------------------------------------------------------------- | -------------------------------------------------------------------------------- | ||
-- Load config. | -- Load config. | ||
local cfg = mw.loadData('Module:Pagetype/config') | local cfg = mw.loadData('Module:Pagetype/config') | ||
-- Load required modules. | -- Load required modules. | ||
local yesno = require('Module:Yesno') | local yesno = require('Module:Yesno') | ||
local p = {} | local p = {} | ||
Line 31: | Line 29: | ||
-- Get argument keys for a title's namespace | -- Get argument keys for a title's namespace | ||
local function getNamespaceArgKeys(title | local function getNamespaceArgKeys(title) | ||
local nsInfo = mw.site.namespaces[title.namespace] | local nsInfo = mw.site.namespaces[title.namespace] | ||
local customAliases = cfg.customNamespaceAliases[title.namespace] or {} | local customAliases = cfg.customNamespaceAliases[title.namespace] or {} | ||
Line 47: | Line 45: | ||
-- Get the argument for a title's namespace, if it was specified in the args table. | -- Get the argument for a title's namespace, if it was specified in the args table. | ||
local function getNamespaceArg(title, args | local function getNamespaceArg(title, args) | ||
if title.isTalkPage then | if title.isTalkPage then | ||
return lookUpNamespaceArg(args, cfg.talk) | return lookUpNamespaceArg(args, cfg.talk) | ||
end | end | ||
for _, key in ipairs(getNamespaceArgKeys(title | for _, key in ipairs(getNamespaceArgKeys(title)) do | ||
local arg = lookUpNamespaceArg(args, mw.ustring.lower(key)) | local arg = lookUpNamespaceArg(args, mw.ustring.lower(key)) | ||
if arg ~= nil then | if arg ~= nil then | ||
Line 61: | Line 59: | ||
-- Look up a page type specific to the title's namespace | -- Look up a page type specific to the title's namespace | ||
local function getExplicitPageType(title | local function getExplicitPageType(title) | ||
if title.isTalkPage then | if title.isTalkPage then | ||
return cfg.talkDefault | return cfg.talkDefault | ||
Line 70: | Line 68: | ||
-- Get a default page type that is not specific to the title's namespace | -- Get a default page type that is not specific to the title's namespace | ||
local function getDefaultPageType(args | local function getDefaultPageType(args) | ||
local other = lookUpNamespaceArg(args, cfg.other) | local other = lookUpNamespaceArg(args, cfg.other) | ||
if type(other) == 'string' then | if type(other) == 'string' then | ||
Line 102: | Line 100: | ||
end | end | ||
local function pluralize(pageType | local function pluralize(pageType) | ||
if cfg.irregularPlurals[pageType] then | if cfg.irregularPlurals[pageType] then | ||
return cfg.irregularPlurals[pageType] | return cfg.irregularPlurals[pageType] | ||
Line 110: | Line 108: | ||
end | end | ||
local function parseContent(title, args, | local function parseContent(title, args, optionsList) | ||
if | if title.namespace==828 and title.subpageText~='doc' -- don't detect modules | ||
or not title.exists -- can't check unless page exists | or not title.exists -- can't check unless page exists | ||
then | then | ||
return nil | return nil | ||
end | end | ||
local content = title:getContent() | local content = title:getContent() | ||
if content == nil then | if content == nil then | ||
return nil | return nil | ||
end | end | ||
content = require('Module:Wikitext Parsing').PrepareText(content) -- disregard templates which do not have any affect | local templates -- lazily evaluated | ||
for _, options in next, optionsList do | |||
local list, parameter, default, articleOnly = unpack(options, 1, 4) | |||
if not articleOnly or title.namespace==0 then -- only check for templates if we should... | |||
local out = lookUpNamespaceArg(args, parameter) | |||
if type(out) == "string" or (out ~= false and default) then -- ...and if we actually have anything to say about them | |||
if not templates then | |||
templates = {} -- do our delayed evaluation now that we are required to | |||
content = require('Module:Wikitext Parsing').PrepareText(content) -- disregard templates which do not have any affect | |||
for template in string.gmatch(content, "{{%s*([^|}]-)%s*[|}]") do | |||
templates[#templates+1] = capitalize(template) | |||
end | |||
end | |||
local wantedTemplates = mw.loadData('Module:Pagetype/' .. list) | |||
local templateFound = false | |||
for _, template in next, templates do | |||
if wantedTemplates[template] then | |||
templateFound = true | |||
break | |||
end | |||
end | |||
if templateFound then | |||
if type(out)=='string' then | |||
return out | |||
elseif out ~= false and default then | |||
return default | |||
end | |||
end | |||
end | |||
end | end | ||
end | end | ||
end | end | ||
-- | -- Find pages which do not exist | ||
local function | local function nonExistent(title, args) | ||
local arg = lookUpNamespaceArg(args, | local arg = lookUpNamespaceArg(args, cfg.ne) | ||
if arg == false then | if arg == false then | ||
return nil | return nil | ||
end | end | ||
if | local exists = false | ||
if title.exists then -- not an article if it does not exist | |||
exists = true | |||
elseif title.namespace==8 and mw.message.new(title.text):exists() then | |||
exists = true | |||
elseif title.namespace==6 and title.fileExists then | |||
exists = true | |||
end | |||
if not exists then | |||
if type(arg) == 'string' then | if type(arg) == 'string' then | ||
return arg | return arg | ||
else | else | ||
return | return cfg.naDefault | ||
end | end | ||
end | end | ||
end | end | ||
-- Get page types for mainspaces pages with an explicit class specified | -- Get page types for mainspaces pages with an explicit class specified | ||
local function getMainNamespaceClassPageType(title, args, cfg) | local function getMainNamespaceClassPageType(title, args) | ||
if | local class = args[1] | ||
if type(class) == 'string' then -- Put in lower case so e.g. "na" and "NA" will both match | |||
class = mw.ustring.lower(class) | |||
end | |||
local arg = lookUpNamespaceArg(args, cfg.na) | |||
if arg == false then -- don't check for this class if it is specifically disallowed | |||
return nil | return nil | ||
end | end | ||
if cfg.naAliases[class] then | |||
if type(arg) == 'string' then | |||
return arg | |||
else | |||
return cfg.naDefault | |||
end | |||
else | |||
return nil | |||
end | end | ||
end | end | ||
-- Get page type specified by an explicit namespace argument. | -- Get page type specified by an explicit namespace argument. | ||
local function getNamespaceArgPageType(title, args | local function getNamespaceArgPageType(title, args) | ||
local namespaceArg = getNamespaceArg(title, args | local namespaceArg = getNamespaceArg(title, args) | ||
if namespaceArg == true then | if namespaceArg == true then | ||
-- Namespace has been explicitly enabled, so return the default for | -- Namespace has been explicitly enabled, so return the default for | ||
-- this namespace | -- this namespace | ||
return getExplicitPageType(title | return getExplicitPageType(title) | ||
elseif namespaceArg == false then | elseif namespaceArg == false then | ||
-- Namespace has been explicitly disabled | -- Namespace has been explicitly disabled | ||
return getDefaultPageType(args | return getDefaultPageType(args) | ||
elseif namespaceArg then | elseif namespaceArg then | ||
-- This namespaces uses custom text | -- This namespaces uses custom text | ||
Line 195: | Line 213: | ||
end | end | ||
-- Get page type not specified or detected by other means | |||
local function getOtherPageType(title, args) | |||
-- Whether the title is in the set of default active namespaces which are looked up in cfg.pagetypes. | -- Whether the title is in the set of default active namespaces which are looked up in cfg.pagetypes. | ||
local | local isInDefaultActiveNamespace = false | ||
local defaultNamespacesKey = args[cfg.defaultns] | local defaultNamespacesKey = args[cfg.defaultns] | ||
if defaultNamespacesKey == cfg.defaultnsAll then | if defaultNamespacesKey == cfg.defaultnsAll then | ||
isInDefaultActiveNamespace = true | |||
else | else | ||
defaultNamespaces = cfg.defaultNamespaces | local defaultNamespaces | ||
if defaultNamespacesKey == cfg.defaultnsExtended then | |||
defaultNamespaces = cfg.extendedNamespaces | |||
elseif defaultNamespacesKey == cfg.defaultnsNone then | |||
defaultNamespaces = {} | |||
else | |||
defaultNamespaces = cfg.defaultNamespaces | |||
end | |||
isInDefaultActiveNamespace = defaultNamespaces[title.namespace] | |||
end | end | ||
if isInDefaultActiveNamespace then | |||
return getExplicitPageType(title) | |||
if isInDefaultActiveNamespace | |||
return getExplicitPageType(title | |||
else | else | ||
return getDefaultPageType(args | return getDefaultPageType(args) | ||
end | end | ||
end | end | ||
function p._main(args) | |||
local title | local title | ||
if args.page then | if args.page then | ||
title = mw.title.new(args.page) | title = mw.title.new(args.page) | ||
else | else | ||
title = mw.title.getCurrentTitle() | title = mw.title.getCurrentTitle() | ||
end | end | ||
if title and not yesno(args.talk, true) and args[cfg.defaultns] ~= cfg.defaultnsAll then | |||
if | title = title.subjectPageTitle | ||
end | end | ||
local pageType = detectRedirects(title, args) | |||
or nonExistent(title, args) | |||
or parseContent(title, args, { | |||
{'softredirect', cfg.softRedirect, cfg.softRedirectDefault}, | |||
{'setindex', cfg.sia, cfg.siaDefault, true}, | |||
{'disambiguation', cfg.dab, cfg.dabDefault, true}, | |||
{'rfd', cfg.rfd, cfg.rfdDefault}, | |||
}) | |||
or (title.namespace == 0 and getMainNamespaceClassPageType(title, args)) | |||
or getNamespaceArgPageType(title, args) | |||
or getOtherPageType(title, args) | |||
if yesno(args.plural, false) then | if yesno(args.plural, false) then | ||
pageType = pluralize(pageType | pageType = pluralize(pageType) | ||
end | end | ||
if yesno(args.caps, false) then | if yesno(args.caps, false) then | ||
Line 270: | Line 270: | ||
function p.main(frame) | function p.main(frame) | ||
local args = getArgs(frame) | local args = require('Module:Arguments').getArgs(frame) | ||
return p._main(args) | return p._main(args) | ||
end | end | ||
return p | return p |