from flask import Flask, jsonify, request, render_template
from flask_sqlalchemy import SQLAlchemy
import re
from werkzeug.security import generate_password_hash, check_password_hash
from sqlalchemy.orm import validates

app = Flask(__name__)
app.config['DEBUG'] = True
app.config['SECRET_KEY'] = '5791628bb0b13ce0c676dfde280ba245'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)

class User(db.Model):
  id = db.Column(db.Integer, primary_key=True)
  username = db.Column(db.String(64), index=True, unique=True, nullable=False)
  email = db.Column(db.String(120), index=True, nullable=False)
  password_hash = db.Column(db.String(128))
  role = db.Column(db.Enum('basic', 'admin', name='user_roles'), default='basic')

  def set_password(self, password):
    self.password_hash = generate_password_hash(password)

  def check_password(self, password):
      return check_password_hash(self.password_hash, password)

  @validates('username')
  def validate_username(self, key, username):
    if not username:
      raise AssertionError('No username provided')

    if User.query.filter(User.username == username).first():
      raise AssertionError('Username is already in use')

    if len(username) < 5 or len(username) > 20:
      raise AssertionError('Username must be between 5 and 20 characters')

    return username

  @validates('email')
  def validate_email(self, key, email):
    if not email:
      raise AssertionError('No email provided')

    if not re.match("[^@]+@[^@]+\.[^@]+", email):
      raise AssertionError('Provided email is not an email address')

    return email

  def set_password(self, password):
    if not password:
      raise AssertionError('Password not provided')

    if not re.match('\d.*[A-Z]|[A-Z].*\d', password):
      raise AssertionError('Password must contain 1 capital letter and 1 number')

    if len(password) < 5 or len(password) > 50:
      raise AssertionError('Password must be between 8 and 50 characters')

    self.password_hash = generate_password_hash(password)


#@app.before_first_request
#def create_db():
#   db.create_all()

@app.route('/')
def home():
  return render_template('index.html')

@app.route('/create', methods=['POST'])
def create():

  data = request.get_json()
  username = data['username']
  password = data['password']
  email = data['email']

  user = User(username=username, email=email)
  user.set_password(password)
  try:
    db.session.add(user)
    db.session.commit()
    return jsonify(msg='User successfully created', user_id=user.id), 200
  except AssertionError as exception_message:
    return jsonify(msg='Error: {}. '.format(exception_message)), 400


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