Module:GAN link

-- This module implements {{GAN link}}.

local p = {}

local function getGANIcon()
 return require('Module:Icon')._main{'GAN'}
end

local function makeWikilink(page, display)
 if display and display ~= page then
  return string.format('[[%s|%s]]', page, display)
 else
  return string.format('[[%s]]', page)
 end
end

local function getNominationPage(article, number)
 return string.format('Talk:%s/GA%d', article, number)
end

local function makeArticleLink(options)
 assert(options.article, 'options.article not specified')
 assert(options.formattedArticle, 'options.formattedArticle not specified')
 local display
 if options.isItalic then
  display = string.format('<i>%s</i>', options.article)
 elseif options.display then
  display = options.display
 else
  display = options.formattedArticle
 end
 return makeWikilink(options.article, display)
end

local function makeExistingNominationLink(nominationPage)
 return makeWikilink(nominationPage, 'nom')
end

local function makeNewNominationLink(nominationPage)
 local url = mw.uri.fullUrl(nominationPage, {
  action    = 'edit',
  editintro = 'Template:GAN/editintro',
  preload   = 'Template:GAN/preload',
 })
 return string.format(
  "'''<span class='plainlinks'>[%s start]</span>'''",
  tostring(url)
 )
end

local function makeNominationLink(nominationPage)
 assert(nominationPage, 'no nominationPage argument given to makeNominationLink')
 local title = mw.title.new(nominationPage)
 if not title then
  error(string.format('%s is not a valid title', nominationPage), 2)
 elseif title.exists then
  return makeExistingNominationLink(nominationPage)
 else
  return makeNewNominationLink(nominationPage)
 end
end

function p._main(args)
 -- Link parameters
 local formattedArticle = args[1]
 if not formattedArticle then
  error('No article specified', 2)
 end
 local article = formattedArticle:gsub("'''", ""):gsub("''", "")
 local display = args[2]

 -- Number
 local number
 if args['#'] then
  number = tonumber(args['#'])
  if not number then
   error("'%s' is not a valid number", args['#'])
  end
 else
  number = 1
 end

 -- Formatting parameters
 local hasIcon = not not args.icon
 local isItalic = not not args.i

 -- Output
 local ret = {}
 if hasIcon then
  ret[#ret + 1] = getGANIcon()
 end
 ret[#ret + 1] = makeArticleLink{
  article = article,
  formattedArticle = formattedArticle,
  display = display,
  isItalic = isItalic,
 }
 ret[#ret + 1] = string.format(
  '(%s)',
  makeNominationLink(getNominationPage(article, number))
 )
 return table.concat(ret, ' ')
end

function p.main(frame)
 local args = require('Module:Arguments').getArgs(frame, {
  wrappers = 'Template:GAN link'
 })
 return p._main(args)
end

return p