From 3500b7ad7c7420a7486dd85e1353607ec8908f45 Mon Sep 17 00:00:00 2001
From: Philipp Balzarek <p.balzarek@googlemail.com>
Date: Thu, 5 Jul 2012 14:24:46 +0200
Subject: [PATCH] add Network.Xmpp.Xep.ServiceDiscovery

---
 pontarius-xmpp.cabal                        |   1 +
 source/Network/Xmpp/Xep/ServiceDiscovery.hs | 137 ++++++++++++++++++++
 2 files changed, 138 insertions(+)
 create mode 100644 source/Network/Xmpp/Xep/ServiceDiscovery.hs

diff --git a/pontarius-xmpp.cabal b/pontarius-xmpp.cabal
index 0238254..81e168f 100644
--- a/pontarius-xmpp.cabal
+++ b/pontarius-xmpp.cabal
@@ -70,6 +70,7 @@ Library
                  , Network.Xmpp.Stream
                  , Network.Xmpp.TLS
                  , Network.Xmpp.Types
+                 , Network.Xmpp.Xep.ServiceDiscovery
   Other-modules:
                  Network.Xmpp.Jid
                  , Network.Xmpp.Concurrent.Types
diff --git a/source/Network/Xmpp/Xep/ServiceDiscovery.hs b/source/Network/Xmpp/Xep/ServiceDiscovery.hs
new file mode 100644
index 0000000..2c2eb18
--- /dev/null
+++ b/source/Network/Xmpp/Xep/ServiceDiscovery.hs
@@ -0,0 +1,137 @@
+{-# LANGUAGE NoMonomorphismRestriction #-}
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE TupleSections #-}
+
+-- XEP 0030: Service Discovery (disco)
+
+module Network.Xmpp.Xep.ServiceDiscovery
+  ( QueryInfoResult(..)
+  , Identity(..)
+  , queryInfo
+  , Item
+  , queryItems
+  , DiscoError(..)
+  )
+  where
+
+import           Control.Applicative((<$>))
+import           Control.Monad.IO.Class
+import           Control.Monad.Error
+
+import qualified Data.Text as Text
+import           Data.XML.Pickle
+import           Data.XML.Types
+
+import           Network.Xmpp.Pickle
+import           Network.Xmpp
+
+data DiscoError = DiscoNoQueryElement
+                | DiscoIQError IQError
+                | DiscoXMLError Element UnpickleError
+
+                deriving (Show)
+
+instance Error DiscoError
+
+data Identity = Ident { iCategory :: Text.Text
+                      , iName     :: Maybe Text.Text
+                      , iType     :: Text.Text
+                      , iLang     :: Maybe LangTag
+                      } deriving Show
+
+data QueryInfoResult = QIR { qiNode       :: Maybe Text.Text
+                           , qiIdentities :: [Identity]
+                           , qiFeatures :: [Text.Text]
+                           } deriving Show
+
+discoInfoNS :: Text.Text
+discoInfoNS = "http://jabber.org/protocol/disco#info"
+
+infoN :: Text.Text -> Name
+infoN name = (Name name (Just discoInfoNS) Nothing)
+
+xpIdentities = xpWrap (map $(\(cat, n, tp, lang) -> Ident cat n tp lang) . fst)
+               (map $ \(Ident cat n tp lang) -> ((cat, n, tp, lang),())) $
+              xpElems (infoN "identity")
+               (xp4Tuple
+                  (xpAttr        "category" xpText)
+                  (xpAttrImplied "name"     xpText)
+                  (xpAttr        "type"     xpText)
+                  xpLangTag
+               )
+               xpUnit
+
+xpFeatures :: PU [Node] [Text.Text]
+xpFeatures = xpWrap (map fst) (map (,())) $
+              xpElems (infoN "feature")
+                (xpAttr "var" xpText)
+                xpUnit
+
+xpQueryInfo = xpWrap (\(nd, (feats, ids)) -> QIR nd ids feats)
+                     (\(QIR nd ids feats) -> (nd, (feats, ids))) $
+              xpElem (infoN "query")
+                (xpAttrImplied "node" xpText)
+                (xp2Tuple
+                     xpFeatures
+                     xpIdentities
+                     )
+
+-- | Query an entity for it's identity and features
+queryInfo ::  Jid -- ^ Entity to query
+          -> Maybe Text.Text -- ^ Node
+          -> Xmpp (Either DiscoError QueryInfoResult)
+queryInfo to node = do
+    res <- sendIQ' (Just to) Get Nothing queryBody
+    return $ case res of
+        Left e -> Left $ DiscoIQError e
+        Right r -> case iqResultPayload r of
+            Nothing -> Left DiscoNoQueryElement
+            Just p -> case unpickleElem xpQueryInfo p of
+                Left e -> Left $ DiscoXMLError p e
+                Right r -> Right r
+  where
+    queryBody = pickleElem xpQueryInfo (QIR node [] [])
+
+--
+-- Items
+--
+
+data Item = Item { itemJid :: Jid
+                 , itemName :: Maybe Text.Text
+                 , itemNode :: Maybe Text.Text
+                 } deriving Show
+
+discoItemsNS :: Text.Text
+discoItemsNS = "http://jabber.org/protocol/disco#items"
+
+itemsN :: Text.Text -> Name
+itemsN n = Name n (Just discoItemsNS) Nothing
+
+xpItem = xpWrap (\(jid, name, node) -> Item jid name node)
+                (\(Item jid name node) -> (jid, name, node)) $
+         xpElemAttrs (itemsN "item")
+           (xp3Tuple
+              (xpAttr "jid" xpPrim)
+              (xpAttrImplied "name" xpText)
+              (xpAttrImplied "node" xpText))
+
+
+xpQueryItems = xpElem (itemsN "query")
+                 (xpAttrImplied "node" xpText)
+                 (xpAll xpItem)
+
+-- | Query an entity for Items of a node
+queryItems :: Jid -- ^ Entity to query
+           -> Maybe Text.Text -- ^ Node
+           -> Xmpp (Either DiscoError (Maybe Text.Text, [Item]))
+queryItems to node = do
+    res <- sendIQ' (Just to) Get Nothing queryBody
+    return $ case res of
+        Left e -> Left $ DiscoIQError e
+        Right r -> case iqResultPayload r of
+            Nothing -> Left DiscoNoQueryElement
+            Just p -> case unpickleElem xpQueryItems p of
+                Left e -> Left $ DiscoXMLError p e
+                Right r -> Right r
+  where
+    queryBody = pickleElem xpQueryItems (node, [])