# -*- coding: utf-8 -*-
from django.conf import settings as settings
from django.contrib.auth.decorators import login_required
from django.core.context_processors import csrf
from django.core.urlresolvers import reverse
from django.db.models import Q
from django.http import HttpResponse, HttpResponseRedirect
from django.utils import simplejson
from django.utils.translation import ugettext_lazy as _
from mitxmako.shortcuts import render_to_response

from multicourse import multicourse_settings

from models import Revision, Article, CreateArticleForm, RevisionFormWithTitle, RevisionForm
import wiki_settings

def view(request, wiki_url):
    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err
    
    if 'coursename' in request.session: coursename = request.session['coursename']
    else: coursename = None

    course_number = multicourse_settings.get_course_number(coursename)

    perm_err = check_permissions(request, article, check_read=True, check_deleted=True)
    if perm_err:
        return perm_err
    d = {'wiki_article': article,
            'wiki_article_revision':article.current_revision,
			'wiki_write': article.can_write_l(request.user),
			'wiki_attachments_write': article.can_attach(request.user),
            'wiki_current_revision_deleted' : not (article.current_revision.deleted == 0),
            'wiki_title' : article.title + " - MITX %s Wiki" % course_number
			}
    d.update(csrf(request))
    return render_to_response('simplewiki_view.html', d)
    
def view_revision(request, revision_number, wiki_url, revision=None):
    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err
    
    try:
        revision = Revision.objects.get(counter=int(revision_number), article=article)
    except:
        d = {'wiki_article': article,
	            'wiki_err_norevision': revision_number,}
        d.update(csrf(request))
        return render_to_response('simplewiki_error.html', d)
        
    
    perm_err = check_permissions(request, article, check_read=True, check_deleted=True, revision=revision)
    if perm_err:
        return perm_err
        
    d = {'wiki_article': article,
            'wiki_article_revision':revision,
			'wiki_write': article.can_write_l(request.user),
			'wiki_attachments_write': article.can_attach(request.user),
            'wiki_current_revision_deleted' : not (revision.deleted == 0),
			}
    d.update(csrf(request))
    return render_to_response('simplewiki_view.html', d)


def root_redirect(request):
    try:
        root = Article.get_root()
    except:
        err = not_found(request, '/')
        return err

    return HttpResponseRedirect(reverse('wiki_view', args=(root.get_url())))

def create(request, wiki_url):
    
    url_path = get_url_path(wiki_url)

    if url_path != [] and url_path[0].startswith('_'):
            d = {'wiki_err_keyword': True,
                'wiki_url': '/'.join(url_path) }
            d.update(csrf(request))
            return render_to_response('simplewiki_error.html', d)        

    # Lookup path
    try:
        # Ensure that the path exists...
        root = Article.get_root()
        # Remove root slug if present in path
        if url_path and root.slug == url_path[0]:
            url_path = url_path[1:]
        
        path = Article.get_url_reverse(url_path[:-1], root)
        if not path:
            d = {'wiki_err_noparent': True,
                'wiki_url_parent': '/'.join(url_path[:-1]) }
            d.update(csrf(request))
            return render_to_response('simplewiki_error.html', d)
        
        perm_err = check_permissions(request, path[-1], check_locked=False, check_write=True, check_deleted=True)
        if perm_err:
            return perm_err
        # Ensure doesn't already exist
        article = Article.get_url_reverse(url_path, root)
        if article:
            return HttpResponseRedirect(reverse('wiki_view', args=(article[-1].get_url(),)))
    
        # TODO: Somehow this doesnt work... 
        #except ShouldHaveExactlyOneRootSlug, (e):
    except:
        if Article.objects.filter(parent=None).count() > 0:
            return HttpResponseRedirect(reverse('wiki_view', args=('/',)))
        # Root not found...
        path = []
        url_path = [""]

    if request.method == 'POST':
        f = CreateArticleForm(request.POST)
        if f.is_valid():
            article = Article()
            article.slug = url_path[-1]
            if not request.user.is_anonymous():
                article.created_by = request.user
            article.title = f.cleaned_data.get('title')
            if path != []:
                article.parent = path[-1]
            a = article.save()
            new_revision = f.save(commit=False)
            if not request.user.is_anonymous():
                new_revision.revision_user = request.user
            new_revision.article = article
            new_revision.save()
            import django.db as db
            return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))
    else:
        f = CreateArticleForm(initial={'title':request.GET.get('wiki_article_name', url_path[-1]),
                                       'contents':_('Headline\n===\n\n')})
        
    d = {'wiki_form': f,
        'wiki_write': True,
        'create_article' : True,
        }
    d.update(csrf(request))

    return render_to_response('simplewiki_edit.html', d)

