Permanently protected module

Module:Autolink

From Yugipedia
Revision as of 06:58, 24 February 2015 by Dinoguy1000 (talk | contribs) (fix and tweak)
Jump to: navigation, search

--
-- implements {{Autolink}} and {{Unlink}}
--

local p = {}

function p.link( frame )
    local args = frame:getParent().args
    -- for testing from console
    -- local args = frame
    -- marker used for {{nolink}} support (doesn't have to be a zero-width non-joiner, that's just what I used in the template version)
    local zwnj = '‌'
    local links, nolink, listmarkup, escape, el, link, text, formatl, formatr

    -- set default to stop errors
    args[1] = args[1] or ''
    args[1] = mw.text.trim( args[1] )

    links = mw.text.split( args[1], '\n' )

    args[2] = #links == 1 and args[2]

    nolink = mw.ustring.find( args[1], zwnj ) == 1
    listmarkup = #links == 1 and ''

    for i=1, #links do
        el = mw.text.trim( links[i] )

        -- catch empty string at the start of lists
        if el ~= '' then
            if listmarkup ~= '' then
                listmarkup = mw.ustring.match( el, '^([*#])' ) or '*'
                el = mw.ustring.gsub( el, '^[*#]', '' )
            end

            if mw.ustring.find( el, zwnj ) or mw.ustring.find( el, '%[%[' ) or nolink then
                el = mw.ustring.gsub( el, '%[%[[Cc]ategory:', '[[:Category:' )
                el = mw.ustring.gsub( el, '%[%[[Ff]ile:', '[[:File:' )
                el = mw.ustring.gsub( el, '%[%[[Ii]mage:', '[[:Image:' )
                links[i] = table.concat( { listmarkup, el }, '' )
            else
                link = el
                text = args[2] or el
                formatl = ''
                formatr = ''

                link = mw.ustring.gsub( link, '""', '' )
                link = mw.ustring.gsub( link, "'''?", '' )

                -- check for formatting that can be moved out of the link entirely
                if mw.ustring.find( text, '^""' ) and mw.ustring.find( text, '""$' ) then
                    formatl = '"'
                    formatr = '"'
                    text = mw.ustring.gsub( text, '""', '' )
                else
                    text = mw.ustring.gsub( text, '""', '"' )
                end
                if mw.ustring.find( text, "^'''" ) and mw.ustring.find( text, "'''$" ) then
                    formatl = formatl .. "'''"
                    formatr = "'''" .. formatr
                    text = mw.ustring.gsub( text, "^'''", '' )
                    text = mw.ustring.gsub( text, "'''$", '' )
                end
                if mw.ustring.find( text, "^''" ) and mw.ustring.find( text, "''$" ) then
                    formatl = formatl .. "''"
                    formatr = "''" .. formatr
                    text = mw.ustring.gsub( text, "^''", '' )
                    text = mw.ustring.gsub( text, "''$", '' )
                end

                if mw.ustring.find( link, '^[Cc]ategory:' ) or
                        mw.ustring.find( link, '^[Ff]ile:' ) or
                        mw.ustring.find( link, '^[Ii]mage:' ) then
                    escape = ':'
                else
                    escape = ''
                end

                if link == text then
                    links[i] = table.concat( { listmarkup, formatl, '[[', escape, link, ']]', formatr }, '' )
                else
                    links[i] = table.concat( { listmarkup, formatl, '[[', escape, link, '|', text, ']]', formatr }, '' )
                end
            end
        end
    end

    links = table.concat( links, '\n' )
    links = mw.ustring.gsub( links, zwnj, '' )
    links = mw.text.trim( links )
    return links
end

function p.unlink( frame )
    local args = frame:getParent().args
    args[1] = args[1] or ''
    return mw.ustring.match( args[1], '%[%[:?(.-)[|%]]' ) or mw.text.trim( args[1] )
end

return p