#!/usr/bin/env python

# File: app_v01.py
# Author: Daniel Douglas, Claire Durand

import sys
import os
from flask import Flask, request, Response, jsonify
from flask_login import LoginManager, UserMixin, login_required

# Move to the directory that contains the interfacing script to import it
# (I'm actually pretty proud of this line because it means I can call server.py from anywhere)
sys.path.append(os.path.join(os.path.dirname(__file__),'..','peer-mentoring-app-interface'))
import database

# Create the application
#
app = Flask(__name__)

# Initialize login manager instance
#
login_manager = LoginManager()
login_manager.init_app(app)


# Define User class. User inherits from UserMixin class
# which we will "overload" (C++ term)
#
class User(UserMixin):
    # Import user data
    #
    user_database = database.data('users')

    # Initialize class parameters
    #
    def __init__(self, username, password):
        self.id = username
        self.password = password

    # Define a classmethod for easy repeatability
    #
    @classmethod
    def get(cls, id):
        return cls.user_database.get(id)


# Provide method for login manager to load a user
#
@login_manager.request_loader
def load_user(request):
    # Look for token in Authorization headers or request
    # arguments
    #
    print 'XXXX', request.authorization
    print 'YYYY', request.headers
    token = request.headers.get('Username'),request.headers.get('Password')
    if token is None:
        token = request.args.get('token')

    # Once we have the token, check for a match
    #
    if token is not None:
        print 'token is ', token
        username, password = token[0],token[1] # ASSUMING A NAIVE TOKEN
        
        # get the user information from class parameters
        #
        user_entry = User.get(username)
        print 'user entry is ', user_entry
        
        if (user_entry is not None):           
            # Create User() object from information
            #
            user = User(user_entry['name'], user_entry['password'])
            
            print 'user is ', user
            # If authentication info is correct, return user instance
            #
            if (user.password == password):
                return user
        # If authenticaton fails, return None
        #
        return None
            

# Default URL operations
#
@app.route('/')
def home_page():
    
    # Print simple welcome message
    #
    return 'Welcome to Peer-Mentoring App\n'


# Data fetch and post operations
#
@app.route('/data/', methods=['POST', 'GET'])
def data():
    
    print 'HEADERS', request.headers

    # If the request method is GET..
    #
    if request.method == 'GET':
        
        # Print the key/value data in the database
        #
        return jsonify(database.data('users'))


    # If the request method is POST
    #
    else:
        # Add the key/values in request to the database
        #
        name = request.headers.get('Username')
        password = request.headers.get('Password')
        print name
        print type(name)
        # If both fields are present in the request,
        # add the user information to the database
        #
        if (name and password):
            return database.post(name,password)
        else:
            # Raise error code 400=BAD REQUEST
            #
            return Response(response="Required Field's Empty", status=200)


# Authenticated user operation
#
@app.route('/private/', methods=['GET', 'POST'])
@login_required    # this line makes the route requre authentication
def private():
    return Response(response='success', status=200)

if __name__ == '__main__':
    # Run the app on all interfaces in debug mode
    # debug mode will restart server is this file changes.
    app.run(host='0.0.0.0', port=22000, debug=True)