#   Copyright 2015 Red Hat, 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.
#

"""Plugin action implementation"""

import logging
import subprocess

from osc_lib.command import command
from tripleoclient import utils


class InstallUndercloud(command.Command):
    """Install and setup the undercloud"""

    auth_required = False
    log = logging.getLogger(__name__ + ".InstallUndercloud")

    def take_action(self, parsed_args):
        self.log.debug("take_action(%s)" % parsed_args)

        utils.ensure_run_as_normal_user()

        subprocess.check_call("instack-install-undercloud")


class UpgradeUndercloud(command.Command):
    """Upgrade undercloud"""

    auth_required = False
    log = logging.getLogger(__name__ + ".UpgradeUndercloud")

    def take_action(self, parsed_args):
        self.log.debug("take action(%s)" % parsed_args)

        utils.ensure_run_as_normal_user()

        subprocess.check_call(['sudo', 'yum', 'update', '-y',
                               'instack-undercloud'])
        subprocess.check_call("instack-pre-upgrade-undercloud")
        subprocess.check_call("instack-upgrade-undercloud")
        # restart nova-api https://bugzilla.redhat.com/show_bug.cgi?id=1315467
        subprocess.check_call(['sudo', 'systemctl', 'restart',
                              'openstack-nova-api'])