def edit(request, wiki_url):
    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err

    # Check write permissions
    perm_err = check_permissions(request, article, check_write=True, check_locked=True, check_deleted=False)
    if perm_err:
        return perm_err

    if wiki_settings.WIKI_ALLOW_TITLE_EDIT:
        EditForm = RevisionFormWithTitle
    else:
        EditForm = RevisionForm
    
    if request.method == 'POST':
        f = EditForm(request.POST)
        if f.is_valid():
            new_revision = f.save(commit=False)
            new_revision.article = article
            
            if request.POST.__contains__('delete'):
                if (article.current_revision.deleted == 1): #This article has already been deleted. Redirect
                    return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))
                new_revision.contents = ""
                new_revision.deleted = 1
            elif not new_revision.get_diff():
                return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))
            
            if not request.user.is_anonymous():
                new_revision.revision_user = request.user
            new_revision.save()
            if wiki_settings.WIKI_ALLOW_TITLE_EDIT:
                new_revision.article.title = f.cleaned_data['title']
                new_revision.article.save()
            return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))
    else:
        startContents = article.current_revision.contents if (article.current_revision.deleted == 0) else 'Headline\n===\n\n'
        
        f = EditForm({'contents': startContents, 'title': article.title})
    d = {'wiki_form': f,
        	'wiki_write': True,
        	'wiki_article': article,
            'wiki_title' : article.title,
        	'wiki_attachments_write': article.can_attach(request.user),
            'create_article' : False,
        	}
    d.update(csrf(request))

    return render_to_response('simplewiki_edit.html', d)

def history(request, wiki_url, page=1):
    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err

    perm_err = check_permissions(request, article, check_read=True, check_deleted=False)
    if perm_err:
        print "returned error " , perm_err
        return perm_err

    page_size = 10
    
    try:
        p = int(page)
    except ValueError:
        p = 1
   
    history = Revision.objects.filter(article__exact = article).order_by('-counter').select_related('previous_revision__counter', 'revision_user', 'wiki_article')
    
    if request.method == 'POST':
        if request.POST.__contains__('revision'): #They selected a version, but they can be either deleting or changing the version
            perm_err = check_permissions(request, article, check_write=True, check_locked=True)
            if perm_err:
                return perm_err
                
            redirectURL = reverse('wiki_view', args=(article.get_url(),))
            try:
                r = int(request.POST['revision'])
                revision = Revision.objects.get(id=r)
                if request.POST.__contains__('change'):
                    article.current_revision = revision
                    article.save()
                elif request.POST.__contains__('view'):
                    redirectURL = reverse('wiki_view_revision', args=(revision.counter, article.get_url(),))
                
                #The rese of these are admin functions
                elif request.POST.__contains__('delete') and request.user.is_superuser:
                    if (revision.deleted == 0):
                         revision.adminSetDeleted(2)
                elif request.POST.__contains__('restore') and request.user.is_superuser:
                    if (revision.deleted == 2):
                        revision.adminSetDeleted(0)
                elif request.POST.__contains__('delete_all') and request.user.is_superuser:
                    Revision.objects.filter(article__exact = article, deleted = 0).update(deleted = 2)
                elif request.POST.__contains__('lock_article'):
                    print "changing locked article " , article.locked
                    article.locked = not article.locked
                    print "changed locked article " , article.locked
                    article.save()
            except:
                pass
            finally:
                return HttpResponseRedirect(redirectURL)
                # 
                # 
                # <input type="submit" name="delete" value="Delete revision"/>
                # <input type="submit" name="restore" value="Restore revision"/>
                # <input type="submit" name="delete_all" value="Delete all revisions">
                # %else:
                # <input type="submit" name="delete_article" value="Delete all revisions">
                # 
    
    page_count = (history.count()+(page_size-1)) / page_size
    if p > page_count:
        p = 1
    beginItem = (p-1) * page_size
    
    next_page = p + 1 if page_count > p else None
    prev_page = p - 1 if p > 1 else None
    
    d = {'wiki_page': p,
	        'wiki_next_page': next_page,
	        'wiki_prev_page': prev_page,
	        'wiki_write': article.can_write_l(request.user),
	        'wiki_attachments_write': article.can_attach(request.user),
	        'wiki_article': article,
            'wiki_title': article.title,
	        'wiki_history': history[beginItem:beginItem+page_size],
            'show_delete_revision' : request.user.is_superuser,}
    d.update(csrf(request))

    return render_to_response('simplewiki_history.html', d)
    
    
