refactor XMPP client

1 files changed, 56 insertions, 5 deletions

diff --git a/Assistant/XMPP.hs b/Assistant/XMPP.hs
index 43bf4ac75..05bc94fa3 100644
--- a/Assistant/XMPP.hs
+++ b/Assistant/XMPP.hs
@@ -7,7 +7,9 @@
 
 module Assistant.XMPP where
 
-import Common.Annex
+import Assistant.Common
+import Annex.UUID
+import Assistant.Pairing
 
 import Network.Protocol.XMPP
 import qualified Data.Text as T
@@ -21,24 +23,44 @@ gitAnnexPresence tag = (emptyPresence PresenceAvailable)
 	extendedAway = Element (Name (T.pack "show") Nothing Nothing) []
 		[NodeContent $ ContentText $ T.pack "xa"]
 
-{- Does a presence contain a gitp-annex tag? -}
+{- Does a presence contain a git-annex tag? -}
 isGitAnnexPresence :: Presence -> Bool
-isGitAnnexPresence p = any matchingtag (presencePayloads p)
-	where
-		matchingtag t = elementName t == gitAnnexTagName
+isGitAnnexPresence p = any isGitAnnexTag (presencePayloads p)
 
 {- Name of a git-annex tag, in our own XML namespace.
  - (Not using a namespace URL to avoid unnecessary bloat.) -}
 gitAnnexTagName :: Name
 gitAnnexTagName  = Name (T.pack "git-annex") (Just $ T.pack "git-annex") Nothing
 
+isGitAnnexTag :: Element -> Bool
+isGitAnnexTag t = elementName t == gitAnnexTagName
+
 {- A git-annex tag, to let other clients know we're a git-annex client too. -}
 gitAnnexSignature :: Element
 gitAnnexSignature = Element gitAnnexTagName [] []
 
+queryAttr :: Name
+queryAttr = Name (T.pack "query") Nothing Nothing
+
 pushAttr :: Name
 pushAttr = Name (T.pack "push") Nothing Nothing
 
+pairingAttr :: Name
+pairingAttr = Name (T.pack "pairing") Nothing Nothing
+
+isAttr :: Name -> (Name, [Content]) -> Bool
+isAttr attr (k, _) = k == attr
+
+getAttr :: Name -> [(Name, [Content])] -> Maybe String
+getAttr wantattr attrs = content <$> headMaybe (filter (isAttr wantattr) attrs)
+  where
+	content (_name, cs) = T.unpack $ T.concat $ map unpack cs
+	unpack (ContentText t) = t
+	unpack (ContentEntity t) = t
+
+uuidAttr :: Name
+uuidAttr = Name (T.pack "uuid") Nothing Nothing
+
 uuidSep :: T.Text
 uuidSep = T.pack ","
 
@@ -61,3 +83,32 @@ decodePushNotification (Element name attrs _nodes)
 	ispush (k, _) = k == pushAttr
 	fromContent (ContentText t) = t
 	fromContent (ContentEntity t) = t
+
+{- A request for other git-annex clients to send presence. -}
+presenceQuery :: Presence
+presenceQuery = gitAnnexPresence $ Element gitAnnexTagName
+	[ (queryAttr, [ContentText T.empty]) ]
+	[]
+
+isPresenceQuery :: Presence -> Bool
+isPresenceQuery p = case filter isGitAnnexTag (presencePayloads p) of
+	[] -> False
+	((Element _name attrs _nodes):_) -> any (isAttr queryAttr) attrs
+
+{- A notification about a stage of pairing. -}
+pairingNotification :: PairStage -> Annex Presence
+pairingNotification pairstage = do
+	u <- getUUID
+	return $ gitAnnexPresence $ Element gitAnnexTagName
+		[ (pairingAttr, [ContentText $ T.pack $ show pairstage])
+		, (uuidAttr, [ContentText $ T.pack $ fromUUID u]) 
+		]
+		[]
+
+isPairingNotification :: Presence -> Maybe (PairStage, UUID)
+isPairingNotification p = case filter isGitAnnexTag (presencePayloads p) of
+	[] -> Nothing
+	((Element _name attrs _nodes):_) ->
+		(,)
+			<$> (readish =<< getAttr pairingAttr attrs)
+			<*> (toUUID <$> getAttr uuidAttr attrs)