Editing Module:Color

local p = {}
local lib = require('Module:Feature')

local aliases = {
	bz        = 'buzz',
	buzzword  = 'buzz',
	neighbor  = 'highlight',
	neighbour = 'highlight',
	green     = 'new',
	red       = 'old',
	tutorial  = 'help',
	item      = 'help',
	wp        = 'whisper',
}

local colors = {
    accent     = 'text-accent',
    buzz       = 'text-buzz',
    highlight  = 'text-highlight',
    menu       = 'text-menu',
    new        = 'text-new',
    old        = 'text-old',
    help       = 'text-help',
    catch      = 'text-catch',
    whisper    = 'text-whisper',
}

-- Main function for wiki usage.
-- If 2 arguments are present, treats the first one as a keyword.
-- If only 1 argument is present, searches it for keywords.
-- Returns the color associated with the keyword, or "inherit" if not found.
function p.main(frame)
    local args = require('Module:Arguments').getArgs(frame)
	local class = ''
	local span = ''
	local text = ''
	local link = args.link
    -- choose variant based on number of arguments
	if args[2] then
		if args[1]:find('#') then
			span = args[1]
		else
			class = p._getKeywordColor(mw.ustring.lower(args[1]))
		end
		text = args[2]
	else
		class = p._searchTextForKeyword(mw.ustring.lower(args[1]))
		text = args[1]
	end

	if link ~= nil then
		if link == '1' or link == 1 then
			text = '[[' .. text .. ']]'
		else
			text = '[[' .. link .. '|' .. text .. ']]'
		end
	else
		text = '' .. text .. ''
	end

	if (args.nobold and lib.isNotEmpty(class)) then
		return '<span class="' .. class .. '">' .. text .. '</span>'
	elseif (args.nobold and lib.isNotEmpty(span)) then
		return '<span style="color:' .. span .. '">' .. text .. '</span>'
	elseif lib.isNotEmpty(span) then
		return '<span style="color:' .. span .. '"><b>' .. text .. '</b></span>'
	else
		return '<span class="' .. class .. '"><b>' .. text .. '</b></span>'
	end

end

-- Library functions usable in other modules

-- Returns the color associated with given keyword,
-- or "inherit" if input is not a keyword.
-- Runs output through nowiki by default,
-- unless noescape is specified to be true.
-- (input must be in lower case.)
function p._getKeywordColor(input, noescape)
    local element = aliases[input] or input
    local color = colors[element]
    if noescape then
    	return color or 'inherit'
	end
    return color and mw.text.nowiki(color) or 'inherit'
end

-- Helper method to search given text for the keys of given table t.
-- If a key is found, returns its value; returns nil otherwise.
local function searchTextForKeys(text, t)
	for key, val in pairs(t) do
		result = mw.ustring.find(text, key, 1, true)
        if result ~= nil then return val end
    end
end

-- Searches given text for keywords and returns the associated color,
-- or "inherit" if no keyword is found.
-- (text must be in lower case.)
function p._searchTextForKeyword(text)
	-- try elements first
	local color = searchTextForKeys(text, colors)
    if color ~= nil then
        return mw.text.nowiki(color)
    end
    -- try aliases afterwards
    local keyword = searchTextForKeys(text, aliases)
    if keyword ~= nil then
    	return mw.text.nowiki(colors[keyword])
	end

    return 'inherit'
end

return p
@@ Line 12: / Line 12: @@
 	item      = 'help',
 	wp        = 'whisper',
-	common    = 'common',
-	gray      = 'common',
-	uncommon  = 'uncommon',
-	blue      = 'uncommon',
-	rare      = 'rare',
-	purple    = 'rare',
-	stellar   = 'stellar',
-	gold      = 'stellar',
-	yellow    = 'stellar',
-	miracle   = 'miracle',
-	rainbow   = 'miracle'
 }
@@ Line 35: / Line 24: @@
      catch      = 'text-catch',
      whisper    = 'text-whisper',
-    common     = 'text-common',
-    uncommon   = 'text-uncommon',
-    rare       = 'text-rare',
-    stellar    = 'text-stellar',
-    miracle    = 'text-miracle',
 }
@@ Line 75: / Line 59: @@
 	end
-	if (args.no_bold and lib.isNotEmpty(class)) then
+	if (args.nobold and lib.isNotEmpty(class)) then
 		return '<span class="' .. class .. '">' .. text .. '</span>'
-	elseif (args.no_bold and lib.isNotEmpty(span)) then
+	elseif (args.nobold and lib.isNotEmpty(span)) then
 		return '<span style="color:' .. span .. '">' .. text .. '</span>'
 	elseif lib.isNotEmpty(span) then
@@ Line 105: / Line 89: @@
 -- Helper method to search given text for the keys of given table t.
 -- If a key is found, returns its value; returns nil otherwise.
--- Checks longer keys first to avoid matching partial keywords (Uncommon getting detected as Common.)
 local function searchTextForKeys(text, t)
-    local keys = {}
+	for key, val in pairs(t) do
+		result = mw.ustring.find(text, key, 1, true)
-    for key in pairs(t) do
+        if result ~= nil then return val end
-        keys[#keys + 1] = key
-    end
-    table.sort(keys, function(a, b)
-        return #a > #b
-    end)
-    for _, key in ipairs(keys) do
-        if mw.ustring.find(text, key, 1, true) ~= nil then
-            return t[key]
-        end
      end
 end
File Pages	==Summary====Licensing=={{Fairuse}}{{File\|}}
Dynamic Page List	¦²«»²{¦}²«»«/»«!----»
Array in DPL	¹^2{}^2^2{¹}^2
Parser Functions	{{#if:\|\|}} {{#if:<condition>\|<result if true>\|<result if false>}} {{#ifeq:\|\|\|}} {{#ifeq:<text1>\|<text2>\|<result if text1=text2>\|<result if text1≠text2>}} {{#expr:}} {{#expr:<mathematical expression>}} {{#switch:\|\|#default=}} {{#switch:<text1>\|<text2>=<result if text1=text2>\|<text3>=<result if text1=text3>\|#default=<result if no valid match>}} {{#replace:\|\|}} {{#replace:<text1>\|<plain text to find in text1>\|<plain text to insert in place of the text found>}} {{#titleparts:}} {{#replace:<page name>}} {{#tag:\|}} {{#tag:<name of html tag>\|<content inside html tag>}}
Markup	{{}}\|[][[]][[Category:]]#REDIRECT [[]] [[Category:Redirect Pages]]<code></code><includeonly></includeonly><noinclude></noinclude><nowiki></nowiki><!----><br>§
Notices	{{Stub\|}}{{Stub Dialogue}}{{Upcoming}}{{Under Construction}}{{Placeholder\|}}
Magic Words	{{!}}{{formatnum:}}{{lc:}}{{uc:}}{{PAGENAME}}{{FULLPAGENAME}}{{ROOTPAGENAME}}{{BASEPAGENAME}}{{SUBPAGENAME}}{{DISPLAYTITLE:\|noreplace}}__TOC____NOTOC____EXPECTUNUSEDCATEGORY____HIDDENCAT__
HTML Entities	— ,