JrGoodle/clowder

"""Clowder command line save controller

.. codeauthor:: Joe DeCapo <joe@polka.cat>

"""

import clowder.util.filesystem as fs
from clowder.util.app import Argument, Subcommand
from clowder.util.console import CONSOLE
from clowder.util.format import Format
from clowder.util.yaml import Yaml

import clowder.util.formatting as fmt
from clowder.controller import (
    CLOWDER_CONTROLLER,
    ClowderRepo,
    print_clowder_name,
    valid_clowder_yaml_required
)
from clowder.environment import clowder_repo_required, ENVIRONMENT
from clowder.util.error import DefaultVersionError, ExistingVersionError


class SaveCommand(Subcommand):
    class Meta:
        name = 'save'
        help = 'Create clowder yaml version for current repos'
        args = [
            Argument('version', help='version to save')
        ]

    @valid_clowder_yaml_required
    @print_clowder_name
    @clowder_repo_required
    def run(self, args) -> None:
        if args.version.lower() == 'default':
            raise DefaultVersionError(f"Version name '{args.version}' is not allowed")

        if ENVIRONMENT.clowder_repo_dir is not None:
            CONSOLE.stdout(ClowderRepo(ENVIRONMENT.clowder_repo_dir).status)
        CLOWDER_CONTROLLER.validate_projects_exist()
        CLOWDER_CONTROLLER.validate_projects_state(CLOWDER_CONTROLLER.projects)

        # TODO: Better validate version name (no spaces, no ~, etc.)
        # Replace path separators with dashes to avoid creating directories
        version_name = args.version.lower().replace('/', '-')

        versions_dir = ENVIRONMENT.clowder_repo_versions_dir
        fs.make_dir(versions_dir, exist_ok=True)

        yml_file = versions_dir / f"{version_name}.clowder.yml"
        yaml_file = versions_dir / f"{version_name}.clowder.yaml"
        version_exists_message = f"Version '{fmt.version(version_name)}' already exists"
        if yml_file.exists():
            raise ExistingVersionError(f"{Format.path(yml_file)}\n{version_exists_message}")
        elif yaml_file.exists():
            raise ExistingVersionError(f"{Format.path(yaml_file)}\n{version_exists_message}")

        CONSOLE.stdout(f" - Save version '{fmt.version(version_name)}'\n{Format.path(yml_file)}")
        yaml = Yaml(yml_file)
        yaml.save(CLOWDER_CONTROLLER.get_yaml(resolved=True))