#
# Copyright (c) 2007 Red Hat
#
# Written by Mauricio Teixeira <mteixeira@webset.net>
#
# This file is part of Smart Package Manager.
#
# Smart Package Manager 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.
#
# Smart Package Manager 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 Smart Package Manager; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
import posixpath
import md5
import os
import ConfigParser
import re

# be compatible with 2.3
import sys
if sys.version_info < (2, 4):
    from sets import Set as set

from smart.channel import *
from smart import *
 
ZYPPER_REPOS_DIR = "/var/cache/zypp/raw/http:/"

import rpmUtils.arch
BASEARCH = rpmUtils.arch.getBaseArch()
RELEASEVER = None

def _getreleasever():
    """
    Get system release and version.
    """
    import rpm
    import rpmUtils.transaction

    rpmroot = sysconf.get("rpm-root", "/")
    ts = rpmUtils.transaction.initReadOnlyTransaction(root=rpmroot)
    ts.pushVSFlags(~(rpm._RPMVSF_NOSIGNATURES|rpm._RPMVSF_NODIGESTS))
    # HACK: we're hard-coding the most used distrosm, will add more if needed
    idx = ts.dbMatch('provides', 'openSUSE-release')
    if idx.count() == 0:
        idx = ts.dbMatch('provides', 'openSUSE-release')
    if idx.count() != 0:
        hdr = idx.next()
        releasever = hdr['version']
        del hdr
    del idx
    del ts
    return str(releasever)

def _replaceStrings(txt):
    """
    Replace some predefined strings that may appear in the repo file.
    """
    retxt = re.sub("\$basearch", "%s" % BASEARCH, txt)
    retxt = re.sub("\$releasever", "%s" % RELEASEVER, retxt)
    return retxt

def _findBaseUrl(mirrorlist, repo):
    """
    Fetches the first suggested mirror from the mirrorlist and use as baseurl.
    """
    iface.debug(_("Yum Sync: trying to locate baseurl from mirrorlist for " \
                     "%s.") % repo)
    import urllib
    list = urllib.urlopen(mirrorlist)
    while 1:
        line = list.readline()
        if line.startswith("#"):
            continue
        elif (line.startswith("http:") or
            line.startswith("ftp:")):
            baseurl = line
            break
    return baseurl

def _loadRepoFile(filename):
    """
    Loads each repository file information.
    """

    file = open(filename)
 
    # The computed aliases we have seen in the given file
    seen = set()

    repofile = ConfigParser.ConfigParser()
    repofile.read(filename)

    for repo in repofile.sections():
        # Iterate through each repo found in file
        alias = "zyppersync-%s" % repo
        name = re.sub("\$basearch", "%s" % BASEARCH,
               repofile.get(repo, 'name'))
        name = re.sub("\$releasever", "%s" % RELEASEVER,
               name)

        # Some repos have baseurl, some have mirrorlist
        if repofile.has_option(repo, 'baseurl'):
            baseurl = _replaceStrings(repofile.get(repo, 'baseurl'))
            baseurl = baseurl.splitlines()[1]
        elif repofile.has_option(repo, 'mirrorlist'):
            mirrorlist = _replaceStrings(repofile.get(repo, 'mirrorlist'))
            baseurl = _findBaseUrl(mirrorlist, repo)
        else:
            iface.warning(_("Zypper channel %s does not contan baseurl or " \
                            "mirrorlist addresses. Not syncing.") % repo)
            return seen

        if repofile.has_option(repo, 'enabled'):
            enabled = not repofile.getboolean(repo, 'enabled')
        else:
            enabled = False

        data = {"type": "rpm-md",
                "name": name,
                "baseurl": baseurl,
                "disabled": enabled}
        seen.add(alias)
 
        try:
            createChannel(alias, data)
        except Error, e:
            iface.error(_("While using %s: %s") % (filename, e))
        else:
            # Store it persistently.
            sysconf.set(("channels", alias), data)

    return seen


def syncZypperRepos(reposdir, force=None):
    """
    Sync Smart channels based on Zypper repositories.
    """

    seen = set()

    if os.path.isdir(reposdir):
        for entry in os.listdir(reposdir):
            if entry.endswith(".repo"):
                filepath = os.path.join(reposdir, entry)
                if os.path.isfile(filepath):
                    seen.update(_loadRepoFile(filepath))

    # Delete the entries which were not seen in current files.
    channels = sysconf.get("channels")
    for alias in sysconf.keys("channels"):
        if alias.startswith("zyppersync-") and alias not in seen:
            sysconf.remove(("channels", alias))


if not sysconf.getReadOnly():
    if sysconf.get("sync-zypper-repos",False):
        # Sync is not enabled by default
        iface.debug(_("Trying to sync Zypper channels..."))
        RELEASEVER = _getreleasever()
        syncYumRepos(sysconf.get("zypper-repos-dir", YUM_REPOS_DIR))

# vim:ts=4:sw=4:et