# Copyright (c) 2003-2006 Infrae. All rights reserved.
# See also LICENSE.txt
# $Id: icon.py,v 1.14 2006/02/01 13:44:32 faassen Exp $

"""Sivla icon registry"""

# python
import os

# zope
import Globals
import OFS.misc_

# Silva
from Products.Silva.interfaces import IIcon, IFile, ISilvaObject
from Products.Silva.interfaces import IGhostContent, IGhostFolder

class RegistryError(Exception):
    """thrown on any error in registry"""

class _IconRegistry:

    def __init__(self):
        self._icon_mapping = {}
    
    def getIcon(self, object):
        if IGhostContent.providedBy(object):
            version = object.getLastVersion()
            if version.get_link_status() == version.LINK_OK:
                kind = 'link_ok'
            else:
                kind = 'link_broken'
            identifier = ('ghost', kind)
        elif IGhostFolder.providedBy(object):
            if object.get_link_status() == object.LINK_OK:
                if object.implements_publication():
                    kind = 'publication'
                else:
                    kind = 'folder'
            else:
                kind = 'link_broken'
            identifier = ('ghostfolder', kind)      
        elif IFile.providedBy(object):
            identifier = ('mime_type', object.get_mime_type())
        elif ISilvaObject.providedBy(object):
            identifier = ('meta_type', object.meta_type)
        else:
            meta_type = getattr(object, 'meta_type', None)
            if meta_type is None:
                raise RegistryError, "Icon not found"
            identifier = ('meta_type', meta_type)
        return self.getIconByIdentifier(identifier)

    def getIconByIdentifier(self, identifier):
        icon = self._icon_mapping.get(identifier, None)
        if icon is None:
            msg = "No icon for %s" % repr(identifier)
            raise RegistryError, msg
        return icon

    def registerIcon(self, identifier, icon, context):
        """register icon

            identifier: icon identifier as returned from getIconIdentifier()
            icon: path to icon (i.e. 'www/root.png')
            context: module context of icon (i.e. globals())

            raises  RegistryError if product doesn't exist
            NOTE: this will overwrite previous icon declarations
        """
        # NOTE: code copied from App.ProductContext, modified though
        product = self._get_module_from_context(context)
        name = os.path.split(icon)[1]
        icon = Globals.ImageFile(icon, context)
        icon.__roles__ = None
        if not hasattr(OFS.misc_.misc_, product):
            msg = "The product %s doesn't exist" % product
            raise RegistryError, msg
        getattr(OFS.misc_.misc_, product)[name] = icon
        self._icon_mapping[identifier] = 'misc_/%s/%s' % (product, name)
            
    def _get_module_from_context(self, context):
        name = context['__name__']
        name_parts = name.split('.')
        if name.startswith('Products.'):
            return name_parts[1]
        return name_parts[0]

registry = _IconRegistry()

