Added dockerfile

Author: jordan-wright

Diff for: .dockerignore

@@ -0,0 +1,3 @@
+Dockerfile
+.env
+.vscode

Diff for: Dockerfile

@@ -0,0 +1,34 @@
+FROM python:3
+
+RUN apt-get update \
+    && apt-get install -y \
+    git \
+    wget \
+    sqlite3 \
+    supervisor \
+    unzip \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Download and install Gophish
+WORKDIR /usr/src/gophish
+RUN wget -O gophish.zip https://getgophish.com/releases/latest/linux/64
+RUN unzip gophish.zip
+RUN rm gophish.zip
+RUN chmod +x gophish
+
+# Expose the admin port to the host
+RUN sed -i 's/127.0.0.1/0.0.0.0/g' config.json
+
+WORKDIR /usr/src/gophish-demo/
+COPY . .
+RUN mv docker/* .
+RUN chmod +x run_demo.sh
+
+# Install depenedencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Setup the supervisor
+RUN mv supervisord.conf /etc/supervisor/conf.d/supervisord.conf
+
+CMD ["/usr/bin/supervisord"]

Diff for: create_demo.py

@@ -198,6 +198,7 @@ def open_email(campaign, result, user_agent=DEFAULT_USER_AGENT):
             create the event
     """
     print('Opening email for {}'.format(result.email))
+    sys.stdout.flush()
     return requests.get(
         '{}/track'.format(campaign.url),
         params={'rid': result.id},
@@ -221,6 +222,7 @@ def click_link(campaign, result, user_agent=DEFAULT_USER_AGENT):
             create the event
     """
     print('Clicking link for {}'.format(result.email))
+    sys.stdout.flush()
     return requests.get(
         campaign.url,
         params={'rid': result.id},
@@ -244,6 +246,7 @@ def submit_data(campaign, result, user_agent=DEFAULT_USER_AGENT):
             create the event
     """
     print('Submitting data for {}'.format(result.email))
+    sys.stdout.flush()
     return requests.post(
         campaign.url,
         params={'rid': result.id},
@@ -271,6 +274,7 @@ def report_email(campaign, result, user_agent=DEFAULT_USER_AGENT):
             create the event
     """
     print('Reporting email for {}'.format(result.email))
+    sys.stdout.flush()
     return requests.get(
         '{}/report'.format(campaign.url),
         params={'rid': result.id},
@@ -331,22 +335,27 @@ def main():
     api = Gophish(api_key=args.api_key, host=args.api_url, verify=False)
 
     # Generate our groups
-    print('Generating groups...')
+    print('Generating Groups...')
+    sys.stdout.flush()
     group_names = generate_groups(
         api, num_groups=args.num_groups, num_members=args.num_members)
     groups = [Group(name=group) for group in group_names]
 
     print('Generating SMTP')
+    sys.stdout.flush()
     smtp = generate_sending_profile(api, '{}:{}'.format(
         smtp.hostname, smtp.port))
 
     print('Generating Template')
+    sys.stdout.flush()
     template = generate_template(api)
 
     print('Generating Landing Page')
+    sys.stdout.flush()
     landing_page = generate_landing_page(api)
 
     print('Generating Campaigns')
+    sys.stdout.flush()
     campaign = Campaign(
         name='Demo Campaign',
         groups=groups,
@@ -366,7 +375,10 @@ def main():
                 )
                 print('Exiting...')
                 sys.exit(1)
-            print('Waiting for emails to finish sending...')
+            print(
+                'Waiting for mock emails to finish sending (this takes a few seconds)...'
+            )
+            sys.stdout.flush()
             time.sleep(1)
             continue
         break
@@ -379,6 +391,12 @@ def main():
         percent_submitted=args.percent_submitted,
         percent_reported=args.percent_reported)
 
+    print(
+        '\n\nAll set! You can now browse to {} to view the Gophish dashboard'.
+        format(args.api_url))
+    print('The credentials are admin:gophish')
+    print('Enjoy!')
+
 
 if __name__ == '__main__':
     main()

Diff for: docker/run_demo.sh

@@ -0,0 +1,21 @@
+#!/bin/sh
+
+DATABASE=/usr/src/gophish/gophish.db
+
+# Wait for the gophish.db file to exist
+until [ -f $DATABASE ]; do
+  >&2 echo "Waiting for database"
+  sleep 1
+done
+
+# Wait for the table to be populated
+until sqlite3 $DATABASE 'select api_key from users limit 1'; do
+   >&2 echo "Waiting for database"
+  sleep 1
+done 
+
+# Get the API key
+export API_KEY=$(sqlite3 $DATABASE 'select api_key from users limit 1');
+
+# Launch the demo
+python create_demo.py --api-key=$API_KEY --num-members 30

Diff for: docker/supervisord.conf

@@ -0,0 +1,21 @@
+[supervisord]
+nodaemon=true
+
+[include]
+files = /etc/supervisor/conf.d/*.conf
+
+[program:gophish]
+command=/usr/src/gophish/gophish
+directory=/usr/src/gophish/
+autostart=true
+autorestart=true
+
+[program:demo]
+command=/usr/src/gophish-demo/run_demo.sh
+directory=/usr/src/gophish-demo/
+autostart=true
+autorestart=false
+stdout_logfile=/dev/stdout
+stdout_logfile_maxbytes=0
+stderr_logfile=/dev/stderr
+stderr_logfile_maxbytes=0