Module:Song video

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

local getArgs   = require('Module:Arguments').getArgs
local utils     = require('Module:Utilities')
local constants = require('Module:Constants')

local p = {}

local color = nil
local colors = {
	['easy preview'] = constants.get_difficulty_color('easy'),
	['normal preview'] = constants.get_difficulty_color('normal'),
	['hard preview'] = constants.get_difficulty_color('hard'),
	['expert preview'] = constants.get_difficulty_color('expert'),
	['master preview'] = constants.get_difficulty_color('master'),
	['original mv'] = '#66dd11',
	['game 2d mv'] = '#66dd11',
	['full 2d mv'] = '#00cdba',
	['3d mv'] = '#66dd11'
}

local function makeHeader(video_type)
	local headers = {
		['easy preview'] = 'Easy Preview',
		['normal preview'] = 'Normal Preview',
		['hard preview'] = 'Hard Preview',
		['expert preview'] = 'Expert Preview',
		['master preview'] = 'Master Preview',
		['original mv'] = 'Original MV',
		['game 2d mv'] = '2D MV (game version)',
		['full 2d mv'] = '2D MV (full version)',
		['3d mv'] = '3D MV'
	}
	
	local header = mw.html.create('div')
		:addClass('header')
		:css({
			['background-color'] = color,
		})
		:wikitext(utils.editIfNil(
			video_type,
			function()
				return utils.editIfNil(headers[video_type:lower()])
			end
		))
		:done()
		
	return header
end

local function makeVideo(frame, link)
	local video = mw.html.create('div')
		:addClass('video')
	
	if link then
		local is_yt = string.find(link, '^' .. ('[a-zA-Z0-9_-]'):rep(11) .. '$') ~= nil
		local is_file = string.find(link, '.+\.mp4') ~= nil
		
		if is_yt then
			video:wikitext(frame:extensionTag{ 
				name = 'youtube',
				args = { width = "320", height= "180" },
				content = link
			})
		elseif is_file then
			video:wikitext('[[File:' .. link .. '|320px]]')
		else
			video:wikitext(utils.editIfNil(nil))
		end
	else
		video:wikitext(utils.editIfNil(nil))
	end
	
	return video
end

local function makeInfo(release_date, illustration, animation, video)
	local infoHeaderCss = {
		['background-color'] = color
	}
	
	local info = mw.html.create('div')
		:addClass('info')

	info:tag('div')
			:css(infoHeaderCss)
			:addClass('ih')
			:wikitext('Release date')
			:done()
		:tag('div')
			:addClass('id')
			:css(infoCellCss)
			:wikitext(utils.editIfNil(release_date))
			:done()
	
	if illustration and illustration ~= '' then
		info:tag('div')
				:css(infoHeaderCss)
				:addClass('ih')
				:wikitext('Illustration')
				:done()
			:tag('div')
				:addClass('id')
				:wikitext(illustration)
				:done()
	end
	
	if animation and animation ~= '' then
		info:tag('div')
				:css(infoHeaderCss)
				:addClass('ih')
				:wikitext('Animation')
				:done()
			:tag('div')
				:addClass('id')
				:wikitext(animation)
				:done()
	end
	
	if video and video ~= '' then
		info:tag('div')
				:css(infoHeaderCss)
				:addClass('ih')
				:wikitext('Video')
				:done()
			:tag('div')
				:addClass('id')
				:wikitext(video)
				:done()
	end
			
	return info
end

function p.main(frame)
	local args = getArgs(frame)
	
	local video_type = args['type']
	local link = args['link']
	local release_date = args['date']
	local illustration = args['illustration']
	local animation = args['animation']
	local video = args['video']
	
	local root = mw.html.create('div')
		:addClass('song-video')
	
	color = colors[video_type] or color
	
	local header = makeHeader(video_type)
	root:node(header)
	
	local body = root:tag('div')
		:addClass('content')
		:node(makeVideo(frame, link))
		:node(makeInfo(release_date, illustration, animation, video))
	
	return mw.getCurrentFrame():extensionTag{
			name = 'templatestyles',
			args = { src = 'Template:Song video/styles.css' }
		} .. tostring(root)
end

return p