from __future__ import absolute_import

from django.utils.translation import ugettext_lazy as _
from django.utils.translation import ugettext
from django.contrib.auth.models import User, Group

from permissions.models import Role
from history.models import History
from django_gpg.runtime import gpg
from documents.models import DocumentType, DocumentTypeFilename, Document
from folders.models import Folder
from taggit.models import Tag
from tags.models import TagProperties
from metadata.models import MetadataType, MetadataSet
from sources.models import WebForm, StagingFolder
from document_indexing.models import Index, IndexTemplateNode
from dynamic_search.models import RecentSearch
# TODO: clear the job queues

bootstrap_options = {}


def nuke_database():
    # Delete all document types
    for obj in DocumentType.objects.all():
        obj.delete()

    # Delete all documents one by one to trigger the document file delete method
    # Should also get rid of document metadata
    for obj in Document.objects.all():
        obj.delete()

    # Delete all metadata types
    for obj in MetadataType.objects.all():
        obj.delete()

    # Delete all metadata sets
    for obj in MetadataSet.objects.all():
        obj.delete()

    # Delete all indexes types, should also delete index nodes
    for obj in Index.objects.all():
        obj.delete()

    # Delete all webforms sources
    for obj in WebForm.objects.all():
        obj.delete()

    # Delete all staging folder sources
    for obj in StagingFolder.objects.all():
        obj.delete()

    # Delete all user groups
    for obj in Group.objects.all():
        obj.delete()

    # Delete all users except superadmins and staff
    for obj in User.objects.all():
        if not obj.is_superuser and not obj.is_staff:
            obj.delete()

    # Delete all user roles
    for obj in Role.objects.all():
        obj.delete()

    # Delete all the remaining history events
    for obj in History.objects.all():
        obj.delete()

    # Delete all tags
    for obj in Tag.objects.all():
        obj.delete()

    # Delete any remaining tag property
    for obj in TagProperties.objects.all():
        obj.delete()

    # Delete all foders
    for obj in Folder.objects.all():
        obj.delete()

    # Delete all recent searches
    for obj in RecentSearch.objects.all():
        obj.delete()

    # Clear the entire key ring (public and private keys)
    gpg.delete_all_keys()


class BootstrapBase(object):
    name = None
    label = ''
    description = ''

    def __unicode__(self):
        return unicode(self.label)


class BootstrapSimple(BootstrapBase):
    name = 'simple'
    label = _(u'Simple')
    description = _(u'A simple setup providing an uploaded date metadata and index plus an alphabetic index based on document filenames.')

    def execute(self):
        # Create metadata types
        upload_date = MetadataType.objects.create(name='upload_date', title=ugettext(u'Upload date'), default='current_date()')

        # Create a segmented date index
        index = Index.objects.create(name='date_tree', title=ugettext(u'Segmented date index'), enabled=True)
        template_root = index.template_root

        # Create index template
        node1 = IndexTemplateNode.objects.create(parent=template_root, index=index, expression='metadata.upload_date[0:4]', enabled=True, link_documents=False)
        node2 = IndexTemplateNode.objects.create(parent=node1, index=index, expression='metadata.upload_date[5:7]', enabled=True, link_documents=False)
        node3 = IndexTemplateNode.objects.create(parent=node2, index=index, expression='metadata.upload_date[8:10]', enabled=True, link_documents=True)


class BootstrapPermit(BootstrapBase):
    name = 'permits'
    label = _(u'Permits')
    description = _(u'A setup for handling permits and related documents.')

    def execute(self):
        # Create document types
        form = DocumentType.objects.create(name=ugettext(u'Form'))
        DocumentTypeFilename.objects.create(document_type=form, filename=ugettext(u'Building construction form'))
        DocumentTypeFilename.objects.create(document_type=form, filename=ugettext(u'Building usage form'))

        blueprint = DocumentType.objects.create(name=ugettext(u'Blueprint'))
        DocumentTypeFilename.objects.create(document_type=blueprint, filename=ugettext(u'Floorplan'))
        DocumentTypeFilename.objects.create(document_type=blueprint, filename=ugettext(u'Plot plan'))

        # Create metadata types
        date = MetadataType.objects.create(name='date', title=ugettext(u'Date'), default='current_date()')
        client = MetadataType.objects.create(name='client', title=ugettext(u'Client'))
        permit = MetadataType.objects.create(name='permit', title=ugettext(u'Permit number'))
        project = MetadataType.objects.create(name='project', title=ugettext(u'Project'))
        user = MetadataType.objects.create(name='user', title=ugettext(u'User'), lookup='sorted([user.get_full_name() or user for user in User.objects.all() if user.is_active])')

        # Create a segmented date index
        index = Index.objects.create(name='main_index', title=ugettext(u'Permit index'), enabled=True)

        # Create index template
        per_permit = IndexTemplateNode.objects.create(parent=index.template_root, index=index, expression='\'%s\'' % ugettext(u'Per permit'), enabled=True, link_documents=False)
        per_permit_child = IndexTemplateNode.objects.create(parent=per_permit, index=index, expression='metadata.permit', enabled=True, link_documents=True)

        per_project = IndexTemplateNode.objects.create(parent=index.template_root, index=index, expression='\'%s\'' % ugettext(u'Per project'), enabled=True, link_documents=False)
        per_project_child = IndexTemplateNode.objects.create(parent=per_project, index=index, expression='metadata.project', enabled=True, link_documents=False)
        per_permit = IndexTemplateNode.objects.create(parent=per_project_child, index=index, expression='\'%s\'' % ugettext(u'Per permit'), enabled=True, link_documents=False)
        per_permit_child = IndexTemplateNode.objects.create(parent=per_permit, index=index, expression='metadata.permit', enabled=True, link_documents=True)

        per_date = IndexTemplateNode.objects.create(parent=index.template_root, index=index, expression='\'%s\'' % ugettext(u'Per date'), enabled=True, link_documents=False)
        per_date_child = IndexTemplateNode.objects.create(parent=per_date, index=index, expression='metadata.date', enabled=True, link_documents=True)

        per_user = IndexTemplateNode.objects.create(parent=index.template_root, index=index, expression='\'%s\'' % ugettext(u'Per user'), enabled=True, link_documents=False)
        per_user_child = IndexTemplateNode.objects.create(parent=per_user, index=index, expression='metadata.user', enabled=True, link_documents=True)

        per_client = IndexTemplateNode.objects.create(parent=index.template_root, index=index, expression='\'%s\'' % ugettext(u'Per client'), enabled=True, link_documents=False)
        per_client_child = IndexTemplateNode.objects.create(parent=per_client, index=index, expression='metadata.client', enabled=True, link_documents=True)


for bootstrap in [BootstrapSimple(), BootstrapPermit()]:
    bootstrap_options[bootstrap.name] = bootstrap