From 83836686398ac66a964c70cc7529e6777aa76cec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?N=C3=A9munaire?= <nemunaire@pomail.fr>
Date: Wed, 30 May 2012 12:22:28 +0200
Subject: [PATCH] Module whereis working in V3

---
 whereis.py => modules/whereis.py | 74 +++++---------------------------
 modules/whereis.xml              | 28 ++++++++++++
 2 files changed, 38 insertions(+), 64 deletions(-)
 rename whereis.py => modules/whereis.py (78%)
 create mode 100644 modules/whereis.xml

diff --git a/whereis.py b/modules/whereis.py
similarity index 78%
rename from whereis.py
rename to modules/whereis.py
index 549bd83..32ea8bb 100644
--- a/whereis.py
+++ b/modules/whereis.py
@@ -10,9 +10,10 @@ from datetime import datetime
 from datetime import date
 from datetime import timedelta
 from urllib.parse import unquote
-from xml.dom.minidom import parse
-from xml.dom.minidom import parseString
-from xml.dom.minidom import getDOMImplementation
+
+from module_state import ModuleState
+
+nemubotversion = 3.0
 
 NS_SERVER = 'ns-server.epita.fr'
 NS_PORT = 4242
@@ -22,7 +23,8 @@ search = list()
 
 class UpdatedStorage:
   def __init__(self):
-    sock = connect_to_ns(NS_SERVER, NS_PORT)
+    sock = connect_to_ns(CONF.getNode("server")["url"],
+                         CONF.getNode("server").getInt("port"))
     self.users = dict()
     if sock != None:
       for l in list_users(sock):
@@ -50,56 +52,10 @@ class User(object):
 
     @property
     def sm(self):
-        if self.ip.startswith('10.200'):
-          ip = self.ip.split('.')
-          if 20 < int(ip[2]) < 30:
-            return 'SM02'
-          else:
-            return 'SM' + self.ip.split('.')[2]
-        elif self.ip.startswith('10.242'):
-            return 'bocal'
-        elif self.ip.startswith('10.247'):
-            return 'pasteur'
-        elif self.ip.startswith('10.248'):
-            return 'srlab'
-        elif self.ip.startswith('10.249'):
-            return 'midlab'
-        elif self.ip.startswith('10.250'):
-            return 'cisco'
-        elif self.ip.startswith('10.41.'):
-            return 'wifi'
-        elif self.ip.startswith('10.223.3.'):
-            return 'épimac'
-        elif self.ip.startswith('10.223.4.'):
-            return 'wcube'
-        elif self.ip.startswith('10.223.5.'):
-            return 'cycom'
-        elif self.ip.startswith('10.223.6.'):
-            return 'epitv'
-        elif self.ip.startswith('10.223.7.'):
-            return 'prologin'
-        elif self.ip.startswith('10.223'):
-            return 'assos'
-        elif self.ip.startswith('10.226.7.'):
-            return 'gistre'
-        elif self.ip.startswith('10.224.1.'):
-            return 'astek'
-        elif self.ip.startswith('10.224.2.'):
-            return 'acu'
-        elif self.ip.startswith('10.224.4.'):
-            return 'lse'
-        elif self.ip.startswith('10.224.8.'):
-            return 'eip'
-        elif self.ip.startswith('10.224.16.'):
-            return 'evolutek'
-        elif self.ip.startswith('10.224.18.'):
-            return 'mslab'
-        elif self.ip.startswith('10.2.'):
-            return 'adm'
-        elif self.ip.startswith('10.10.'):
-            return 'vpn'
-        else:
-            return None
+      for sm in CONF.getNodes("sm"):
+        if self.ip.startswith(sm["ip"]):
+            return sm["name"]
+      return None
 
     @property
     def poste(self):
@@ -141,15 +97,6 @@ def list_users(sock):
     return buf.split('\n')[:-2]
 
 
-def load_module(datas_path):
-  """Load this module"""
-  return
-
-def save_module():
-  """Save the module state"""
-  return
-
-
 def help_tiny ():
   """Line inserted in the response to the command !help"""
   return "Find a user on the PIE"
@@ -207,7 +154,6 @@ def whoison(msg):
         continue
       else:
         found = list()
-        print (len(datas.users))
         for userC in datas.users:
           for user in datas.users[userC]:
             if (msg.cmd[0] == "whoison" and (user.ip[:len(pc)] == pc or user.location.lower() == pc)) or (msg.cmd[0] == "whoisin" and user.sm == pc):
diff --git a/modules/whereis.xml b/modules/whereis.xml
new file mode 100644
index 0000000..90b2c2f
--- /dev/null
+++ b/modules/whereis.xml
@@ -0,0 +1,28 @@
+<?xml version="1.0" ?>
+<nemubotmodule name="whereis">
+  <server url="ns-server.epita.fr" port="4242" />
+  <sm name="adm" ip="10.2." />
+  <sm name="vpn" ip="10.10." />
+  <sm name="wifi" ip="10.41" />
+  <sm name="sm02" ip="10.200.2" />
+  <sm name="sm14" ip="10.200.14." />
+  <sm name="sm15" ip="10.200.15." />
+  <sm name="épimac" ip="10.223.3." />
+  <sm name="wcube" ip="10.223.4." />
+  <sm name="cycom" ip="10.223.5" />
+  <sm name="eptv" ip="10.223.6" />
+  <sm name="prologin" ip="10.223.7" />
+  <sm name="assos" ip="10.223" />
+  <sm name="gistre" ip="10.226.7." />
+  <sm name="astek" ip="10.224.1." />
+  <sm name="acu" ip="10.224.2." />
+  <sm name="lse" ip="10.224.4." />
+  <sm name="eip" ip="10.224.8." />
+  <sm name="evolutek" ip="10.224.16." />
+  <sm name="mslab" ip="10.224.18." />
+  <sm name="bocal" ip="10.242" />
+  <sm name="pasteur" ip="10.247" />
+  <sm name="srlab" ip="10.248" />
+  <sm name="midlab" ip="10.249" />
+  <sm name="cisco" ip="10.250" />
+</nemubotmodule>
\ No newline at end of file