### This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.

## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.

## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

import sos.plugintools
import os
import commands

class yum(sos.plugintools.PluginBase):
    """yum information
    """

    optionList = [("yumlist", "list repositories and packages", "slow", False)]
    optionList = [("yumdebug", "gather yum debugging data", "slow", False)]

    def checkenabled(self):
        if self.cInfo["policy"].pkgByName("yum") or os.path.exists("/etc/yum.conf"):
            return True
        return False

    def diagnose(self):
        # FIXME: diagnose should only report actual problems, disabling this for now.
        return True
        # repo sanity checking
        # TODO: elaborate/validate actual repo files, however this directory should
        # be empty on RHEL 5+ systems.
        try: rhelver = self.cInfo["policy"].rhelVersion()
        except: rhelver = None
        
        if rhelver >= 5:
            if len(os.listdir('/etc/yum.repos.d/')):
                self.addDiagnose("/etc/yum.repos.d/ contains additional repository "+
                                 "information and can cause rpm conflicts.")

    def setup(self):
        # Pull all yum related information
        self.addCopySpec("/etc/yum")
        self.addCopySpec("/etc/yum.repos.d")
        self.addCopySpec("/etc/yum.conf")
        self.addCopySpec("/var/log/yum.log")

        if self.isOptionEnabled("yumlist"):
            # Get a list of channels the machine is subscribed to.
            self.collectExtOutput("/bin/echo \"repo list\" | /usr/bin/yum shell")
            # List various information about available packages
            self.collectExtOutput("/usr/bin/yum list")

        if self.isOptionEnabled("yumdebug") and self.cInfo["policy"].pkgByName('yum-utils'):
            for output in commands.getoutput("/usr/bin/yum-debug-dump").split("\n"):
                if "Output written to:" in output:
                    try:
                        self.collectExtOutput("/bin/zcat %s" % (output.split()[-1:][0],))
                    except IndexError:
                        pass
        return

    def postproc(self):
        self.doRegexSub("/etc/yum.repos.d/*",
                        r"(\s*proxy_password\s*=\s*)\S+", r"\1*******")