local z = {}
local WikitextLC = require( 'Module:WikitextLC' )
function Tcode( args )
if args.T == nil or args.T == '' then
return ''
end
local div = mw.html.create( 'div' )
:attr( 'class', 'noteTA-title' )
:attr( 'data-noteta-code', args.T )
:wikitext( WikitextLC.title( args.T ) )
if args.dt ~= nil and args.dt ~= '' then
div:attr( 'data-noteta-desc', args.dt )
end
return tostring( div )
end
function group( name, frame )
if name == nil or name == '' then
return ''
end
local moduleTitle = mw.title.makeTitle( 'Module', 'CGroup/' .. name )
if moduleTitle and moduleTitle.exists then
local data = mw.loadData( 'Module:CGroup/' .. name )
local pieces = {}
if data.content then
local matched_indices = z.match_text_with_cgroup(mw.title.getCurrentTitle(), data.content)
-- local matched_indices = z.match_text_with_cgroup(mw.title.new('User:GnolizX/肖申克的救赎/新'), data.content)
for i, v in ipairs( data.content ) do
if matched_indices[i] and v.type == 'item' and v.rule then
table.insert( pieces, '-{H|' .. v.rule .. '}-' )
end
end
return tostring( mw.html.create( 'div' )
:attr( 'data-noteta-group-source', 'module' )
:attr( 'data-noteta-group', data.name or name )
:wikitext( table.concat( pieces ) ) )
end
end
local templateTitle = mw.title.makeTitle( 'Template', 'CGroup/' .. name )
if templateTitle and templateTitle.exists then
return frame:expandTemplate{ title = templateTitle }
end
return tostring( mw.html.create( 'div' )
-- :attr( 'id', 'noteTA-group-' .. mw.uri.anchorEncode( name ) )
:attr( 'data-noteta-group-source', 'none' )
:attr( 'data-noteta-group', name ) )
end
function Gcode( args, frame )
local code = {}
for i = 1, 30 do
table.insert( code, group( args['G' .. i], frame ) )
end
code = table.concat( code )
if code ~= '' then
code = tostring( mw.html.create( 'div' )
:attr( 'class', 'noteTA-group' )
:wikitext( code ) )
if args.G31 ~= nil then
code = code .. '[[Category:NoteTA模板参数使用数量超过限制的页面|G]]'
end
end
return code
end
function local_( i, code, desc )
if code == nil or code == '' then
return ''
end
local div = mw.html.create( 'div' )
-- :attr( 'id', 'noteTA-local-' .. i )
:attr( 'data-noteta-code', code )
:wikitext( WikitextLC.hidden( code ) )
if desc ~= nil and desc ~= '' then
div:attr( 'data-noteta-desc', desc )
end
return tostring( div )
end
function Lcode( args )
local code = {}
for i = 1, 30 do
table.insert( code, local_( i, args[i], args['d' .. i] ) )
end
code = table.concat( code )
if code ~= '' then
code = tostring( mw.html.create( 'div' )
:attr( 'class', 'noteTA-local' )
:wikitext( code ) )
if args[31] ~= nil then
code = code .. '[[Category:NoteTA模板参数使用数量超过限制的页面|L]]'
end
end
return code
end
function z.main( frame )
local args
if frame == mw.getCurrentFrame() then
-- Being called from {{noteTA}}
args = frame:getParent().args
else
-- Being called from another module
args = frame
frame = mw.getCurrentFrame()
end
local Tc = Tcode( args )
local Gc = Gcode( args, frame )
local Lc = Lcode( args )
local code = Tc .. Gc .. Lc
if code ~= '' then
local hash = require( 'Module:Crc32lua' ).crc32( mw.dumpObject( args ) )
code = frame:extensionTag{
name = 'indicator',
content = '[[File:Zh conversion icon m.svg|35px|本页使用了标题或全文手工转换|link=|class=skin-invert]]',
args = { name = string.format( 'noteTA-%x', hash ) },
} .. tostring( mw.html.create( 'div' )
:attr( 'id', string.format( 'noteTA-%x', hash ) )
:attr( 'class', 'noteTA' )
:wikitext( code ) )
if mw.title.getCurrentTitle():inNamespace( 'Template' ) then
code = code .. '[[Category:放置于模板的noteTA]]'
end
end
return code
end
--------------------------------------------------------------------------------
-- Construct a trie from CGroup rules for faster matching.
-- A leaf node is [index] = <index of its item in CGroup data>.
function z.build_trie(cgroup_content)
local matchers = {}
for index, content in ipairs(cgroup_content) do
if content.type == 'item' then
for rules in string.gmatch(content.rule, '([^;]+)') do
if mw.text.trim(rules) ~= '' then
local split = mw.text.split(rules, '=>')[1]
local matcher = split:match(":(.*)") or split
matchers[mw.text.trim(matcher)] = index
end
end
end
end
local trie = {}
for matcher, index in pairs(matchers) do
local current_node = trie
for i = 1, #matcher do
local byte = matcher:sub(i,i)
current_node[byte] = current_node[byte] or {}
current_node = current_node[byte]
end
current_node.index = index
end
return trie
end
-- Match the content of an article against CGroup rules using a trie.
function z.match_text_with_cgroup(article_title, cgroup_content)
local trie = z.build_trie(cgroup_content)
local content = article_title:getContent()
local matches = {}
for start_pos = 1, #content do
local current_node = trie
local matched_index = nil
for i = start_pos, #content do
local byte = content:sub(i, i)
current_node = current_node[byte]
if not current_node then
break
end
if current_node.index then
matched_index = current_node.index
end
end
if matched_index then
matches[matched_index] = true
end
end
return matches
end
--------------------------------------------------------------------------------
return z
