TorBot GUI

.gitignore

@@ -6,6 +6,10 @@ modules/.pytest_cache
 modules/.ropeproject/
 modules/lib/*.so
 
+# JS Ignores
+node_modules
+build
+
 # Tests Ignores
 tests/.pytest_cache
 tests/*.pyc

.hound.yml

@@ -1,2 +1,5 @@
 flake8:
-  enabled: true
+  enabled: true
+eslint:
+  enbaled: true
+  config_file: ./ui/.eslintrc.json

modules/wsserver.py

@@ -0,0 +1,12 @@
+import asyncio
+import websockets
+
+async def handle_msg(websocket, path):
+    msg = await websocket.recv()
+    print(path)
+    print(msg)
+
+def startWSServer():
+    start_server = websockets.serve(handle_msg, 'localhost', '8080')
+    asyncio.get_event_loop().run_until_complete(start_server)
+    asyncio.get_event_loop().run_forever()

requirements.txt

@@ -8,3 +8,4 @@ pyinstaller==3.4.0
 ete3==3.1.1
 PyQt5==5.11.3
 validators==0.12.2
+websockets==7.0.0

torBot.py

@@ -14,6 +14,7 @@
 from modules.updater import updateTor
 from modules.savefile import saveJson
 from modules.info import execute_all
+from modules.wsserver import startWSServer
 
 # GLOBAL CONSTS
 LOCALHOST = "127.0.0.1"
@@ -118,6 +119,8 @@ def get_args():
                         help="Visualizes tree of data gathered.")
     parser.add_argument("-d", "--download", action="store_true",
                         help="Downloads tree of data gathered.")
+    parser.add_argument("-g", "--gui", action="store_true",
+                        help="Display GUI for TorBot.")
     return parser.parse_args()
 
 
@@ -126,7 +129,8 @@ def main():
     TorBot's Core
     """
     args = get_args()
-    connect(args.ip, args.port)
+    if not args.gui:
+        connect(args.ip, args.port)
     try:
         node = LinkNode(args.url, tld=args.extension)
     except (ValueError, HTTPError, ConnectionError) as err:
@@ -164,6 +168,8 @@ def main():
             tree = LinkTree(node, tld=node.tld)
             file_name = str(input("File Name (.pdf/.png/.svg): "))
             tree.save(file_name)
+        elif args.gui:
+            startWSServer()
         else:
             LinkIO.display_children(node)
             if args.save:

ui/.eslintrc.json

@@ -0,0 +1,33 @@
+{
+    "env": {
+        "node": true,
+        "browser": true,
+        "commonjs": true,
+        "es6": true
+    },
+    "extends": ["plugin:react/recommended","eslint:recommended"],
+    "parserOptions": {
+        "ecmaFeatures": {
+            "jsx": true
+        },
+        "ecmaVersion": 2018,
+        "sourceType": "module"
+    },
+    "plugins": [
+        "react"
+    ],
+    "rules": {
+        "linebreak-style": [
+            "error",
+            "unix"
+        ],
+        "quotes": [
+            "error",
+            "single"
+        ],
+        "semi": [
+            "error",
+            "always"
+        ]
+    }
+}

ui/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "torbot",
+  "version": "0.1.0",
+  "private": true,
+  "dependencies": {
+    "electron-reload-webpack-plugin": "^2.0.4",
+    "extract-text-webpack-plugin": "^4.0.0-beta.0",
+    "react": "^16.6.0",
+    "react-dom": "^16.6.0",
+    "react-scripts": "2.0.5"
+  },
+  "scripts": {
+    "start": "webpack --mode development --watch",
+    "build": "webpack --mode production",
+    "test": "react-scripts test",
+    "eject": "react-scripts eject",
+    "electron": "electron ."
+  },
+  "homepage": "./",
+  "main": "src/electron-main.js",
+  "browserslist": [
+    ">0.2%",
+    "not dead",
+    "not ie <= 11",
+    "not op_mini all"
+  ],
+  "devDependencies": {
+    "@babel/preset-env": "^7.1.0",
+    "babel-eslint": "^10.0.1",
+    "babel-loader": "^8.0.4",
+    "css-loader": "^1.0.1",
+    "electron": "^3.0.8",
+    "electron-connect": "^0.6.3",
+    "electron-reload": "^1.2.5",
+    "eslint": "^5.9.0",
+    "eslint-config-prettier": "^3.3.0",
+    "eslint-plugin-prettier": "^3.0.0",
+    "eslint-plugin-react": "^7.11.1",
+    "html-webpack-plugin": "^3.2.0",
+    "mini-css-extract-plugin": "^0.4.4",
+    "prettier": "^1.15.2",
+    "webpack-cli": "^3.1.2"
+  }
+}

ui/src/app.js

@@ -0,0 +1,11 @@
+import React from 'react';
+import Home from './components/home';
+import './app.css';
+
+class App extends React.Component {
+    render() {
+        return <Home/>;
+    }
+}
+
+export default App;

ui/src/components/home.js

@@ -0,0 +1,37 @@
+import React from 'react';
+
+
+class Home extends React.Component {
+
+    constructor(props) {
+        super(props);
+        this.state = {url: ''};
+        this.onSubmit = this.onSubmit.bind(this);
+        this.onUrlChange = this.onUrlChange.bind(this);
+    }
+
+    onUrlChange(event) {
+        if (event.key === 'Enter') this.onSubmit();
+        this.setState({url: event.target.value});
+    }
+
+    onSubmit(event) {
+        let scope = this;
+        let ws = new WebSocket('ws://localhost:8080')
+        ws.onopen = function() {
+            ws.send(scope.state.url);
+        };
+    }
+
+    render() {
+        return (
+            <React.Fragment>
+                <h1 align='center'>TorBot</h1>
+                <input onKeyDown={this.onUrlChange} type='text'/>
+                <button onClick={this.onSubmit}></button>
+            </React.Fragment>
+        );
+    }
+}
+
+export default Home;

ui/src/electron-main.js

@@ -0,0 +1,60 @@
+// Modules to control application life and create native browser window
+const { app, BrowserWindow } = require('electron');
+
+// Keep a global reference of the window object, if you don't, the window will
+// be closed automatically when the JavaScript object is garbage collected.
+let mainWindow;
+
+// Used in configuring path to load html from React
+const path = require('path');
+const url = require('url');
+
+function createWindow () {
+    let browserOptions = {width: 800, height: 600};
+    // Create the browser window.
+    mainWindow = new BrowserWindow(browserOptions);
+
+    // Configuring locaiton for HTML outputted after React
+    const startURL = url.format({
+      pathname: path.join(__dirname, '../build/index.html'),
+      protocol: 'file:', 
+      slashes: true
+    });
+
+    // Loads html
+    mainWindow.loadURL(startURL);
+
+    // Open the DevTools
+    mainWindow.webContents.openDevTools();
+
+    // Emitted when the window is closed.
+    mainWindow.on('closed', function () {
+    // Dereference the window object, usually you would store windows
+    // in an array if your app supports multi windows, this is the time
+    // when you should delete the corresponding element.
+    mainWindow = null;
+    });
+}
+
+// This method will be called when Electron has finished
+// initialization and is ready to create browser windows.
+// Some APIs can only be used after this event occurs.
+app.on('ready', createWindow);
+
+
+// Quit when all windows are closed.
+app.on('window-all-closed', function () {
+    // On OS X it is common for applications and their menu bar
+    // to stay active until the user quits explicitly with Cmd + Q
+    if (process.platform !== 'darwin') {
+        app.quit();
+    }
+});
+
+app.on('activate', function () {
+    // On OS X it's common to re-create a window in the app when the
+    // dock icon is clicked and there are no other windows open.
+    if (mainWindow === null) {
+        createWindow();
+    }
+});

ui/src/global.css

@@ -0,0 +1,14 @@
+body {
+  margin: 0;
+  padding: 0;
+  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "Roboto", "Oxygen",
+    "Ubuntu", "Cantarell", "Fira Sans", "Droid Sans", "Helvetica Neue",
+    sans-serif;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
+}
+
+code {
+  font-family: source-code-pro, Menlo, Monaco, Consolas, "Courier New",
+    monospace;
+}

ui/src/index.html

@@ -0,0 +1,15 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
+    <meta name="theme-color" content="#000000">
+    <title>TorBot</title>
+  </head>
+  <body>
+    <noscript>
+      You need to enable JavaScript to run this app.
+    </noscript>
+    <div id="root"></div>
+  </body>
+</html>

ui/src/renderer.js

@@ -0,0 +1,8 @@
+import React from 'react';
+import ReactDOM from 'react-dom';
+import './global.css';
+import App from './app';
+
+
+ReactDOM.render(<App/>, document.getElementById('root'));
+

ui/webpack.config.js

@@ -0,0 +1,73 @@
+const MiniCssExtractTextPlugin = require('mini-css-extract-plugin');
+const createElectronReloadWebpackPlugin = require('electron-reload-webpack-plugin');
+const ElectronReloadWebpackPlugin = createElectronReloadWebpackPlugin({
+    path: './',
+    logLevel: 0
+});
+const HtmlWebpackPlugin = require('html-webpack-plugin');
+
+const path = require('path');
+module.exports = {
+
+    watch: true,
+
+    target: 'electron-main',
+
+    entry: ['./src/renderer.js'],
+
+    output: {
+        path: __dirname + '/build',
+        filename: 'bundle.js'
+    },
+
+    module: {
+        rules: [
+            {
+                test: /\.jsx?$/,
+                loader: 'babel-loader',
+                options: {
+                    presets: ['@babel/react']
+                }
+            },
+            {
+                test: /\.css$/,
+                use: [
+                {
+                    loader: MiniCssExtractTextPlugin.loader
+                },
+                'css-loader'
+                ]
+            },
+            {
+                test: /\.(png|jpg|gif|svg)$/,
+                loader: 'file-loader',
+                query: {
+                    name: '[name].[ext]?[hash]'
+                }
+            }
+        ]
+    },
+
+    resolve: {
+        alias: {
+            images: path.resolve(__dirname, 'images')
+        }
+    },
+
+    plugins: [
+        new MiniCssExtractTextPlugin({
+            filename: 'bundle.css',
+        }),
+        new ElectronReloadWebpackPlugin(),
+        new HtmlWebpackPlugin({
+            title: 'TorBot',
+            template: 'src/index.html'
+
+        })
+    ],
+
+    resolve: {
+      extensions: ['.js', '.json', '.jsx']
+    }
+
+};