from flask import Flask, request, abort
import pots

app = Flask(__name__)
pot = pots.CoffeePot("TestPot")


@app.route('/pot', methods=['GET'])
def get():
    order = pot.get()
    if order is None:
        abort('404', description='No orders in queue')
    return order

@app.route('/pot/order', methods=['GET'])
def popOrder():
    order = pot.popOrder()
    if order is None:
        abort('404', description='No orders in queue')
    return order

@app.route('/pot/ordersInQueue', methods=['GET'])
def getOrderCount():
    return pot.getOrderCount() + ' orders in queue'


@app.route('/pot', methods=['BREW', 'POST'])
def brew():
    return pot.brew(request.headers)

# TODO: THIS NEEDS DOING WITH SOMETHING THAT ACTUALLY SUPPORTS CUSTOM HTTP METHODS
@app.route('/pot', methods=['WHEN'])
def when():
    return pot.when()


if __name__ == '__main__':
    app.run(port=8081)