Editing Module:Video

local p = {}

function p.file( frame )
	-- looks for a webm file that matches the first and second arguments separated by a dash and ends in '-voice'
	if frame.args[2] and frame.args[2] ~= 'none' then
		fileexists = frame:expandTemplate{title = 'exist', args = {'file:'..frame.args[1]..'-'..frame.args[2]..'-voice.webm', 'yes'}}
		if fileexists == 'yes' then
			return frame.args[1]..'-'..frame.args[2]..'-voice.webm'
		end	
	end
	
	-- looks for a webm file that matches:
		-- a matched source at the beginning of the first argument,
		-- and appends the matched source to the first argument separated by a dash,
		-- and ends in '-voice'
	for _, source in pairs({'RoE', 'AB', 'SoD', 'HC', 'HotA'}) do
		if frame.args[1]:match('^' .. source) == source then
			fileexists = frame:expandTemplate{title = 'exist', args = {'file:'..frame.args[1]..'-'..source..'-voice.webm', 'yes'}}
			if fileexists == 'yes' then
				return frame.args[1]..'-'..source..'-voice.webm'
			end
			break
		end
	end

	-- looks for a webm file that matches the first argument by itself
	fileexists = frame:expandTemplate{title = 'exist', args = {'file:'..frame.args[1]..'.webm', 'yes'}}
	if fileexists == 'yes' then
		return frame.args[1]..'.webm'
	end
	
	-- returns a gif using the first argument with its last two parts separated by dashes removed
	local lines = p.split(frame.args[1], '-')
	local parent2 = ''
	for i, line in ipairs(lines) do
	   parent2 = parent2 .. line
		if i + 2 >= #lines then
			break
		else
			parent2 = parent2 .. '-'
		end
	end
	return parent2..'.gif'
end

function p.split(str, sep)
   local result = {}
   local regex = ("([^%s]+)"):format(sep)
   for each in str:gmatch(regex) do
      table.insert(result, each)
   end
   return result
end

return p