fix for no organization owner
[redakcja.git] / apps / catalogue / views.py
index ef042de..4204e77 100644 (file)
@@ -1,44 +1,56 @@
-from datetime import datetime
+# -*- coding: utf-8 -*-
+#
+# This file is part of MIL/PEER, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
 import logging
 import os
-from StringIO import StringIO
+import shutil
+import subprocess
+from tempfile import NamedTemporaryFile
+from xml.sax.saxutils import escape as escape_xml
 
+from django.conf import settings
 from django.contrib import auth
 from django.contrib.auth.models import User
 from django.contrib.auth.decorators import login_required
+from django.contrib.sites.models import Site
 from django.core.urlresolvers import reverse
-from django.db.models import Count, Q
 from django import http
-from django.http import Http404
-from django.shortcuts import get_object_or_404, render
+from django.http import Http404, HttpResponse, HttpResponseForbidden
+from django.shortcuts import get_object_or_404, render, redirect
+from django.utils.encoding import force_str
 from django.utils.http import urlquote_plus
-from django.utils.translation import ugettext_lazy as _
 from django.views.decorators.http import require_POST
-from django.views.generic.simple import direct_to_template
-
-import librarian.html
-import librarian.text
+from unidecode import unidecode
 
 from catalogue import forms
-from catalogue import helpers
+from catalogue.filters import DocumentFilterSet
+from catalogue.forms import TagMultipleForm, TagSingleForm
 from catalogue.helpers import active_tab
-from catalogue.models import Book, Chunk, BookPublishRecord, ChunkPublishRecord
-from catalogue import xml_tools
+from catalogue.models import Category
+from librarian import BuildError
+from librarian.utils import Context
+from librarian.document import Document as SST
+from librarian.formats.html import HtmlFormat
+from librarian.formats.pdf import PdfFormat
+from librarian.formats.epub import EpubFormat
+from redakcja.utlis import send_notify_email
+from .constants import STAGES
+from .models import Document, Plan
+from dvcs.models import Revision
+from organizations.models import Organization
+from fileupload.views import UploadView
 
 #
 # Quick hack around caching problems, TODO: use ETags
 #
 from django.views.decorators.cache import never_cache
+# from fnpdjango.utils.text.slughifi import slughifi
 
 logger = logging.getLogger("fnp.catalogue")
 
 
-@active_tab('all')
-@never_cache
-def document_list(request):
-    return render(request, 'catalogue/document_list.html')
-
-
 @never_cache
 def user(request, username):
     user = get_object_or_404(User, username=username)
@@ -50,22 +62,11 @@ def user(request, username):
 @never_cache
 def my(request):
     return render(request, 'catalogue/my_page.html', {
-        'last_books': sorted(request.session.get("wiki_last_books", {}).items(),
-                        key=lambda x: x[1]['time'], reverse=True),
-        })
+        'last_books': sorted(
+            request.session.get("wiki_last_books", {}).items(), key=lambda x: x[1]['time'], reverse=True),
 
-
-@active_tab('users')
-def users(request):
-    return direct_to_template(request, 'catalogue/user_list.html', extra_context={
-        'users': User.objects.all().annotate(count=Count('chunk')).order_by(
-            '-count', 'last_name', 'first_name'),
-    })
-
-
-@active_tab('activity')
-def activity(request):
-    return render(request, 'catalogue/activity.html')
+        "logout_to": '/',
+        })
 
 
 @never_cache
@@ -74,323 +75,451 @@ def logout_then_redirect(request):
     return http.HttpResponseRedirect(urlquote_plus(request.GET.get('next', '/'), safe='/?='))
 
 
+@login_required
 @active_tab('create')
-def create_missing(request, slug=None):
-    if slug is None:
-        slug = ''
-    slug = slug.replace(' ', '-')
-
+def create_missing(request):
     if request.method == "POST":
         form = forms.DocumentCreateForm(request.POST, request.FILES)
-        if form.is_valid():
+        tag_forms = [
+            (TagMultipleForm if category.multiple else TagSingleForm)(
+                category=category, data=request.POST, prefix=category.dc_tag)
+            for category in Category.objects.all()]
+        if form.is_valid() and all(tag_form.is_valid() for tag_form in tag_forms):
             
             if request.user.is_authenticated():
                 creator = request.user
             else:
                 creator = None
