from zope.component import queryUtility
from Products.CMFCore.utils import getToolByName
from plone.app.redirector.interfaces import IRedirectionStorage

from Acquisition import aq_base


def objectMoved(obj, event):
    """Tell the redirection storage that an object moved
    """

    # Unfortunately, IObjectMoved is a rather generic event...
    if event.oldParent is not None and event.newParent is not None and event.oldName is not None:
        storage = queryUtility(IRedirectionStorage)
        if storage is not None:
            old_path = "%s/%s" % ('/'.join(event.oldParent.getPhysicalPath()), event.oldName)
            new_path = '/'.join(obj.getPhysicalPath())

            # This event gets redispatched to children, and we should keep track of them as well
            # In this case, event.object is not the same as obj, and the old_path should actually
            # include obj.id

            if aq_base(event.object) is not aq_base(obj):
                new_path_of_moved = '/'.join(event.object.getPhysicalPath())
                old_path = old_path + new_path[len(new_path_of_moved):]

            # XXX: Special case - don't remember anything happening inside portal_factory
            if '/portal_factory/' in old_path:
                return

            # If we have the 'UI' package for redirector installed, check its
            # config and see if we should create redirects for the contenttype
            rt = getToolByName(obj, 'portal_redirection', None)
            if rt is not None:
                allowedtypes = rt.getRedirectionAllowedForTypes()
                if not obj.portal_type in allowedtypes:
                    return

            # Special case: don't remember object when it was just created...
            putils = getToolByName(obj, 'plone_utils')
            if putils is not None and putils.isIDAutoGenerated(event.oldName):
                request = getattr(obj, 'REQUEST', None)
                if request is not None and 'ACTUAL_URL' in request:
                    url = request['ACTUAL_URL'].split('/')
                    if 'portal_factory' in url and event.oldName in url:
                        return

            storage.add(old_path, new_path)


def objectRemoved(obj, event):
    """Tell the redirection storage that the object was removed
    """
    storage = queryUtility(IRedirectionStorage)
    if storage is not None:
        path = '/'.join(obj.getPhysicalPath())
        storage.destroy(path)