## $Id: fileviewlets.py 14871 2017-10-13 08:48:46Z henrik $
##
## Copyright (C) 2014 Uli Fouquet & Henrik Bettermann
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##

import grok
from zope.component import getUtility
from waeup.kofa.interfaces import IExtFileStore
from waeup.kofa.interfaces import MessageFactory as _
from waeup.kofa.browser.layout import UtilityView
from waeup.kofa.browser.fileviewlets import (
    FileDisplay, FileUpload, Image)
from waeup.kofa.utils.helpers import file_size

from waeup.kofa.documents.document import PDFDocument
from waeup.kofa.documents.browser import (
    DocumentManageFormPage, DocumentDisplayFormPage)

from waeup.kofa.documents.workflow import PUBLISHED


# File viewlets for documents

class PDFScanManageUpload(FileUpload):
    """Scan upload viewlet for officers.
    """
    grok.view(DocumentManageFormPage)
    grok.context(PDFDocument)
    grok.require('waeup.manageDocuments')
    label = _(u'PDF File')
    title = _(u'PDF File')
    mus = 1.5 * 1024*1024
    download_name = u'file.pdf'
    tab_redirect = '#tab2'

    @property
    def download_filename(self):
        file = getUtility(IExtFileStore).getFileByContext(
            self.context, attr=self.download_name)
        fs = file_size(file)/1024.0
        return u"%s.pdf (%.1f kB)" % (self.context.document_id, fs)


class PDFScanDisplay(FileDisplay):
    """Scan display viewlet.
    """
    grok.order(1)
    grok.context(PDFDocument)
    grok.require('waeup.viewDocuments')
    grok.view(DocumentDisplayFormPage)
    label = _(u'PDF Scan')
    title = _(u'PDF Scan')
    download_name = u'file.pdf'

    @property
    def download_filename(self):
        file = getUtility(IExtFileStore).getFileByContext(
            self.context, attr=self.download_name)
        fs = file_size(file)/1024.0
        return u"%s.pdf (%.1f kB)" % (self.context.document_id, fs)


class PDFScanImage(UtilityView, Image):
    """Scan document.
    """
    grok.name('file.pdf')
    grok.context(PDFDocument)
    grok.require('waeup.Public')
    download_name = u'file.pdf'

    @property
    def download_filename(self):
        return self.context.document_id

    def update(self):
        if self.context.state != PUBLISHED:
            self.flash(_('The document requested has not yet been published.'),
                type="warning")
            self.redirect(self.application_url())
        return