##########################################################################
#
# Package.py: store the Package object information.
#
#  Python-CDD is a library to make easier to build applications to
#  Custom Debian Distributions.
#  See http://projetos.ossystems.com.br/python-cdd for more information.
#    
# ====================================================================
# Copyright (c) 2002-2005 O.S. Systems.  All rights reserved.
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution.
#
#########################################################################
# Authors: Otavio Salvador <otavio@ossystems.com.br>
#          Nat Budin <natb@brandeis.edu>
#          Free Ekanayaka <free@agnula.org>
"""
Classes to represent source and binary packages
"""

import apt_pkg
import logging
log = logging.getLogger("cdd.FileSystem")


class Package:
    """
    This class encapsulates a binary debian package as represented in an apt
    Packages file.
    """
    def __init__(self, section):
        log.debug("Package: constructor")
        self._fields = {}
        self._fields.update(section)

        # Handle dependencie fields
        for field in ('Suggests', 'Recommends',
                      'Depends', 'Pre-Depends', 'Conflicts'):
            if self._fields.has_key(field):
                self._fields[field + "-orig"] = self._fields[field]
                self._fields[field] = apt_pkg.ParseDepends(self._fields[field])

        if self._fields.has_key('Provides'):
            self._fields['Provides-orig'] = self._fields['Provides']
            self._fields['Provides'] = self._fields['Provides'].split(', ')

        # Point to the functions
        self.has_key = self._fields.has_key
        self.keys = self._fields.keys
        self.__getitem__ = self._fields.get
        
    def dump(self):
        """
        Return a string containing the package stanza
        """
        log.debug("Package.dump")
        
        keys = self.keys()
        stanza = ""

        for field in ['Package', 'Priority', 'Section', 'Installed-Size',
                      'Maintainer', 'Architecture', 'Version', 'Replaces',
                      'Provides-orig', 'Depends-orig', 'Suggests-orig',
                      'Conflicts-orig', 'Filename', 'Size', 'MD5sum',
                      'Description', 'Tag']:
            if self._fields.has_key(field):
                if field.endswith('-orig'):
                    stanza += field[:-5] + ": " + self._fields[field] + "\n"
                else:
                    stanza += field + ": " + self._fields[field] + "\n"
                keys.remove(field)


        return stanza + "\n"
            
class SourcePackage(Package):
    """
    This class encapsulates a debian source package as represented in an apt
    Sources file.
    
    The class transforms the `Files` field provided by the Sources file into a
    list of dictionaries with three fields: `name`, `md5sum` and `size`.
    """
    def __init__(self, section):
        log.debug("SoucePackage: constructor")
        # First, we load the default information
        Package.__init__(self, section)

        # Handle Build-Depends field
        for field in ('Build-Depends', 'Build-Depends-Indep',
                      'Build-Conflicts', 'Build-Conflicts-Indep'):
            if self._fields.has_key(field):
                self._fields[field + "-orig"] = self._fields[field]
                self._fields[field] = \
                                    apt_pkg.ParseSrcDepends(self._fields[field])

        for field in ('Binary', 'Uploaders'):
            if self._fields.has_key(field):
                self._fields[field + "-orig"] = self._fields[field]
                self._fields[field] = self._fields[field].split(', ')

        # Now, let's do the trick ;-)
        self._fields['Files-orig'] = "\n " + self._fields['Files']
        files = self._fields['Files'].split(' ')
        self._fields['Files'] = []
        for i in range(0, len(files), 3):
            tmpdict = {}
            tmpdict['md5sum'] = files[i]
            tmpdict['size'] = files[i+1]
            if files[i+2][len(files[i+2])-1:len(files[i+2])] == "\n":
                tmpdict['name'] = files[i+2][:-1]
            else:
                tmpdict['name'] = files[i+2]
            self._fields['Files'].append(tmpdict)

    def dump(self):
        """
        Return a string containing the package stanza
        """
        log.debug("SourcePackage.dump")
        
        keys = self.keys()
        stanza = ""

        for field in ['Package', 'Binary-orig', 'Version',
                      'Priority', 'Section', 'Maintainer',
                      'Build-Depends-orig', 'Build-Depends-Indep-orig',
                      'Build-Conflicts-orig', 'Build-Conflicts-Indep-orig',
                      'Architecture', 'Standards-Version', 'Format',
                      'Directory', 'Files-orig', 'Uploaders-orig']:
            if self._fields.has_key(field):
                if field.endswith('-orig') and field != 'Files-orig':
                    stanza += field[:-5] + ": " + self._fields[field] + "\n"
                elif field == 'Files-orig':
                    stanza += field[:-5] + ":" + self._fields[field] + "\n"
                else:
                    stanza += field + ": " + self._fields[field] + "\n"
                keys.remove(field)


        return stanza + "\n"