# 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.

from __future__ import print_function

import logging

from cliff import command as cmd
from fuelclient.objects import environment as environment_obj
from fuelclient.objects import release as release_obj

from octane.util import env as env_util
from octane.util import helpers

LOG = logging.getLogger(__name__)


def load_network_template(network_template):
    try:
        data = helpers.load_yaml(network_template)
    except Exception:
        LOG.exception("Cannot open network template from %s",
                      network_template)
        raise
    return data


def upgrade_env(env_id, release_id, network_template=None):
    release = release_obj.Release(release_id)
    seed_id = env_util.clone_env(env_id, release)
    env_util.copy_fuel_keys(env_id, seed_id)
    if network_template:
        network_template_data = load_network_template(network_template)
        seed_env = environment_obj.Environment(seed_id)
        seed_env.set_network_template_data(network_template_data)
    return seed_id


class UpgradeEnvCommand(cmd.Command):
    """Create upgrade seed env for env ENV_ID and copy settings to it"""

    def get_parser(self, prog_name):
        parser = super(UpgradeEnvCommand, self).get_parser(prog_name)
        parser.add_argument(
            'env_id', type=int, metavar='ENV_ID',
            help="ID of environment to be upgraded")
        parser.add_argument(
            'release_id', type=int, metavar='RELEASE_ID',
            help="ID of a release for the new environment.")
        parser.add_argument(
            '--template', type=str, metavar='TEMPLATE_FILE',
            help="Use network template from file")
        return parser

    def take_action(self, parsed_args):
        seed_id = upgrade_env(parsed_args.env_id, parsed_args.release_id,
                              network_template=parsed_args.template)
        print(seed_id)  # TODO: This shouldn't be needed