dda-devops-build/src/main/python/ddadevops/release_mixin/release_mixin.py

# TODO: jem, zam - 2023_04_14: mv file one dir up
from typing import Optional
from pybuilder.core import Project 
from ..devops_build import DevopsBuild
from .infrastructure import ReleaseRepository, ReleaseTypeRepository, VersionRepository
from .infrastructure_api import GitApi, EnvironmentApi
from .services import PrepareReleaseService, TagAndPushReleaseService
from .domain import EnvironmentKeys
from src.main.python.ddadevops.domain.release_config import ReleaseConfig

def create_release_mixin_config(config_file, main_branch) -> dict:
    config = {}
    config.update(
        {'ReleaseMixin':
            {'main_branch': main_branch,
             'config_file': config_file}})
    return config

def add_versions(config, release_version, bump_version) -> dict:
    config['ReleaseMixin'].update(
        {'release_version': release_version,
         'bump_version': bump_version})
    return config

class ReleaseMixin(DevopsBuild):
    def __init__(self, project: Project, config: Optional[dict] = None, release_config: Optional[ReleaseConfig] = None):
        if not release_config:
            if not config:
                raise Exception("Release parameters could not be set.")
            super().__init__(project, config=config)
            release_mixin_config = config['ReleaseMixin']
            release_config = ReleaseConfig(
                main_branch = release_mixin_config['main_branch'],
                config_file = release_mixin_config['config_file'],
                devops=self.repo.get_devops(project)
            )
        else:
            super().__init__(project, devops=release_config.devops)
        self.repo.set_release(self.project, release_config)

        git_api = GitApi()
        environment_api = EnvironmentApi()
        env_key = EnvironmentKeys.DDADEVOPS_RELEASE_TYPE.name
        environment_val_set = environment_api.get(env_key) is not "" and environment_api.get(env_key) is not None
        
        if environment_val_set:
            release_type_repo = ReleaseTypeRepository.from_environment(environment_api)
        else:
            release_type_repo = ReleaseTypeRepository.from_git(git_api)

        version_repo = VersionRepository(release_config.config_file)
        release_repo = ReleaseRepository(version_repo, release_type_repo, release_config.main_branch)

        self.prepare_release_service = PrepareReleaseService(release_repo)
        self.tag_and_push_release_service = TagAndPushReleaseService(git_api, release_repo)

    def prepare_release(self):
        self.prepare_release_service.write_and_commit_release()
        self.prepare_release_service.write_and_commit_bump()

    def tag_and_push_release(self):
        self.tag_and_push_release_service.tag_release()
        self.tag_and_push_release_service.push_release()