模块:FileNameFix

来自The Land of StarLight

此模块的文档可以在模块:FileNameFix/doc创建

--[[
{{#invoke:FileNameFix|filefix|可能有问题的文件名}}
{{#invoke:FileNameFix|urlfix|可能有问题的文件名}}
这个东西主要用于修复包含“ \ / * : ? " < > | ”这些字符的页面自动生成的文件名
映射表:
“\ / * |”:“-”
“:”:“:”
“?”:“?”
“"”:“'”
]]
local nf = {}

function nf.filefix(frame)
	local new_args = nf._getParameters( frame.args, { 'name' } );
	local fname = new_args['name'] or '';
	fname = mw.ustring.gsub(fname, '/', '-');
	fname = mw.ustring.gsub(fname, '\\', '-');
	fname = mw.ustring.gsub(fname, '*', '-');
	fname = mw.ustring.gsub(fname, '|', '-');
	fname = mw.ustring.gsub(fname, ':', ':');
	fname = mw.ustring.gsub(fname, '?', '?');
	fname = mw.ustring.gsub(fname, '"', '\'');
	return fname
end

function nf.urlfix(frame)
	local new_args = nf._getParameters( frame.args, { 'name' } );
	local fname = new_args['name'] or '';
	fname = mw.ustring.gsub(fname, ' ', '_');
	fname = mw.ustring.gsub(fname, '\'', '%%27');
	fname = mw.ustring.gsub(fname, '&', '%%26');
	fname = mw.ustring.gsub(fname, '%%26#38;', '%%26');
	fname = mw.ustring.gsub(fname, '%%26#39;', '%%27');
	return fname
end

function nf._getParameters( frame_args, arg_list )
    local new_args = {};
    local index = 1;
    local value;
    
    for i,arg in ipairs( arg_list ) do
        value = frame_args[arg]
        if value == nil then
            value = frame_args[index];
            index = index + 1;
        end
        new_args[arg] = value;
    end
    
    return new_args;
end 

return nf