def revision_feed(request, page=1):
    page_size = 10
    
    try:
        p = int(page)
    except ValueError:
        p = 1
   
    history = Revision.objects.order_by('-revision_date').select_related('revision_user', 'article', 'previous_revision')
    
    page_count = (history.count()+(page_size-1)) / page_size
    if p > page_count:
        p = 1
    beginItem = (p-1) * page_size
    
    next_page = p + 1 if page_count > p else None
    prev_page = p - 1 if p > 1 else None
    
    d = {'wiki_page': p,
	        'wiki_next_page': next_page,
	        'wiki_prev_page': prev_page,
	        'wiki_history': history[beginItem:beginItem+page_size],
            'show_delete_revision' : request.user.is_superuser,}
    d.update(csrf(request))

    return render_to_response('simplewiki_revision_feed.html', d)

def search_articles(request):
    # blampe: We should check for the presence of other popular django search
    # apps and use those if possible. Only fall back on this as a last resort.
    # Adding some context to results (eg where matches were) would also be nice.
    
    # todo: maybe do some perm checking here
    
    if request.method == 'POST':
        querystring = request.POST['value'].strip()
    else:
        querystring = ""
        
        
    results = Article.objects.all()
    
    if request.user.is_superuser:
        results = results.order_by('current_revision__deleted')
    else:
        results = results.filter(current_revision__deleted = 0)
    

    if querystring:
        for queryword in querystring.split():
            # Basic negation is as fancy as we get right now
            if queryword[0] == '-' and len(queryword) > 1:
                results._search = lambda x: results.exclude(x)
                queryword = queryword[1:]
            else:
                results._search = lambda x: results.filter(x)
                    
            results = results._search(Q(current_revision__contents__icontains = queryword) | \
                                      Q(title__icontains = queryword))
        
    results = results.select_related('current_revision__deleted')
    
    results = sorted(results, key=lambda article: (article.current_revision.deleted, article.get_url().lower()) )

    if len(results) == 1 and querystring:
        return HttpResponseRedirect(reverse('wiki_view', args=(results[0].get_url(),)))
    else:
        d = {'wiki_search_results': results,
            	'wiki_search_query': querystring,}
        d.update(csrf(request))
        return render_to_response('simplewiki_searchresults.html', d)
        

def search_add_related(request, wiki_url):
    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err

    perm_err = check_permissions(request, article, check_read=True)
    if perm_err:
        return perm_err

    search_string = request.GET.get('query', None)
    self_pk = request.GET.get('self', None)
    if search_string:
        results = []
        related = Article.objects.filter(title__istartswith = search_string)
        others = article.related.all()
        if self_pk:
            related = related.exclude(pk=self_pk)
        if others:
            related = related.exclude(related__in = others)
        related = related.order_by('title')[:10]
        for item in related:
            results.append({'id': str(item.id),
                            'value': item.title,
                            'info': item.get_url()})
    else:
        results = []
    
    json = simplejson.dumps({'results': results})
    return HttpResponse(json, mimetype='application/json')

