CommandCenter.py

#!/usr/bin/python
from flask import Flask, render_template, request, jsonify, redirect, session, flash, make_response
import os
import json
import requests

app = Flask(__name__)
app.secret_key = "bulkingseason"

AUTH_API_URL = 'http://atlas.whiteteam.ritsec:9000'

def validate_session(token):
    """
    Sends token to auth server to validate, should recieve
    associated team number if it is valid

    :param token: the session token to validate

    :return team_id: the id of the team the token is attached to
    """
    post_data = dict()
    post_data['token'] = token
    resp = requests.post("{}/validate-session".format(AUTH_API_URL, data=post_data))
    if resp.status_code != 200:
        raise Exception("Bad status code")

    if 'success' not in resp.json():
        raise Exception(resp.json()['error'])

    team_id = resp['success']
    return team_id

@app.route('/')
def home():
    if session.get('token'):
        token = session.get('token')
        team_id = session.get('team_id')
        try:
            validate_session(token)
        except:
            redirect('/login')

        return render_template('index.html', team_id=team_id)
    else:
        return render_template('login.html')


@app.route('/login', methods=['POST'])
def login():
    password = request.form['password']
    username = request.form['username']
    data = {'username': username, 'password': password}
    r = requests.post("{}/login".format(AUTH_API_URL), data=data)
    if r.status_code != 200:
        # error
        pass
    elif 'error' in r.json():
        # error
        pass
    else:
        token = r.json()['token']
        team_id = r.json()['team_id']
        session['token'] = token
        session['team_id'] = team_id
        response = make_response(redirect('/'))
        response.set_cookie('token', token)
        return response
    #print(r.json())
    return home()

@app.route('/logout', methods=['GET'])
def logout():
    data = {'token': session['token']}
    requests.post("{}/expire-session".format(AUTH_API_URL), data=data)
    session.clear()
    return redirect('/')


@app.route('/hq_coms', methods=['POST'])
def hq_coms():
    host1 = request.form['server1']
    host2 = request.form['server2']
    host3 = request.form['server3']
    host4 = request.form['server4']
    host5 = request.form['server5']
    host6 = request.form['server6']
    host7 = request.form['server7']
    host8 = request.form['server8']
    host9 = request.form['server9']
    host10 = request.form['server10']
    return jsonify({
        'reachable1': [host1, ping(host1)],
        'reachable2': [host2, ping(host2)],
        'reachable3': [host3, ping(host3)],
        'reachable4': [host4, ping(host4)],
        'reachable5': [host5, ping(host5)],
        'reachable6': [host6, ping(host6)],
        'reachable7': [host7, ping(host7)],
        'reachable8': [host8, ping(host8)],
        'reachable9': [host9, ping(host9)],
        'reachable10': [host10, ping(host10)]
    })


"""
        Ansible
"""
@app.route('/ansible_playbook1', methods=['POST'])
def ansible_playbook1():
    command = "ansible-playbook /var/www/ISTS16-ansible/predeploy.yml -t jenkins -i /var/www/ISTS16-ansible/hosts"
    os.system(command)
    return redirect('/')


@app.route('/ansible_playbook2', methods=['POST'])
def ansible_playbook2():
    command = "ansible-playbook /var/www/ISTS16-ansible/predeploy.yml -t ecommerce -i /var/www/ISTS16-ansible/hosts"
    os.system(command)
    return redirect('/')


@app.route('/ansible_playbook3', methods=['POST'])
def ansible_playbook3():
    command = "ansible-playbook /var/www/ISTS16-ansible/predeploy.yml -t mail -i /var/www/ISTS16-ansible/hosts"
    os.system(command)
    return redirect('/')

@app.route('/ansible_playbook4', methods=['POST'])
def ansible_playbook4():
    command = "ansible-playbook /var/www/ISTS16-ansible/predeploy.yml -t jenkinswin -i /var/www/ISTS16-ansible/hosts"
    os.system(command)
    return redirect('/')



"""
        Helpers
"""
def ping(host):
    if host.lower() == "luna":
        host = "localhost"

    response = os.system("ping -c 1 -W 1 " + host)
    if response == 0:
        return 'UP'
    else:
        return 'DOWN'


if __name__ == '__main__':
    app.secret_key = os.urandom(12)
    app.run(host='0.0.0.0', port=80, debug=True)