wizard_mail_compose_message.go

package mail

import (
	"github.com/hexya-erp/hexya/src/models"
	"github.com/hexya-erp/pool/h"
	"github.com/hexya-erp/pool/q"
)

//import base64
//import re
//EXPRESSION_PATTERN = re.compile('(\$\{.+?\})')
func Reopen(self interface{}, res_id interface{}, model interface{}, context interface{}) {
	//    context = dict(context or {}, default_model=model)
	//    return {'type': 'ir.actions.act_window',
	//            'view_mode': 'form',
	//            'view_type': 'form',
	//            'res_id': res_id,
	//            'res_model': self._name,
	//            'target': 'new',
	//            'context': context,
	//            }
}
func init() {
	h.MailComposeMessage().DeclareModel()

	//    _batch_size = 500
	h.MailComposeMessage().Methods().DefaultGet().Extend(
		` Handle composition mode. Some details about context keys:
            - comment: default mode, model and ID of a
record the user comments
                - default_model or active_model
                - default_res_id or active_id
            - reply: active_id of a message the user replies to
                - default_parent_id or message_id or active_id:
ID of the
                    mail.message we reply to
                - message.res_model or default_model
                - message.res_id or default_res_id
            - mass_mail: model and IDs of records the user mass-mails
                - active_ids: record IDs
                - default_model or active_model
        `,
		func(rs m.MailComposeMessageSet, fields interface{}) {
			//        result = super(MailComposer, self).default_get(fields)
			//        result['composition_mode'] = result.get(
			//            'composition_mode', self._context.get('mail.compose.message.mode', 'comment'))
			//        result['model'] = result.get(
			//            'model', self._context.get('active_model'))
			//        result['res_id'] = result.get('res_id', self._context.get('active_id'))
			//        result['parent_id'] = result.get(
			//            'parent_id', self._context.get('message_id'))
			//        if 'no_auto_thread' not in result and (result['model'] not in self.env or not hasattr(self.env[result['model']], 'message_post')):
			//            result['no_auto_thread'] = True
			//        if result['composition_mode'] == 'reply':
			//            result['composition_mode'] = 'comment'
			//        vals = {}
			//        if 'active_domain' in self._context:  # not context.get() because we want to keep global [] domains
			//            vals['use_active_domain'] = True
			//            vals['active_domain'] = '%s' % self._context.get('active_domain')
			//        if result['composition_mode'] == 'comment':
			//            vals.update(self.get_record_data(result))
			//        for field in vals:
			//            if field in fields:
			//                result[field] = vals[field]
			//        if result['model'] == 'res.users' and result['res_id'] == self._uid:
			//            result['model'] = 'res.partner'
			//            result['res_id'] = self.env.user.partner_id.id
			//        if fields is not None:
			//            [result.pop(field, None)
			//             for field in result.keys() if field not in fields]
			//        return result
		})
	h.MailComposeMessage().Methods().GetCompositionModeSelection().DeclareMethod(
		`GetCompositionModeSelection`,
		func(rs m.MailComposeMessageSet) {
			//        return [('comment', 'Post on a document'),
			//                ('mass_mail', 'Email Mass Mailing'),
			//                ('mass_post', 'Post on Multiple Documents')]
		})
	h.MailComposeMessage().AddFields(map[string]models.FieldDefinition{
		"CompositionMode": models.SelectionField{
			//selection=_get_composition_mode_selection
			String:  "Composition mode",
			Default: models.DefaultValue("comment"),
		},
		"PartnerIds": models.Many2ManyField{
			RelationModel:    h.Partner(),
			M2MLinkModelName: "",
			M2MOurField:      "",
			M2MTheirField:    "",
			String:           "Additional Contacts",
		},
		"UseActiveDomain": models.BooleanField{
			String: "Use active domain",
		},
		"ActiveDomain": models.TextField{
			String:   "Active domain",
			ReadOnly: true,
		},
		"AttachmentIds": models.Many2ManyField{
			RelationModel:    h.Attachment(),
			M2MLinkModelName: "",
			M2MOurField:      "",
			M2MTheirField:    "",
			String:           "Attachments",
		},
		"IsLog": models.BooleanField{
			String: "Log an Internal Note",
			Help:   "Whether the message is an internal note (comment mode only)",
		},
		"Subject": models.CharField{
			Default: models.DefaultValue(false),
		},
		"Notify": models.BooleanField{
			String: "Notify followers",
			Help:   "Notify followers of the document (mass post only)",
		},
		"AutoDelete": models.BooleanField{
			String: "Delete Emails",
			Help:   "Delete sent emails (mass mailing only)",
		},
		"AutoDeleteMessage": models.BooleanField{
			String: "Delete Message Copy",
			Help: "Do not keep a copy of the email in the document communication" +
				"history (mass mailing only)",
		},
		"TemplateId": models.Many2OneField{
			RelationModel: h.MailTemplate(),
			String:        "Use template",
			Index:         true,
			Filter:        q.Model().Equals(model),
		},
		"MessageType": models.SelectionField{
			Default: models.DefaultValue("comment"),
		},
		"SubtypeId": models.Many2OneField{
			Default: func(env models.Environment) interface{} { return self.sudo().env.ref().id },
		},
	})
	h.MailComposeMessage().Methods().CheckAccessRule().DeclareMethod(
		` Access rules of mail.compose.message:
            - create: if
                - model, no res_id, I create a message in mass mail mode
            - then: fall back on mail.message acces rules
        `,
		func(rs m.MailComposeMessageSet, operation interface{}) {
			//        if operation == 'create' and self._uid != SUPERUSER_ID:
			//            # read mail_compose_message.ids to have their values
			//            message_values = {}
			//            self._cr.execute(
			//                'SELECT DISTINCT id, model, res_id FROM "%s" WHERE id = ANY (%%s) AND res_id = 0' % self._table, (self.ids))
			//            for mid, rmod, rid in self._cr.fetchall():
			//                message_values[mid] = {'model': rmod, 'res_id': rid}
			//            # remove from the set to check the ids that mail_compose_message accepts
			//            author_ids = [mid for mid, message in message_values.iteritems()
			//                          if message.get('model') and not message.get('res_id')]
			//            # not sure slef = ...
			//            self = self.browse(list(set(self.ids) - set(author_ids)))
			//        return super(MailComposer, self).check_access_rule(operation)
		})
	h.MailComposeMessage().Methods().Notify().DeclareMethod(
		` Override specific notify method of mail.message, because we do
            not want that feature in the wizard. `,
		func(rs m.MailComposeMessageSet, force_send interface{}, user_signature interface{}) {
			//        return
		})
	h.MailComposeMessage().Methods().GetRecordData().DeclareMethod(
		` Returns a defaults-like dict with initial values for the composition
        wizard when sending an email related a previous
email (parent_id) or
        a document (model, res_id). This is based on previously
computed default
        values. `,
		func(rs m.MailComposeMessageSet, values interface{}) {
			//        result, subject = {}, False
			//        if values.get('parent_id'):
			//            parent = self.env['mail.message'].browse(values.get('parent_id'))
			//            result['record_name'] = parent.record_name,
			//            subject = tools.ustr(parent.subject or parent.record_name or '')
			//            if not values.get('model'):
			//                result['model'] = parent.model
			//            if not values.get('res_id'):
			//                result['res_id'] = parent.res_id
			//            partner_ids = values.get('partner_ids', list(
			//            )) + [(4, id) for id in parent.partner_ids.ids]
			//            # check message is private then add author also in partner list.
			//            if self._context.get('is_private') and parent.author_id:
			//                partner_ids += [(4, parent.author_id.id)]
			//            result['partner_ids'] = partner_ids
			//        elif values.get('model') and values.get('res_id'):
			//            doc_name_get = self.env[values.get('model')].browse(
			//                values.get('res_id')).name_get()
			//            result['record_name'] = doc_name_get and doc_name_get[0][1] or ''
			//            subject = tools.ustr(result['record_name'])
			//        re_prefix = _('Re:')
			//        if subject and not (subject.startswith('Re:') or subject.startswith(re_prefix)):
			//            subject = "%s %s" % (re_prefix, subject)
			//        result['subject'] = subject
			//        return result
		})
	h.MailComposeMessage().Methods().SendMailAction().DeclareMethod(
		`SendMailAction`,
		func(rs m.MailComposeMessageSet) {
			//        return self.send_mail()
		})
	h.MailComposeMessage().Methods().SendMail().DeclareMethod(
		` Process the wizard content and proceed with sending the related
            email(s), rendering any template patterns on
the fly if needed. `,
		func(rs m.MailComposeMessageSet, auto_commit interface{}) {
			//        for wizard in self:
			//            # Duplicate attachments linked to the email.template.
			//            # Indeed, basic mail.compose.message wizard duplicates attachments in mass
			//            # mailing mode. But in 'single post' mode, attachments of an email template
			//            # also have to be duplicated to avoid changing their ownership.
			//            if wizard.attachment_ids and wizard.composition_mode != 'mass_mail' and wizard.template_id:
			//                new_attachment_ids = []
			//                for attachment in wizard.attachment_ids:
			//                    if attachment in wizard.template_id.attachment_ids:
			//                        new_attachment_ids.append(attachment.copy(
			//                            {'res_model': 'mail.compose.message', 'res_id': wizard.id}).id)
			//                    else:
			//                        new_attachment_ids.append(attachment.id)
			//                    wizard.write(
			//                        {'attachment_ids': [(6, 0, new_attachment_ids)]})
			//
			//            # Mass Mailing
			//            mass_mode = wizard.composition_mode in ('mass_mail', 'mass_post')
			//
			//            Mail = self.env['mail.mail']
			//            ActiveModel = self.env[wizard.model if wizard.model else 'mail.thread']
			//            if wizard.template_id:
			//                # template user_signature is added when generating body_html
			//                # mass mailing: use template auto_delete value -> note, for emails mass mailing only
			//                Mail = Mail.with_context(mail_notify_user_signature=False)
			//                ActiveModel = ActiveModel.with_context(
			//                    mail_notify_user_signature=False, mail_auto_delete=wizard.template_id.auto_delete)
			//            if not hasattr(ActiveModel, 'message_post'):
			//                ActiveModel = self.env['mail.thread'].with_context(
			//                    thread_model=wizard.model)
			//            if wizard.composition_mode == 'mass_post':
			//                # do not send emails directly but use the queue instead
			//                # add context key to avoid subscribing the author
			//                ActiveModel = ActiveModel.with_context(
			//                    mail_notify_force_send=False, mail_create_nosubscribe=True)
			//            # wizard works in batch mode: [res_id] or active_ids or active_domain
			//            if mass_mode and wizard.use_active_domain and wizard.model:
			//                res_ids = self.env[wizard.model].search(
			//                    safe_eval(wizard.active_domain)).ids
			//            elif mass_mode and wizard.model and self._context.get('active_ids'):
			//                res_ids = self._context['active_ids']
			//            else:
			//                res_ids = [wizard.res_id]
			//
			//            batch_size = int(self.env['ir.config_parameter'].sudo(
			//            ).get_param('mail.batch_size')) or self._batch_size
			//            sliced_res_ids = [res_ids[i:i + batch_size]
			//                              for i in range(0, len(res_ids), batch_size)]
			//
			//            # log a note: subtype is False
			//            if wizard.composition_mode == 'mass_mail' or wizard.is_log or (wizard.composition_mode == 'mass_post' and not wizard.notify):
			//                subtype_id = False
			//            elif wizard.subtype_id:
			//                subtype_id = wizard.subtype_id.id
			//            else:
			//                subtype_id = self.sudo().env.ref('mail.mt_comment', raise_if_not_found=False).id
			//
			//            for res_ids in sliced_res_ids:
			//                batch_mails = Mail
			//                all_mail_values = wizard.get_mail_values(res_ids)
			//                for res_id, mail_values in all_mail_values.iteritems():
			//                    if wizard.composition_mode == 'mass_mail':
			//                        batch_mails |= Mail.create(mail_values)
			//                    else:
			//                        ActiveModel.browse(res_id).message_post(
			//                            message_type=wizard.message_type,
			//                            subtype_id=subtype_id,
			//                            **mail_values)
			//
			//                if wizard.composition_mode == 'mass_mail':
			//                    batch_mails.send(auto_commit=auto_commit)
			//        return {'type': 'ir.actions.act_window_close'}
		})
	h.MailComposeMessage().Methods().GetMailValues().DeclareMethod(
		`Generate the values that will be used by send_mail to create
mail_messages
        or mail_mails. `,
		func(rs m.MailComposeMessageSet, res_ids interface{}) {
			//        self.ensure_one()
			//        results = dict.fromkeys(res_ids, False)
			//        rendered_values = {}
			//        mass_mail_mode = self.composition_mode == 'mass_mail'
			//        if mass_mail_mode and self.model:
			//            rendered_values = self.render_message(res_ids)
			//        reply_to_value = dict.fromkeys(res_ids, None)
			//        if mass_mail_mode and not self.no_auto_thread:
			//            # reply_to_value = self.env['mail.thread'].with_context(thread_model=self.model).browse(res_ids).message_get_reply_to(default=self.email_from)
			//            reply_to_value = self.env['mail.thread'].with_context(
			//                thread_model=self.model).message_get_reply_to(res_ids, default=self.email_from)
			//        for res_id in res_ids:
			//            # static wizard (mail.message) values
			//            mail_values = {
			//                'subject': self.subject,
			//                'body': self.body or '',
			//                'parent_id': self.parent_id and self.parent_id.id,
			//                'partner_ids': [partner.id for partner in self.partner_ids],
			//                'attachment_ids': [attach.id for attach in self.attachment_ids],
			//                'author_id': self.author_id.id,
			//                'email_from': self.email_from,
			//                'record_name': self.record_name,
			//                'no_auto_thread': self.no_auto_thread,
			//                'mail_server_id': self.mail_server_id.id,
			//            }
			//
			//            # mass mailing: rendering override wizard static values
			//            if mass_mail_mode and self.model:
			//                if self.model in self.env and hasattr(self.env[self.model], 'message_get_email_values'):
			//                    mail_values.update(self.env[self.model].browse(
			//                        res_id).message_get_email_values())
			//                # keep a copy unless specifically requested, reset record name (avoid browsing records)
			//                mail_values.update(notification=not self.auto_delete_message,
			//                                   model=self.model, res_id=res_id, record_name=False)
			//                # auto deletion of mail_mail
			//                if self.auto_delete or self.template_id.auto_delete:
			//                    mail_values['auto_delete'] = True
			//                # rendered values using template
			//                email_dict = rendered_values[res_id]
			//                mail_values['partner_ids'] += email_dict.pop('partner_ids', [])
			//                mail_values.update(email_dict)
			//                if not self.no_auto_thread:
			//                    mail_values.pop('reply_to')
			//                    if reply_to_value.get(res_id):
			//                        mail_values['reply_to'] = reply_to_value[res_id]
			//                if self.no_auto_thread and not mail_values.get('reply_to'):
			//                    mail_values['reply_to'] = mail_values['email_from']
			//                # mail_mail values: body -> body_html, partner_ids -> recipient_ids
			//                mail_values['body_html'] = mail_values.get('body', '')
			//                mail_values['recipient_ids'] = [
			//                    (4, id) for id in mail_values.pop('partner_ids', [])]
			//
			//                # process attachments: should not be encoded before being processed by message_post / mail_mail create
			//                mail_values['attachments'] = [(name, base64.b64decode(
			//                    enc_cont)) for name, enc_cont in email_dict.pop('attachments', list())]
			//                attachment_ids = []
			//                for attach_id in mail_values.pop('attachment_ids'):
			//                    new_attach_id = self.env['ir.attachment'].browse(
			//                        attach_id).copy({'res_model': self._name, 'res_id': self.id})
			//                    attachment_ids.append(new_attach_id.id)
			//                mail_values['attachment_ids'] = self.env['mail.thread']._message_preprocess_attachments(
			//                    mail_values.pop('attachments', []),
			//                    attachment_ids, 'mail.message', 0)
			//
			//            results[res_id] = mail_values
			//        return results
		})
	h.MailComposeMessage().Methods().OnchangeTemplateIdWrapper().DeclareMethod(
		`OnchangeTemplateIdWrapper`,
		func(rs m.MailComposeMessageSet) {
			//        self.ensure_one()
			//        values = self.onchange_template_id(
			//            self.template_id.id, self.composition_mode, self.model, self.res_id)['value']
			//        for fname, value in values.iteritems():
			//            setattr(self, fname, value)
		})
	h.MailComposeMessage().Methods().OnchangeTemplateId().DeclareMethod(
		` - mass_mailing: we cannot render, so return the template values
            - normal mode: return rendered values
            /!\ for x2many field, this onchange return
command instead of ids
        `,
		func(rs m.MailComposeMessageSet, template_id interface{}, composition_mode interface{}, model interface{}, res_id interface{}) {
			//        if template_id and composition_mode == 'mass_mail':
			//            template = self.env['mail.template'].browse(template_id)
			//            fields = ['subject', 'body_html',
			//                      'email_from', 'reply_to', 'mail_server_id']
			//            values = dict((field, getattr(template, field))
			//                          for field in fields if getattr(template, field))
			//            if template.attachment_ids:
			//                values['attachment_ids'] = [
			//                    att.id for att in template.attachment_ids]
			//            if template.mail_server_id:
			//                values['mail_server_id'] = template.mail_server_id.id
			//            if template.user_signature and 'body_html' in values:
			//                signature = self.env.user.signature
			//                values['body_html'] = tools.append_content_to_html(
			//                    values['body_html'], signature, plaintext=False)
			//        elif template_id:
			//            values = self.generate_email_for_composer(
			//                template_id, [res_id])[res_id]
			//            # transform attachments into attachment_ids; not attached to the document because this will
			//            # be done further in the posting process, allowing to clean database if email not send
			//            Attachment = self.env['ir.attachment']
			//            for attach_fname, attach_datas in values.pop('attachments', []):
			//                data_attach = {
			//                    'name': attach_fname,
			//                    'datas': attach_datas,
			//                    'datas_fname': attach_fname,
			//                    'res_model': 'mail.compose.message',
			//                    'res_id': 0,
			//                    'type': 'binary',  # override default_type from context, possibly meant for another model!
			//                }
			//                values.setdefault('attachment_ids', list()).append(
			//                    Attachment.create(data_attach).id)
			//        else:
			//            default_values = self.with_context(default_composition_mode=composition_mode, default_model=model, default_res_id=res_id).default_get(
			//                ['composition_mode', 'model', 'res_id', 'parent_id', 'partner_ids', 'subject', 'body', 'email_from', 'reply_to', 'attachment_ids', 'mail_server_id'])
			//            values = dict((key, default_values[key]) for key in [
			//                          'subject', 'body', 'partner_ids', 'email_from', 'reply_to', 'attachment_ids', 'mail_server_id'] if key in default_values)
			//        if values.get('body_html'):
			//            values['body'] = values.pop('body_html')
			//        values = self._convert_to_write(values)
			//        return {'value': values}
		})
	h.MailComposeMessage().Methods().SaveAsTemplate().DeclareMethod(
		` hit save as template button: current form value will be a new
            template attached to the current document. `,
		func(rs m.MailComposeMessageSet) {
			//        for record in self:
			//            models = self.env['ir.model'].search(
			//                [('model', '=', record.model or 'mail.message')])
			//            model_name = ''
			//            if models:
			//                model_name = models.name
			//            template_name = "%s: %s" % (model_name, tools.ustr(record.subject))
			//            values = {
			//                'name': template_name,
			//                'subject': record.subject or False,
			//                'body_html': record.body or False,
			//                'model_id': models.id or False,
			//                'attachment_ids': [(6, 0, [att.id for att in record.attachment_ids])],
			//            }
			//            template = self.env['mail.template'].create(values)
			//            # generate the saved template
			//            record.write({'template_id': template.id})
			//            record.onchange_template_id_wrapper()
			//            return _reopen(self, record.id, record.model, context=self._context)
		})
	h.MailComposeMessage().Methods().RenderMessage().DeclareMethod(
		`Generate template-based values of wizard, for the document records given
        by res_ids. This method is meant to be inherited
by email_template that
        will produce a more complete dictionary, using Jinja2 templates.

        Each template is generated for all res_ids, allowing
to parse the template
        once, and render it multiple times. This is useful
for mass mailing where
        template rendering represent a significant part of the process.

        Default recipients are also computed, based on
mail_thread method
        message_get_default_recipients. This allows to
ensure a mass mailing has
        always some recipients specified.

        :param browse wizard: current mail.compose.message browse record
        :param list res_ids: list of record ids

        :return dict results: for each res_id, the generated
template values for
                              subject, body, email_from and reply_to
        `,
		func(rs m.MailComposeMessageSet, res_ids interface{}) {
			//        self.ensure_one()
			//        multi_mode = True
			//        if isinstance(res_ids, (int, long)):
			//            multi_mode = False
			//            res_ids = [res_ids]
			//        subjects = self.render_template(self.subject, self.model, res_ids)
			//        bodies = self.render_template(
			//            self.body, self.model, res_ids, post_process=True)
			//        emails_from = self.render_template(
			//            self.email_from, self.model, res_ids)
			//        replies_to = self.render_template(self.reply_to, self.model, res_ids)
			//        default_recipients = {}
			//        if not self.partner_ids:
			//            default_recipients = self.env['mail.thread'].message_get_default_recipients(
			//                res_model=self.model, res_ids=res_ids)
			//        results = dict.fromkeys(res_ids, False)
			//        for res_id in res_ids:
			//            results[res_id] = {
			//                'subject': subjects[res_id],
			//                'body': bodies[res_id],
			//                'email_from': emails_from[res_id],
			//                'reply_to': replies_to[res_id],
			//            }
			//            results[res_id].update(default_recipients.get(res_id, dict()))
			//        if self.template_id:
			//            template_values = self.generate_email_for_composer(
			//                self.template_id.id, res_ids,
			//                fields=['email_to', 'partner_to', 'email_cc', 'attachment_ids', 'mail_server_id'])
			//        else:
			//            template_values = {}
			//        for res_id in res_ids:
			//            if template_values.get(res_id):
			//                # recipients are managed by the template
			//                results[res_id].pop('partner_ids')
			//                results[res_id].pop('email_to')
			//                results[res_id].pop('email_cc')
			//                # remove attachments from template values as they should not be rendered
			//                template_values[res_id].pop('attachment_ids', None)
			//            else:
			//                template_values[res_id] = dict()
			//            # update template values by composer values
			//            template_values[res_id].update(results[res_id])
			//        return multi_mode and template_values or template_values[res_ids[0]]
		})
	h.MailComposeMessage().Methods().GenerateEmailForComposer().DeclareMethod(
		` Call email_template.generate_email(), get fields relevant for
            mail.compose.message, transform email_cc and
email_to into partner_ids `,
		func(rs m.MailComposeMessageSet, template_id interface{}, res_ids interface{}, fields interface{}) {
			//        multi_mode = True
			//        if isinstance(res_ids, (int, long)):
			//            multi_mode = False
			//            res_ids = [res_ids]
			//        if fields is None:
			//            fields = ['subject', 'body_html', 'email_from', 'email_to', 'partner_to',
			//                      'email_cc',  'reply_to', 'attachment_ids', 'mail_server_id']
			//        returned_fields = fields + ['partner_ids', 'attachments']
			//        values = dict.fromkeys(res_ids, False)
			//        template_values = self.env['mail.template'].with_context(
			//            tpl_partners_only=True).browse(template_id).generate_email(res_ids, fields=fields)
			//        for res_id in res_ids:
			//            res_id_values = dict((field, template_values[res_id][field])
			//                                 for field in returned_fields if template_values[res_id].get(field))
			//            res_id_values['body'] = res_id_values.pop('body_html', '')
			//            values[res_id] = res_id_values
			//        return multi_mode and values or values[res_ids[0]]
		})
	h.MailComposeMessage().Methods().RenderTemplate().DeclareMethod(
		`RenderTemplate`,
		func(rs m.MailComposeMessageSet, template interface{}, model interface{}, res_ids interface{}, post_process interface{}) {
			//        return self.env['mail.template'].render_template(template, model, res_ids, post_process=post_process)
		})
}