Merge pull request #4 from mightyBroccoli/dev

+ added !xep feature

3 files changed, 94 insertions, 4 deletions

diff --git a/classes/functions.py b/classes/functions.py
index 08d6146..a8ed356 100644
--- a/classes/functions.py
+++ b/classes/functions.py
@@ -93,7 +93,7 @@ class ContactInfo:
 
 # class handeling XMPPError exeptions
 class HandleError:
-	def __init__(self, error, msg, key, target):
+	def __init__(self, error, msg, key, target="target missing"):
 		self.error = error
 		self.message = msg
 		self.key = key
diff --git a/classes/strings.py b/classes/strings.py
index ade0520..6866a31 100644
--- a/classes/strings.py
+++ b/classes/strings.py
@@ -12,7 +12,8 @@ class StaticAnswers:
 			'help':		'!help -- display this text',
 			'version':	'!version domain.tld  -- receive XMPP server version',
 			'uptime':	'!uptime domain.tld -- receive XMPP server uptime',
-			'contact':	'!contact domain.tld -- receive XMPP server contact address info'}
+			'contact':	'!contact domain.tld -- receive XMPP server contact address info',
+			'xep': 		'!xep XEP Number -- recieve information about the specified XEP'}
 		self.possible_answers = {
 			'1': 'I heard that, %s.',
 			'2': 'I am sorry for that %s.',
@@ -22,8 +23,10 @@ class StaticAnswers:
 			'2': 'not a valid target'
 		}
 		self.keywords = {
-			"keywords": ["!help", "!uptime", "!version", "!contact"],
-			"no_arg_keywords": ["!help"]
+			"keywords": ["!help", "!uptime", "!version", "!contact", "!xep"],
+			"domain_keywords": ["!uptime", "!version", "!contact"],
+			"no_arg_keywords": ["!help"],
+			"number_keywords": ["!xep"]
 		}
 
 	def keys(self, arg="", keyword='keywords'):
diff --git a/classes/xep.py b/classes/xep.py
new file mode 100644
index 0000000..9e4f61f
--- /dev/null
+++ b/classes/xep.py
@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+import requests
+import defusedxml.ElementTree as ET
+
+
+class XEPRequest:
+	def __init__(self, msg, xepnumber):
+		"""
+		class which requests the header of the referenced xep
+		:param xepnumber: number int or str to request the xep for
+		"""
+		self.message_type = msg['type']
+		self.muc_nick = msg['mucnick']
+
+		self.reqxep = str(xepnumber)
+		self.xeplist = None
+		self.acceptedxeps = list()
+
+	def req_xeplist(self):
+		"""
+		query and save the current xep list to reduce network bandwidth
+		"""
+		try:
+			with open(".etag") as file:
+				local_etag = file.read()
+		except FileNotFoundError:
+			local_etag = ""
+
+		with requests.Session() as s:
+			s.headers.update({'Accept': 'application/xml'})
+			head = s.head("https://xmpp.org/extensions/xeplist.xml")
+			etag = head.headers['etag']
+
+			if local_etag == etag:
+				with open("xeplist.xml", "r") as file:
+					self.xeplist = ET.fromstring(file.read())
+			else:
+				r = s.get("https://xmpp.org/extensions/xeplist.xml")
+				r.encoding = 'utf-8'
+				local_etag = head.headers['etag']
+
+				with open("xeplist.xml", "w") as file:
+					file.write(r.content.decode())
+					self.xeplist = ET.fromstring(r.content.decode())
+
+				with open('.etag', 'w') as string:
+					string.write(local_etag)
+
+		# populate xep comparison list
+		for xep in self.xeplist.findall(".//*[@accepted='true']/number"):
+			self.acceptedxeps.append(xep.text)
+
+	def get(self):
+		"""
+		function to query the xep entry if xepnumber is present in xeplist
+		:return: nicely formatted xep header information
+		"""
+		# check if xeplist is accurate
+		self.req_xeplist()
+
+		result = list()
+		# if requested number is inside acceptedxeps continou
+		if self.reqxep in self.acceptedxeps:
+			searchstring = ".//*[@accepted='true']/[number='%s']" % self.reqxep
+
+			for item in self.xeplist.findall(searchstring):
+				for x in range(1,5):
+					result.append(item[x].tag + " : " + item[x].text)
+
+		else:
+			if self.message_type == "groupchat":
+				result.append(self.muc_nick + " : " + "XEP-" + str(self.reqxep) + " : is not available.")
+			else:
+				result.append("XEP-" + str(self.reqxep) + " : is not available.")
+
+		return result
+
+	def format(self):
+		reply = self.get()
+		if self.message_type == "groupchat":
+			text = "%s: " % self.muc_nick
+			reply[0] = text + reply[0]
+
+		text = '\n'.join(reply)
+
+		return text