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 '<div style="font-size: 0.875em;">' .. args.notes .. '</div>' or '<div style="font-size: 0.875em;">N/A</div>'
        
        local categoryLink = site.network and site.network ~= '' and string.format(' <small>([[:Category:%s|%s]])</small>', site.network, site.network) or ''

        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" | %s\n| style="text-align:center;vertical-align:middle;" | %s%s\n| style="text-align:center;vertical-align:middle;" | %s\n| style="text-align:center;vertical-align:middle;" | %s\n| style="width: 0px; padding: 0px; border: none;" |\n',
		    style,
		    args[1] == 'Studio' and 'Studio' or 'Website',
		    string.format('[%s %s]', url, site.label),
		    categoryLink,
		    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 string.format('|-\n| style="text-align:center;vertical-align:middle;" | Error: invalid site parameter: %s. See [[PBC:SiteList|PBC:SiteList for documentation]]', args.site)
    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