from bpcms.models import Document, Folder
from django.conf import settings
from django.core.cache import cache
import re

def content_menu(request):

    menu_items = cache.get('bpcms_mainmenu')

    if not menu_items:
        items = Document.objects.filter(folder__isnull=True, islive__exact=True)
        menu_items = []
        for item in items:
            if item.slug != 'index':
                menu_items.append(
                    {
                        'title': item.title,
                        'url': item.get_absolute_url(),
                    }
                )
        folders = Folder.objects.filter(parent__isnull=True)
        for folder in folders:
            try:
                items = Document.objects.filter(folder__exact=folder, islive__exact=True, slug__exact='index')
                item = items[0]
                menu_items.append(
                    {
                        'title': item.title,
                        'url': folder.get_absolute_url(),
                    }
                )
            except:
                pass

        menu_items.sort( \
            key=lambda x: x['title'].lower())
        cache.set('bpcms_mainmenu', menu_items, 3600)

    return {
        'content_menu': menu_items,
    }

def content_submenu(request):
    # find where we are in the tree first
    folderpath = request.path
    folderpath = re.sub(u'^%s' %(settings.BPCMS_ROOT), '', folderpath)
    # right - that should be any prefix got rid of, now traverse the tree till
    # we've got something
    menu_items = cache.get('bpcms_contentmenu_%s' %(folderpath))

    if not menu_items:
        folder = None
        folderpath = folderpath.split('/')
        for part in folderpath:
            try:
                if folder:
                    f = Folder.objects.get(parent__exact=folder, slug__exact=part)
                else:
                    f = Folder.objects.get(parent__isnull=True, slug__exact=part)
                folder = f
            except:
                break

        # We have the folder (in theory)
        menu_items = []
        if folder:
            items = Document.objects.filter(folder__exact=folder, \
                islive__exact=True)
            for item in items:
                if item.slug != 'index':
                    menu_items.append(
                        {
                            'title': item.title,
                            'url': item.get_absolute_url(),
                        }
                    )
            folders = Folder.objects.filter(parent__exact=folder)
            for folder in folders:
                items = Document.objects.filter(folder__exact=folder, \
                    islive__exact=True, slug__exact='index')
                for item in items:
                    menu_items.append(
                        {
                            'title': item.title,
                            'url': folder.get_absolute_url(),
                        }
                    )

        menu_items.sort( \
            key=lambda x: x['title'].lower())

        cache.set('bpcms_contentmenu_%s' %("/".join(folderpath)), menu_items)

    return {
        'content_submenu': menu_items,
    }

def content_breadcrumb(request):
    if settings.BPCMS_ROOT == "/":
        breadcrumb = [{'url': settings.BPCMS_ROOT, 'title': "Home"}]
    else:
        breadcrumb = [{'url': settings.BPCMS_ROOT, 'title': settings.BPCMS_ROOT},]
    # find where we are in the tree first
    folderpath = request.path
    folderpath = re.sub(u'^%s' %(settings.BPCMS_ROOT), '', folderpath)
    # right - that should be any prefix got rid of, now traverse the tree till
    # we've got something
    folder = None
    folderpath = folderpath.split('/')
    for part in folderpath:
        try:
            if folder:
                f = Folder.objects.get(parent__exact=folder, slug__exact=part)
            else:
                f = Folder.objects.get(parent__isnull=True, slug__exact=part)
            folder = f
            try:
                item = Document.objects.get(folder__exact=folder,
                    slug__exact=u'index', islive__exact=True)
                breadcrumb.append({'url': folder.get_absolute_url(),
                    'title': item.title})
            except:
                breadcrumb.append({'url': folder.get_absolute_url(),
                    'title': folder.title})
        except:
            # must have hit the end of the tree, lets see if we can get an
            # item...
            try:
                if folder:
                    item = Document.objects.get(folder__exact=folder, \
                        slug__exact=part, islive__exact=True)
                else:
                    item = Document.objects.get(folder__isnull=True, \
                        slug__exact=part, islive__exact=True)
                breadcrumb.append({'url': item.get_absolute_url(), 'title': item.title})
            except:
                pass
            break

    return {
        'content_breadcrumb': breadcrumb,
    }