Difference between revisions of "Module:Documentation"

From The HILLSIDE
Jump to navigation Jump to search
en>Hlm Z.
en>Mr. Stradivarius
(replace content with a port of English Wikipedia's Template:Documentation)
Line 1: Line 1:
--This module implements {{Documentation}}.
+
-- This module implements {{documentation}}.
+
 
 +
-- Get required modules.
 +
local getArgs = require('Module:Arguments').getArgs
 +
local htmlBuilder = require('Module:HtmlBuilder')
 +
local messageBox = require('Module:Message box')
 +
 
 
local p = {}
 
local p = {}
+
 
function p.corps(frame)
+
-- Constants.
    args = frame:getParent().args
+
local currentTitle = mw.title.getCurrentTitle()
    local page = mw.title.getCurrentTitle()
+
local subjectSpace = mw.site.namespaces[currentTitle.namespace].subject.id
    doc = p.docname(page)
+
 
    local corps = {}
+
----------------------------------------------------------------------------
+
-- Helper functions
    if page.subpageText == 'sandbox' then
+
----------------------------------------------------------------------------
        table.insert(corps, '<div style="clear:both />')
+
 
        table.insert(corps, frame:preprocess('{{Template sandbox notice}}'))
+
local function makeWikilink(page, display)
    end
+
if display then
+
return mw.ustring.format('[[%s|%s]]', page, display)
    table.insert(corps, p.header(page))
+
else
    table.insert(corps, p.content(frame, page))
+
return mw.ustring.format('[[%s]]', page)
    table.insert(corps, p.footer(page))
+
end
 
    if args.raw then
 
        return frame:preprocess('<nowiki>' .. table.concat(corps) .. '</nowiki>\n:' .. os.clock())
 
    end
 
 
    return table.concat(corps)
 
 
end
 
end
+
 
function p.docname(page)
+
local function makeUrlLink(url, display)
    if not page.isSubpage then return page.subjectNsText .. ":" .. page.text .. "/doc" end
+
return mw.ustring.format('[%s %s]', url, display)
 
    if page.subpageText == 'doc'
 
    or page.subpageText == 'sandbox'
 
    or page.subpageText == 'testcases' then
 
        return page.subjectNsText .. ":" .. page.baseText .. "/doc"
 
    else
 
        return page.subjectNsText .. ":" .. page.text .. "/doc"
 
    end
 
 
end
 
end
+
 
function p.ifexist(page)
+
----------------------------------------------------------------------------
    if not page then return false end
+
-- Main functions
    if mw.title.new(page).exists then return true end
+
----------------------------------------------------------------------------
    return false
+
 
 +
function p.main(frame)
 +
local args = getArgs(frame, {
 +
valueFunc = function (key, value)
 +
if type(value) == 'string' then
 +
value = value:match('^%s*(.-)%s*$') -- Remove whitespace.
 +
if key == 'heading' or value ~= '' then
 +
return value
 +
else
 +
return nil
 +
end
 +
else
 +
return value
 +
end
 +
end
 +
})
 +
return p._main(args)
 
end
 
end
+
 
function p.header(page)
+
function p._main(args)
    local header = {'<div class="template-documentation"'}
+
local root = htmlBuilder.create()
+
root
    if args.color then
+
.wikitext(p.protectionTemplate())
        table.insert(header, ' style="background:')
+
.wikitext(p.sandboxNotice(args))
        table.insert(header, args.color .. '"')
+
-- This div tag is from {{documentation/start box}}, but moving it here so that we don't have to worry about unclosed tags.
    end
+
.tag('div')
+
.attr('id', 'template-documentation')
    table.insert(header, '><div style="margin-bottom:1ex; border-bottom:1px solid #aaa; padding-bottom:3px;">')
+
.addClass('template-documentation iezoomfix')
    table.insert(header, '[[File:Template-info.png|50px|alt=Template documentation|link=]]')
+
.wikitext(p.startBox(args))
    table.insert(header, '<span style="font-weight:bold; font-size:125%">&nbsp;')
+
.wikitext(p.content(args))
+
.tag('div')
    if args.heading then table.insert(header, args.heading)
+
.css('clear', 'both') -- So right or left floating items don't stick out of the doc box.
    else table.insert(header, 'Template documentation') end
+
.done()
    table.insert(header, '</span>')
+
.done()
+
.wikitext(p.endBox(args))
    if not args.content then
+
.wikitext(p.addTrackingCategories())
        table.insert(header, '<span class="mw-editsection plainlinks">&#91;[')
+
return tostring(root)
        local arg = mw.title.new(args[1] or doc)
 
 
        if args[1] and p.ifexist(args[1]) or p.ifexist(doc) then
 
            table.insert(header, arg:fullUrl('action=view') .. ' view]')
 
            table.insert(header, '&#93;&#32;&#91;[')
 
            table.insert(header, arg:fullUrl('action=edit') .. ' edit]')
 
            table.insert(header, '&#93;&#32;&#91;[')
 
            table.insert(header, arg:fullUrl('action=history') .. ' history]')
 
            table.insert(header, '&#93;&#32;&#91;[')
 
            table.insert(header, page:fullUrl('action=purge') .. ' purge]')
 
        else
 
            table.insert(header, arg:fullUrl({["action"]="edit", ["preload"]="Template:Documentation/preload"}))
 
            table.insert(header, ' create]')
 
        end
 
        table.insert(header, '&#93;</span>')
 
    end
 
 
    table.insert(header, '</div>')
 
    return table.concat(header)
 
 
end
 
end
+
 
function p.content(frame, page)
+
function p.sandboxNotice(args)
    local content = {}
+
if currentTitle.subpageText == 'sandbox' then
    local arg = args[1] or doc
+
local frame = mw.getCurrentFrame()
    if args.content then
+
local root = htmlBuilder.create()
        table.insert(content, '\n')
+
root
        table.insert(content, args.content)
+
.tag('div')
    else
+
.css('clear', 'both')
        table.insert(content, frame:preprocess('<nowiki />'))
+
.done()
        table.insert(content, '\n')
+
.wikitext(frame:expandTemplate{title = 'template sandbox notice', args = {args.livepage}})
        if args[1] and p.ifexist(args[1]) or p.ifexist(doc) then
+
return tostring(root)
            table.insert(content, frame:preprocess('{{' .. arg .. '}}'))
+
else
        end
+
return nil
    end
+
end
 
    table.insert(content, '\n')
 
    table.insert(content, frame:preprocess('<nowiki />'))
 
    table.insert(content, '<div style="clear:both" />\n')
 
    return table.concat(content)
 
 
end
 
end
+
 
function p.footer(page)
+
function p.protectionTemplate()
    local footer = {}
+
if currentTitle.namespace == 10 then -- We are in the template namespace.
    local arg = mw.title.new(args[1] or doc)
+
local frame = mw.getCurrentFrame()
+
 
    table.insert(footer, '</div><div class="template-documentation plainlinks" ')
+
local function getProtectionLevel(protectionType)
    table.insert(footer, 'style="font-style:italic; margin:2px 0px 0px; padding: 0.35em 0.9em')
+
-- Gets the protection level for the current page.
+
local level = frame:callParserFunction('PROTECTIONLEVEL', protectionType)
    if args.color then
+
if level ~= '' then
        table.insert(footer, '; background:')
+
return level
        table.insert(footer, args.color)
+
else
    end
+
return nil -- The parser function returns the blank string if there is no match.
    table.insert(footer, ';">\n')
+
end
+
end
    if args.content then
+
 
        if args["link box"] then
+
if getProtectionLevel('move') == 'sysop' or getProtectionLevel('edit') then
            table.insert(footer, args["link box"])
+
-- The page is full-move protected, or full, template, or semi-protected.
        else
+
return frame:expandTemplate{title = 'pp-template', args = {docusage = 'yes'}}
            table.insert(footer, 'This documentation is directly included in this page.')
+
end
        end
+
end
    else
+
return nil
        if args[1] and p.ifexist(args[1]) or p.ifexist(doc) then
 
            table.insert(footer, 'The above [[Wikipedia:Template documentation|documentation]] is ')
 
            table.insert(footer, '[[Wikipedia:Transclusion|transcluded]] from [[')
 
            table.insert(footer, tostring(arg))  
 
            table.insert(footer, ']]&nbsp;<span style="font-size:89%; font-style:normal;">([')
 
            table.insert(footer, arg:fullUrl('action=edit'))
 
            table.insert(footer, ' edit]&nbsp;|&nbsp;[')
 
            table.insert(footer, arg:fullUrl('action=history') .. ' history])</span>.<br />')
 
        end
 
        table.insert(footer, 'Editors can experiment in this templates ')
 
 
        local sandbox = arg.subjectNsText .. ":" .. arg.baseText .. "/sandbox"
 
        local argsandbox = mw.title.new(sandbox)
 
        if p.ifexist(sandbox) then
 
            table.insert(footer, '[[' .. sandbox .. '|sandbox]]')
 
            table.insert(footer, '&nbsp;<span style="font-size:89%; font-style:normal;">([')
 
            table.insert(footer, argsandbox:fullUrl('action=edit'))
 
            table.insert(footer, ' edit])</span>')
 
        else
 
            table.insert(footer, 'sandbox&nbsp;<span style="font-size:89%; font-style:normal;">([')
 
            table.insert(footer, argsandbox:fullUrl({["action"]="edit", ["preload"]="Template:Documentation/preload-sandbox"}))
 
            table.insert(footer, ' create])</span>')
 
        end
 
        table.insert(footer, ' and ')
 
 
        local test = arg.subjectNsText .. ":" .. arg.baseText .. "/testcases"
 
        local argtest = mw.title.new(test)
 
        if p.ifexist(test) then
 
            table.insert(footer, '[[' .. test .. '|testcases]]')
 
            table.insert(footer, '&nbsp;<span style="font-size:89%; font-style:normal;">([')
 
            table.insert(footer, argtest:fullUrl('action=edit'))
 
            table.insert(footer, ' edit])</span>')
 
        else
 
            table.insert(footer, 'testcases&nbsp;<span style="font-size:89%; font-style:normal;">([')
 
            table.insert(footer, argtest:fullUrl({["action"]="edit", ["preload"]="Template:Documentation/preload-testcases"}))
 
            table.insert(footer, ' create])</span>')
 
        end
 
 
        table.insert(footer, ' pages.<br />Please add categories to the ')
 
        table.insert(footer, '<span class="plainlinks">[')
 
 
        if args[1] and p.ifexist(args[1]) or p.ifexist(doc) then
 
            table.insert(footer, arg:fullUrl('action=edit'))
 
        else
 
            table.insert(footer, arg:fullUrl({["action"]="edit", ["preload"]="Template:Documentation/preload"}))
 
        end
 
 
        table.insert(footer, ' /doc] subpage.')
 
    end
 
    table.insert(footer, '</div>')
 
    return table.concat(footer)
 
 
end
 
end
+
 
 +
function p.startBox(args)
 +
-- Arg processing from {{documentation}}.
 +
local preload = args.preload -- Allow custom preloads.
 +
local heading = args.heading -- Blank values are not removed.
 +
local headingStyle = args['heading-style']
 +
local content = args.content
 +
local docspace = p.docspace()
 +
local docname = args[1] -- Other docname, if fed.
 +
local templatePage = p.templatePage()
 +
 
 +
-- Arg processing from {{documentation/start box2}}.
 +
local docpage
 +
if docname then
 +
docpage = docname
 +
else
 +
local namespace = docspace or currentTitle.nsText
 +
local pagename = templatePage or currentTitle.text
 +
docpage = namespace .. ':' .. pagename .. '/doc'
 +
end
 +
local docTitle = mw.title.new(docpage)
 +
local docExist = docTitle.exists
 +
 +
-- Output from {{documentation/start box}}.
 +
 
 +
-- First, check the heading parameter.
 +
if heading == '' then
 +
-- Heading is defined but blank, so do nothing.
 +
return nil
 +
end
 +
 
 +
-- Build the start box div.
 +
local sbox = htmlBuilder.create('div')
 +
sbox
 +
.css('padding-bottom', '3px')
 +
.css('border-bottom', '1px solid #aaa')
 +
.css('margin-bottom', '1ex')
 +
 
 +
-- Make the heading.
 +
local hspan = sbox.tag('span')
 +
if headingStyle then
 +
hspan.cssText(headingStyle)
 +
elseif subjectSpace == 10 then
 +
-- We are in the template or template talk namespaces.
 +
hspan
 +
.css('font-weight', 'bold')
 +
.css('fond-size', '125%')
 +
else
 +
hspan.css('font-size', '150%')
 +
end
 +
if heading then
 +
-- "heading" has data.
 +
hspan.wikitext(heading)
 +
elseif subjectSpace == 10 then -- Template namespace
 +
hspan.wikitext('[[File:Template-info.png|50px|link=|alt=Documentation icon]] Template documentation')
 +
elseif subjectSpace == 828 then -- Module namespace
 +
hspan.wikitext('[[File:Template-info.png|50px|link=|alt=Documentation icon]] Module documentation')
 +
elseif subjectSpace == 6 then -- File namespace
 +
hspan.wikitext('Summary')
 +
else
 +
hspan.wikitext('Documentation')
 +
end
 +
 
 +
-- Add the [view][edit][history][purge] or [create] links.
 +
-- Check for the content parameter first, as we don't need the links if the documentation
 +
-- content is being entered directly onto the template page.
 +
if not content then
 +
local lspan = sbox.tag('span') -- lspan is short for "link span".
 +
lspan
 +
.addClass('mw-editsection plainlinks')
 +
.attr('id', 'doc_editlinks')
 +
if docExist then
 +
local viewLink = makeWikilink(docpage, 'view')
 +
local editLink = makeUrlLink(docTitle:fullUrl{action = 'edit'}, 'edit')
 +
local historyLink = makeUrlLink(docTitle:fullUrl{action = 'history'}, 'history')
 +
local purgeLink = makeUrlLink(docTitle:fullUrl{action = 'purge'}, 'purge')
 +
local text = '[%s] [%s] [%s] [%s]'
 +
text = text:gsub('%[', '&#91;') -- Replace square brackets with HTML entities.
 +
text = text:gsub('%]', '&#93;')
 +
lspan.wikitext(mw.ustring.format(text, viewLink, editLink, historyLink, purgeLink))
 +
else
 +
if preload then
 +
preload = mw.uri.encode(preload)
 +
elseif subjectSpace == 6 then -- File namespace
 +
preload = 'Template:Documentation/preload-filespace'
 +
else
 +
preload = 'Template:Documentation/preload'
 +
end
 +
lspan.wikitext(makeUrlLink(docTitle:fullUrl{action = 'edit', preload = preload}, 'create'))
 +
end
 +
end
 +
 
 +
return tostring(sbox)
 +
end
 +
 
 +
function p.content(args)
 +
local content = args.content
 +
if not content then
 +
local docpage = args[1]
 +
if docpage and mw.title.new(docpage).exists then
 +
local frame = mw.getCurrentFrame()
 +
content = frame:preprocess('{{ ' .. docpage .. ' }}')
 +
else
 +
docpage = p.docspace() .. ':' .. p.templatePage() .. '/doc'
 +
if mw.title.new(docpage).exists then
 +
local frame = mw.getCurrentFrame()
 +
content = frame:preprocess('{{ ' .. docpage .. ' }}')
 +
end
 +
end
 +
end
 +
-- The line breaks below are necessary so that "=== Headings ===" at the start and end
 +
-- of docs are interpreted correctly.
 +
return '\n' .. (content or '') .. '\n'
 +
end
 +
 
 +
function p.endBox(args)
 +
-- Argument processing in {{documentation}}.
 +
local preload = args.preload -- Allow custom preloads.
 +
local content = args.content
 +
local linkBox = args['link box'] -- So "link box=off" works.
 +
local docspace = p.docspace()
 +
local docname = args[1] -- Other docname, if fed.
 +
local templatePage = p.templatePage()
 +
 
 +
-- Argument processing in {{documentation/end box2}}.
 +
local docpageRoot = (docspace or currentTitle.nsText) .. ':' .. (templatePage or currentTitle.text)
 +
local docpage
 +
if docname then
 +
docpage = docname
 +
else
 +
docpage = docpageRoot .. '/doc'
 +
end
 +
local docTitle = mw.title.new(docpage)
 +
local docExist = docTitle.exists
 +
local docnameFed = args[1] and true
 +
local sandbox = docpageRoot .. '/sandbox'
 +
local testcases = docpageRoot .. '/testcases'
 +
templatePage = currentTitle.nsText .. ':' .. templatePage
 +
 
 +
-- Output from {{documentation/end box}}
 +
 +
-- First, check whether we should output the end box at all. Add the end box by default if the documentation
 +
-- exists or if we are in the user, module or template namespaces.
 +
if linkBox == 'off' or not (docExist or subjectSpace == 2 or subjectSpace == 828 or subjectSpace == 10) then
 +
return nil
 +
end
 +
 
 +
-- Assemble the arguments for {{fmbox}}.
 +
local fmargs = {}
 +
fmargs.id = 'documentation-meta-data'
 +
fmargs.image = 'none'
 +
fmargs.style = 'background-color: #ecfcf4'
 +
fmargs.textstyle = 'font-style: italic;'
 +
 
 +
-- Assemble the fmbox text field.
 +
local text = ''
 +
if linkBox then
 +
-- Use custom link box content if it is defined.
 +
text = text .. linkBox
 +
else
 +
if docExist then
 +
-- /doc exists; link to it.
 +
local docLink = makeWikilink(docpage)
 +
local editLink = makeUrlLink(docTitle:fullUrl{action = 'edit'}, 'edit')
 +
local historyLink = makeUrlLink(docTitle:fullUrl{action = 'history'}, 'history')
 +
text = text .. 'The above [[Wikipedia:Template documentation|documentation]] is [[Wikipedia:Transclusion|transcluded]] from ' .. docLink .. '.'
 +
.. ' <small style="font-style: normal;">(' .. editLink .. ' &#124; ' .. historyLink .. ')</small> <br />'
 +
elseif subjectSpace == 828 then
 +
-- /doc does not exist; ask to create it.
 +
local createLink = makeUrlLink(docTitle:fullUrl{action = 'edit', preload = 'Template:Documentation/preload-module-doc'}, 'create')
 +
text = text .. 'You might want to ' .. createLink .. ' a documentation page for this [[Wikipedia:Lua|Scribunto module]].<br />'
 +
end
 +
-- Add links to /sandbox and /testcases when appropriate.
 +
if subjectSpace == 2 or subjectSpace == 828 or subjectSpace == 10 then
 +
-- We are in the user, module or template namespaces.
 +
local pagePossessive = subjectSpace == 828 and "module's" or "template's"
 +
text = text .. 'Editors can experiment in this ' .. pagePossessive .. ' '
 +
local sandboxTitle = mw.title.new(sandbox)
 +
if sandboxTitle.exists then
 +
local sandboxLink = makeWikilink(sandbox, 'sandbox')
 +
local sandboxEditLink = makeUrlLink(sandboxTitle:fullUrl{action = 'edit'}, 'edit')
 +
local compareLink = makeUrlLink(mw.title.new('Special:ComparePages'):fullUrl{page1 = mw.uri.encode(templatePage), page2 = mw.uri.encode(sandbox)}, 'diff')
 +
text = text .. sandboxLink .. ' <small style="font-style: normal">(' .. sandboxEditLink .. ' | ' .. compareLink .. ')</small>'
 +
else
 +
local sandboxPreload = 'Template:Documentation/preload-' .. (subjectSpace == 828 and 'module-' or '') .. 'sandbox'
 +
local sandboxCreateLink = makeUrlLink(sandboxTitle:fullUrl{action = 'edit', preload = sandboxPreload}, 'create')
 +
local mirrorPreload = mw.uri.encode(templatePage)
 +
local mirrorSummary = mw.uri.encode('Create sandbox version of ' .. makeWikilink(templatePage))
 +
local mirrorLink = makeUrlLink(sandboxTitle:fullUrl{action = 'edit', preload = mirrorPreload, summary = mirrorSummary}, 'mirror')
 +
text = text .. 'sandbox <small style="font-style: normal">(' .. sandboxCreateLink .. ' | ' .. mirrorLink .. ')</small>'
 +
end
 +
text = text .. ' and '
 +
local testcaseTitle = mw.title.new(testcases)
 +
if testcaseTitle.exists then
 +
local testcasesLink = makeWikilink(testcases, 'testcases')
 +
local testcasesEditLink = makeUrlLink(testcaseTitle:fullUrl{action = 'edit'}, 'edit')
 +
text = text .. testcasesLink .. ' <small style="font-style: normal">(' .. testcasesEditLink .. ')</small>'
 +
else
 +
local testcasesPreload = 'Template:Documentation/preload-' .. (subjectSpace == 828 and 'module-' or '') .. 'testcases'
 +
local testcasesCreateLink = makeUrlLink(testcaseTitle:fullUrl{action = 'edit', preload = testcasesPreload}, 'create')
 +
text = text .. 'testcases <small style="font-style: normal">(' .. testcasesCreateLink .. ')</small>'
 +
end
 +
text = text .. ' pages. <br />'
 +
-- Show the categories text, but not if "content" fed or "docname fed" since then it is unclear where to add the categories.
 +
if not content and not docnameFed then
 +
text = text .. 'Please add categories to the ' .. makeWikilink(docpage, '/doc') .. ' subpage.'
 +
end
 +
-- Show the "subpages" link.
 +
if subjectSpace == 828 then -- Module space.
 +
text = text .. ' ' .. makeWikilink('Special:PrefixIndex/' .. templatePage .. '/', 'Subpages of this module')
 +
elseif subjectSpace == 10 then -- Template space.
 +
text = text .. ' ' .. makeWikilink('Special:PrefixIndex/' .. templatePage .. '/', 'Subpages of this template')
 +
elseif subjectSpace ~= 6 then -- Don't show the link in file space.
 +
text = text .. ' ' .. makeWikilink('Special:PrefixIndex/' .. templatePage .. '/', 'Subpages of this page')
 +
end
 +
-- Show the "print" link if it exists.
 +
local printPage = templatePage .. '/Print'
 +
local printTitle = mw.title.new(printPage)
 +
if printTitle.exists then
 +
text = text .. '<br />A [[Help:Books/for experts#Improving the book layout|print version]] of this template exists at ' .. makeWikilink(printPage, '/Print') .. '.'
 +
.. 'If you make a change to this template, please update the print version as well.[[Category:Templates with print versions]]'
 +
end
 +
end
 +
end
 +
fmargs.text = text
 +
 
 +
-- Return the fmbox output.
 +
return messageBox.main('fmbox', fmargs)
 +
end
 +
 
 +
function p.addTrackingCategories()
 +
-- Check if {{documentation}} is transcluded on a /doc or /testcases page.
 +
local ret = ''
 +
local subpage = currentTitle.subpageText
 +
if subpage == 'doc' or subpage == 'testcases' then
 +
local sort = (currentTitle.namespace == 0 and 'Main:' or '') .. currentTitle.prefixedText -- Sort on namespace.
 +
ret = ret .. makeWikilink('Category:Wikipedia pages with strange ((documentation)) usage', sort)
 +
end
 +
return ret
 +
end
 +
 
 +
function p.docspace()
 +
-- Determines the namespace of the documentation.
 +
if subjectSpace == 0 or subjectSpace == 6 or subjectSpace == 8 or subjectSpace == 14 then
 +
-- Pages in the Article, File, MediaWiki or Category namespaces must have their
 +
-- /doc, /sandbox and /testcases pages in talk space.
 +
return mw.site.namespaces[subjectSpace].talk.name
 +
else
 +
return currentTitle.subjectNsText
 +
end
 +
end
 +
 
 +
function p.templatePage()
 +
-- Determines the template page. No namespace or interwiki prefixes are included.
 +
local subpage = currentTitle.subpageText
 +
if subpage == 'sandbox' or subpage == 'testcases' then
 +
return currentTitle.baseText
 +
else
 +
return currentTitle.text
 +
end
 +
end
 +
 
 
return p
 
return p

Revision as of 05:54, 28 December 2013

This module displays a blue box containing documentation for templates, Lua modules, or other pages. The {{documentation}} template invokes it.

Normal usage

For most uses, you should use the {{documentation}} template; please see that template's page for its usage instructions and parameters.

Use in other modules

To use this module from another Lua module, first load it with require:

local documentation = require('Module:Documentation').main

Then you can simply call it using a table of arguments.

documentation{content = 'Some documentation', ['link box'] = 'My custom link box'}

Please refer to the template documentation for usage instructions and a list of parameters.

Porting to other wikis

The module has a configuration file at Module:Documentation/config which is intended to allow easy translation and porting to other wikis. Please see the code comments in the config page for instructions. If you have any questions, or you need a feature which is not currently implemented, please leave a message at Template talk:Documentation to get the attention of a developer.


-- This module implements {{documentation}}.

-- Get required modules.
local getArgs = require('Module:Arguments').getArgs
local htmlBuilder = require('Module:HtmlBuilder')
local messageBox = require('Module:Message box')

local p = {}

-- Constants.
local currentTitle = mw.title.getCurrentTitle()
local subjectSpace = mw.site.namespaces[currentTitle.namespace].subject.id

----------------------------------------------------------------------------
-- Helper functions
----------------------------------------------------------------------------

local function makeWikilink(page, display)
	if display then
		return mw.ustring.format('[[%s|%s]]', page, display)
	else
		return mw.ustring.format('[[%s]]', page)
	end
end

local function makeUrlLink(url, display)
	return mw.ustring.format('[%s %s]', url, display)
end

----------------------------------------------------------------------------
-- Main functions
----------------------------------------------------------------------------

function p.main(frame)
	local args = getArgs(frame, {
		valueFunc = function (key, value)
			if type(value) == 'string' then
				value = value:match('^%s*(.-)%s*$') -- Remove whitespace.
				if key == 'heading' or value ~= '' then
					return value
				else
					return nil
				end
			else
				return value
			end
		end
	})
	return p._main(args)
end

function p._main(args)
	local root = htmlBuilder.create()
	root
		.wikitext(p.protectionTemplate())
		.wikitext(p.sandboxNotice(args))
		 -- This div tag is from {{documentation/start box}}, but moving it here so that we don't have to worry about unclosed tags.
		.tag('div')
			.attr('id', 'template-documentation')
			.addClass('template-documentation iezoomfix')
			.wikitext(p.startBox(args))
			.wikitext(p.content(args))
			.tag('div')
				.css('clear', 'both') -- So right or left floating items don't stick out of the doc box.
				.done()
			.done()
		.wikitext(p.endBox(args))
		.wikitext(p.addTrackingCategories())
	return tostring(root)
end

function p.sandboxNotice(args)
	if currentTitle.subpageText == 'sandbox' then
		local frame = mw.getCurrentFrame()
		local root = htmlBuilder.create()
		root
			.tag('div')
				.css('clear', 'both')
				.done()
			.wikitext(frame:expandTemplate{title = 'template sandbox notice', args = {args.livepage}})
		return tostring(root)
	else
		return nil
	end
end

function p.protectionTemplate()
	if currentTitle.namespace == 10 then -- We are in the template namespace.
		local frame = mw.getCurrentFrame()

		local function getProtectionLevel(protectionType)
			-- Gets the protection level for the current page.
			local level = frame:callParserFunction('PROTECTIONLEVEL', protectionType)
			if level ~= '' then
				return level
			else
				return nil -- The parser function returns the blank string if there is no match.
			end
		end

		if getProtectionLevel('move') == 'sysop' or getProtectionLevel('edit') then
			-- The page is full-move protected, or full, template, or semi-protected.
			return frame:expandTemplate{title = 'pp-template', args = {docusage = 'yes'}}
		end
	end
	return nil
end

function p.startBox(args)
	-- Arg processing from {{documentation}}.
	local preload = args.preload -- Allow custom preloads.
	local heading = args.heading -- Blank values are not removed.
	local headingStyle = args['heading-style']
	local content = args.content
	local docspace = p.docspace()
	local docname = args[1] -- Other docname, if fed.
	local templatePage = p.templatePage()

	-- Arg processing from {{documentation/start box2}}.
	local docpage
	if docname then
		docpage = docname
	else
		local namespace = docspace or currentTitle.nsText
		local pagename = templatePage or currentTitle.text
		docpage = namespace .. ':' .. pagename .. '/doc'
	end
	local docTitle = mw.title.new(docpage)
	local docExist = docTitle.exists
	
	-- Output from {{documentation/start box}}.

	-- First, check the heading parameter.
	if heading == '' then
		-- Heading is defined but blank, so do nothing.
		return nil
	end

	-- Build the start box div.
	local sbox = htmlBuilder.create('div')
	sbox
		.css('padding-bottom', '3px')
		.css('border-bottom', '1px solid #aaa')
		.css('margin-bottom', '1ex')

	-- Make the heading.
	local hspan = sbox.tag('span')
	if headingStyle then
		hspan.cssText(headingStyle)
	elseif subjectSpace == 10 then
		-- We are in the template or template talk namespaces.
		hspan
			.css('font-weight', 'bold')
			.css('fond-size', '125%')
	else
		hspan.css('font-size', '150%')
	end
	if heading then
		-- "heading" has data.
		hspan.wikitext(heading)
	elseif subjectSpace == 10 then -- Template namespace
		hspan.wikitext('[[File:Template-info.png|50px|link=|alt=Documentation icon]] Template documentation')
	elseif subjectSpace == 828 then -- Module namespace
		hspan.wikitext('[[File:Template-info.png|50px|link=|alt=Documentation icon]] Module documentation')
	elseif subjectSpace == 6 then -- File namespace
		hspan.wikitext('Summary')
	else
		hspan.wikitext('Documentation')
	end

	-- Add the [view][edit][history][purge] or [create] links.
	-- Check for the content parameter first, as we don't need the links if the documentation
	-- content is being entered directly onto the template page.
	if not content then
		local lspan = sbox.tag('span') -- lspan is short for "link span".
		lspan
			.addClass('mw-editsection plainlinks')
			.attr('id', 'doc_editlinks')
		if docExist then
			local viewLink = makeWikilink(docpage, 'view')
			local editLink = makeUrlLink(docTitle:fullUrl{action = 'edit'}, 'edit')
			local historyLink = makeUrlLink(docTitle:fullUrl{action = 'history'}, 'history')
			local purgeLink = makeUrlLink(docTitle:fullUrl{action = 'purge'}, 'purge')
			local text = '[%s] [%s] [%s] [%s]'
			text = text:gsub('%[', '&#91;') -- Replace square brackets with HTML entities.
			text = text:gsub('%]', '&#93;')
			lspan.wikitext(mw.ustring.format(text, viewLink, editLink, historyLink, purgeLink))
		else
			if preload then
				preload = mw.uri.encode(preload)
			elseif subjectSpace == 6 then -- File namespace
				preload = 'Template:Documentation/preload-filespace'
			else
				preload = 'Template:Documentation/preload'
			end
			lspan.wikitext(makeUrlLink(docTitle:fullUrl{action = 'edit', preload = preload}, 'create'))
		end
	end

	return tostring(sbox)
end

function p.content(args)
	local content = args.content
	if not content then
		local docpage = args[1]
		if docpage and mw.title.new(docpage).exists then
			local frame = mw.getCurrentFrame()
			content = frame:preprocess('{{ ' .. docpage .. ' }}')
		else
			docpage = p.docspace() .. ':' .. p.templatePage() .. '/doc'
			if mw.title.new(docpage).exists then
				local frame = mw.getCurrentFrame()
				content = frame:preprocess('{{ ' .. docpage .. ' }}')
			end
		end
	end
	-- The line breaks below are necessary so that "=== Headings ===" at the start and end
	-- of docs are interpreted correctly.
	return '\n' .. (content or '') .. '\n' 
end

function p.endBox(args)
	-- Argument processing in {{documentation}}.
	local preload = args.preload -- Allow custom preloads.
	local content = args.content
	local linkBox = args['link box'] -- So "link box=off" works.
	local docspace = p.docspace()
	local docname = args[1] -- Other docname, if fed.
	local templatePage = p.templatePage()

	-- Argument processing in {{documentation/end box2}}.
	local docpageRoot = (docspace or currentTitle.nsText) .. ':' .. (templatePage or currentTitle.text)
	local docpage
	if docname then
		docpage = docname
	else
		docpage = docpageRoot .. '/doc'
	end
	local docTitle = mw.title.new(docpage)
	local docExist = docTitle.exists
	local docnameFed = args[1] and true
	local sandbox = docpageRoot .. '/sandbox'
	local testcases = docpageRoot .. '/testcases'
	templatePage = currentTitle.nsText .. ':' .. templatePage

	-- Output from {{documentation/end box}}
	
	-- First, check whether we should output the end box at all. Add the end box by default if the documentation
	-- exists or if we are in the user, module or template namespaces.
	if linkBox == 'off' or not (docExist or subjectSpace == 2 or subjectSpace == 828 or subjectSpace == 10) then
		return nil
	end

	-- Assemble the arguments for {{fmbox}}.
	local fmargs = {}
	fmargs.id = 'documentation-meta-data'
	fmargs.image = 'none'
	fmargs.style = 'background-color: #ecfcf4'
	fmargs.textstyle = 'font-style: italic;'

	-- Assemble the fmbox text field.
	local text = ''
	if linkBox then
		-- Use custom link box content if it is defined.
		text = text .. linkBox
	else
		if docExist then
			-- /doc exists; link to it.
			local docLink = makeWikilink(docpage)
			local editLink = makeUrlLink(docTitle:fullUrl{action = 'edit'}, 'edit')
			local historyLink = makeUrlLink(docTitle:fullUrl{action = 'history'}, 'history')
			text = text .. 'The above [[Wikipedia:Template documentation|documentation]] is [[Wikipedia:Transclusion|transcluded]] from ' .. docLink .. '.'
				.. ' <small style="font-style: normal;">(' .. editLink .. ' &#124; ' .. historyLink .. ')</small> <br />'
		elseif subjectSpace == 828 then
			-- /doc does not exist; ask to create it.
			local createLink = makeUrlLink(docTitle:fullUrl{action = 'edit', preload = 'Template:Documentation/preload-module-doc'}, 'create')
			text = text .. 'You might want to ' .. createLink .. ' a documentation page for this [[Wikipedia:Lua|Scribunto module]].<br />'
		end
		-- Add links to /sandbox and /testcases when appropriate.
		if subjectSpace == 2 or subjectSpace == 828 or subjectSpace == 10 then
			-- We are in the user, module or template namespaces. 
			local pagePossessive = subjectSpace == 828 and "module's" or "template's"
			text = text .. 'Editors can experiment in this ' .. pagePossessive .. ' '
			local sandboxTitle = mw.title.new(sandbox)
			if sandboxTitle.exists then
				local sandboxLink = makeWikilink(sandbox, 'sandbox')
				local sandboxEditLink = makeUrlLink(sandboxTitle:fullUrl{action = 'edit'}, 'edit')
				local compareLink = makeUrlLink(mw.title.new('Special:ComparePages'):fullUrl{page1 = mw.uri.encode(templatePage), page2 = mw.uri.encode(sandbox)}, 'diff')
				text = text .. sandboxLink .. ' <small style="font-style: normal">(' .. sandboxEditLink .. ' | ' .. compareLink .. ')</small>'
			else
				local sandboxPreload = 'Template:Documentation/preload-' .. (subjectSpace == 828 and 'module-' or '') .. 'sandbox'
				local sandboxCreateLink = makeUrlLink(sandboxTitle:fullUrl{action = 'edit', preload = sandboxPreload}, 'create')
				local mirrorPreload = mw.uri.encode(templatePage)
				local mirrorSummary = mw.uri.encode('Create sandbox version of ' .. makeWikilink(templatePage))
				local mirrorLink = makeUrlLink(sandboxTitle:fullUrl{action = 'edit', preload = mirrorPreload, summary = mirrorSummary}, 'mirror')
				text = text .. 'sandbox <small style="font-style: normal">(' .. sandboxCreateLink .. ' | ' .. mirrorLink .. ')</small>'
			end
			text = text .. ' and '
			local testcaseTitle = mw.title.new(testcases)
			if testcaseTitle.exists then
				local testcasesLink = makeWikilink(testcases, 'testcases')
				local testcasesEditLink = makeUrlLink(testcaseTitle:fullUrl{action = 'edit'}, 'edit')
				text = text .. testcasesLink .. ' <small style="font-style: normal">(' .. testcasesEditLink .. ')</small>'
			else
				local testcasesPreload = 'Template:Documentation/preload-' .. (subjectSpace == 828 and 'module-' or '') .. 'testcases'
				local testcasesCreateLink = makeUrlLink(testcaseTitle:fullUrl{action = 'edit', preload = testcasesPreload}, 'create')
				text = text .. 'testcases <small style="font-style: normal">(' .. testcasesCreateLink .. ')</small>'
			end
			text = text .. ' pages. <br />'
			-- Show the categories text, but not if "content" fed or "docname fed" since then it is unclear where to add the categories.
			if not content and not docnameFed then
				text = text .. 'Please add categories to the ' .. makeWikilink(docpage, '/doc') .. ' subpage.'
			end
			-- Show the "subpages" link.
			if subjectSpace == 828 then -- Module space.
				text = text .. ' ' .. makeWikilink('Special:PrefixIndex/' .. templatePage .. '/', 'Subpages of this module')
			elseif subjectSpace == 10 then -- Template space.
				text = text .. ' ' .. makeWikilink('Special:PrefixIndex/' .. templatePage .. '/', 'Subpages of this template')
			elseif subjectSpace ~= 6 then -- Don't show the link in file space.
				text = text .. ' ' .. makeWikilink('Special:PrefixIndex/' .. templatePage .. '/', 'Subpages of this page')
			end
			-- Show the "print" link if it exists.
			local printPage = templatePage .. '/Print'
			local printTitle = mw.title.new(printPage)
			if printTitle.exists then
				text = text .. '<br />A [[Help:Books/for experts#Improving the book layout|print version]] of this template exists at ' .. makeWikilink(printPage, '/Print') .. '.'
					.. 'If you make a change to this template, please update the print version as well.[[Category:Templates with print versions]]'
			end
		end
	end
	fmargs.text = text

	-- Return the fmbox output.
	return messageBox.main('fmbox', fmargs)
end

function p.addTrackingCategories()
	-- Check if {{documentation}} is transcluded on a /doc or /testcases page.
	local ret = ''
	local subpage = currentTitle.subpageText
	if subpage == 'doc' or subpage == 'testcases' then
		local sort = (currentTitle.namespace == 0 and 'Main:' or '') .. currentTitle.prefixedText -- Sort on namespace.
		ret = ret .. makeWikilink('Category:Wikipedia pages with strange ((documentation)) usage', sort)
	end
	return ret
end

function p.docspace()
	-- Determines the namespace of the documentation.
	if subjectSpace == 0 or subjectSpace == 6 or subjectSpace == 8 or subjectSpace == 14 then
		-- Pages in the Article, File, MediaWiki or Category namespaces must have their
		-- /doc, /sandbox and /testcases pages in talk space.
		return mw.site.namespaces[subjectSpace].talk.name 
	else
		return currentTitle.subjectNsText
	end
end

function p.templatePage()
	-- Determines the template page. No namespace or interwiki prefixes are included.
	local subpage = currentTitle.subpageText
	if subpage == 'sandbox' or subpage == 'testcases' then
		return currentTitle.baseText
	else
		return currentTitle.text
	end
end

return p