Change resolver to nslookup

Author: Nitr4x

README.md

@@ -43,7 +43,7 @@ baseplate_proxy.add_child(LegoNettools)
 The nettools Lego could be triggered by various prefix.
 
 - [x] Whois
-- [x] Resolver
+- [x] Nslookup
 - [x] DNS
 - [ ] Trace
 - [x] Ping
@@ -62,9 +62,9 @@ The nettools Lego could be triggered by various prefix.
 
 ` !ping {target}`
 
-#### Resolver
+#### Nslookup
 
-` !resolver {--host | --nslookup} {target}`
+` !nslookup {target}`
 
 #### Qualys
 

legos/tools/Resolver.py renamed to legos/tools/Nslookup.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 
+import re
 import socket
 
 from Legobot.Utilities import Utilities as utils
@@ -9,7 +10,7 @@
 __copyright__ = "Copyright 2017, Legobot"
 
 
-class Resolver(ToolScheme):
+class Nslookup(ToolScheme):
     """This class allows resolving the IP address from a domain name as well as
        the reverse process.
     """
@@ -18,23 +19,16 @@ def __init__(self, args):
         if utils.isNotEmpty(args):
             super().__init__(args)
 
-            self.fncs = {
-                'host': self._host,
-                'nslookup': self._nslookup
-            }
-
     def run(self):
         if utils.isNotEmpty(self.target):
-            if len(self.cmds) > 0:
-                for cmd in self.cmds:
-                    try:
-                        return self.fncs[cmd]()
-                    except KeyError:
-                        return 'Command unknown: ' + cmd
+            if re.match(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$", self.target):
+                return self._getIP()
+            else:
+                return self._getDomainName()
         else:
             return self.getHelp()
 
-    def _host(self):
+    def _getDomainName(self):
         """Retrieve the IP address corresponding to a domain name.
 
         Args:
@@ -48,7 +42,7 @@ def _host(self):
         except:
             return 'Host cannot be resolved'
 
-    def _nslookup(self):
+    def _getIP(self):
         """Retrieve the domain name corresponding to an IP address.
 
         Args:
@@ -63,4 +57,4 @@ def _nslookup(self):
             return 'IP address cannot be resolved'
 
     def getHelp(self):
-        return " !resolver {--host | --nslookup} {target}"
+        return " !nslookup {target}"