Editing Module:Dictionary

local p = {}
local dictionary_data = mw.loadJsonData("Data:Dictionary")
local lang_codes_set = mw.loadJsonData("Data:LangCodes.json")
--Used by [[Template:Translate]], see documentation there

-- Get a localized string by the raw key
p.translate = function(key, lang_code_override, fallback_str, show_needs_translation)
	-- If called internally (direct Lua call), args will be passed directly.
    -- If called from wikitext, `key` will be the `frame` object, and we get args from `frame.args`.

    -- Handle the case where it's called via #invoke (i.e., from wikitext)
    if type(key) == "table" and key.args then
        local frame = key
        key = frame.args[1]
        lang_code_override = frame.args["lang_code_override"]
        fallback_str = frame.args["fallback_str"]
        show_needs_translation = frame.args["show_needs_translation"]
    end
    
    --default to showing needs translation
    if (show_needs_translation == nil) then
    	show_needs_translation = 'true' 
    end
    
	local lang_code = lang_code_override
	if (lang_code == '' or lang_code == nil) then
    	lang_code = get_lang_code()
	end
	
	local translations = dictionary_data[key]
	if (translations == nil) then return string.format("Key '%s' is not in Data:Dictionary", key) end
	
	translation = translations[lang_code]
	if (translation == nil) then
		-- Determine the appended tooltip
		local needs_translation_expand = ''
		if (show_needs_translation == 'true' and key ~= 'MissingValveTranslationTooltip' and key ~= 'NeedsTranslationTooltip' and key ~= 'NotesNeedsTranslation') then
			local template_args = {}
			if key == 'Notes' then
				template_args[1] = 'NotesNeedsTranslation'
			end
			needs_translation_expand = mw.getCurrentFrame():expandTemplate{title = 'NeedsTranslationTooltip', args=template_args}
		end
		
		-- Determine the fallback string to use
		--default to english fallback if none specified or if en specified
		if (fallback_str == 'en' or fallback_str == nil) then
			translation = translations["en"]
		else
			translation = fallback_str

			-- If english translation is missing, default to the key instead of blank string
			if (translation == nil) then
				translation = key
			end
		end
		
		-- Add the tooltip
		if (translation ~= '' and translation ~= nil) then
			translation = translation .. needs_translation_expand
		end
	end
	
	return translation
end

function p.translate_embed(key, var1, var2, var3, var4, var5)
    -- If called internally (direct Lua call), args will be passed directly.
    -- If called from wikitext, `key` will be the `frame` object, and we get args from `frame.args`.

	local vars
    -- Handle the case where it's called via #invoke (i.e., from wikitext)
    if type(key) == "table" and key.args then
        local frame = key
        key = frame.args[1]
        
        -- Collect all the variables from the frame (starting from the second argument)
	    vars = {}
	    local i = 2
	    while frame.args[i] do
	        table.insert(vars, frame.args[i])
	        i = i + 1
	    end
	else
		vars = {[1] = var1, [2] = var2, [3] = var3, [4] = var4, [5] = var5}
    end
    
    
    -- Retrieve the translation string using the provided key
    local translation = p.translate(key)
    
    if not translation then
        return string.format("No translation found for key '%s'", key)
    end

    -- Now we use string.format to insert variables into the translation string
    local formatted_translation = string.format(translation, unpack(vars))

    -- Return the final string with embedded variables
    return formatted_translation
end

function get_lang_code()
    local title = mw.title.getCurrentTitle()
    local lang_code = title.fullText:match(".*/(.*)$")
	
	-- if the last part the url is not two letters, then its not a lang code,
	-- so default to english
	if (lang_code == nil or lang_codes_set[lang_code] == nil) then
		return 'en'	
	end
		
    return lang_code
end



return p