module Network.Protocol.XMPP.Client
( runClient
, bindJID
) where
import Control.Monad ((>=>))
import Control.Monad.Error (throwError)
import Control.Monad.Trans (liftIO)
import Data.ByteString (ByteString)
import qualified Data.Text.Lazy as T
import Network (connectTo)
import qualified System.IO as IO
import qualified Network.Protocol.XMPP.Client.Authentication as A
import qualified Network.Protocol.XMPP.Connections as C
import qualified Network.Protocol.XMPP.Client.Features as F
import qualified Network.Protocol.XMPP.Handle as H
import qualified Network.Protocol.XMPP.JID as J
import qualified Network.Protocol.XMPP.Monad as M
import qualified Network.Protocol.XMPP.XML as X
import Network.Protocol.XMPP.ErrorT
import Network.Protocol.XMPP.Stanza
runClient :: C.Server
-> J.JID
-> T.Text
-> T.Text
-> M.XMPP a
-> IO (Either M.Error a)
runClient server jid username password xmpp = do
let C.Server sjid host port = server
rawHandle <- connectTo host port
IO.hSetBuffering rawHandle IO.NoBuffering
let handle = H.PlainHandle rawHandle
M.startXMPP handle "jabber:client" $ do
features <- newStream sjid
tryTLS sjid features $ \tlsFeatures -> do
let mechanisms = authenticationMechanisms tlsFeatures
A.authenticate mechanisms jid sjid username password
M.restartXMPP Nothing (newStream sjid >> xmpp)
newStream :: J.JID -> M.XMPP [F.Feature]
newStream jid = do
M.putBytes $ C.xmlHeader "jabber:client" jid
M.readEvents C.startOfStream
F.parseFeatures `fmap` M.getElement
tryTLS :: J.JID -> [F.Feature] -> ([F.Feature] -> M.XMPP a) -> M.XMPP a
tryTLS sjid features m
| not (streamSupportsTLS features) = m features
| otherwise = do
M.putElement xmlStartTLS
M.getElement
h <- M.getHandle
eitherTLS <- liftIO $ runErrorT $ H.startTLS h
case eitherTLS of
Left err -> throwError $ M.TransportError err
Right tls -> M.restartXMPP (Just tls) $ newStream sjid >>= m
authenticationMechanisms :: [F.Feature] -> [ByteString]
authenticationMechanisms = step where
step [] = []
step (f:fs) = case f of
(F.FeatureSASL ms) -> ms
_ -> step fs
bindJID :: J.JID -> M.XMPP J.JID
bindJID jid = do
M.putStanza . bindStanza . J.jidResource $ jid
bindResult <- M.getStanza
let getJID =
X.elementChildren
>=> X.isNamed (X.Name "jid" (Just "urn:ietf:params:xml:ns:xmpp-bind") Nothing)
>=> X.elementNodes
>=> X.isContent
>=> return . X.contentText
let maybeJID = do
iq <- case bindResult of
ReceivedIQ x -> Just x
_ -> Nothing
payload <- iqPayload iq
case getJID payload of
[] -> Nothing
(str:_) -> J.parseJID str
returnedJID <- case maybeJID of
Just x -> return x
Nothing -> throwError $ M.InvalidBindResult bindResult
M.putStanza sessionStanza
M.getStanza
M.putStanza $ emptyPresence PresenceAvailable
M.getStanza
return returnedJID
bindStanza :: Maybe J.Resource -> IQ
bindStanza resource = (emptyIQ IQSet) { iqPayload = Just payload } where
payload = X.nselement "urn:ietf:params:xml:ns:xmpp-bind" "bind" [] requested
requested = case fmap J.strResource resource of
Nothing -> []
Just x -> [X.NodeElement $ X.element "resource" []
[X.NodeContent $ X.ContentText x]]
sessionStanza :: IQ
sessionStanza = (emptyIQ IQSet) { iqPayload = Just payload } where
payload = X.nselement "urn:ietf:params:xml:ns:xmpp-session" "session" [] []
streamSupportsTLS :: [F.Feature] -> Bool
streamSupportsTLS = any isStartTLS where
isStartTLS (F.FeatureStartTLS _) = True
isStartTLS _ = False
xmlStartTLS :: X.Element
xmlStartTLS = X.nselement "urn:ietf:params:xml:ns:xmpp-tls" "starttls" [] []