Module:Flag

local p = {}

function p._main(frame, name, mode, check)
 local categories = {
  ['Flag icon'] = '[[Category:Flag icons missing country data templates]]',
  ['Flag decoration'] = '[[Category:Flag decoration missing country data templates]]',
 }

    local yesno = require('Module:Yesno')

 --Get input arguments
 local args = require('Module:Arguments').getArgs(frame,{frameOnly = ((frame.args.frameonly or '') ~= ''), valueFunc =
 function(key,value)
  if value then
   value = mw.text.trim(value)
   --Change empty string to nil for all args except 'image' and 'border'
   if key=="image" or key=="border" or value~="" then
    return value
   end
  end
  return nil
 end
 })
 
 local preftext = ""
 
 if name == 'Flag+link' then
  args.pref = args[1]
 elseif name == 'Flag athlete' then
  args.preftext = args[1]
  args[2] = args.code or args[2] or ""
  args.variant = args.variant or args[3]
 elseif name == 'Flag medalist' then
  preftext = (args[1] or "") .. "<br/>"
  args[2] = args.code or args[2] or ""
  args.variant = args.variant or args[3]
 elseif name == 'Flag link' then
  args.suff = args[2]
  args[2] = args[1] or ""
  args.variant = args.variant or args[3]
 else
  sports = {fb = 'fb', ['fb-rt'] = 'fb', fbicon = 'fb', ru = 'ru'}
  if sports[name] then
   args.avar = sports[name]
   if name == 'fb' and yesno(args.align) == true then
    args.align='c'
   end
  end
  args.variant = args.variant or args[2]
  args[2] = args[1] or ""
 end

 args[1] = mode .. (args.opts or '')
 args.missingcategory = categories[name]
 args.noredlink = args.noredlink or 'notext'

 if args.placeholder and yesno(args.placeholder) ~= true then
  args[1] = args[1] .. 'o'
 end

 if mw.title.getCurrentTitle().namespace == 0 then
  local unknown = '[[Category:Pages using ' .. mw.ustring.lower(name) .. ' template with unknown parameters|_VALUE_' ..
    frame:getParent():getTitle() .. ']]'
 else
  local unknown = ''
 end

 if check then
  local opts = {
            unknown=unknown,
            preview='Page using [[Template:' .. name .. ']] with unknown parameter "_VALUE_"',
            ignoreblank='y',
            '1',
            '2',
            'variant',
            'image',
            'size',
            'sz',
            'border',
            'align',
            'al',
            'width',
            'w',
            'alt',
            'ilink',
            'noredlink',
            'missingcategory',
            'name',
            'clink',
            'link',
            'pref',
            'suff',
            'plink',
            'the',
            'section',
            'altvar',
            'avar',
            'age',
            'nalign',
            'nal',
            'text',
            'nodata',
            'opts',
            'placeholder',
            'getParent',
            'frameonly'
  }

  check = require('Module:Check for unknown parameters')._check(opts,args)
 else
  check = ''
 end

 return preftext .. require('Module:Flagg').luaMain(frame,args) .. check
end

function p.main(frame) return p._main(frame,       'Flag',            'unc',    false) end
p.flag = p.main
p[''] = p.main
function p.decoration(frame) return p._main(frame, 'Flag decoration', 'uxx',    false) end
p.deco = p.decoration
p.flagdeco = p.decoration
p['flag decoration'] = p.decoration
function p.icon(frame) return p._main(frame,       'Flag icon',       'cxxl',   true) end
p.flagicon = p.icon
p['flag icon'] = p.icon
function p.link(frame) return p._main(frame,       'Flag link',       'unpe',    false) end
p.flaglink = p.link
p['flag link'] = p.link
function p.pluslink(frame) return p._main(frame,   'Flag+link',       'unpefo', false) end
p['+link'] = p.pluslink
p['flag+link'] = p.pluslink
function p.country(frame) return p._main(frame,    'Flag country',    'unce',   false) end
p.flagcountry = p.country
p['flag country'] = p.country
function p.athlete(frame) return p._main(frame,    'Flag athlete',    'unape',  false) end
p.flagathlete = p.athlete
p['flag athlete'] = p.athlete
function p.medalist(frame) return p._main(frame,   'Flag medalist',   'unce',   false) end
p.flagmedalist = p.medalist
p['flag medalist'] = p.medalist
function p.fb(frame) return p._main(frame,         'fb',              'unpe',   false) end
function p.fbrt(frame) return p._main(frame,       'fb-rt',           'unpre',  false) end
p['fb-rt'] = p.fbrt
function p.fbicon(frame) return p._main(frame,     'fbicon',          'pxxl', false) end
function p.ru(frame) return p._main(frame,         'ru',              'unpe', false) end
function p.IOCathlete(frame) return require('Module:Country_alias').flagXYZ(frame:newChild{ title = 'Template:Flag IOC athlete', args = {type = 'athlete-O'} }) end
p.flagIOCathlete = p.IOCathlete
p['IOC athlete'] = p.IOCathlete
p['flag IOC athlete'] = p.IOCathlete
return p