Модуль:URL

Для документации этого модуля может быть создана страница Модуль:URL/doc

--
-- This module implements {{URL}}
--
-- See unit tests at [[Module:URL/tests]]

local p = {}
 
local function safeUri(s)
    local success, uri = pcall(function()
        return mw.uri.new(s)
    end)
    if success then
        return uri
    end
end

function p._url(url, text)
    url = mw.text.trim(url or '')
    text = mw.text.trim(text or '')
    
    if url == '' then
        return text
    end
    
    -- If the URL contains any unencoded spaces, encode them, because MediaWiki will otherwise interpret a space as the end of the URL.
    url = mw.ustring.gsub(url, '%s', function(s) return mw.uri.encode(s, 'PATH') end)
    
    -- If there is an empty query string or fragment id, remove it as it will cause mw.uri.new to throw an error
    url = mw.ustring.gsub(url, '#$', '')
    url = mw.ustring.gsub(url, '%?$', '')
    
    -- If it's an HTTP[S] URL without the double slash, fix it.
    url = mw.ustring.gsub(url, '^[Hh][Tt][Tt][Pp]([Ss]?):(/?)([^/])', 'http%1://%3')

    -- Handle URLs from Wikidata of the format http://
    url = mw.ustring.gsub(url, '^[Hh][Tt][Tt][Pp]([Ss]?)://', 'http%1://')
    
    local uri = safeUri(url)
    
    -- Handle URL's without a protocol and URL's that are protocol-relative, 
    -- e.g. www.example.com/foo or www.example.com:8080/foo, and //www.example.com/foo
    if uri and (not uri.protocol or (uri.protocol and not uri.host)) and url:sub(1, 2) ~= '//' then
        url = 'http://' .. url
        uri = safeUri(url)
    end
    
    if text == '' then
        if uri then
            if uri.path == '/' then uri.path = '' end
            
            local port = ''
            if uri.port then port = ':' .. uri.port end
            
            text = mw.ustring.lower(uri.host or '') .. port .. (uri.relativePath or '')

			-- Add <wbr> before _/.-# sequences
			text = mw.ustring.gsub(text,"(/+)","<wbr/>%1")      -- This entry MUST be the first. "<wbr/>" has a "/" in it, you know.
			text = mw.ustring.gsub(text,"(%.+)","<wbr/>%1")
			-- text = mw.ustring.gsub(text,"(%-+)","<wbr/>%1") 	-- DISABLED for now
			text = mw.ustring.gsub(text,"(%#+)","<wbr/>%1")
			text = mw.ustring.gsub(text,"(_+)","<wbr/>%1")
        else -- URL is badly-formed, so just display whatever was passed in
            text = url
        end
    end

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

function p.url(frame)
    local templateArgs = frame.args
	local parentArgs = frame:getParent().args;
    local url = templateArgs[1] or parentArgs[1] or ''
    local text = templateArgs[2] or parentArgs[2]
    if not text then
    	url = url or extractUrl(templateArgs) or extractUrl(parentArgs);
	end
	text = text or ''
    return p._url(url, text)
end
function extractUrl(args)
	for name, val in pairs(args) do
		local url = name .. "=" .. val;
		url = mw.ustring.gsub(url, '^[Hh][Tt][Tt][Pp]([Ss]?):(/?)([^/])', 'http%1://%3')
		local uri = safeUri(url);
		if uri and uri.host then
			return url
		end
	end
end

function startsWith( source, substring )
	if mw.ustring.len( substring ) > mw.ustring.len( source ) then
		return false
	end
	return mw.ustring.sub( source, 1, mw.ustring.len( substring ) ) == substring
end

