Module:URL

From Zoophilia Wiki
Revision as of 22:01, 25 February 2013 by meta>Toohool (use unicode string functions)
Jump to navigationJump to search

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

--
-- This module implements {{URL}}
--

local p = {}
 
function trim(s)
    return (mw.ustring.gsub(s, "^%s*(.-)%s*$", "%1"))
end

function p._url(url, text)
    url = trim(url or '')
    text = trim(text or '')
    
    if url == '' then
        if text == '' then
            return mw.getCurrentFrame():expandTemplate{ title = 'tlx', args = { 'URL', "''example.com''", "''optional display text''" } }
        else
            return text
        end
    end
    
    local uri = mw.uri.new(url)
    if uri.protocol == nil then
        url = 'http://' .. url
        uri = mw.uri.new(url)
    end
    
    if text == '' then
        if uri.path == '/' then uri.path = '' end
        
        text = mw.ustring.lower(uri.host or '') .. uri.relativePath
    end

    return mw.ustring.format('<span class="url">[%s %s]</span>', url, text)
end

function p.url(frame)
    local templateArgs = frame:getParent().args
    local url = templateArgs[1] or ''
    local text = templateArgs[2] or ''
    return p._url(url, text)
end

return p