# vim: tabstop=4 shiftwidth=4 softtabstop=4

#    Copyright (C) 2012 Yahoo! Inc. All Rights Reserved.
#    Copyright (C) 2012 New Dream Network, LLC (DreamHost) All Rights Reserved.
#
#    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.

"""Platform-specific logic for RedHat Enterprise Linux v6 components.
"""

from devstack import log as logging
from devstack import shell as sh
from devstack import utils

from devstack.components import db
from devstack.components import horizon
from devstack.components import nova

from devstack.packaging import yum

LOG = logging.getLogger(__name__)

SOCKET_CONF = "/etc/httpd/conf.d/wsgi-socket-prefix.conf"
HTTPD_CONF = '/etc/httpd/conf/httpd.conf'

# See: http://wiki.libvirt.org/page/SSHPolicyKitSetup
# FIXME: take from distro config??
LIBVIRT_POLICY_FN = "/etc/polkit-1/localauthority/50-local.d/50-libvirt-access.pkla"
LIBVIRT_POLICY_CONTENTS = """
[libvirt Management Access]
Identity={idents}
Action=org.libvirt.unix.manage
ResultAny=yes
ResultInactive=yes
ResultActive=yes
"""
DEF_IDENT = 'unix-group:libvirtd'


class DBInstaller(db.DBInstaller):

    def _configure_db_confs(self):
        LOG.info("Fixing up %r mysql configs.", self.distro.name)
        fc = sh.load_file('/etc/my.cnf')
        lines = fc.splitlines()
        new_lines = list()
        for line in lines:
            if line.startswith('skip-grant-tables'):
                line = '#' + line
            new_lines.append(line)
        fc = utils.joinlinesep(*new_lines)
        with sh.Rooted(True):
            sh.write_file('/etc/my.cnf', fc)


class HorizonInstaller(horizon.HorizonInstaller):

    def _config_fixups(self):
        (user, group) = self._get_apache_user_group()
        # This is recorded so it gets cleaned up during uninstall
        self.tracewriter.file_touched(SOCKET_CONF)
        LOG.info("Fixing up %r and %r files" % (SOCKET_CONF, HTTPD_CONF))
        with sh.Rooted(True):
            # Fix the socket prefix to someplace we can use
            fc = "WSGISocketPrefix %s" % (sh.joinpths(self.log_dir, "wsgi-socket"))
            sh.write_file(SOCKET_CONF, fc)
            # Now adjust the run user and group (of httpd.conf)
            new_lines = list()
            for line in sh.load_file(HTTPD_CONF).splitlines():
                if line.startswith("User "):
                    line = "User %s" % (user)
                if line.startswith("Group "):
                    line = "Group %s" % (group)
                new_lines.append(line)
            sh.write_file(HTTPD_CONF, utils.joinlinesep(*new_lines))


class NovaInstaller(nova.NovaInstaller):

    def _get_policy(self, ident_users):
        fn = LIBVIRT_POLICY_FN
        contents = LIBVIRT_POLICY_CONTENTS.format(idents=(";".join(ident_users)))
        return (fn, contents)

    def _get_policy_users(self):
        ident_users = set()
        ident_users.add(DEF_IDENT)
        ident_users.add('unix-user:%s' % (sh.getuser()))
        return ident_users

    def configure(self):
        configs_made = nova.NovaInstaller.configure(self)
        driver_canon = nova.canon_virt_driver(self.cfg.get('nova', 'virt_driver'))
        if driver_canon == 'libvirt':
            (fn, contents) = self._get_policy(self._get_policy_users())
            dirs_made = list()
            with sh.Rooted(True):
                # TODO check if this dir is restricted before assuming it isn't?
                dirs_made.extend(sh.mkdirslist(sh.dirname(fn)))
                sh.write_file(fn, contents)
            self.tracewriter.cfg_file_written(fn)
            self.tracewriter.dirs_made(*dirs_made)
            configs_made += 1
        return configs_made


class YumPackagerWithRelinks(yum.YumPackager):

    def _remove(self, pkg):
        response = yum.YumPackager._remove(self, pkg)
        if response:
            options = pkg.get('packager_options', {})
            links = options.get('links', [])
            for (_, tgt) in links:
                if sh.islink(tgt):
                    sh.unlink(tgt)
        return response

    def install(self, pkg):
        yum.YumPackager.install(self, pkg)
        options = pkg.get('packager_options', {})
        links = options.get('links', [])
        for src, tgt in links:
            if not sh.islink(tgt):
                # This is actually a feature, EPEL must not conflict
                # with RHEL, so X pkg installs newer version in
                # parallel.
                #
                # This of course doesn't work when running from git
                # like devstack does....
                sh.symlink(src, tgt)
        return True