tasks.py

import os
import regex
import shutil
import stat

import invoke
import jinja2
import yaml

import util


USERNAME = os.getenv("USERNAME", "")
USERCODE_PATH = os.path.join(
    "C:\\", "Users", USERNAME, "projects", "eco-mods-public", "Mods", "UserCode"
)

BUNWULF_CONSTRUCTION_PATH = os.path.join(
    "C:\\",
    "Users",
    USERNAME,
    "projects",
    "eco-mods-public",
    "Mods",
    "UserCode",
    "BunWulfConstruction",
    "Recipes",
)

BUNWULF_EDUCATIONAL_PATH = os.path.join(
    "C:\\",
    "Users",
    USERNAME,
    "projects",
    "eco-mods-public",
    "Mods",
    "UserCode",
    "BunWulfEducational",
    "Recipes",
)

LINUX_SERVER_PATH = os.path.join(
    "/home",
    "kai",
    "Steam",
    "steamapps",
    "common",
    "EcoServer",
).replace("\\", "/")

WINDOWS_SERVER_PATH = os.path.join(
    "C:\\",
    "Program Files (x86)",
    "Steam",
    "steamapps",
    "common",
    "Eco",
    "Eco_Data",
    "Server",
)


def server_path():
    if "windows" in os.getenv("OS", "").lower():
        return WINDOWS_SERVER_PATH
    else:
        return LINUX_SERVER_PATH


class RemovalException(Exception):
    pass


def handleRemoveReadonly(func, path, _):
    if not os.access(path, os.W_OK):
        os.chmod(path, stat.S_IWUSR)
        func(path)
    else:
        raise RemovalException("could not handle path")


def copy_paths(origin_path, target_path):
    if not os.path.isdir(origin_path):
        return
    if os.path.exists(target_path) and os.path.isdir(target_path):
        print(f"\tRemoving {target_path}")
        shutil.rmtree(target_path, ignore_errors=False, onerror=handleRemoveReadonly)
    if os.path.isdir(origin_path):
        print(f"\tCopying {origin_path} to {target_path}")
        shutil.copytree(origin_path, target_path)


@invoke.task
def copy_assets(ctx: invoke.Context, branch=""):
    print("Cleaning out assets folder")
    if os.path.exists("./eco-server/assets"):
        shutil.rmtree(
            "./eco-server/assets", ignore_errors=False, onerror=handleRemoveReadonly
        )

    # get assets from git
    branch_flag = ""
    if branch != "":
        branch_flag = f"-b {branch}"
    ctx.run(
        f"git clone --depth 1 {branch_flag} -- git@github.com:coilysiren/eco-mods-assets.git ./eco-server/assets",
        echo=True,
    )
    shutil.rmtree(
        "./eco-server/assets/.git", ignore_errors=False, onerror=handleRemoveReadonly
    )

    for build in os.listdir("./eco-server/assets/Builds/Mods/UserCode/"):
        origin_path = os.path.join(
            "./eco-server/assets/Builds/Mods/UserCode", build, "Assets"
        )
        target_path = os.path.join("./Mods/UserCode", build, "Assets")
        copy_paths(origin_path, target_path)


@invoke.task
def zip_assets(ctx: invoke.Context, mod):
    if os.path.exists(f"{mod}.zip"):
        os.remove(f"{mod}.zip")

    if os.path.exists(os.path.join("./Mods/UserCode", mod, "bin")):
        shutil.rmtree(
            f"./Mods/UserCode/{mod}/bin",
            ignore_errors=False,
            onerror=handleRemoveReadonly,
        )

    if os.path.exists(os.path.join("./Mods/UserCode", mod, "obj")):
        shutil.rmtree(
            f"./Mods/UserCode/{mod}/obj",
            ignore_errors=False,
            onerror=handleRemoveReadonly,
        )

    ctx.run(f"zip -r {mod}.zip ./Mods/UserCode/{mod}")


@invoke.task
def push_asset(ctx: invoke.Context, mod):
    zip_assets(ctx, mod)
    # TODO: delete target folder first
    ctx.run(f"scp {mod}.zip kai@kai-server:{LINUX_SERVER_PATH}")
    ctx.run(f'ssh -t kai@kai-server "cd {LINUX_SERVER_PATH} && unzip -o {mod}.zip"')


#######################
# SPECIALITY SPECIFIC #
#######################


@invoke.task
def bunwulf_agricultural(_: invoke.Context):
    plants = os.path.join(server_path(), "Mods", "__core__", "AutoGen", "Plant")
    plant = os.listdir(plants)

    plant_entity_pattern = r".*public partial class (\w+) : PlantEntity.*"
    plant_species_pattern = r".*public partial class (\w+) : PlantSpecies.*"
    tree_species_pattern = r".*public partial class (\w+) : TreeSpecies.*"

    templates = jinja2.Environment(loader=jinja2.FileSystemLoader("templates/"))
    template = templates.get_template("plant.template")

    # Read in every plant file
    for p in plant:
        with open(os.path.join(plants, p), "r", encoding="utf-8") as f:
            data = f.read()

        # Skip anything produces plant fibers
        if "PlantFibersItem" in data:
            continue

        # Skip trees
        if regex.match(tree_species_pattern, data, regex.DOTALL):
            continue

        else:
            # Pull out all the data we need
            plant_entity = regex.search(plant_entity_pattern, data, regex.DOTALL).group(
                1
            )
            plant_species = regex.search(
                plant_species_pattern, data, regex.DOTALL
            ).group(1)

            # Render and write the template
            print(f"Writing {plant_entity} to BunWulfAgricultural")
            content = template.render(
                plant_entity=plant_entity, plant_species=plant_species
            )
            with open(
                os.path.join(USERCODE_PATH, "BunWulfAgricultural", "Plant", p),
                "w",
                encoding="utf-8",
            ) as f:
                f.write(content)


@invoke.task
def bunwulf_structural(_: invoke.Context):
    with open("recipes.yml", "r", encoding="utf-8") as recipes:
        recipe_data = yaml.safe_load(recipes)["BunWulfStructural"]

    util.process_recipes(
        recipe_data, os.path.join(USERCODE_PATH, "BunWulfStructural", "Recipes")
    )