function formatUrlImpl( source, title, length )
	local scheme, host, path
	local postfix = ''
	local arg1, arg2 = source, title
	local isTestPage = mw.title.getCurrentTitle().prefixedText == 'Модуль:URL'
	
	-- Две квадратные скобки подряд — [[вики-ссылка]] вместо [ссылки] — возвращаем вход как есть.
	if string.find( arg1, "[[", 1, true ) then
		local result = arg1
		if not isTestPage then
			result = result .. '[[Категория:Википедия:Статьи с вики-ссылкой, переданной в Модуль:URL]]'
			if arg2 then
				-- Если есть arg2, а мы распарсить ссылку не смогли, и значит заменить title не сможем корректно, это есть ошибка.
				result = result .. '[[Категория:Википедия:Статьи с ошибочной работой Модуль:URL]]'
			end
		end
		return result
	end
	
	-- Более одной квадратной скобки — скорее всего, задано более одного URL — тоже возвращаем как есть.
	if select(2, string.gsub( arg1, "%[", "" )) > 1 then
		local result = arg1
		if not isTestPage then
			result = result .. '[[Категория:Википедия:Статьи со сложным входом в Модуль:URL]]'
			if arg2 then
				-- Если есть arg2, а мы распарсить ссылку не смогли, и значит заменить title не сможем корректно, это есть ошибка.
				result = result .. '[[Категория:Википедия:Статьи с ошибочной работой Модуль:URL]]'
			end
		end
		return result
	end
	
	source = mw.text.trim( source, "%[%] " )
	local titleDelimeterPosition = mw.ustring.find( source, " ", 1 )
	if titleDelimeterPosition then
		if not title or title == "" then
			title = mw.ustring.sub( source, titleDelimeterPosition + 1 )
			local postfixDelimeterPosition = mw.ustring.find( title, "%]", 1 )
			if postfixDelimeterPosition then
				postfix = mw.ustring.sub( title, postfixDelimeterPosition + 1 )
				title = mw.ustring.sub( title, 1, postfixDelimeterPosition - 1 )
			end
		end
		source = mw.ustring.sub( source, 1, titleDelimeterPosition - 1 )
	end
	
	local hostStartPosition
	local schemeDelimeterPosition = mw.ustring.find( source, "://", 1, true )
	if schemeDelimeterPosition then
		scheme = mw.ustring.sub( source, 1, schemeDelimeterPosition + 2)
		hostStartPosition = schemeDelimeterPosition + 3
	elseif mw.ustring.find( source, "^//", 1 ) then
		scheme = "//"
		hostStartPosition = 3
	elseif mw.ustring.find( source, "^mailto:", 1 ) then
		scheme = "mailto:"
		hostStartPosition = 8
	elseif mw.ustring.find( source, "@", 1 ) then
		scheme = "mailto:"
		source = scheme .. source
		hostStartPosition = 8
	else
		scheme = "http://"
		source = scheme .. source
		hostStartPosition = 8
	end

	if title then
		local finds = mw.ustring.find( arg1, "[", 1, true )
		if titleDelimeterPosition and finds and finds > titleDelimeterPosition + 1 then
			-- Если titleDelimeterPosition промазал мимо скобки и нашел пробел раньше неё, к примеру "a [b  c]",
			-- то свернуть всю нашу хиромантию и выдать первый аргумент без изменений.
			if arg2 == nil then
				return arg1 .. (isTestPage and '' or '[[Категория:Википедия:Статьи со сложным входом в Модуль:URL]]')
			-- Если есть arg2, а мы распарсить ссылку не смогли, и значит заменить title не сможем корректно, это есть ошибка.
			-- С другой стороны, если arg2 нет, а arg1 очень сложный, то возможно это нормальный ход вещей,
			-- и на вход в модуль дана уже очень сильно оформленная ссылка.
			else
				return arg1 .. (isTestPage and '' or '[[Категория:Википедия:Статьи с ошибочной работой Модуль:URL]]')
			end
		end
		return '[' .. source .. ' ' .. title .. ']' .. postfix
	end

	local hostDelimeterPosition = mw.ustring.find( source, "/", hostStartPosition, true )
	if hostDelimeterPosition then
		host = mw.ustring.sub( source, hostStartPosition, hostDelimeterPosition - 1 )
		if hostDelimeterPosition == mw.ustring.len( source ) then
			path = nil
		else
			path = mw.ustring.sub( source, hostDelimeterPosition + 1 )
		end
	else
		host = mw.ustring.sub( source, hostStartPosition )
	end

	-- post-split format options
	if startsWith( host, 'www.' ) then
		host = mw.ustring.sub( host, 5 )
	end
	host = mw.language.new( 'en' ):lc( host )

	if path and path ~= '' and path ~= '/' then
		local title = host .. '/' .. path
		if length and #title > length then
			title = host .. '/' .. mw.ustring.sub( path, 1, length - #title - 2 ) .. '…'
		end
		return '[' .. source .. ' ' .. title .. ']' .. postfix
	else
		return '[' .. source .. ' ' .. host .. ']' .. postfix
	end
end

function p.formatUrl( frame )
	local url = frame.args[1] or ''
	local title = frame.args[2] or ''
	local length = frame.args['length'] and tonumber( frame.args['length'] )

	url = mw.text.trim( url )
	title = mw.text.trim( title )

	if url == '' then
		return nil
	end
	if title == '' then
		title = nil
	end
	return formatUrlImpl( url, title, length )
end

function p.formatUrlSingle( context, options, url )
	url = mw.text.trim( url )
	if url == '' then
		return nil
	end
	local title = nil
	if ( options['text'] and options['text'] ~= '' ) then
		title = options['text']
	end
	local length = options['length'] and tonumber( options['length'] )

	return formatUrlImpl( url, title, length )
end

return p