from Products.CacheSetup import config
from Products.CacheSetup.Extensions import policy_1, policy_2

POLICIES = (policy_1, policy_2)

def updateOldCachePolicy(portal, out):
    ct = getattr(portal, config.CACHE_TOOL_ID)
    rules = getattr(ct, config.RULES_ID, None)
    if rules is not None and rules.portal_type == 'RuleFolder':
        # this is an old policy so make it a new-style policy folder
        oldpolicy_id = "old-policy"
        oldpolicy_title = "Old Cache Policy"
        ct.addPolicyFolder(oldpolicy_id, oldpolicy_title)
        oldpolicy = getattr(ct, oldpolicy_id)
        folder_ids = [config.RULES_ID, config.HEADERSETS_ID]
        cp = ct.manage_copyObjects(folder_ids)
        oldpolicy.manage_pasteObjects(cp)
        for folder in oldpolicy.objectValues():
            folder.unmarkCreationFlag()
            for item in folder.objectValues():
                item.unmarkCreationFlag()
        ct.setActivePolicyId(oldpolicy_id)
        ct.manage_delObjects(folder_ids)
        # let's also migrate cacheConfig
        cache_config = ct.getCacheConfig()
        if cache_config in ('zserver','apache'):
            ct.setProxyPurgeConfig('no-purge')
        elif cache_config == 'squid':
            ct.setProxyPurgeConfig('no-rewrite')
        else:
            ct.setProxyPurgeConfig('custom-rewrite')
    else:
        # this is cachefu version 1.1
        # let's rename these policies to avoid future collisions
        policy_ids = ['default-cache-policy','no-proxy-cache','default-cache-policy-v2','squid-without-vary']
        found_ids = []
        new_ids = []
        for policy_id in policy_ids:
            policy = getattr(ct, policy_id, None)
            if policy is not None:
                title = policy.Title()
                suffix = "(v1.1)"
                if policy_id in ["default-cache-policy-v2","squid-without-vary"]:
                    suffix = "(v1.1, deprecated)"
                if title == "Squid without Vary header (Note: Turn off compression)":
                    title = "Squid without Vary header"
                policy.setTitle('%s %s' % (title,suffix))
                found_ids.append(policy_id)
                new_ids.append('%s-v1.1' % policy_id)
        ct.manage_renameObjects(found_ids, new_ids)
        # Should work in a fallback in case any of the new ids
        # are already taken.
        

def addCachePolicies(portal, out):
    # We'll extend this later
    # preferably using GenericSetup profiles
    ct = getattr(portal, config.CACHE_TOOL_ID)
    for p in POLICIES:
        # fix any id collisions
        if getattr(ct, p.POLICY_ID, None) is not None:
            count = 1
            while getattr(ct, '%s-backup%s' % (p.POLICY_ID, count), None) is not None:
                count = count + 1
            newId = '%s-backup%s' % (p.POLICY_ID, count)
            ct.manage_renameObjects([p.POLICY_ID], [newId])
            oldpolicy = getattr(ct, newId)
            oldpolicy.setTitle('%s (backup%s)' %(oldpolicy.Title(), count))
        # now add the new policy
        ct.addPolicyFolder(p.POLICY_ID, p.POLICY_TITLE)
        rules = ct.getRules(p.POLICY_ID)
        p.addCacheRules(rules)
        header_sets = ct.getHeaderSets(p.POLICY_ID)
        p.addHeaderSets(header_sets)
    # Lets move the new policies to the top of the list
    policy_ids = [p.POLICY_ID for p in POLICIES]
    ct.moveObjectsToTop(policy_ids)