# Copyright 2017 The Armada Authors.
#
# 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.

import click
from oslo_config import cfg

from armada.cli import CliAction
from armada.handlers.tiller import Tiller

CONF = cfg.CONF


@click.group()
def tiller():
    """ Tiller Services actions

    """


DESC = """
This command gets Tiller information

The tiller command uses flags to obtain information from Tiller services

To obtain Armada deployed releases:

    $ armada tiller --releases

To obtain Tiller service status/information:

    $ armada tiller --status

"""

SHORT_DESC = "Command gets Tiller information."


@tiller.command(name='tiller', help=DESC, short_help=SHORT_DESC)
@click.option('--tiller-host', help="Tiller host IP.", default=None)
@click.option(
    '--tiller-port', help="Tiller host port.", type=int, default=None)
@click.option(
    '--tiller-namespace',
    '-tn',
    help="Tiller namespace.",
    type=str,
    default=None)
@click.option('--releases', help="List of deployed releases.", is_flag=True)
@click.option('--status', help="Status of Tiller services.", is_flag=True)
@click.option('--bearer-token', help="User bearer token.", default=None)
@click.option('--debug', help="Enable debug logging.", is_flag=True)
@click.pass_context
def tiller_service(ctx, tiller_host, tiller_port, tiller_namespace, releases,
                   status, bearer_token, debug):
    CONF.debug = debug
    TillerServices(ctx, tiller_host, tiller_port, tiller_namespace, releases,
                   status, bearer_token).safe_invoke()


class TillerServices(CliAction):

    def __init__(self, ctx, tiller_host, tiller_port, tiller_namespace,
                 releases, status, bearer_token):
        super(TillerServices, self).__init__()
        self.ctx = ctx
        self.tiller_host = tiller_host
        self.tiller_port = tiller_port
        self.tiller_namespace = tiller_namespace
        self.releases = releases
        self.status = status
        self.bearer_token = bearer_token

    def invoke(self):

        with Tiller(
                tiller_host=self.tiller_host,
                tiller_port=self.tiller_port,
                tiller_namespace=self.tiller_namespace,
                bearer_token=self.bearer_token) as tiller:

            self.handle(tiller)

    def handle(self, tiller):
        if self.status:
            if not self.ctx.obj.get('api', False):
                self.logger.info('Tiller Service: %s', tiller.tiller_status())
                self.logger.info('Tiller Version: %s', tiller.tiller_version())
            else:
                client = self.ctx.obj.get('CLIENT')
                query = {
                    'tiller_host': self.tiller_host,
                    'tiller_port': self.tiller_port,
                    'tiller_namespace': self.tiller_namespace
                }
                resp = client.get_status(query=query)
                tiller_status = resp.get('tiller').get('state', False)
                tiller_version = resp.get('tiller').get('version')

                self.logger.info("Tiller Service: %s", tiller_status)
                self.logger.info("Tiller Version: %s", tiller_version)

        if self.releases:
            if not self.ctx.obj.get('api', False):
                for release in tiller.list_releases():
                    self.logger.info("Release %s in namespace: %s",
                                     release.name, release.namespace)
            else:
                client = self.ctx.obj.get('CLIENT')
                query = {
                    'tiller_host': self.tiller_host,
                    'tiller_port': self.tiller_port,
                    'tiller_namespace': self.tiller_namespace
                }
                resp = client.get_releases(query=query)
                for namespace in resp.get('releases'):
                    for release in resp.get('releases').get(namespace):
                        self.logger.info('Release %s in namespace: %s',
                                         release, namespace)