import sqlite3
import os, base64

from flask import render_template, request, session, redirect, url_for, g, flash, abort

import flask_bcrypt as Bcrypt

from .forms import LoginForm, SignUpForm

from .dbhelper import DBHelper

from .passwdbuilder import PassBuilder

DB = DBHelper()
PB = PassBuilder()

from storeapp import app

app.config.update(dict(
	DATABASE=os.path.join(app.root_path, 'store.db'),
	SECRET_KEY='vqE/O0iNhARuC1e6c9AM9mg0C2DLUGkHfrZDwN3/qgHJFdYP14TRmIuZngPrrwKVd1KcD+KfyEkh/yxxkPyi5nhlyk8OF32wC6HM',
	USERNAME='admin',
	PASSWORD='default'
))
app.config.from_envvar('STOREAPP_SETTINGS', silent=True)

def connect_db():
	#connects to the db
	rv = sqlite3.connect(app.config['DATABASE'])
	rv.row_factory = sqlite3.Row
	return rv

def init_db():
	db = get_db()
	with app.open_resource('schema.sql', mode='r') as f:
		db.cursor().executescript(f.read())
	db.commit()

@app.cli.command('initdb')
def initdb_command():
	#inits the db
	init_db()
	print('Initialized the database')


def get_db():
	#Opens a new DB connection if there is none for the app
	if not hasattr(g, 'sqlite_db'):
		g.sqlite_db=connect_db()
	return g.sqlite_db

@app.teardown_appcontext
def close_db(error):
	#closes the db at end of request
	if hasattr(g, 'sqlite_db'):
		g.sqlite_db.close()




@app.route('/')
def index():
	db = get_db()
	cur = db.execute('select title, author from books order by id desc')
	books = cur.fetchall()

	return render_template('index.html', books=books)


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

	db = get_db()

	cur = db.cursor()

	form = SignUpForm()
	if request.method == 'POST':
		if form.validate_on_submit():
			username = request.form.get("username")
			email = request.form.get("email")
			password1 = request.form.get("password1")
			password2 = request.form.get("password2")


			if password1 != password2:
				flash('Sorry your passwords must match, please try again')
				return redirect(url_for('signup'))



			salt = PB.SaltBuilder()
			print(salt)

			hashed_pw = PB.HashBuilder(password1) + salt
			print(hashed_pw)

			try:

				db.execute('insert into users (username, email, password) values (?, ?, ?)',
					[username, email, hashed_pw])

				db.commit()

				flash('Thanks for registering')

				return redirect(url_for('index'))

			except sqlite3.IntegrityError:
				flash('Username already taken')


	return render_template('signup.html', form=form)




@app.route('/login', methods=('GET', 'POST'))
def login():

	form = LoginForm(request.form)
	if request.method == 'POST':
		username = request.form.get("username")
		password = request.form.get("password")
		completion = validate(username, password)
		if completion == False:
			flash('Invalid login, please try again. Are you registered?')
		else:
			return redirect(url_for('members'))
	return render_template('login.html', form=form)

def validate(username, password):
	print('Inside validate')
	db = get_db()
	completion = False
	with db:
		cur = db.cursor()
		cur.execute('SELECT * FROM users')
		rows = cur.fetchall()
		for row in rows:
			dbUser = row[1]
			print(dbUser)
			dbPass = row[3]
			print(dbPass)
			if dbUser==username:
				if Bcrypt.check_password_hash(dbPass, password) == True:
					completion == True
			else:
				flash('Invalid login!')
	return completion



@app.route('/logout')
def logout():
	session.pop('logged_in', None)
	flash('You were logged out')
	return redirect(url_for('index'))

@app.route('/members')
def members():

	if not session.get('logged_in'):
		abort(401)

	return render_template('members.html')

@app.route('/books')
def books():

	db = get_db()
	cur = db.execute('select title, author, category from books order by id desc')
	books = cur.fetchall()


	return render_template('books.html', books=books)

@app.route('/add', methods=['GET','POST'])
def addbook():
	if not session.get('logged_in'):
		abort(401)
	db = get_db()
	db.execute('insert into books (title, author, category) values (?, ?, ?)',
				[request.form['title'], request.form['author'], request.form['category']])

	db.commit()
	flash('New book added!')
	return redirect(url_for('books'))

@app.route('/users')
def showusers():
	#if not session.get('logged_in'):
		#abort(401)
	db = get_db()
	cur = db.execute('select username from users order by id desc')
	members = cur.fetchall()

	return render_template('users.html', members=members)