From 4360585272f9c32913bee2b4b203e9588717ddbb Mon Sep 17 00:00:00 2001
From: Filipp Lepalaan <f@230.to>
Date: Thu, 13 Feb 2014 18:43:39 +0200
Subject: Better?

---
 app.py | 43 ++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 38 insertions(+), 5 deletions(-)

(limited to 'app.py')

diff --git a/app.py b/app.py
index 79b3fa9..cca136c 100644
--- a/app.py
+++ b/app.py
@@ -1,5 +1,6 @@
 import re
 import logging
+import sqlite3
 import subprocess
 import tornado.ioloop
 import tornado.web
@@ -14,17 +15,49 @@ class MainHandler(tornado.web.RequestHandler):
         
 
 class ScanHandler(tornado.web.RequestHandler):
+    def initialize(self):
+        conn = sqlite3.connect(":memory:")
+        self.cursor = conn.cursor()
+        self.cursor.execute('CREATE TABLE notes kw text value text')
+        
+    def get(self):
+        commands = ('nmap', 'help', '?', '!',)
+
+        out = {'results': list()}
+        cmd = self.get_argument('cmd', '')
+
+        if cmd not in commands or cmd == 'help':
+            out['results'].append({'raw': 'Available commands:'})
+            out['results'].append({'raw': '* nmap - scan your IP'})
+            out['results'].append({'raw': '* ?<keyword> - query keyword'})
+            out['results'].append({'raw': '* !<keyword> definition - learn <keyword>'})
+            out['results'].append({'raw': '* !!<keyword> - forget <keyword>'})
+            out['results'].append({'raw': '* help - shows this help message'})
+
+        if cmd == 'nmap':
+            logging.debug('SCANNING!')
+            result = subprocess.check_output(['nmap', self.request.remote_ip])
+            for r in re.finditer(r'(\d+/[a-z]{3})\s([a-z]+)\s+(.+)', result):
+                t = r.groups()
+                out['results'].append({
+                    'port': t[0],
+                    'state': t[1],
+                    'service': t[2],
+                    'raw': "    ".join(r.groups(0))
+                })
+
+        self.write(out)
+
+
+class HelpHandler(tornado.web.RequestHandler):
     def get(self):
-        logging.debug('SCANNING!')
-        result = subprocess.check_output(['nmap', self.request.remote_ip])
-        for r in re.finditer(r'(\d+/[a-z]{3})\s([a-z]+)\s+(.+)', result):
-            t = r.groups()
-            self.write({'port': t[0], 'state': t[1], 'service': t[2]})
+        out = {'results': list()}
 
 
 application = tornado.web.Application([
     (r'/', MainHandler),
     (r'/scan/', ScanHandler),
+    (r'/help/', HelpHandler),
 ])
 
 if __name__ == '__main__':
-- 
cgit v1.2.3