#    Copyright 2016 Mirantis, Inc.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.
"""Script to prepare shell script to generate target image"""


def execute():
    """Function to prepare shell script to generate target image"""
    import sys

    import six

    from nailgun.settings import NailgunSettings
    from nailgun.objects.release import Release
    from nailgun import consts
    from nailgun.orchestrator import tasks_templates

    settings = NailgunSettings()
    master_ip = settings.config['MASTER_IP']
    release_id = sys.argv[1]

    rel = Release.get_by_uid(release_id)

    packages_str = \
        rel.attributes_metadata['editable']['provision']['packages']['value']
    packages = list(
        six.moves.filter(bool, (s.strip() for s in packages_str.split('\n'))))
    task = tasks_templates.make_provisioning_images_task(
        [consts.MASTER_NODE_UID],
        rel.attributes_metadata['editable']['repo_setup']['repos']['value'],
        rel.attributes_metadata['generated']['provision'],
        'prepare_release_ubuntu',
        packages)

    release_str = 'release_{release_id}'.format(release_id=release_id)
    with open('build_image.sh', 'w') as cmd_file:
        cmd_file.write(task['parameters']['cmd'].replace(
            "{cluster.release.environment_version}",
            rel.environment_version).replace(
                '{cluster.release.version}',
                rel.version).replace(
                    '{settings.MASTER_IP}',
                    master_ip).replace(
                        "{cluster.id}",
                        release_str))


if __name__ == '__main__':
    execute()