#!/usr/bin/python3
# Copyright Magomed Kostoev
# Published under MIT license

import os

from lib.logging import log

def generate_script_executing_script(script_to_execute):
    script_to_execute = script_to_execute.replace("\\", "\\\\")
    contents = ""
    contents += "from importlib.machinery import SourceFileLoader\n"
    contents += f"SourceFileLoader('__main__', '{script_to_execute}').load_module()\n"
    return contents

def create_workspace_script(name, script_to_execute):
    log(f"Installing {name}... ", end = "")

    script_contents = generate_script_executing_script(script_to_execute)
    with open(name, "w") as f:
        f.write(script_contents)

    log("Done")

if __name__ == "__main__":
    tools_get_started_py = os.path.abspath(__file__)
    tools = os.sep.join(tools_get_started_py.split(os.sep)[:-1])
    tools_workspace = os.sep.join([tools, "workspace"])
    # Create (in current directory) scripts that execute
    # the same named scripts from _tools/workspace
    tools_workspace_run_py = os.path.join(tools_workspace, "run.py")
    tools_workspace_build_py = os.path.join(tools_workspace, "build.py")
    create_workspace_script("run.py", tools_workspace_run_py)
    create_workspace_script("build.py", tools_workspace_build_py)
    # Initalize tup here
    # TODO: Do anything if tup doesn't exist
    os.system("tup init")