-            book = Book.create(
-                text=form.cleaned_data['text'],
-                creator=creator,
-                slug=form.cleaned_data['slug'],
-                title=form.cleaned_data['title'],
+
+            title = form.cleaned_data['title']
+            try:
+                org = request.user.membership_set.get(
+                    organization=int(form.cleaned_data['owner_organization'])).organization
+                kwargs = {'owner_organization': org}
+            except:
+                kwargs = {'owner_user': request.user}
+
+            doc = Document.objects.create(**kwargs)
+
+            for tag_form in tag_forms:
+                tag_form.save(instance=doc)
+
+            cover = request.FILES.get('cover')
+            if cover:
+                uppath = 'uploads/%d/' % doc.pk
+                path = settings.MEDIA_ROOT + uppath
+                if not os.path.isdir(path):
+                    os.makedirs(path)
+                cover.name = unidecode(cover.name)
+                dest_path = path + cover.name
+                if not os.path.abspath(dest_path).startswith(os.path.abspath(path)):
+                    raise Http404
+                with open(dest_path, 'w') as destination:
+                    for chunk in cover.chunks():
+                        destination.write(chunk)
+                cover_url = 'http://milpeer.eu/media/dynamic/' + uppath + cover.name
+            else:
+                cover_url = ''
+
+            text = '''<section xmlns="http://nowoczesnapolska.org.pl/sst#" xmlns:dc="http://purl.org/dc/elements/1.1/">
+                <metadata>
+                    <dc:publisher>%s</dc:publisher>
+                    <dc:description>%s</dc:description>
+                    %s
+                    <dc:relation.coverImage.url>%s</dc:relation.coverImage.url>
+                </metadata>
+                <header>%s</header>
+                <div class="p"> </div>
+                </section>''' % (
+                    escape_xml(form.cleaned_data['publisher']),
+                    escape_xml(form.cleaned_data['description']),
+                    '\n'.join(tag_form.metadata_rows() for tag_form in tag_forms),
+                    escape_xml(cover_url),
+                    escape_xml(title))
+
+            doc.commit(
+                text=text,
+                author=creator
             )
+            doc.assigned_to = request.user
+            doc.save()
 
-            return http.HttpResponseRedirect(reverse("wiki_editor", args=[book.slug]))
+            return http.HttpResponseRedirect(reverse("wiki_editor", args=[doc.pk]))
     else:
-        form = forms.DocumentCreateForm(initial={
-                "slug": slug,
-                "title": slug.replace('-', ' ').title(),
-        })
+        org_pk = request.GET.get('organization')
+        if org_pk:
+            try:
+                org = Organization.objects.get(pk=org_pk)
+            except Organization.DoesNotExist:
+                org = None
+            else:
+                if not org.is_member(request.user):
+                    org = None
+        else:
+            org = None
+        if org is not None:
+            org = org.pk
 
-    return direct_to_template(request, "catalogue/document_create_missing.html", extra_context={
-        "slug": slug,
+        form = forms.DocumentCreateForm(initial={'owner_organization': org})
+
+        tag_forms = [
+            (TagMultipleForm if category.multiple else TagSingleForm)(
+                category=category, tutorial_no=i, prefix=category.dc_tag)
+            for i, category in enumerate(Category.objects.all(), start=2)]
+
+    return render(request, "catalogue/document_create_missing.html", {
         "form": form,
+        "tag_forms": tag_forms,
+
+        "logout_to": '/',
     })
 
 
-@active_tab('upload')
-def upload(request):
-    if request.method == "POST":
-        form = forms.DocumentsUploadForm(request.POST, request.FILES)
-        if form.is_valid():
-            import slughifi
+@never_cache
+def book_html(request, pk, rev_pk=None, preview=False):
+    doc = get_object_or_404(Document, pk=pk, deleted=False)
 
-            if request.user.is_authenticated():
-                creator = request.user
-            else:
-                creator = None
+    try:
+        published_revision = doc.publish_log.all()[0].revision
+    except IndexError:
+        published_revision = None
 
-            zip = form.cleaned_data['zip']
-            skipped_list = []
-            ok_list = []
-            error_list = []
-            slugs = {}
-            existing = [book.slug for book in Book.objects.all()]
-            for filename in zip.namelist():
-                if filename[-1] == '/':
-                    continue
-                title = os.path.basename(filename)[:-4]
-                slug = slughifi(title)
-                if not (slug and filename.endswith('.xml')):
-                    skipped_list.append(filename)
-                elif slug in slugs:
-                    error_list.append((filename, slug, _('Slug already used for %s' % slugs[slug])))
-                elif slug in existing:
-                    error_list.append((filename, slug, _('Slug already used in repository.')))
-                else:
-                    try:
-                        zip.read(filename).decode('utf-8') # test read
-                        ok_list.append((filename, slug, title))
-                    except UnicodeDecodeError:
-                        error_list.append((filename, title, _('File should be UTF-8 encoded.')))
-                    slugs[slug] = filename
-
-            if not error_list:
-                for filename, slug, title in ok_list:
-                    book = Book.create(
-                        text=zip.read(filename).decode('utf-8'),
-                        creator=creator,
-                        slug=slug,
-                        title=title,
-                    )
-
-            return direct_to_template(request, "catalogue/document_upload.html", extra_context={
-                "form": form,
-                "ok_list": ok_list,
-                "skipped_list": skipped_list,
-                "error_list": error_list,
-            })
+    if rev_pk is None:
+        if preview:
+            revision = doc.revision
+        else:
+            if published_revision is not None:
+                revision = published_revision
+            else:
+                # No published version, fallback to preview mode.
+                preview = True
+                revision = doc.revision
     else:
-        form = forms.DocumentsUploadForm()
+        revision = get_object_or_404(Revision, pk=rev_pk)
 
-    return direct_to_template(request, "catalogue/document_upload.html", extra_context={
-        "form": form,
+    was_published = revision == published_revision or doc.publish_log.filter(revision=revision).exists()
+
+    try:
+        sst = SST.from_string(revision.materialize())
+    except ValueError as e:
+        html = e
+    else:
+        html = HtmlFormat(sst).build(
+            files_path='http://%s/media/dynamic/uploads/%s/' % (request.get_host(), pk)).get_string()
+
+    # response = http.HttpResponse(html, content_type='text/html', mimetype='text/html')
+    # return response
+    # book_themes = {}
+    # for fragment in book.fragments.all().iterator():
+    #     for theme in fragment.tags.filter(category='theme').iterator():
+    #         book_themes.setdefault(theme, []).append(fragment)
+
+    # book_themes = book_themes.items()
+    # book_themes.sort(key=lambda s: s[0].sort_key)
+    return render(request, 'catalogue/book_text.html', {
+        'doc': doc,
+        'preview': preview,
+        'revision': revision,
+        'published_revision': published_revision,
+        'specific': rev_pk is not None,
+        'html': html,
+        'can_edit': doc.can_edit(request.user) if doc else None,
+        'was_published': was_published,
     })
 
 
 @never_cache
-def book_xml(request, slug):
-    xml = get_object_or_404(Book, slug=slug).materialize()
+def book_pdf(request, pk, rev_pk):
+    doc = get_object_or_404(Document, pk=pk)
+    rev = get_object_or_404(Revision, pk=rev_pk)
+    # Test
+
+    try:
+        sst = SST.from_string(rev.materialize())
+    except ValueError as e:
+        return HttpResponse(content=force_str(e.message), content_type='text/plain', status='400')
+    
+    ctx = Context(
+        files_path='http://%s/media/dynamic/uploads/%s/' % (request.get_host(), pk),
+        source_url='http://%s%s' % (request.get_host(), reverse('catalogue_html', args=[doc.pk])),
+        organization=doc.owner_organization.name if doc.owner_organization else doc.owner_user.get_full_name(),
+    )
+    if doc.owner_organization is not None and doc.owner_organization.logo:
+        ctx.cover_logo = 'http://%s%s' % (request.get_host(), doc.owner_organization.logo.url)
+    try:
+        pdf_file = PdfFormat(sst).build(ctx)
+    except BuildError as e:
+        return HttpResponse(content=force_str(e.message), content_type='text/plain', status='400')
 
-    response = http.HttpResponse(xml, content_type='application/xml', mimetype='application/wl+xml')
-    response['Content-Disposition'] = 'attachment; filename=%s.xml' % slug
-    return response
+    from catalogue.ebook_utils import serve_file
+    return serve_file(pdf_file.get_filename(), '%d.pdf' % doc.pk, 'application/pdf')
 
 
 @never_cache
-def book_txt(request, slug):
-    xml = get_object_or_404(Book, slug=slug).materialize()
-    output = StringIO()
-    # errors?
-    librarian.text.transform(StringIO(xml), output)
-    text = output.getvalue()
-    response = http.HttpResponse(text, content_type='text/plain', mimetype='text/plain')
-    response['Content-Disposition'] = 'attachment; filename=%s.txt' % slug
-    return response
+def book_epub(request, pk, rev_pk):
+    doc = get_object_or_404(Document, pk=pk)
+    rev = get_object_or_404(Revision, pk=rev_pk)
+    # Test
 
+    try:
+        sst = SST.from_string(rev.materialize())
+    except ValueError as e:
+        return HttpResponse(content=force_str(e.message), content_type='text/plain', status='400')
+
+    ctx = Context(
+        files_path='http://%s/media/dynamic/uploads/%s/' % (request.get_host(), pk),
+        source_url='http://%s%s' % (request.get_host(), reverse('catalogue_html', args=[doc.pk])),
+    )
+    if doc.owner_organization is not None and doc.owner_organization.logo:
+        ctx.cover_logo = 'http://%s%s' % (request.get_host(), doc.owner_organization.logo.url)
+    try:
+        epub_file = EpubFormat(sst).build(ctx)
+    except BuildError as e:
+        return HttpResponse(content=force_str(e.message), content_type='text/plain', status='400')
 
-@never_cache
-def book_html(request, slug):
-    xml = get_object_or_404(Book, slug=slug).materialize()
-    output = StringIO()
-    # errors?
-    librarian.html.transform(StringIO(xml), output, parse_dublincore=False,
-                             flags=['full-page'])
-    html = output.getvalue()
-    response = http.HttpResponse(html, content_type='text/html', mimetype='text/html')
-    return response
+    from catalogue.ebook_utils import serve_file
+    return serve_file(epub_file.get_filename(), '%d.epub' % doc.pk, 'application/epub+zip')
 
 
 @never_cache
-def revision(request, slug, chunk=None):
+def book_mobi(request, pk, rev_pk):
+    doc = get_object_or_404(Document, pk=pk)
+    rev = get_object_or_404(Revision, pk=rev_pk)
+
     try:
-        doc = Chunk.get(slug, chunk)
-    except (Chunk.MultipleObjectsReturned, Chunk.DoesNotExist):
-        raise Http404
-    return http.HttpResponse(str(doc.revision()))
-
-
-def book(request, slug):
-    book = get_object_or_404(Book, slug=slug)
-
-    # TODO: most of this should go somewhere else
-
-    # do we need some automation?
-    first_master = None
-    chunks = []
-    need_fixing = False
-    choose_master = False
-
-    length = book.chunk_set.count()
-    for i, chunk in enumerate(book):
-        chunk_dict = {
-            "chunk": chunk,
-            "fix": [],
-            "grade": ""
-            }
-        graded = xml_tools.GradedText(chunk.materialize())
-        if graded.is_wl():
-            master = graded.master()
-            if first_master is None:
-                first_master = master
-            elif master != first_master:
-                chunk_dict['fix'].append('bad-master')
-
-            if i > 0 and not graded.has_trim_begin():
-                chunk_dict['fix'].append('trim-begin')
-            if i < length - 1 and not graded.has_trim_end():
-                chunk_dict['fix'].append('trim-end')
-
-            if chunk_dict['fix']:
-                chunk_dict['grade'] = 'wl-fix'
-            else:
-                chunk_dict['grade'] = 'wl'
+        sst = SST.from_string(rev.materialize())
+    except ValueError as e:
+        return HttpResponse(content=force_str(e.message), content_type='text/plain', status='400')
+
+    ctx = Context(
+        files_path='http://%s/media/dynamic/uploads/%s/' % (request.get_host(), pk),
+        source_url='http://%s%s' % (request.get_host(), reverse('catalogue_html', args=[doc.pk])),
+    )
+    if doc.owner_organization is not None and doc.owner_organization.logo:
+        ctx.cover_logo = 'http://%s%s' % (request.get_host(), doc.owner_organization.logo.url)
+    try:
+        epub_file = EpubFormat(sst).build(ctx)
+    except BuildError as e:
+        return HttpResponse(content=force_str(e.message), content_type='text/plain', status='400')
 
-        elif graded.is_broken_wl():
-            chunk_dict['grade'] = 'wl-broken'
-        elif graded.is_xml():
-            chunk_dict['grade'] = 'xml'
-        else:
-            chunk_dict['grade'] = 'plain'
-            chunk_dict['fix'].append('wl')
-            choose_master = True
+    output_file = NamedTemporaryFile(prefix='librarian', suffix='.mobi', delete=False)
+    output_file.close()
+    subprocess.check_call(
+        ['ebook-convert', epub_file.get_filename(), output_file.name, '--no-inline-toc'])
 
-        if chunk_dict['fix']:
-            need_fixing = True
-        chunks.append(chunk_dict)
+    from catalogue.ebook_utils import serve_file
+    return serve_file(output_file.name, '%d.mobi' % doc.pk, 'application/epub+zip')
 
-    if first_master or not need_fixing:
-        choose_master = False
 
-    if request.method == "POST":
-        form = forms.ChooseMasterForm(request.POST)
-        if not choose_master or form.is_valid():
-            if choose_master:
-                first_master = form.cleaned_data['master']
-
-            # do the actual fixing
-            for c in chunks:
-                if not c['fix']:
-                    continue
-
-                text = c['chunk'].materialize()
-                for fix in c['fix']:
-                    if fix == 'bad-master':
-                        text = xml_tools.change_master(text, first_master)
-                    elif fix == 'trim-begin':
-                        text = xml_tools.add_trim_begin(text)
-                    elif fix == 'trim-end':
-                        text = xml_tools.add_trim_end(text)
-                    elif fix == 'wl':
-                        text = xml_tools.basic_structure(text, first_master)
-                author = request.user if request.user.is_authenticated() else None
-                description = "auto-fix: " + ", ".join(c['fix'])
-                c['chunk'].commit(text=text, author=author, 
-                    description=description)
-
-            return http.HttpResponseRedirect(book.get_absolute_url())
-    elif choose_master:
-        form = forms.ChooseMasterForm()
+# @never_cache
+# def revision(request, slug, chunk=None):
+#     try:
+#         doc = Chunk.get(slug, chunk)
+#     except (Chunk.MultipleObjectsReturned, Chunk.DoesNotExist):
+#         raise Http404
+#     if not doc.book.accessible(request):
+#         return HttpResponseForbidden("Not authorized.")
+#     return http.HttpResponse(str(doc.revision()))
+
+
+@login_required
+def book_schedule(request, pk):
+    book = get_object_or_404(Document, pk=pk, deleted=False)
+    if not book.can_edit(request.user):
+        return HttpResponseForbidden("Not authorized.")
+    if request.method == 'POST':
+        Plan.objects.filter(document=book).delete()
+        for i, (s, name) in enumerate(STAGES):
+            user_id = request.POST.get('s%d-user' % i)
+            deadline = request.POST.get('s%d-deadline' % i) or None
+            Plan.objects.create(document=book, stage=s, user_id=user_id, deadline=deadline)
+
+        book.set_stage(request.POST.get('stage', ''))
+        return redirect('catalogue_user')
+
+    current = {}
+    for p in Plan.objects.filter(document=book):
+        current[p.stage] = (getattr(p.user, 'pk', None), (p.deadline.isoformat() if p.deadline else None))
+
+    schedule = [(i, s, current.get(s, ())) for i, (s, name) in enumerate(STAGES)]
+    
+    if book.owner_organization:
+        people = [m.user for m in book.owner_organization.membership_set.exclude(status='pending')]
     else:
-        form = None
-
-    return direct_to_template(request, "catalogue/book_detail.html", extra_context={
-        "book": book,
-        "chunks": chunks,
-        "need_fixing": need_fixing,
-        "choose_master": choose_master,
-        "first_master": first_master,
-        "form": form,
+        people = [book.owner_user]
+    return render(request, 'catalogue/book_schedule.html', {
+        'book': book,
+        'schedule': schedule,
+        'people': people,
     })
 
 
-def chunk_add(request, slug, chunk):
-    try:
-        doc = Chunk.get(slug, chunk)
-    except (Chunk.MultipleObjectsReturned, Chunk.DoesNotExist):
-        raise Http404
-
-    if request.method == "POST":
-        form = forms.ChunkAddForm(request.POST, instance=doc)
-        if form.is_valid():
-            if request.user.is_authenticated():
-                creator = request.user
+@login_required
+def book_owner(request, pk):
+    doc = get_object_or_404(Document, pk=pk, deleted=False)
+    if not doc.can_edit(request.user):
+        return HttpResponseForbidden("Not authorized.")
+
+    error = ''
+
+    if request.method == 'POST':
+        # TODO: real form
+        new_org_pk = request.POST.get('owner_organization')
+        if not new_org_pk:
+            doc.owner_organization = None
+            doc.owner_user = request.user
+            doc.save()
+        else:
+            org = Organization.objects.get(pk=new_org_pk)
+            if not org.is_member(request.user):
+                error = 'Bad organization'
             else:
-                creator = None
-            doc.split(creator=creator,
-                slug=form.cleaned_data['slug'],
-                title=form.cleaned_data['title'],
-            )
+                doc.owner_organization = org
+                doc.owner_user = None
+                doc.save()
+        if not error:
+            return redirect('catalogue_user')
+
+    return render(request, 'catalogue/book_owner.html', {
+        'doc': doc,
+        'error': error,
+    })
 
-            return http.HttpResponseRedirect(doc.book.get_absolute_url())
-    else:
-        form = forms.ChunkAddForm(initial={
-                "slug": str(doc.number + 1),
-                "title": "cz. %d" % (doc.number + 1, ),
-        })
 
-    return direct_to_template(request, "catalogue/chunk_add.html", extra_context={
-        "chunk": doc,
-        "form": form,
+@login_required
+def book_delete(request, pk):
+    doc = get_object_or_404(Document, pk=pk, deleted=False)
+    if not doc.can_edit(request.user):
+        return HttpResponseForbidden("Not authorized.")
+
+    if request.method == 'POST':
+        doc.deleted = True
+        doc.save()
+        return redirect('catalogue_user')
+
+    return render(request, 'catalogue/book_delete.html', {
+        'doc': doc,
     })
 
 
-def chunk_edit(request, slug, chunk):
-    try:
-        doc = Chunk.get(slug, chunk)
-    except (Chunk.MultipleObjectsReturned, Chunk.DoesNotExist):
-        raise Http404
-    if request.method == "POST":
-        form = forms.ChunkForm(request.POST, instance=doc)
-        if form.is_valid():
-            form.save()
-            return http.HttpResponseRedirect(doc.book.get_absolute_url())
+@require_POST
+@login_required
+def publish(request, pk):
+    from wiki import forms
+    from .models import PublishRecord
+    from dvcs.models import Revision
+
+    doc = get_object_or_404(Document, pk=pk, deleted=False)
+    if not doc.can_edit(request.user):
+        return HttpResponseForbidden("Not authorized.")
+    form = forms.DocumentTextPublishForm(request.POST, prefix="textpublish")
+    if form.is_valid():
+        rev = Revision.objects.get(pk=form.cleaned_data['revision'])
+        # FIXME: check if in tree
+        # if PublishRecord.objects.filter(revision=rev, document=doc).exists():
+        #     return http.HttpResponse('exists')
+        if not doc.published:
+            site = Site.objects.get_current()
+            send_notify_email(
+                'New published document in MIL/PEER',
+                '''New published document in MIL/PEER: %s. View it in browser: https://%s%s.
+
+--
+MIL/PEER team.''' % (doc.meta()['title'], site.domain, reverse('catalogue_html', args=[doc.pk])))
+        PublishRecord.objects.create(revision=rev, document=doc, user=request.user)
+        doc.published = True
+        doc.save()
+        if request.is_ajax():
+            return http.HttpResponse('ok')
+        else:
+            return redirect('catalogue_html', doc.pk)
     else:
-        form = forms.ChunkForm(instance=doc)
-    return direct_to_template(request, "catalogue/chunk_edit.html", extra_context={
-        "chunk": doc,
-        "form": form,
-    })
+        if request.is_ajax():
+            return http.HttpResponse('error')
+        else:
+            try:
+                return redirect('catalogue_preview_rev', doc.pk, form.cleaned_data['revision'])
+            except KeyError:
+                return redirect('catalogue_preview', doc.pk)
 
 
-def book_append(request, slug):
-    book = get_object_or_404(Book, slug=slug)
-    if request.method == "POST":
-        form = forms.BookAppendForm(request.POST)
-        if form.is_valid():
-            append_to = form.cleaned_data['append_to']
-            append_to.append(book)
-            return http.HttpResponseRedirect(append_to.get_absolute_url())
+@require_POST
+@login_required
+def unpublish(request, pk):
+    doc = get_object_or_404(Document, pk=pk, deleted=False)
+    if not doc.can_edit(request.user):
+        return HttpResponseForbidden("Not authorized.")
+
+    doc.publish_log.all().delete()
+    if request.is_ajax():
+        return http.HttpResponse('ok')
     else:
-        form = forms.BookAppendForm()
-    return direct_to_template(request, "catalogue/book_append_to.html", extra_context={
-        "book": book,
-        "form": form,
-    })
+        return redirect('catalogue_html', doc.pk)
+
+
+class GalleryMixin(object):
+    def get_directory(self):
+        # return "%s%s/" % (settings.IMAGE_DIR, 'org%d' % self.org.pk if self.org is not None else self.request.user.pk)
+        return "uploads/%d/" % self.doc.pk
+
 
+class GalleryView(GalleryMixin, UploadView):
 
-def book_edit(request, slug):
-    book = get_object_or_404(Book, slug=slug)
+    def breadcrumbs(self):
+        return [
+                (self.doc.meta()['title'], '/documents/%d/' % self.doc.pk),
+            ]
+
+    def get_object(self, request, pk=None):
+        self.doc = Document.objects.get(pk=pk, deleted=False)
+
+
+@login_required
+def fork(request, pk):
+    doc = get_object_or_404(Document, pk=pk, deleted=False)
     if request.method == "POST":
-        form = forms.BookForm(request.POST, instance=book)
+        form = forms.DocumentForkForm(request.POST, request.FILES)
         if form.is_valid():
-            form.save()
-            return http.HttpResponseRedirect(book.get_absolute_url())
+            try:
+                org = request.user.membership_set.get(
+                    organization=int(form.cleaned_data['owner_organization'])).organization
+                kwargs = {'owner_organization': org}
+            except:
+                kwargs = {'owner_user': request.user}
+
+            new_doc = Document.objects.create(revision=doc.revision, **kwargs)
+
+            if os.path.isdir(settings.MEDIA_ROOT + "uploads/%d" % doc.pk):
+                shutil.copytree(
+                    settings.MEDIA_ROOT + "uploads/%d" % doc.pk,
+                    settings.MEDIA_ROOT + "uploads/%d" % new_doc.pk
+                )
+
+            new_doc.assigned_to = request.user
+            new_doc.save()
+
+            return http.HttpResponseRedirect(reverse("wiki_editor", args=[new_doc.pk]))
     else:
-        form = forms.BookForm(instance=book)
-    return direct_to_template(request, "catalogue/book_edit.html", extra_context={
-        "book": book,
+        form = forms.DocumentForkForm()
+
+    return render(request, "catalogue/document_fork.html", {
         "form": form,
+
+        "logout_to": '/',
     })
 
 
-@require_POST
-@login_required
-def publish(request, slug):
-    book = get_object_or_404(Book, slug=slug)
-    try:
-        book.publish(request.user)
-    except BaseException, e:
-        return http.HttpResponse(e)
-    else:
-        return http.HttpResponseRedirect(book.get_absolute_url())
+def upcoming(request):
+    f = DocumentFilterSet(request.GET, queryset=Document.objects.filter(deleted=False).filter(publish_log=None))
+    return render(request, "catalogue/upcoming.html", {
+        'filter_set': f,
+        'link_url': 'catalogue_preview',
+    })
+
+
+def finished(request):
+    f = DocumentFilterSet(request.GET, queryset=Document.objects.filter(deleted=False).exclude(publish_log=None))
+    return render(request, "catalogue/finished.html", {
+        'filter_set': f,
+        'link_url': 'catalogue_html',
+    })