{-# LANGUAGE TupleSections #-}
{-# OPTIONS_GHC -Wno-unrecognised-pragmas #-}

-- This file is part of the Wire Server implementation.
--
-- Copyright (C) 2022 Wire Swiss GmbH <opensource@wire.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 <https://www.gnu.org/licenses/>.

module Galley.API.Push
  ( -- * Message pushes
    MessagePush (..),

    -- * Executing message pushes
    BotMap,
    newMessagePush,
    runMessagePush,
  )
where

import Control.Lens (set)
import Data.Id
import Data.Json.Util
import Data.List1 qualified as List1
import Data.Map qualified as Map
import Data.Qualified
import Galley.Data.Services
import Galley.Effects.ExternalAccess
import Gundeck.Types.Push (RecipientClients (RecipientClientsSome), Route (..))
import Imports
import Polysemy
import Polysemy.TinyLog
import System.Logger.Class qualified as Log
import Wire.API.Event.Conversation
import Wire.API.Message
import Wire.NotificationSubsystem

data MessagePush
  = MessagePush (Maybe ConnId) MessageMetadata [Recipient] [BotMember] Event

type BotMap = Map UserId BotMember

class ToRecipient a where
  toRecipient :: a -> Recipient

instance ToRecipient (UserId, ClientId) where
  toRecipient :: (UserId, ClientId) -> Recipient
toRecipient (UserId
u, ClientId
c) = UserId -> RecipientClients -> Recipient
Recipient UserId
u (List1 ClientId -> RecipientClients
RecipientClientsSome (ClientId -> List1 ClientId
forall a. a -> List1 a
List1.singleton ClientId
c))

instance ToRecipient Recipient where
  toRecipient :: Recipient -> Recipient
toRecipient = Recipient -> Recipient
forall a. a -> a
id

newMessagePush ::
  (ToRecipient r) =>
  BotMap ->
  Maybe ConnId ->
  MessageMetadata ->
  [r] ->
  Event ->
  MessagePush
newMessagePush :: forall r.
ToRecipient r =>
BotMap
-> Maybe ConnId -> MessageMetadata -> [r] -> Event -> MessagePush
newMessagePush BotMap
botMap Maybe ConnId
mconn MessageMetadata
mm [r]
userOrBots Event
event =
  let toPair :: Recipient -> ([Recipient], [BotMember])
toPair Recipient
r = case UserId -> BotMap -> Maybe BotMember
forall k a. Ord k => k -> Map k a -> Maybe a
Map.lookup (Recipient -> UserId
recipientUserId Recipient
r) BotMap
botMap of
        Just BotMember
botMember -> ([], [BotMember
botMember])
        Maybe BotMember
Nothing -> ([Recipient
r], [])
      ([Recipient]
recipients, [BotMember]
botMembers) = (r -> ([Recipient], [BotMember]))
-> [r] -> ([Recipient], [BotMember])
forall m a. Monoid m => (a -> m) -> [a] -> m
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap (Recipient -> ([Recipient], [BotMember])
toPair (Recipient -> ([Recipient], [BotMember]))
-> (r -> Recipient) -> r -> ([Recipient], [BotMember])
forall b c a. (b -> c) -> (a -> b) -> a -> c
. r -> Recipient
forall a. ToRecipient a => a -> Recipient
toRecipient) [r]
userOrBots
   in Maybe ConnId
-> MessageMetadata
-> [Recipient]
-> [BotMember]
-> Event
-> MessagePush
MessagePush Maybe ConnId
mconn MessageMetadata
mm [Recipient]
recipients [BotMember]
botMembers Event
event

runMessagePush ::
  forall x r.
  ( Member ExternalAccess r,
    Member TinyLog r,
    Member NotificationSubsystem r
  ) =>
  Local x ->
  Maybe (Qualified ConvId) ->
  MessagePush ->
  Sem r ()
runMessagePush :: forall x (r :: EffectRow).
(Member ExternalAccess r, Member TinyLog r,
 Member NotificationSubsystem r) =>
Local x -> Maybe (Qualified ConvId) -> MessagePush -> Sem r ()
runMessagePush Local x
loc Maybe (Qualified ConvId)
mqcnv mp :: MessagePush
mp@(MessagePush Maybe ConnId
_ MessageMetadata
_ [Recipient]
_ [BotMember]
botMembers Event
event) = do
  [Push] -> Sem r ()
forall (r :: EffectRow).
Member NotificationSubsystem r =>
[Push] -> Sem r ()
pushNotifications ([Push] -> Sem r ()) -> [Push] -> Sem r ()
forall a b. (a -> b) -> a -> b
$ Maybe Push -> [Push]
forall a. Maybe a -> [a]
maybeToList (Maybe Push -> [Push]) -> Maybe Push -> [Push]
forall a b. (a -> b) -> a -> b
$ MessagePush -> Maybe Push
toPush MessagePush
mp
  Maybe (Qualified ConvId)
-> (Qualified ConvId -> Sem r ()) -> Sem r ()
forall (t :: * -> *) (f :: * -> *) a b.
(Foldable t, Applicative f) =>
t a -> (a -> f b) -> f ()
for_ Maybe (Qualified ConvId)
mqcnv ((Qualified ConvId -> Sem r ()) -> Sem r ())
-> (Qualified ConvId -> Sem r ()) -> Sem r ()
forall a b. (a -> b) -> a -> b
$ \Qualified ConvId
qcnv ->
    if Local x -> Domain
forall (t :: QTag) a. QualifiedWithTag t a -> Domain
tDomain Local x
loc Domain -> Domain -> Bool
forall a. Eq a => a -> a -> Bool
/= Qualified ConvId -> Domain
forall a. Qualified a -> Domain
qDomain Qualified ConvId
qcnv
      then Bool -> Sem r () -> Sem r ()
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
unless ([BotMember] -> Bool
forall a. [a] -> Bool
forall (t :: * -> *) a. Foldable t => t a -> Bool
null [BotMember]
botMembers) (Sem r () -> Sem r ()) -> Sem r () -> Sem r ()
forall a b. (a -> b) -> a -> b
$ do
        (Msg -> Msg) -> Sem r ()
forall msg (r :: EffectRow).
Member (Logger msg) r =>
msg -> Sem r ()
warn ((Msg -> Msg) -> Sem r ()) -> (Msg -> Msg) -> Sem r ()
forall a b. (a -> b) -> a -> b
$ ByteString -> Msg -> Msg
forall a. ToBytes a => a -> Msg -> Msg
Log.msg (ByteString
"Ignoring messages for local bots in a remote conversation" :: ByteString) (Msg -> Msg) -> (Msg -> Msg) -> Msg -> Msg
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ByteString -> String -> Msg -> Msg
forall a. ToBytes a => ByteString -> a -> Msg -> Msg
Log.field ByteString
"conversation" (Qualified ConvId -> String
forall a. Show a => a -> String
show Qualified ConvId
qcnv)
      else ConvId -> [(BotMember, Event)] -> Sem r ()
forall (r :: EffectRow) (f :: * -> *).
(Member ExternalAccess r, Foldable f) =>
ConvId -> f (BotMember, Event) -> Sem r ()
deliverAndDeleteAsync (Qualified ConvId -> ConvId
forall a. Qualified a -> a
qUnqualified Qualified ConvId
qcnv) ((BotMember -> (BotMember, Event))
-> [BotMember] -> [(BotMember, Event)]
forall a b. (a -> b) -> [a] -> [b]
map (,Event
event) [BotMember]
botMembers)

toPush :: MessagePush -> Maybe Push
toPush :: MessagePush -> Maybe Push
toPush (MessagePush Maybe ConnId
mconn MessageMetadata
mm [Recipient]
rs [BotMember]
_ Event
event) =
  let usr :: UserId
usr = Qualified UserId -> UserId
forall a. Qualified a -> a
qUnqualified (Event -> Qualified UserId
evtFrom Event
event)
   in Maybe UserId -> Object -> [Recipient] -> Maybe Push
newPush (UserId -> Maybe UserId
forall a. a -> Maybe a
Just UserId
usr) (Event -> Object
forall a. ToJSONObject a => a -> Object
toJSONObject Event
event) [Recipient]
rs
        Maybe Push -> (Push -> Push) -> Maybe Push
forall (f :: * -> *) a b. Functor f => f a -> (a -> b) -> f b
<&> ASetter Push Push (Maybe ConnId) (Maybe ConnId)
-> Maybe ConnId -> Push -> Push
forall s t a b. ASetter s t a b -> b -> s -> t
set ASetter Push Push (Maybe ConnId) (Maybe ConnId)
Lens' Push (Maybe ConnId)
pushConn Maybe ConnId
mconn
        (Push -> Push) -> (Push -> Push) -> Push -> Push
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ASetter Push Push (Maybe Priority) (Maybe Priority)
-> Maybe Priority -> Push -> Push
forall s t a b. ASetter s t a b -> b -> s -> t
set ASetter Push Push (Maybe Priority) (Maybe Priority)
Lens' Push (Maybe Priority)
pushNativePriority (MessageMetadata -> Maybe Priority
mmNativePriority MessageMetadata
mm)
        (Push -> Push) -> (Push -> Push) -> Push -> Push
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ASetter Push Push Route Route -> Route -> Push -> Push
forall s t a b. ASetter s t a b -> b -> s -> t
set ASetter Push Push Route Route
Lens' Push Route
pushRoute (Route -> Route -> Bool -> Route
forall a. a -> a -> Bool -> a
bool Route
RouteDirect Route
RouteAny (MessageMetadata -> Bool
mmNativePush MessageMetadata
mm))
        (Push -> Push) -> (Push -> Push) -> Push -> Push
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ASetter Push Push Bool Bool -> Bool -> Push -> Push
forall s t a b. ASetter s t a b -> b -> s -> t
set ASetter Push Push Bool Bool
Lens' Push Bool
pushTransient (MessageMetadata -> Bool
mmTransient MessageMetadata
mm)