Module:SiteList

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

local p = {}

local sites = {
    website1 = {
        label = 'Website 1',
        aliases = {'website1', 'website1.com', 'website1com'},
        url = 'https://website1.com/',
        updatesUrl = 'https://website1.com/updates/',
        modelsUrl = 'https://website1.com/models/',
        category = 'Website 1 models',
        network = 'Network1 Network',
    },
    website2 = {
        label = 'Website 2',
        aliases = {'website2', 'website2.com', 'website2com'},
        url = 'https://website2.com/',
        updatesUrl = 'https://website2.com/scene/',
        modelsUrl = 'https://website2.com/people/',
        category = 'Website 2 models',
        network = 'Network2 Network',
    },
    website3 = {
        label = 'Website 3',
        aliases = {'website3', 'website3.com', 'website3com'},
        url = 'https://website3.com/',
        updatesUrl = 'https://website3.com/videos/',
        modelsUrl = 'https://website3.com/performers/',
        category = 'Website 3 models',
        network = 'Network3 Network',
    },
    website4 = {
        label = 'Website 4',
        aliases = {'website4', 'website4.com', 'website4com'},
        url = 'https://website4.com/',
        updatesUrl = 'https://website4.com/videos/',
        modelsUrl = 'https://website4.com/performers/',
        category = 'Website 4 models',
        network = '',
    },
}

function p.site(frame)
    local args = require('Module:Arguments').getArgs(frame, {wrappers = 'Template:SiteList'})
    local site = getSite(args.site)

    if site then
		local url = site.url
		if args.sceneid then
		    url = site.updatesUrl .. args.sceneid .. '.html'
		elseif args.id then
		    url = site.modelsUrl .. args.id .. '.html'
		end
		
		local alias = args.alias or mw.title.getCurrentTitle().text
		local notes = args.notes and '<small><span style="font-size: 0.875em;">' .. args.notes .. '</span></small>' or '<small><span>N/A</span></small>'
		
		local categoryLink = site.network and site.network ~= '' and string.format(' <small>([[:Category:%s|%s]])</small>', site.network, site.network) or '<small><span>N/A</span></small>'

        local style
        if args[1] == 'Studio' then
            style = 'background-color:#DADADA;text-align:center;color:black'
        else
            style = 'background-color:#E7E7E7;text-align:center;color:black'
        end

		local result = string.format(
		    '|-\n| style="%s" | <span>%s</span>\n| style="text-align:center;" | <span>%s</span>\n| style="text-align:center;" | <span>%s</span>\n| style="text-align:center;" | %s\n',
		    style,
		    args[1] == 'Studio' and 'Studio' or 'Website',
		    string.format('[%s %s] <small>([[:Category:%s|%s]])</small>', url, site.label, site.category, site.category),
		    alias,
		    notes
		)

        local ns = mw.title.getCurrentTitle().namespace
        if ns == 0 then
            result = result .. '[[Category:' .. site.category .. ']]'
            result = result .. '[[Category:Articles in the mainspace using Template:SiteList]]'
        elseif ns == 2 then
            result = result .. '[[Category:Articles in the userspace using Template:SiteList]]'
        elseif ns == 118 then
            result = result .. '[[Category:Drafts using Template:SiteList]]'
        end

        return result
    else
        return mw.text.nowiki('{{SiteList}}') .. ' Error: invalid site parameter'
    end
end

function table.contains(table, element)
  for _, value in pairs(table) do
    if value == element then
      return true
    end
  end
  return false
end

function getSite(name)
    for _, site in pairs(sites) do
        if site.label:lower() == name:lower() or table.contains(site.aliases, name:lower()) then
            return site
        end
    end
    return nil
end

return p