#!/bin/python
import os
import os.path

START_SCRIPT = "main.lua"

dir_paths = []
lua_paths = []

for (root, dirs, files) in os.walk("src", topdown=True, followlinks=True):
    dirs[:] = [d for d in dirs if not d[0] == '.']

    for elem in dirs:
        dir_paths.append('"' + root + "/" + elem + '"')

    for elem in files:
        if (elem.lower().endswith(".lua")):
            lua_paths.append('"' + root + "/" + elem + '"')

SCRIPT = \
"""local module = "{}"
local dirs = {{{}}}
local files = {{{}}}

local function getLatest(url) do
    shell.run("clear")

    shell.run("rm", "src")
    shell.run("mkdir", "src")

    for idx, elem in pairs(dirs) do
        shell.run("mkdir", elem)
    end

    for idx, elem in pairs(files) do
        shell.run("wget", url .. "/" .. elem, elem)
    end

    print("==== STARTING ====\\n")

    shell.run(\"src/{}\")
end
end

return {{ getLatest = getLatest }}
"""

module = os.path.basename(os.path.dirname(os.path.normpath(__file__)))

with open("updater.lua", "w") as f:
    f.write(SCRIPT.format(module, ", ".join(dir_paths), ", ".join(lua_paths), START_SCRIPT))