luvune.luau

local fs = require("@lune/fs")
local process = require("@lune/process")

-- 1. Initialize the table FIRST
local Luvune = {}

local KEYWORD_MAP = {
    ["def"] = "function",
    ["elif"] = "elseif",
    ["if"] = "if",
    ["else"] = "else",
    ["while"] = "while",
    ["for"] = "for",
    ["None"] = "nil",
    ["True"] = "true",
    ["False"] = "false",
    ["import"] = "local",
    ["pass"] = "-- pass",
    ["print"] = "print",
}

-- 2. Define the conversion logic
function Luvune.convertLine(line: string): string
    -- Handle Python comments first so they don't break Luau
    local converted = line:gsub("#", "--")

    -- Replace keywords
    for pyKey, luauKey in pairs(KEYWORD_MAP) do
        converted = converted:gsub("%f[%a]" .. pyKey .. "%f[%A]", luauKey)
    end

    -- Handle Block Headers (Python ":" to Luau "then/do")
    if converted:match("^%s*if") or converted:match("^%s*elseif") then
        converted = converted:gsub(":", " then")
    elseif converted:match("^%s*while") or converted:match("^%s*for") then
        converted = converted:gsub(":", " do")
    elseif converted:match("^%s*function") then
        converted = converted:gsub(":", "") 
    end

    return converted
end

-- 3. Execution Logic
local function main()
    local args = process.args
    local inputPath = args[1]

    if not inputPath then
        print("Usage: lune run luvune.luau <filename.py>")
        return
    end

    if not fs.isFile(inputPath) then
        print("Error: File not found -> " .. inputPath)
        return
    end

    local sourceCode = fs.readFile(inputPath)
    local outputLines = {}

    print("Luvune is processing: " .. inputPath)

    for line in sourceCode:gmatch("[^\r\n]+") do
        -- Now Luvune.convertLine is guaranteed to exist
        table.insert(outputLines, Luvune.convertLine(line))
    end

    local outputPath = inputPath:gsub("%.py$", "") .. ".luau"
    fs.writeFile(outputPath, table.concat(outputLines, "\n"))
    
    print("Successfully created: " .. outputPath)
end

main()