def add_related(request, wiki_url):

    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err
    
    perm_err = check_permissions(request, article, check_write=True, check_locked=True)
    if perm_err:
        return perm_err
    
    try:
        related_id = request.POST['id']
        rel = Article.objects.get(id=related_id)
        has_already = article.related.filter(id=related_id).count()
        if has_already == 0 and not rel == article:
            article.related.add(rel)
            article.save()
    except:
        pass
    finally:
        return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))

def remove_related(request, wiki_url, related_id):
    (article, path, err) = fetch_from_url(request, wiki_url)
    if err:
        return err

    perm_err = check_permissions(request, article, check_write=True, check_locked=True)
    if perm_err:
        return perm_err

    try:
        rel_id = int(related_id)
        rel = Article.objects.get(id=rel_id)
        article.related.remove(rel)
        article.save()
    except:
        pass
    finally:
        return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))

def random_article(request):
    from random import randint
    num_arts = Article.objects.count()
    article = Article.objects.all()[randint(0, num_arts-1)]
    return HttpResponseRedirect(reverse('wiki_view', args=(article.get_url(),)))

def encode_err(request, url):
    d = {'wiki_err_encode': True}
    d.update(csrf(request))
    return render_to_response('simplewiki_error.html', d)
    
def not_found(request, wiki_url):
    """Generate a NOT FOUND message for some URL"""
    d = {'wiki_err_notfound': True,
         'wiki_url': wiki_url}
    d.update(csrf(request))
    return render_to_response('simplewiki_error.html', d)

def get_url_path(url):
    """Return a list of all actual elements of a url, safely ignoring
    double-slashes (//) """
    return filter(lambda x: x!='', url.split('/'))

def fetch_from_url(request, url):
    """Analyze URL, returning the article and the articles in its path
    If something goes wrong, return an error HTTP response"""

    err = None
    article = None
    path = None
    
    url_path = get_url_path(url)

    try:
        root = Article.get_root()
    except:
        err = not_found(request, '/')
        return (article, path, err)

    if url_path and root.slug == url_path[0]:
        url_path = url_path[1:]

    path = Article.get_url_reverse(url_path, root)
    if not path:
        err = not_found(request, '/' + '/'.join(url_path))
    else:
        article = path[-1]
    return (article, path, err)


def check_permissions(request, article, check_read=False, check_write=False, check_locked=False, check_deleted=False, revision = None):    
    read_err = check_read and not article.can_read(request.user)
    
    write_err = check_write and not article.can_write(request.user)
    
    locked_err = check_locked and article.locked
    
    if revision == None:
        revision = article.current_revision
    deleted_err = check_deleted and not (revision.deleted == 0)
    if (request.user.is_superuser):
        deleted_err = False
        locked_err = False
    
    if read_err or write_err or locked_err or deleted_err:
        d = {'wiki_article': article,
	            'wiki_err_noread': read_err,
	            'wiki_err_nowrite': write_err,
	            'wiki_err_locked': locked_err,
                'wiki_err_deleted': deleted_err,}
        d.update(csrf(request))
        # TODO: Make this a little less jarring by just displaying an error
        #       on the current page? (no such redirect happens for an anon upload yet)
        # benjaoming: I think this is the nicest way of displaying an error, but
        # these errors shouldn't occur, but rather be prevented on the other pages.
        return render_to_response('simplewiki_error.html', d)
    else:
        return None

####################
# LOGIN PROTECTION #
####################

if wiki_settings.WIKI_REQUIRE_LOGIN_VIEW:
    view               = login_required(view)
    history            = login_required(history)
    search_articles    = login_required(search_articles)
    root_redirect      = login_required(root_redirect)
    revision_feed      = login_required(revision_feed)
    random_article     = login_required(random_article)
    search_add_related = login_required(search_add_related)
    not_found          = login_required(not_found)
    view_revision      = login_required(view_revision)
    
if wiki_settings.WIKI_REQUIRE_LOGIN_EDIT:
    create          = login_required(create)
    edit            = login_required(edit)
    add_related     = login_required(add_related)
    remove_related  = login_required(remove_related)

if wiki_settings.WIKI_CONTEXT_PREPROCESSORS:
    settings.TEMPLATE_CONTEXT_PROCESSORS += wiki_settings.WIKI_CONTEXT_PREPROCESSORS
