#!/usr/bin/env python

# Output override files for apt-ftparchive and indices/
# Copyright (C) 2000, 2001  James Troup <james@nocrew.org>
# $Id: cindy,v 1.7 2002-02-12 22:13:06 troup Exp $

# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

################################################################################

# X-Listening-To: Sanitarium / Master of the Puppets - Metallica

################################################################################

import pg, sys;
import utils, db_access;
import apt_pkg;

################################################################################

Cnf = None;
projectB = None;
override = {}

################################################################################

def process(suite, component, type):
    global override;

    suite_id = db_access.get_suite_id(suite);
    if suite_id == -1:
        utils.fubar("Suite '%s' not recognised." % (suite));

    component_id = db_access.get_component_id(component);
    if component_id == -1:
        utils.fubar("Component '%s' not recognised." % (component));

    type_id = db_access.get_override_type_id(type);
    if type_id == -1:
        utils.fubar("Type '%s' not recognised. (Valid types are deb, udeb and dsc)" % (type));
    dsc_type_id = db_access.get_override_type_id("dsc");

    if type == "deb" or type == "udeb":
        packages = {};
        q = projectB.query("SELECT DISTINCT b.package FROM binaries b, bin_associations ba WHERE b.id = ba.bin AND ba.suite = %s" % (suite_id));
        for i in q.getresult():
            packages[i[0]] = "";

    src_packages = {};
    q = projectB.query("SELECT DISTINCT s.source FROM source s, src_associations sa WHERE s.id = sa.source AND sa.suite = %s" % (suite_id));
    for i in q.getresult():
        src_packages[i[0]] = "";

    q = projectB.query("SELECT package, priority, section, maintainer FROM override WHERE suite = %s AND component = %s AND type = %s" % (suite_id, component_id, type_id));
    projectB.query("BEGIN WORK");
    for i in q.getresult():
        package = i[0];
        if type == "deb" or type == "udeb":
            if not packages.has_key(package):
                if not src_packages.has_key(package):
                    print "DELETING: %s" % (package);
                    #projectB.query("DELETE FROM override WHERE package = '%s' AND suite = %s AND component = %s AND type = %s"
                    #% (package, suite_id, component_id, type_id));
                else:
                    print "MAKING SOURCE: %s" % (package);
                    #projectB.query("DELETE FROM override WHERE package = '%s' AND suite = %s AND component = %s AND type = %s"
                    #% (package, suite_id, component_id, type_id));
                    # Then if source doesn't already have a copy, insert it into source
                    q = projectB.query("SELECT package FROM override WHERE package = '%s' AND suite = %s AND component = %s AND type = %s" % (package, suite_id, component_id, dsc_type_id));
                    if not q.getresult():
                        #projectB.query("INSERT INTO override (package, suite, component, priority, section, type, maintainer) VALUES ('%s', %s, %s, %s, %s, %s, '%s')" % (package, suite_id, component_id, i[1], i[2], dsc_type_id, i[3]));
                        print "(nop)"
        else: # dsc
            if not src_packages.has_key(package):
                print "DELETING: %s" % (package);
                #projectB.query("DELETE FROM override WHERE package = '%s' AND suite = %s AND component = %s AND type = %s"
                #% (package, suite_id, component_id, type_id));
    projectB.query("COMMIT WORK");


################################################################################

def main ():
    global Cnf, projectB, override;

    Cnf = utils.get_conf()
    apt_pkg.ParseCommandLine(Cnf,[],sys.argv);

    projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
    db_access.init(Cnf, projectB);

    for suite in [ "stable", "unstable" ]:
        print "Processing %s..." % (suite);
        for component in Cnf.SubTree("Component").List():
            if component == "mixed":
                continue; # Ick
            for type in Cnf.SubTree("OverrideType").List():
                print "Processing %s [%s - %s]..." % (suite, component, type);
		process(suite, component, type);

#######################################################################################

if __name__ == '__main__':
    main()