Added Docker support

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:2.7
+
+COPY . /securitybot
+
+USER securitybot
+
+env PYTHONPATH $PYTHONPATH:/securitybot
+
+RUN apt-get update && \
+    apt-get install -y mysql-client && \
+    pip install -r /securitybot/requirements.txt
+
+WORKDIR /securitybot
+
+ENTRYPOINT ["/securitybot/docker_entrypoint.sh"]

README.md

@@ -33,6 +33,44 @@ If the following were all generated successfully, Securitybot should be up and r
 To test it, message the bot user it's assigned to and say `hi`.
 To test the process of dealing with an alert, message `test` to test the bot.
 
+#### Environment variable
+To prevent having to modify the scripts, you can set the following environment variables to configure the bot:
+
+|Variable|Setting|
+|--------|-------|
+|SLACK_API_TOKEN|Slack API Token|
+|REPORTING_CHANNEL|Slack Channel ID|
+|DUO_INTEGRATION_KEY|Duo Integration Key|
+|DUO_SECRET_KEY|Duo Secret Key|
+|DUO_ENDPOINT|Duo Endpoint|
+|DB_HOST|MySQL Hostname|
+|DB_USER|MySQL Username|
+|DB_PASS|MySQL Password|
+|DB_NAME|MySQL Database name|
+
+#### Docker
+
+Dockerfile is included to generate a Docker Image to run the bot.  Entrypoint script will wait for database startup and initialize database if it does not already exist.  Entrypoint takes one of two arguments:
+
+* bot: Starts the main bot
+* frontend: starts the frontend and API server
+
+Run configuration will be based on the environment variables above.
+
+Example:
+
+Bot:
+```
+docker build --tag securitybot
+docker run -e DB_NAME=securitybot DB_NAME=securitybot -e DB_USER=root -e DB_HOST=127.0.0.1 -e DB_PASS=password -e SLACK_API_TOKEN=<your api token> -e DUO_INTEGRATION_KEY=<your integration key> -e DUO_SECRET_KEY=<your secret key> -e DUO_ENDPOINT=<your endpoint> securitybot bot
+```
+
+Frontend:
+```
+docker build --tag securitybot
+docker run -p 8888:8888 -e DB_NAME=securitybot -e DB_USER=root -e DB_HOST=127.0.0.1 -e DB_PASS=password securitybot frontend
+```
+
 ## Architecture
 Securitybot was designed to be as modular as possible.
 This means that it's possible to easily swap out chat systems, 2FA providers, and alerting data sources.

docker_entrypoint.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+set -e
+
+
+end=$((SECONDS + 120))
+
+# Wait max of 2 minutes for database to come up
+until mysql --user $DB_USER --password=$DB_PASS --host=$DB_HOST -e"quit" || [ $SECONDS -ge $end ]; do echo "Waiting for DB"; sleep 2; done;
+
+# Check if database is initialized
+if ! mysql --user $DB_USER --password=$DB_PASS --host=$DB_HOST -e"use $DB_NAME"; then
+  echo "Creating database"
+  mysql --user $DB_USER --password=$DB_PASS --host=$DB_HOST -e"create database $DB_NAME"
+  echo "Populating database"
+  python /securitybot/util/db_up.py
+fi
+
+
+if [ "$1" = 'bot' ]; then
+  echo "Starting bot"
+  shift
+  exec python /securitybot/main.py "$@"
+elif [ "$1" = 'frontend' ]; then
+  echo "Starting frontend"
+  shift
+  exec python /securitybot/frontend/securitybot_frontend.py "$@"
+fi
+
+exec "$@"

main.py

@@ -6,14 +6,16 @@
 from securitybot.tasker.sql_tasker import SQLTasker
 from securitybot.auth.duo import DuoAuth
 from securitybot.sql import init_sql
+from os import getenv
+
 import duo_client
 
 CONFIG = {}
-SLACK_KEY = 'slack_api_token'
-DUO_INTEGRATION = 'duo_integration_key'
-DUO_SECRET = 'duo_secret_key'
-DUO_ENDPOINT = 'duo_endpoint'
-REPORTING_CHANNEL = 'some_slack_channel_id'
+SLACK_KEY = getenv('SLACK_API_TOKEN', 'slack_api_token')
+DUO_INTEGRATION = getenv('DUO_INTEGRATION_KEY', 'duo_integration_key')
+DUO_SECRET = getenv('DUO_SECRET_KEY', 'duo_secret_key')
+DUO_ENDPOINT = getenv('DUO_ENDPOINT', 'duo_endpoint')
+REPORTING_CHANNEL = getenv('REPORTING_CHANNEL', 'some_slack_channel_id')
 ICON_URL = 'https://dl.dropboxusercontent.com/s/t01pwfrqzbz3gzu/securitybot.png'
 
 def init():

securitybot/sql.py

@@ -4,8 +4,15 @@
 import MySQLdb
 import logging
 
+from os import getenv
 from typing import Any, Sequence
 
+DB_HOST = getenv('DB_HOST', 'localhost')
+DB_USER = getenv('DB_USER', 'root')
+DB_PASS = getenv('DB_PASS', '')
+DB_NAME = getenv('DB_NAME', 'securitybot')
+
+
 class SQLEngine(object):
     # Whether the singleton has been instantiated
     _host = None # type: str
@@ -90,4 +97,4 @@ class SQLEngineException(Exception):
 def init_sql():
     # type: () -> None
     '''Initializes SQL.'''
-    SQLEngine('localhost', 'root', '', 'securitybot')
+    SQLEngine(DB_HOST, DB_USER, DB_PASS, DB_NAME)

util/db_up.py

@@ -1,16 +1,18 @@
 #!/usr/bin/env python
 import MySQLdb
 import sys
+from os import getenv
 
 # DB CONFIG GOES HERE
-host = 'localhost'
-user = 'root'
-passwd= ''
+host = getenv('DB_HOST', 'localhost')
+user = getenv('DB_USER', 'root')
+passwd=getenv('DB_PASS', '')
+dbname=getenv('DB_NAME', 'securitybot')
 
 db = MySQLdb.connect(host=host,
                      user=user,
                      passwd=passwd,
-                     db='securitybot')
+                     db=dbname)
 
 cur = db.cursor()