# Part of Odoo. See LICENSE file for full copyright and licensing details. import base64 from collections import defaultdict from hashlib import sha512 from secrets import choice from markupsafe import Markup from datetime import timedelta from odoo import _, api, fields, models, tools, Command from odoo.addons.base.models.avatar_mixin import get_hsl_from_seed from odoo.addons.mail.tools.discuss import Store from odoo.exceptions import AccessError, UserError, ValidationError from odoo.osv import expression from odoo.tools import format_list, get_lang, html_escape channel_avatar = ''' ''' group_avatar = ''' ''' class Channel(models.Model): _description = 'Discussion Channel' _name = 'discuss.channel' _mail_flat_thread = False _mail_post_access = 'read' _inherit = ["mail.thread", "bus.listener.mixin"] MAX_BOUNCE_LIMIT = 10 @api.model def _generate_random_token(self): # Built to be shared on invitation link. It uses non-ambiguous characters and it is of a # reasonable length: enough to avoid brute force, but short enough to be shareable easily. # This token should not contain "mail.guest"._cookie_separator value. return ''.join(choice('abcdefghijkmnopqrstuvwxyzABCDEFGHIJKLMNPQRSTUVWXYZ23456789') for _i in range(10)) # description name = fields.Char('Name', required=True) active = fields.Boolean(default=True, help="Set active to false to hide the channel without removing it.") channel_type = fields.Selection([ ('chat', 'Chat'), ('channel', 'Channel'), ('group', 'Group')], string='Channel Type', required=True, default='channel', readonly=True, help="Chat is private and unique between 2 persons. Group is private among invited persons. Channel can be freely joined (depending on its configuration).") is_editable = fields.Boolean('Is Editable', compute='_compute_is_editable') default_display_mode = fields.Selection(string="Default Display Mode", selection=[('video_full_screen', "Full screen video")], help="Determines how the channel will be displayed by default when opening it from its invitation link. No value means display text (no voice/video).") description = fields.Text('Description') image_128 = fields.Image("Image", max_width=128, max_height=128) avatar_128 = fields.Image("Avatar", max_width=128, max_height=128, compute='_compute_avatar_128') avatar_cache_key = fields.Char(compute="_compute_avatar_cache_key") channel_partner_ids = fields.Many2many( 'res.partner', string='Partners', compute='_compute_channel_partner_ids', inverse='_inverse_channel_partner_ids', search='_search_channel_partner_ids') channel_member_ids = fields.One2many('discuss.channel.member', 'channel_id', string='Members') parent_channel_id = fields.Many2one("discuss.channel", help="Parent channel", ondelete="cascade", index=True, auto_join=True, readonly=True) sub_channel_ids = fields.One2many("discuss.channel", "parent_channel_id", string="Sub Channels", readonly=True) from_message_id = fields.Many2one("mail.message", help="The message the channel was created from.", readonly=True) pinned_message_ids = fields.One2many('mail.message', 'res_id', domain=[('model', '=', 'discuss.channel'), ('pinned_at', '!=', False)], string='Pinned Messages') sfu_channel_uuid = fields.Char(groups="base.group_system") sfu_server_url = fields.Char(groups="base.group_system") rtc_session_ids = fields.One2many('discuss.channel.rtc.session', 'channel_id', groups="base.group_system") is_member = fields.Boolean('Is Member', compute='_compute_is_member', search='_search_is_member') member_count = fields.Integer(string="Member Count", compute='_compute_member_count', compute_sudo=True) last_interest_dt = fields.Datetime("Last Interest", index=True, help="Contains the date and time of the last interesting event that happened in this channel. This updates itself when new message posted.") group_ids = fields.Many2many( 'res.groups', string='Auto Subscription', help="Members of those groups will automatically added as followers. " "Note that they will be able to manage their subscription manually " "if necessary.") # access uuid = fields.Char('UUID', size=50, default=_generate_random_token, copy=False) group_public_id = fields.Many2one('res.groups', string='Authorized Group', compute='_compute_group_public_id', readonly=False, store=True) invitation_url = fields.Char('Invitation URL', compute='_compute_invitation_url') allow_public_upload = fields.Boolean(default=False) _sql_constraints = [ ('channel_type_not_null', 'CHECK(channel_type IS NOT NULL)', 'The channel type cannot be empty'), ("from_message_id_unique", "UNIQUE(from_message_id)", "Messages can only be linked to one sub-channel"), ( "sub_channel_no_group_public_id", "CHECK(parent_channel_id IS NULL OR group_public_id IS NULL)", "Group public id should not be set on sub-channels as access is based on parent channel", ), ('uuid_unique', 'UNIQUE(uuid)', 'The channel UUID must be unique'), ('group_public_id_check', "CHECK (channel_type = 'channel' OR group_public_id IS NULL)", 'Group authorization and group auto-subscription are only supported on channels.') ] # CONSTRAINTS @api.constrains("from_message_id") def _constraint_from_message_id(self): # sudo: discuss.channel - skipping ACL for constraint, more performant and no sensitive information is leaked if failing_channels := self.sudo().filtered( lambda c: c.from_message_id and ( c.from_message_id.res_id != c.parent_channel_id.id or c.from_message_id.model != "discuss.channel" ) ): raise ValidationError( _( "Cannot create %(channels)s: initial message should belong to parent channel.", channels=format_list(self.env, failing_channels.mapped("name")), ) ) @api.constrains("parent_channel_id") def _constraint_parent_channel_id(self): # sudo: discuss.channel - skipping ACL for constraint, more performant and no sensitive information is leaked if failing_channels := self.sudo().filtered( lambda c: c.parent_channel_id and ( c.parent_channel_id.parent_channel_id or c.parent_channel_id.channel_type != "channel" ) ): raise ValidationError( _( "Cannot create %(channels)s: parent should not be a sub-channel and should be of type 'channel'.", channels=format_list(self.env, failing_channels.mapped("name")), ), ) @api.constrains('channel_member_ids') def _constraint_partners_chat(self): # sudo: discuss.channel - skipping ACL for constraint, more performant and no sensitive information is leaked for ch in self.sudo().filtered(lambda ch: ch.channel_type == 'chat'): if len(ch.channel_member_ids) > 2: raise ValidationError(_("A channel of type 'chat' cannot have more than two users.")) @api.constrains('group_public_id', 'group_ids') def _constraint_group_id_channel(self): # sudo: discuss.channel - skipping ACL for constraint, more performant and no sensitive information is leaked failing_channels = self.sudo().filtered(lambda channel: channel.channel_type != 'channel' and (channel.group_public_id or channel.group_ids)) if failing_channels: raise ValidationError(_("For %(channels)s, channel_type should be 'channel' to have the group-based authorization or group auto-subscription.", channels=', '.join([ch.name for ch in failing_channels]))) # COMPUTE / INVERSE @api.depends("channel_type", "is_member", "group_public_id") @api.depends_context("uid") def _compute_is_editable(self): for channel in self: channel.is_editable = channel.has_access("write") @api.depends('channel_type', 'image_128', 'uuid') def _compute_avatar_128(self): for record in self: record.avatar_128 = record.image_128 or record._generate_avatar() @api.depends('avatar_128') def _compute_avatar_cache_key(self): for channel in self: if not channel.avatar_128: channel.avatar_cache_key = 'no-avatar' else: channel.avatar_cache_key = sha512(channel.avatar_128).hexdigest() def _generate_avatar(self): if self.channel_type not in ('channel', 'group'): return False avatar = group_avatar if self.channel_type == 'group' else channel_avatar bgcolor = get_hsl_from_seed(self.uuid) avatar = avatar.replace('fill="#875a7b"', f'fill="{bgcolor}"') return base64.b64encode(avatar.encode()) @api.depends('channel_member_ids.partner_id') def _compute_channel_partner_ids(self): for channel in self: channel.channel_partner_ids = channel.channel_member_ids.partner_id def _inverse_channel_partner_ids(self): new_members = [] outdated = self.env['discuss.channel.member'] for channel in self: current_members = channel.channel_member_ids partners = channel.channel_partner_ids partners_new = partners - current_members.partner_id new_members += [{ 'channel_id': channel.id, 'partner_id': partner.id, } for partner in partners_new] outdated += current_members.filtered(lambda m: m.partner_id not in partners) if new_members: self.env['discuss.channel.member'].create(new_members) if outdated: outdated.unlink() def _search_channel_partner_ids(self, operator, operand): return [('channel_member_ids', 'any', [('partner_id', operator, operand)])] @api.depends_context('uid', 'guest') @api.depends('channel_member_ids') def _compute_is_member(self): if not self: return members = self.env['discuss.channel.member'].search([('channel_id', 'in', self.ids), ('is_self', '=', True)]) is_member_channels = members.channel_id for channel in self: channel.is_member = channel in is_member_channels def _search_is_member(self, operator, operand): is_in = (operator == '=' and operand) or (operator == '!=' and not operand) # Separate query to fetch candidate channels because the sub-select that _search would # generate leads psql query plan to take bad decisions. When candidate ids are explicitly # given it doesn't need to make (incorrect) guess, at the cost of one extra but fast query. # It is expected to return hundreds of channels, a thousand at most, which is acceptable. # A "join" would be ideal, but the ORM is currently not able to generate it from the domain. current_partner, current_guest = self.env["res.partner"]._get_current_persona() if current_guest: # sudo: discuss.channel - sudo for performance, just checking existence channels = current_guest.sudo().channel_ids elif current_partner: # sudo: discuss.channel - sudo for performance, just checking existence channels = current_partner.sudo().channel_ids else: channels = self.env["discuss.channel"] return [('id', "in" if is_in else "not in", channels.ids)] @api.depends('channel_member_ids') def _compute_member_count(self): read_group_res = self.env['discuss.channel.member']._read_group(domain=[('channel_id', 'in', self.ids)], groupby=['channel_id'], aggregates=['__count']) member_count_by_channel_id = {channel.id: count for channel, count in read_group_res} for channel in self: channel.member_count = member_count_by_channel_id.get(channel.id, 0) @api.depends('channel_type') def _compute_group_public_id(self): channels = self.filtered(lambda channel: channel.channel_type == 'channel') channels.filtered( lambda channel: not channel.parent_channel_id and not channel.group_public_id ).group_public_id = self.env.ref("base.group_user") (self - channels).group_public_id = None @api.depends('uuid') def _compute_invitation_url(self): for channel in self: channel.invitation_url = f"/chat/{channel.id}/{channel.uuid}" # ------------------------------------------------------------ # CRUD # ------------------------------------------------------------ @api.model_create_multi def create(self, vals_list): for vals in vals_list: # find partners to add from partner_ids partner_ids_cmd = vals.get('channel_partner_ids') or [] if any(cmd[0] not in (4, 6) for cmd in partner_ids_cmd): raise ValidationError(_('Invalid value when creating a channel with members, only 4 or 6 are allowed.')) partner_ids = [cmd[1] for cmd in partner_ids_cmd if cmd[0] == 4] partner_ids += [cmd[2] for cmd in partner_ids_cmd if cmd[0] == 6] # find partners to add from channel_member_ids membership_ids_cmd = vals.get('channel_member_ids', []) for cmd in membership_ids_cmd: if cmd[0] != 0: raise ValidationError(_('Invalid value when creating a channel with memberships, only 0 is allowed.')) for field_name in cmd[2]: if field_name not in ["partner_id", "guest_id", "unpin_dt", "last_interest_dt", "fold_state"]: raise ValidationError( _( "Invalid field ā€œ%(field_name)sā€ when creating a channel with members.", field_name=field_name, ) ) membership_pids = [cmd[2]['partner_id'] for cmd in membership_ids_cmd if cmd[0] == 0] partner_ids_to_add = partner_ids # always add current user to new channel to have right values for # is_pinned + ensure they have rights to see channel if not self.env.context.get('install_mode') and not self.env.user._is_public(): partner_ids_to_add = list(set(partner_ids + [self.env.user.partner_id.id])) vals['channel_member_ids'] = membership_ids_cmd + [ (0, 0, {'partner_id': pid}) for pid in partner_ids_to_add if pid not in membership_pids ] # clean vals vals.pop('channel_partner_ids', False) # Create channel and alias channels = super(Channel, self.with_context(mail_create_bypass_create_check=self.env['discuss.channel.member']._bypass_create_check, mail_create_nolog=True, mail_create_nosubscribe=True)).create(vals_list) # pop the mail_create_bypass_create_check key to avoid leaking it outside of create) channels = channels.with_context(mail_create_bypass_create_check=None) channels._subscribe_users_automatically() return channels @api.ondelete(at_uninstall=False) def _unlink_except_all_employee_channel(self): # Delete discuss.channel try: all_emp_group = self.env.ref('mail.channel_all_employees') except ValueError: all_emp_group = None if all_emp_group and all_emp_group in self: raise UserError(_('You cannot delete those groups, as the Whole Company group is required by other modules.')) for channel in self: channel._bus_send("discuss.channel/delete", {"id": channel.id}) def write(self, vals): if 'channel_type' in vals: failing_channels = self.filtered(lambda channel: channel.channel_type != vals.get('channel_type')) if failing_channels: raise UserError(_('Cannot change the channel type of: %(channel_names)s', channel_names=', '.join(failing_channels.mapped('name')))) if {"from_message_id", "parent_channel_id"} & set(vals): raise UserError( _( "Cannot change initial message nor parent channel of: %(channels)s.", channels=format_list(self.env, self.mapped("name")), ) ) old_vals = {channel: channel._channel_basic_info() for channel in self} result = super().write(vals) for channel in self: info = channel._channel_basic_info() diff = {} for key, value in info.items(): if value != old_vals[channel][key]: diff[key] = value if diff: channel._bus_send_store(channel, diff) if vals.get('group_ids'): self._subscribe_users_automatically() return result def init(self): self._cr.execute('SELECT indexname FROM pg_indexes WHERE indexname = %s', ('discuss_channel_member_seen_message_id_idx',)) if not self._cr.fetchone(): self._cr.execute('CREATE INDEX discuss_channel_member_seen_message_id_idx ON discuss_channel_member (channel_id,partner_id,seen_message_id)') # ------------------------------------------------------------ # MEMBERS MANAGEMENT # ------------------------------------------------------------ def _subscribe_users_automatically(self): new_members = self._subscribe_users_automatically_get_members() if new_members: to_create = [ {'channel_id': channel_id, 'partner_id': partner_id} for channel_id in new_members for partner_id in new_members[channel_id] ] # sudo: discuss.channel.member - adding member of other users based on channel auto-subscribe self.env['discuss.channel.member'].sudo().create(to_create) for channel in self: channel.group_ids._bus_send_store( channel, {**channel._channel_basic_info(), "is_pinned": True} ) def _subscribe_users_automatically_get_members(self): """ Return new members per channel ID """ return dict( (channel.id, ((channel.group_ids.users.partner_id.filtered(lambda p: p.active) - channel.channel_partner_ids).ids)) for channel in self ) def action_unfollow(self): self._action_unfollow(self.env.user.partner_id) def _action_unfollow(self, partner=None, guest=None): self.ensure_one() self.message_unsubscribe(partner.ids) custom_store = Store(self, {"is_pinned": False, "isLocallyPinned": False}) member = self.env["discuss.channel.member"].search( [ ("channel_id", "=", self.id), ("partner_id", "=", partner.id) if partner else ("guest_id", "=", guest.id), ] ) if not member: target = partner or guest target._bus_send_store(custom_store, notification_type="discuss.channel/leave") return notification = Markup('
%s
') % _( "left the channel" ) # sudo: mail.message - post as sudo since the user just unsubscribed from the channel member.channel_id.sudo().message_post( body=notification, subtype_xmlid="mail.mt_comment", author_id=partner.id ) # send custom store after message_post to avoid is_pinned reset to True member._bus_send_store(custom_store, notification_type="discuss.channel/leave") member.unlink() self._bus_send_store( self, { "channelMembers": Store.many(member, "DELETE", only_id=True), "memberCount": self.member_count, }, ) def add_members(self, partner_ids=None, guest_ids=None, invite_to_rtc_call=False, open_chat_window=False, post_joined_message=True): """ Adds the given partner_ids and guest_ids as member of self channels. """ current_partner, current_guest = self.env["res.partner"]._get_current_persona() partners = self.env['res.partner'].browse(partner_ids or []).exists() guests = self.env['mail.guest'].browse(guest_ids or []).exists() all_new_members = self.env["discuss.channel.member"] for channel in self: members_to_create = [] existing_members = self.env['discuss.channel.member'].search(expression.AND([ [('channel_id', '=', channel.id)], expression.OR([ [('partner_id', 'in', partners.ids)], [('guest_id', 'in', guests.ids)] ]) ])) members_to_create += [{ 'partner_id': partner.id, 'channel_id': channel.id, } for partner in partners - existing_members.partner_id] members_to_create += [{ 'guest_id': guest.id, 'channel_id': channel.id, } for guest in guests - existing_members.guest_id] new_members = self.env['discuss.channel.member'].create(members_to_create) all_new_members += new_members for member in new_members: payload = { "channel": { **member.channel_id._channel_basic_info(), "model": "discuss.channel", "is_pinned": True, }, "open_chat_window": open_chat_window, } if not member.is_self and not self.env.user._is_public(): payload["invited_by_user_id"] = self.env.user.id member._bus_send("discuss.channel/joined", payload) if post_joined_message: notification = ( _("joined the channel") if member.is_self else _("invited %s to the channel", member._get_html_link(for_persona=True)) ) member.channel_id.message_post( body=Markup('
%s
') % notification, message_type="notification", subtype_xmlid="mail.mt_comment", ) if new_members: channel._bus_send_store( Store(channel, {"memberCount": channel.member_count}).add(new_members) ) if existing_members and (current_partner or current_guest): # If the current user invited these members but they are already present, notify the current user about their existence as well. # In particular this fixes issues where the current user is not aware of its own member in the following case: # create channel from form view, and then join from discuss without refreshing the page. (current_partner or current_guest)._bus_send_store( Store(channel, {"memberCount": channel.member_count}).add(existing_members) ) if invite_to_rtc_call: for channel in self: current_channel_member = self.env['discuss.channel.member'].search([('channel_id', '=', channel.id), ('is_self', '=', True)]) # sudo: discuss.channel.rtc.session - reading rtc sessions of current user if current_channel_member and current_channel_member.sudo().rtc_session_ids: # sudo: discuss.channel.rtc.session - current user can invite new members in call current_channel_member.sudo()._rtc_invite_members(member_ids=new_members.ids) return all_new_members # ------------------------------------------------------------ # RTC # ------------------------------------------------------------ def _rtc_cancel_invitations(self, member_ids=None): """ Cancels the invitations of the RTC call from all invited members, if member_ids is provided, only the invitations of the specified members are canceled. :param list member_ids: list of the members ids from which the invitation has to be removed """ self.ensure_one() channel_member_domain = [ ('channel_id', '=', self.id), ('rtc_inviting_session_id', '!=', False), ] if member_ids: channel_member_domain = expression.AND([channel_member_domain, [('id', 'in', member_ids)]]) members = self.env['discuss.channel.member'].search(channel_member_domain) members.rtc_inviting_session_id = False members._bus_send_store(self, {"rtcInvitingSession": False}) if members: self._bus_send_store( self, { "invitedMembers": Store.many( members, "DELETE", fields={"channel": [], "persona": ["name", "im_status"]}, ), }, ) # ------------------------------------------------------------ # MAILING # ------------------------------------------------------------ def _notify_get_recipients(self, message, msg_vals, **kwargs): """ Override recipients computation as channel is not a standard mail.thread document. Indeed there are no followers on a channel. Instead of followers it has members that should be notified. :param message: see ``MailThread._notify_get_recipients()``; :param msg_vals: see ``MailThread._notify_get_recipients()``; :param kwargs: see ``MailThread._notify_get_recipients()``; :return recipients: structured data holding recipients data. See ``MailThread._notify_thread()`` for more details about its content and use; """ # get values from msg_vals or from message if msg_vals doen't exists message_type = msg_vals.get('message_type', 'comment') if msg_vals else message.message_type pids = msg_vals.get('partner_ids', []) if msg_vals else message.partner_ids.ids # notify only user input (comment, whatsapp messages or incoming / outgoing emails) if message_type not in ('comment', 'email', 'email_outgoing', 'whatsapp_message'): return [] recipients_data = [] author_id = msg_vals.get("author_id") or message.author_id.id if pids: email_from = tools.email_normalize(msg_vals.get('email_from') or message.email_from) self.env['res.partner'].flush_model(['active', 'email', 'partner_share']) self.env['res.users'].flush_model(['notification_type', 'partner_id']) sql_query = """ SELECT DISTINCT ON (partner.id) partner.id, partner.lang, partner.partner_share, users.id as uid, COALESCE(users.notification_type, 'email') as notif, COALESCE(users.share, FALSE) as ushare FROM res_partner partner LEFT JOIN res_users users on partner.id = users.partner_id WHERE partner.active IS TRUE AND partner.email != %s AND partner.id = ANY(%s) AND partner.id != ANY(%s)""" self.env.cr.execute( sql_query, (email_from or '', list(pids), [author_id] if author_id else [], ) ) for partner_id, lang, partner_share, uid, notif, ushare in self._cr.fetchall(): # ocn_client: will add partners to recipient recipient_data. more ocn notifications. We neeed to filter them maybe recipients_data.append({ 'active': True, 'id': partner_id, 'is_follower': False, 'groups': [], 'lang': lang, 'notif': notif, 'share': partner_share, 'type': 'user' if not partner_share and notif else 'customer', 'uid': uid, 'ushare': ushare, }) domain = expression.AND([ [("channel_id", "=", self.id)], [("partner_id", "!=", author_id)], [("partner_id.active", "=", True)], [("mute_until_dt", "=", False)], [("partner_id.user_ids.res_users_settings_ids.mute_until_dt", "=", False)], expression.OR([ [("channel_id.channel_type", "!=", "channel")], expression.AND([ [("channel_id.channel_type", "=", "channel")], expression.OR([ [("custom_notifications", "=", "all")], expression.AND([ [("custom_notifications", "=", False)], [("partner_id.user_ids.res_users_settings_ids.channel_notifications", "=", "all")], ]), expression.AND([ [("custom_notifications", "=", "mentions")], [("partner_id", "in", pids)], ]), expression.AND([ [("custom_notifications", "=", False)], [("partner_id.user_ids.res_users_settings_ids.channel_notifications", "=", False)], [("partner_id", "in", pids)], ]), ]), ]), ]), ]) # sudo: discuss.channel.member - read to get the members of the channel and res.users.settings of the partners members = self.env["discuss.channel.member"].sudo().search(domain) for member in members: recipients_data.append({ "active": True, "id": member.partner_id.id, "is_follower": False, "groups": [], "lang": member.partner_id.lang, "notif": "web_push", "share": member.partner_id.partner_share, "type": "customer", "uid": False, "ushare": False, }) return recipients_data def _notify_get_recipients_groups(self, message, model_description, msg_vals=None): """ All recipients of a message on a channel are considered as partners. This means they will receive a minimal email, without a link to access in the backend. Mailing lists should indeed send minimal emails to avoid the noise. """ groups = super()._notify_get_recipients_groups( message, model_description, msg_vals=msg_vals ) for (index, (group_name, _group_func, group_data)) in enumerate(groups): if group_name != 'customer': groups[index] = (group_name, lambda partner: False, group_data) return groups def _get_notify_valid_parameters(self): return super()._get_notify_valid_parameters() | {"silent"} def _notify_thread(self, message, msg_vals=False, **kwargs): # link message to channel rdata = super()._notify_thread(message, msg_vals=msg_vals, **kwargs) payload = {"data": Store(message).get_result(), "id": self.id} if temporary_id := self.env.context.get("temporary_id"): payload["temporary_id"] = temporary_id if kwargs.get("silent"): payload["silent"] = True self._bus_send_store(self, {"is_pinned": True}, subchannel="members") self._bus_send("discuss.channel/new_message", payload) return rdata def _notify_by_web_push_prepare_payload(self, message, msg_vals=False): payload = super()._notify_by_web_push_prepare_payload(message, msg_vals=msg_vals) payload['options']['data']['action'] = 'mail.action_discuss' record_name = msg_vals.get('record_name') if msg_vals and 'record_name' in msg_vals else message.record_name author_id = [msg_vals["author_id"]] if msg_vals and msg_vals.get("author_id") else message.author_id.ids author = self.env["res.partner"].browse(author_id) or self.env["mail.guest"].browse( msg_vals.get("author_guest_id", message.author_guest_id.id) ) if self.channel_type == 'chat': payload['title'] = author.name elif self.channel_type == 'channel': payload['title'] = "#%s - %s" % (record_name, author.name) elif self.channel_type == 'group': if not record_name: member_names = self.channel_member_ids.mapped(lambda m: m.partner_id.name if m.partner_id else m.guest_id.name) record_name = f"{', '.join(member_names[:-1])} and {member_names[-1]}" if len(member_names) > 1 else member_names[0] if member_names else "" payload['title'] = "%s - %s" % (record_name, author.name) else: payload['title'] = "#%s" % (record_name) return payload def _notify_thread_by_web_push(self, message, recipients_data, msg_vals=False, **kwargs): # only notify "web_push" recipients in discuss channels. # exclude "inbox" recipients in discuss channels as inbox and web push can be mutually exclusive. # the user can turn off the web push but receive notifs via inbox if they want to. super()._notify_thread_by_web_push(message, [r for r in recipients_data if r["notif"] == "web_push"], msg_vals=msg_vals, **kwargs) def _message_receive_bounce(self, email, partner): """ Override bounce management to unsubscribe bouncing addresses """ for p in partner: if p.message_bounce >= self.MAX_BOUNCE_LIMIT: self._action_unfollow(p) return super()._message_receive_bounce(email, partner) def _message_compute_author(self, author_id=None, email_from=None, raise_on_email=True): return super()._message_compute_author(author_id=author_id, email_from=email_from, raise_on_email=False) def _message_compute_parent_id(self, parent_id): # super() unravels the chain of parents to set parent_id as the first # ancestor. We don't want that in channel. if not parent_id: return parent_id return self.env['mail.message'].search( [('id', '=', parent_id), ('model', '=', self._name), ('res_id', '=', self.id) ]).id def _get_allowed_message_post_params(self): return super()._get_allowed_message_post_params() | {"special_mentions", "parent_id"} @api.returns('mail.message', lambda value: value.id) def message_post(self, *, message_type='notification', **kwargs): if (not self.env.user or self.env.user._is_public()) and self.is_member: # sudo: discuss.channel - guests don't have access for creating mail.message self = self.sudo() # sudo: discuss.channel - write to discuss.channel is not accessible for most users self.sudo().last_interest_dt = fields.Datetime.now() if "everyone" in kwargs.pop("special_mentions", []): kwargs["partner_ids"] = list( set(kwargs["partner_ids"] + self.channel_member_ids.partner_id.ids) ) # mail_post_autofollow=False is necessary to prevent adding followers # when using mentions in channels. Followers should not be added to # channels, and especially not automatically (because channel membership # should be managed with discuss.channel.member instead). # The current client code might be setting the key to True on sending # message but it is only useful when targeting customers in chatter. # This value should simply be set to False in channels no matter what. return super(Channel, self.with_context(mail_create_nosubscribe=True, mail_post_autofollow=False)).message_post(message_type=message_type, **kwargs) def _message_post_after_hook(self, message, msg_vals): """ Automatically set the message posted by the current user as seen for themselves. """ if (current_channel_member := self.env["discuss.channel.member"].search([ ("channel_id", "=", self.id), ("is_self", "=", True) ])) and message.is_current_user_or_guest_author: current_channel_member._set_last_seen_message(message, notify=False) current_channel_member._set_new_message_separator(message.id + 1, sync=True) return super()._message_post_after_hook(message, msg_vals) def _check_can_update_message_content(self, message): """ We don't call super in this override as we want to ignore the mail.thread behavior completely """ if not message.message_type == 'comment': raise UserError(_("Only messages type comment can have their content updated on model 'discuss.channel'")) def _create_attachments_for_post(self, values_list, extra_list): # Create voice metadata from meta information attachments = super()._create_attachments_for_post(values_list, extra_list) voice = attachments.env['ir.attachment'] # keep env, notably for potential sudo for attachment, (_cid, _name, _token, info) in zip(attachments, extra_list): if info.get('voice'): voice += attachment if voice: voice._set_voice_metadata() return attachments def _message_subscribe(self, partner_ids=None, subtype_ids=None, customer_ids=None): """ Do not allow follower subscription on channels. Only members are considered. """ raise UserError(_('Adding followers on channels is not possible. Consider adding members instead.')) # ------------------------------------------------------------ # BROADCAST # ------------------------------------------------------------ # Anonymous method def _broadcast(self, partner_ids): """ Broadcast the current channel header to the given partner ids :param partner_ids : the partner to notify """ for partner in self.env['res.partner'].browse(partner_ids): user_id = partner.user_ids and partner.user_ids[0] or False if user_id: user_channels = self.with_user(user_id).with_context( # sudo: res.company - context is required by ir.rules allowed_company_ids=user_id.sudo().company_ids.ids ) partner._bus_send_store(user_channels) # ------------------------------------------------------------ # INSTANT MESSAGING API # ------------------------------------------------------------ # A channel header should be broadcasted: # - when adding user to channel (only to the new added partners) # - when folding/minimizing a channel (only to the user making the action) # A message should be broadcasted: # - when a message is posted on a channel (to the channel, using _notify() method) # ------------------------------------------------------------ def set_message_pin(self, message_id, pinned): """ (Un)pin a message on the channel and send a notification to the members. :param message_id: id of the message to be pinned. :param pinned: whether the message should be pinned or unpinned. """ self.ensure_one() message_to_update = self.env['mail.message'].search([ ['id', '=', message_id], ['model', '=', 'discuss.channel'], ['res_id', '=', self.id], ['pinned_at', '=' if pinned else '!=', False] ]) if not message_to_update: return message_to_update.flush_recordset(['pinned_at']) # Use SQL because by calling write method, write_date is going to be updated, but we don't want pin/unpin # a message changes the write_date self.env.cr.execute("UPDATE mail_message SET pinned_at=%s WHERE id=%s", (fields.Datetime.now() if pinned else None, message_to_update.id)) message_to_update.invalidate_recordset(['pinned_at']) self._bus_send_store(message_to_update, {"pinned_at": message_to_update.pinned_at}) if pinned: notification_text = '''
%(user_pinned_a_message_to_this_channel)s %(see_all_pins)s
''' notification = Markup(notification_text) % { 'user_pinned_a_message_to_this_channel': Markup('%s') % ( message_id, _('%(user_name)s pinned a message to this channel.', user_name=self.env.user.display_name), ), 'see_all_pins': _('See all pinned messages.'), } self.message_post(body=notification, message_type="notification", subtype_xmlid="mail.mt_comment") def _find_or_create_member_for_self(self): self.ensure_one() domain = [("channel_id", "=", self.id), ("is_self", "=", True)] member = self.env["discuss.channel.member"].search(domain) if member: return member if not self.env.user._is_public(): return self.add_members(partner_ids=self.env.user.partner_id.ids) guest = self.env["mail.guest"]._get_guest_from_context() if guest: return self.add_members(guest_ids=guest.ids) return self.env["discuss.channel.member"] def _find_or_create_persona_for_channel(self, guest_name, timezone, country_code, post_joined_message=True): """ :param channel: channel to add the persona to :param guest_name: name of the persona :param post_joined_message: whether to post a message to the channel to notify that the persona joined :return tuple(partner, guest): """ self.ensure_one() guest = self.env["mail.guest"] member = self.env["discuss.channel.member"].search([("channel_id", "=", self.id), ("is_self", "=", True)]) if member: return member.partner_id, member.guest_id if not self.env.user._is_public(): self.add_members([self.env.user.partner_id.id], post_joined_message=post_joined_message) else: guest = self.env["mail.guest"]._get_guest_from_context() if not guest: guest = self.env["mail.guest"].create( { "country_id": self.env["res.country"].search([("code", "=", country_code)]).id, "lang": get_lang(self.env).code, "name": guest_name, "timezone": timezone, } ).sudo(False) guest._set_auth_cookie() self = self.with_context(guest=guest) self.add_members(guest_ids=guest.ids, post_joined_message=post_joined_message) return self.env.user.partner_id if not guest else self.env["res.partner"], guest @api.model def _get_channels_as_member(self): # 2 different queries because the 2 sub-queries together with OR are less efficient member_domain = [("channel_type", "in", ("channel", "group")), ("is_member", "=", True)] pinned_member_domain = [ ("channel_type", "not in", ("channel", "group")), ("channel_member_ids", "any", [("is_self", "=", True), ("is_pinned", "=", True)]), ] channels = self.env["discuss.channel"].search(member_domain) channels += self.env["discuss.channel"].search(pinned_member_domain) return channels def _channel_basic_info(self): self.ensure_one() data = self._read_format( [ "allow_public_upload", "channel_type", "create_uid", "description", "last_interest_dt", "name", "uuid", ], load=False, )[0] data["authorizedGroupFullName"] = self.group_public_id.full_name data["avatarCacheKey"] = self.avatar_cache_key data["defaultDisplayMode"] = self.default_display_mode data["group_based_subscription"] = bool(self.group_ids) data["memberCount"] = self.member_count return data def _to_store(self, store: Store): """Adds channel data to the given store.""" if not self: return [] # sudo: bus.bus: reading non-sensitive last id bus_last_id = self.env["bus.bus"].sudo()._bus_last_id() current_partner, current_guest = self.env["res.partner"]._get_current_persona() self.env['discuss.channel'].flush_model() self.env['discuss.channel.member'].flush_model() # Query instead of ORM for performance reasons: "LEFT JOIN" is more # efficient than "id IN" for the cross-table condition between channel # (for channel_type) and member (for other fields). self.env.cr.execute(""" SELECT discuss_channel_member.id FROM discuss_channel_member LEFT JOIN discuss_channel ON discuss_channel.id = discuss_channel_member.channel_id AND discuss_channel.channel_type != 'channel' WHERE discuss_channel_member.channel_id in %(channel_ids)s AND ( discuss_channel.id IS NOT NULL OR discuss_channel_member.rtc_inviting_session_id IS NOT NULL OR discuss_channel_member.partner_id = %(current_partner_id)s OR discuss_channel_member.guest_id = %(current_guest_id)s ) ORDER BY discuss_channel_member.id ASC """, {'channel_ids': tuple(self.ids), 'current_partner_id': current_partner.id or None, 'current_guest_id': current_guest.id or None}) all_needed_members = self.env['discuss.channel.member'].browse([m['id'] for m in self.env.cr.dictfetchall()]) Store(all_needed_members) # prefetch in batch members_by_channel = defaultdict(lambda: self.env['discuss.channel.member']) invited_members_by_channel = defaultdict(lambda: self.env['discuss.channel.member']) member_of_current_user_by_channel = defaultdict(lambda: self.env['discuss.channel.member']) for member in all_needed_members: members_by_channel[member.channel_id] += member if member.rtc_inviting_session_id: invited_members_by_channel[member.channel_id] += member if (current_partner and member.partner_id == current_partner) or (current_guest and member.guest_id == current_guest): member_of_current_user_by_channel[member.channel_id] = member for channel in self: member = member_of_current_user_by_channel.get(channel, self.env['discuss.channel.member']).with_prefetch([m.id for m in member_of_current_user_by_channel.values()]) info = channel._channel_basic_info() info["is_editable"] = channel.is_editable info["fetchChannelInfoState"] = "fetched" info["parent_channel_id"] = Store.one(channel.parent_channel_id) info["from_message_id"] = Store.one(channel.from_message_id) # find the channel member state if current_partner or current_guest: info['message_needaction_counter'] = channel.message_needaction_counter info["message_needaction_counter_bus_id"] = bus_last_id if member: store.add( member, extra_fields={ "last_interest_dt": True, "message_unread_counter": True, "message_unread_counter_bus_id": bus_last_id, "new_message_separator": True }, ) info['state'] = member.fold_state or 'closed' info['custom_notifications'] = member.custom_notifications info['mute_until_dt'] = fields.Datetime.to_string(member.mute_until_dt) info['custom_channel_name'] = member.custom_channel_name info['is_pinned'] = member.is_pinned if member.rtc_inviting_session_id: # sudo: discuss.channel.rtc.session - reading sessions of accessible channel is acceptable info["rtcInvitingSession"] = Store.one(member.rtc_inviting_session_id.sudo()) # add members info if channel.channel_type != 'channel': # avoid sending potentially a lot of members for big channels # exclude chat and other small channels from this optimization because they are # assumed to be smaller and it's important to know the member list for them store.add(members_by_channel[channel] - member) # add RTC sessions info invited_members = invited_members_by_channel[channel] info["invitedMembers"] = Store.many( invited_members, "ADD", fields={"channel": [], "persona": ["name", "im_status"]} ) # sudo: discuss.channel.rtc.session - reading sessions of accessible channel is acceptable info["rtcSessions"] = Store.many(channel.sudo().rtc_session_ids, "ADD", extra=True) store.add(channel, info) # User methods @api.model @api.returns('self', lambda channels: Store(channels).get_result()) def channel_get(self, partners_to, pin=True, force_open=False): """ Get the canonical private channel between some partners, create it if needed. To reuse an old channel (conversation), this one must be private, and contains only the given partners. :param partners_to : list of res.partner ids to add to the conversation :param pin : True if getting the channel should pin it for the current user :param force_open : True if getting the channel should open it for the current user :returns: channel_info of the created or existing channel :rtype: dict """ if self.env.user.partner_id.id not in partners_to: partners_to.append(self.env.user.partner_id.id) if len(partners_to) > 2: raise UserError(_("A chat should not be created with more than 2 persons. Create a group instead.")) # determine type according to the number of partner in the channel self.flush_model() self.env['discuss.channel.member'].flush_model() self.env.cr.execute(""" SELECT M.channel_id FROM discuss_channel C, discuss_channel_member M WHERE M.channel_id = C.id AND M.partner_id IN %s AND C.channel_type LIKE 'chat' AND NOT EXISTS ( SELECT 1 FROM discuss_channel_member M2 WHERE M2.channel_id = C.id AND M2.partner_id NOT IN %s ) GROUP BY M.channel_id HAVING ARRAY_AGG(DISTINCT M.partner_id ORDER BY M.partner_id) = %s LIMIT 1 """, (tuple(partners_to), tuple(partners_to), sorted(list(partners_to)),)) result = self.env.cr.dictfetchall() if result: # get the existing channel between the given partners channel = self.browse(result[0].get('channel_id')) # pin or open the channel for the current partner if pin or force_open: member = self.env['discuss.channel.member'].search([('partner_id', '=', self.env.user.partner_id.id), ('channel_id', '=', channel.id)]) vals = {'last_interest_dt': fields.Datetime.now()} if pin: vals['unpin_dt'] = False if force_open: vals['fold_state'] = "open" member.write(vals) channel._broadcast(self.env.user.partner_id.ids) else: # create a new one channel = self.create({ 'channel_member_ids': [ Command.create({ 'partner_id': partner_id, # only pin for the current user, so the chat does not show up for the correspondent until a message has been sent # manually set the last_interest_dt to make sure that it works well with the default last_interest_dt (datetime.now()) 'unpin_dt': False if partner_id == self.env.user.partner_id.id else fields.Datetime.now(), 'last_interest_dt': fields.Datetime.now() if partner_id == self.env.user.partner_id.id else fields.Datetime.now() - timedelta(seconds=30), }) for partner_id in partners_to ], 'channel_type': 'chat', 'name': ', '.join(self.env['res.partner'].browse(partners_to).mapped('name')), }) channel._broadcast(partners_to) return channel def channel_pin(self, pinned=False): self.ensure_one() member = self.env['discuss.channel.member'].search( [('partner_id', '=', self.env.user.partner_id.id), ('channel_id', '=', self.id), ('is_pinned', '!=', pinned)]) if member: member.write({'unpin_dt': False if pinned else fields.Datetime.now()}) if not pinned: self.env.user._bus_send("discuss.channel/unpin", {"id": self.id}) else: self.env.user._bus_send_store(self) def _types_allowing_seen_infos(self): """ Return the channel types which allow sending seen infos notification on the channel """ return ["chat", "group"] def channel_fetched(self): """ Broadcast the channel_fetched notification to channel members """ for channel in self: if not channel.message_ids.ids: continue # a bit not-modular but helps understanding code if channel.channel_type not in {'chat', 'whatsapp'}: continue last_message_id = channel.message_ids.ids[0] # zero is the index of the last message member = self.env['discuss.channel.member'].search([('channel_id', '=', channel.id), ('partner_id', '=', self.env.user.partner_id.id)], limit=1) if not member: # member not a part of the channel continue if member.fetched_message_id.id == last_message_id: # last message fetched by user is already up-to-date continue # Avoid serialization error when multiple tabs are opened. query = """ UPDATE discuss_channel_member SET fetched_message_id = %s WHERE id IN ( SELECT id FROM discuss_channel_member WHERE id = %s FOR NO KEY UPDATE SKIP LOCKED ) """ self.env.cr.execute(query, (last_message_id, member.id)) channel._bus_send( "discuss.channel.member/fetched", { "channel_id": channel.id, "id": member.id, "last_message_id": last_message_id, "partner_id": self.env.user.partner_id.id, }, ) def channel_set_custom_name(self, name): self.ensure_one() member = self.env['discuss.channel.member'].search([('partner_id', '=', self.env.user.partner_id.id), ('channel_id', '=', self.id)]) member.write({'custom_channel_name': name}) member._bus_send_store(self, {"custom_channel_name": name}) def channel_rename(self, name): self.ensure_one() self.write({'name': name}) def channel_change_description(self, description): self.ensure_one() self.write({'description': description}) def channel_join(self): """Shortcut to add the current user as member of self channels. Prefer calling add_members() directly when possible. """ self.add_members(self.env.user.partner_id.ids) @api.model @api.returns('self', lambda channels: Store(channels).get_result()) def channel_create(self, name, group_id): """ Create a channel and add the current partner, broadcast it (to make the user directly listen to it when polling) :param name : the name of the channel to create :param group_id : the group allowed to join the channel. :return dict : channel header """ # create the channel vals = { 'channel_type': 'channel', 'name': name, } new_channel = self.create(vals) group = self.env['res.groups'].search([('id', '=', group_id)]) if group_id else None new_channel.group_public_id = group.id if group else None notification = Markup('
%s
') % _("created this channel.") new_channel.message_post(body=notification, message_type="notification", subtype_xmlid="mail.mt_comment") self.env.user._bus_send_store(new_channel) return new_channel @api.model @api.returns('self', lambda channels: Store(channels).get_result()) def create_group(self, partners_to, default_display_mode=False, name=''): """ Creates a group channel. :param partners_to : list of res.partner ids to add to the conversation :param str default_display_mode: how the channel will be displayed by default :param str name: group name. default name is computed client side from the list of members if no name is set :returns: channel_info of the created channel :rtype: dict """ partners_to = set(partners_to) channel = self.create({ 'channel_member_ids': [Command.create({'partner_id': partner_id}) for partner_id in partners_to], 'channel_type': 'group', 'default_display_mode': default_display_mode, 'name': name, }) channel._broadcast(channel.channel_member_ids.partner_id.ids) return channel def _create_sub_channel(self, from_message_id=None, name=None): self.ensure_one() message = self.env["mail.message"] if from_message_id: message = self.env["mail.message"].search([("id", "=", from_message_id)]) sub_channel = self.create( { "channel_member_ids": [Command.create({"partner_id": self.env.user.partner_id.id})], "channel_type": "channel", "from_message_id": message.id, "name": name or (message.body.striptags()[:30] if message else _("New Thread")), "parent_channel_id": self.id, } ) self.env.user.partner_id._bus_send_store(Store(sub_channel)) notification = ( Markup('
%s
') % _( "%(user)s started a thread: %(goto)s%(thread_name)s%(goto_end)s. %(goto_all)sSee all threads%(goto_all_end)s." ) ) % { "user": self.env.user.display_name, "goto": Markup( "" ) % sub_channel.id, "goto_end": Markup(""), "goto_all": Markup(""), "goto_all_end": Markup(""), "thread_name": sub_channel.name, } self.message_post( body=notification, message_type="notification", subtype_xmlid="mail.mt_comment" ) return sub_channel @api.readonly @api.model def get_mention_suggestions(self, search, limit=8): """ Return 'limit'-first channels' id, name, channel_type and authorizedGroupFullName fields such that the name matches a 'search' string. Exclude channels of type chat (DM) and group. """ domain = expression.AND([ [('name', 'ilike', search)], [('channel_type', '=', 'channel')], [('channel_partner_ids', 'in', [self.env.user.partner_id.id])] ]) channels = self.search(domain, limit=limit) return [{ 'authorizedGroupFullName': channel.group_public_id.full_name, 'channel_type': channel.channel_type, 'model': "discuss.channel", 'id': channel.id, 'name': channel.name, 'parent_channel_id': { 'id': channel.parent_channel_id.id, 'model': 'discuss.channel' } if channel.parent_channel_id else False, } for channel in channels] def _get_last_messages(self): """ Return the last message for each of the given channels.""" if not self: return self.env["mail.message"] self.env['mail.message'].flush_model() self.env.cr.execute( """ SELECT last_message_id FROM discuss_channel LEFT JOIN LATERAL ( SELECT id FROM mail_message WHERE mail_message.model = 'discuss.channel' AND mail_message.res_id = discuss_channel.id ORDER BY id DESC LIMIT 1 ) AS t(last_message_id) ON TRUE WHERE discuss_channel.id IN %(ids)s GROUP BY discuss_channel.id, t.last_message_id ORDER BY discuss_channel.id """, {"ids": tuple(self.ids)}, ) return self.env["mail.message"].browse([mid for (mid,) in self.env.cr.fetchall() if mid]) def _load_more_members(self, known_member_ids): self.ensure_one() unknown_members = self.env['discuss.channel.member'].search( domain=[('id', 'not in', known_member_ids), ('channel_id', '=', self.id)], limit=100 ) count = self.env['discuss.channel.member'].search_count( domain=[('channel_id', '=', self.id)], ) return Store(unknown_members).add(self, {"memberCount": count}).get_result() # ------------------------------------------------------------ # COMMANDS # ------------------------------------------------------------ def execute_command_help(self, **kwargs): self.ensure_one() if self.channel_type == 'channel': msg = _( "You are in channel %(bold_start)s#%(channel_name)s%(bold_end)s.", bold_start=Markup(""), bold_end=Markup(""), channel_name=self.name, ) else: if members := self.channel_member_ids.filtered(lambda m: not m.is_self): msg = _( "You are in a private conversation with %(member_names)s.", member_names=html_escape( format_list(self.env, [f"%(member_{member.id})s" for member in members]) ) % { f"member_{member.id}": member._get_html_link(for_persona=True) for member in members }, ) else: msg = _("You are alone in a private conversation.") msg += self._execute_command_help_message_extra() self.env.user._bus_send_transient_message(self, msg) def _execute_command_help_message_extra(self): msg = _( "%(new_line)s" "%(new_line)sType %(bold_start)s@username%(bold_end)s to mention someone, and grab their attention." "%(new_line)sType %(bold_start)s#channel%(bold_end)s to mention a channel." "%(new_line)sType %(bold_start)s/command%(bold_end)s to execute a command." "%(new_line)sType %(bold_start)s:shortcut%(bold_end)s to insert a canned response in your message.", bold_start=Markup(""), bold_end=Markup(""), new_line=Markup("
"), ) return msg def execute_command_leave(self, **kwargs): if self.channel_type in ('channel', 'group'): self.action_unfollow() else: self.channel_pin(False) def execute_command_who(self, **kwargs): if all_other_members := self.channel_member_ids.filtered(lambda m: not m.is_self): members = all_other_members[:30] list_params = [f"%(member_{member.id})s" for member in members] if len(all_other_members) != len(members): list_params.append(_("more")) else: list_params.append(_("you")) msg = _( "Users in this channel: %(members)s.", members=html_escape(format_list(self.env, list_params)) % { f"member_{member.id}": member._get_html_link(for_persona=True) for member in members }, ) else: msg = _("You are alone in this channel.") self.env.user._bus_send_transient_message(self, msg)