#!/usr/bin/python -tt
# 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 Library 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
# (c) 2007 Red Hat, Inc
# Written by skvidal@fedoraproject.org

import yum
import re
from yum.plugins import TYPE_CORE

requires_api_version = '2.5'
plugin_type = (TYPE_CORE,)

#def init_hook(conduit):

def config_hook(conduit):
    # get the priority per repo from the repo config 
    yum.config.RepoConf.score = yum.config.IntOption(0)


def exclude_hook(conduit):
    # iterate through each repo based on their priority and 
    # exclude packages from repos with lesser priorities.
    # packages must match exactly

    pkgdict = {}
    for pkg in conduit.getPackages():
        if not pkgdict.has_key(pkg.pkgtup):
            pkgdict[pkg.pkgtup] = []
        pkgdict[pkg.pkgtup].append(pkg)
        
    
    for (pkgtup, pkglist) in pkgdict.items():
        if len(pkglist) > 1:
            pkglist.sort(sortPkgsByRepoScore)
            for pkg in pkglist[1:]:
                pkg.repo.sack.delPackage(pkg)

    
    
def sortPkgsByRepoScore(a, b):
    #higher score wins and longer/bigger name
    if a.repo.score > b.repo.score:
        return -1
    elif a.repo.score == b.repo.score:
        if a.repo.id > b.repo.id:
            return 1
        elif a.repo.id < b.repo.id:
            return -1
        else:
            return 0
    elif a.repo.score < b.repo.score:
        return 1