# -*- encoding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    This module copyright (C) 2014 Savoir-faire Linux
#    (<http://www.savoirfairelinux.com>).
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

from openerp.osv import orm, fields
from openerp.addons.connector.session import ConnectorSession
from openerp.addons.connector.queue.job import job
import base64
import logging
_logger = logging.getLogger(__name__)


class ir_attachment(orm.Model):
    _inherit = 'ir.attachment'

    _columns = {
        'id_dms': fields.char('Id of Dms', size=256, help="Id of Dms."),
    }

    def create(self, cr, uid, values, context=None):
        metadata_obj = self.pool.get('metadata')
        user_obj = self.pool.get('res.users')
        user_login = user_obj.browse(cr, uid, uid, context=context).login
        session = ConnectorSession(cr, uid, context=context)
        value = {
            'name': values.get('name'),
            'datas_fname': values.get('datas_fname'),
            'file_type': values.get('file_type') or '',
            'datas': values.get('datas'),
            'description': values.get('description') or '',

        }

        metadata_ids = metadata_obj.search(cr, uid, [], context=context)
        dict_metadata = {}
        list_fields = []
        # Get list of metadata
        if values.get('res_model'):
            for line in metadata_obj.browse(cr, uid, metadata_ids,
                                            context=context):
                if line.model_id.model == values.get('res_model'):
                    if line.metadata_list_ids:
                        for one_field in line.metadata_list_ids:
                            list_fields.append(one_field.field_id.name)
            result = self.pool.get(values.get('res_model')).read(cr, uid, [
                values.get('res_id')], list_fields, context=context)[0]

            for one_field in list_fields:
                dict_metadata['cmis:' + one_field] = result[one_field]
        values['datas'] = None
        res = super(ir_attachment, self).create(cr, uid, values,
                                                context=context)
        # Create Job
        # if bool_testdoc in context, we don't need to create
        # the doc in the DMS
        if not context.get('bool_testdoc'):
            create_doc_in_edm.delay(
                session, 'ir.attachment', value, res, dict_metadata,
                user_login)
        return res


@job
def create_doc_in_edm(session, model_name, value, res,
                      dict_metadata, user_login, filters=None):
    ir_attach_obj = session.pool.get('ir.attachment')
    cmis_backend_obj = session.pool.get('cmis.backend')
    if session.context is None:
        session.context = {}
    # login with the cmis account
    repo = cmis_backend_obj._auth(session.cr, session.uid,
                                  context=session.context)
    root = repo.rootFolder
    ids = cmis_backend_obj.search(session.cr, session.uid, [], session.context)

    folder_path = cmis_backend_obj.read(
        session.cr, session.uid,
        ids,
        ['initial_directory_write'],
        context=session.context)[0]['initial_directory_write']
    # Document properties
    if value['name']:
        file_name = value['name']
    elif value['datas_fname']:
        file_name = value['datas_fname']
    else:
        file_name = value['datas_fname']
    props = {
        'cmis:name': file_name,
        'cmis:description': value['description'],
        'cmis:createdBy': user_login,
    }
    # Add list of metadata in props
    if len(dict_metadata):
        for k, v in dict_metadata.iteritems():
            props[k] = v
    if folder_path:
        sub1 = repo.getObjectByPath(folder_path)
    else:
        sub1 = root

    someDoc = sub1.createDocumentFromString(value['datas_fname'],
                                            contentString=base64.b64decode(
                                                value['datas']),
                                            contentType=value['file_type'])
    # TODO: create custom properties on a document (Alfresco)
    #someDoc.getProperties().update(props)
    # Updating ir.attachment object with the new id
    # of document generated by DMS
    ir_attach_obj.write(session.cr, session.uid, res, {
        'id_dms': someDoc.getObjectId()}, session.context)
    return True

# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: