Module:Pronoun

Documentation for this module may be created at Module:Pronoun/doc

local p = {}

-- Checks if a verb needs an 'es' ending (for third-person singular simple present tense)
local function needsEsEnding(verb)
    return verb:match("[oschx]$") or verb:match("sh$") or verb:match("ss$")
end

-- Special cases for verb conjugation that don't follow the regular 's' or 'es' pattern
local specialCases = {
    go = "goes",
}

-- Adjusts the verb based on the pronoun, verb tense, and maintains original capitalization
local function adjustVerb(originalPronoun, verb)
    local pronoun = originalPronoun:lower() -- Use lowercase for logic
    local originalVerb = verb -- Preserve the original verb for later comparison
    verb = verb:lower() -- Use lowercase for logic

    -- Continuous tense (verbs ending in 'ing')
    if verb:sub(-3) == "ing" then
        if pronoun == "he" or pronoun == "she" or pronoun == "it" then
            return originalPronoun .. " is " .. originalVerb
        elseif pronoun == "they" then
            return originalPronoun .. " are " .. originalVerb
        end
    else
        -- Simple present tense for singular third-person
        if pronoun == "he" or pronoun == "she" or pronoun == "it" then
            verb = specialCases[verb] or verb -- Apply special case if exists
            if not verb:match("s$") then
                if needsEsEnding(verb) then
                    verb = verb .. "es"
                else
                    verb = verb .. "s"
                end
            end
        elseif pronoun == "they" then
            -- For 'they', ensure verb is in base form, adjusting special cases as needed
            for base, conjugated in pairs(specialCases) do
                if verb == conjugated then
                    verb = base -- revert special case to base form
                end
            end
            -- Remove 'es' or 's' if present (from a previously conjugated form)
            if verb:match("es$") then
                verb = verb:sub(1, -3)
            elseif verb:match("s$") then
                verb = verb:sub(1, -2)
            end
        end
        -- Recombine with original pronoun, using modified verb if it was changed
        return originalPronoun .. " " .. (verb == originalVerb:lower() and originalVerb or verb)
    end
end

function p.main(frame)
    local originalPronoun = frame.args[1]
    local verb = frame.args[2]
    return adjustVerb(originalPronoun, verb) -- Adjust and return result
end

return p