Module:CopyEmbeddedStories

From Seven Sages of Rome
Revision as of 20:00, 13 November 2024 by Noeth (talk | contribs)

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

-- Module:CopyEmbeddedStories
local p = {}

-- Helper function to extract all EmbeddedStory templates from text
local function extractEmbeddedStories(text)
    local stories = {}
    -- Match all EmbeddedStory templates
    for story in mw.ustring.gmatch(text, "{{EmbeddedStory.-}}") do
        table.insert(stories, story)
    end
    return stories
end

-- Helper function to append templates to a page
local function appendToPage(title, templates)
    local page = mw.title.new(title)
    if not page then return false, "Invalid page title" end
    
    local status, err = pcall(function()
        local pageText = mw.text.trim(mw.title.new(title):getContent() or "")
        -- Add a newline if the page isn't empty
        if pageText ~= "" then
            pageText = pageText .. "\n\n"
        end
        -- Append all templates
        for _, template in ipairs(templates) do
            pageText = pageText .. template .. "\n"
        end
        
        -- Save the page
        local success = page:update({
            text = pageText,
            summary = "Copied EmbeddedStory templates",
            minor = false
        })
        if not success then
            error("Failed to save page")
        end
    end)
    
    return status, err
end

-- Main function to copy templates
function p.copyTemplates(frame)
    local sourcePage = frame.args[1]
    local targetPage = frame.args[2]
    
    if not sourcePage or not targetPage then
        return "Error: Source and target pages must be specified"
    end
    
    -- Get source page content
    local sourceTitle = mw.title.new(sourcePage)
    if not sourceTitle then
        return "Error: Invalid source page title"
    end
    
    local sourceContent = sourceTitle:getContent()
    if not sourceContent then
        return "Error: Could not read source page"
    end
    
    -- Extract templates
    local stories = extractEmbeddedStories(sourceContent)
    if #stories == 0 then
        return "No EmbeddedStory templates found on source page"
    end
    
    -- Copy templates to target page
    local success, err = appendToPage(targetPage, stories)
    if not success then
        return "Error copying templates: " .. (err or "unknown error")
    end
    
    return string.format("Successfully copied %d templates to %s", #stories, targetPage)
end

-- Function to generate form
function p.showForm(frame)
    local currentPage = mw.title.getCurrentTitle().text
    
    -- Create the HTML form using mw.html builder
    local builder = mw.html.create()
    
    -- Add the form container
    builder
        :tag('div')
            :addClass('template-copy-form')
            :tag('h3')
                :wikitext('Copy Embedded Stories')
            :done()
            :tag('div')
                :css('margin', '10px 0')
                :tag('input')
                    :attr('type', 'text')
                    :attr('id', 'targetPage')
                    :attr('placeholder', 'Enter target page name')
                    :css('width', '300px')
                    :css('padding', '5px')
                :done()
                :tag('button')
                    :attr('onclick', 'copyTemplates()')
                    :css('margin-left', '10px')
                    :css('padding', '5px 10px')
                    :wikitext('Copy Stories')
                :done()
            :done()
            :tag('div')
                :attr('id', 'copyResult')
                :css('margin-top', '10px')
            :done()
        :done()

    -- Add the JavaScript
    local script = string.format([[
<script>
function copyTemplates() {
    var targetPage = document.getElementById('targetPage').value;
    if (!targetPage) {
        mw.notify('Please enter a target page name');
        return;
    }
    var currentPage = %s;
    
    // Call the Lua module via API
    new mw.Api().post({
        action: 'parse',
        text: '{{#invoke:CopyEmbeddedStories|copyTemplates|' + currentPage + '|' + targetPage + '}}',
        contentmodel: 'wikitext'
    }).done(function(response) {
        mw.notify(response.parse.text['*']);
        document.getElementById('copyResult').innerHTML = response.parse.text['*'];
    }).fail(function(error) {
        mw.notify('Error copying templates');
        document.getElementById('copyResult').innerHTML = 'Error copying templates';
    });
}
</script>
]], mw.text.jsonEncode(currentPage))

    return tostring(builder) .. script
end

return p