from flask import render_template, request, redirect, url_for
from flask_login import login_user, login_required, logout_user, current_user
from app import app, db, bcrypt, login_manager
from models import User, Listing, Booking, Image
from datetime import datetime
from utils import generate_password, send_email

@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))

@app.route('/')
def home():
    property_type = request.args.get('property_type')
    listings = Listing.query.all()
    if property_type:
        listings = [l for l in listings if l.property_type == property_type]
    return render_template('index.html', listings=listings)

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        email = request.form['email']
        password = request.form['password']
        user = User.query.filter_by(email=email).first()
        if user and bcrypt.check_password_hash(user.password, password):
            login_user(user)
            return redirect(url_for('dashboard'))
    return render_template('login.html')

@app.route('/logout/')
@login_required
def logout():
    logout_user()
    return redirect(url_for('login'))

@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        email = request.form['email']
        password_plain = generate_password()
        password = bcrypt.generate_password_hash(password_plain).decode('utf-8')
        phone = request.form['phone']
        user = User(email=email, password=password, phone=phone)
        db.session.add(user)
        db.session.commit()
        send_email(email, 'Your Account Password', f'Your password is: {password_plain}')
        login_user(user)
        return redirect(url_for('dashboard'))
    return render_template('register.html')

@app.route('/dashboard')
@login_required
def dashboard():
    if current_user.is_host:
        listings = Listing.query.filter_by(host_id=current_user.id).all()
        return render_template('dashboard.html', listings=listings)
    return redirect(url_for('home'))

@app.route('/create-listing', methods=['GET', 'POST'])
@login_required
def create_listing():
    if not current_user.is_host:
        return redirect(url_for('home'))

    if request.method == 'POST':
        title = request.form['title']
        description = request.form['description']
        price = request.form['price']
        location = request.form['location']
        property_type = request.form['property_type']
        latitude = request.form.get('latitude')
        longitude = request.form.get('longitude')
        guests = request.form['guests']
        listing = Listing(title=title, description=description, price=price, location=location, property_type=property_type, host_id=current_user.id, latitude=latitude, longitude=longitude, guests=guests)
        db.session.add(listing)
        db.session.flush()

        image_urls = request.form.getlist('images')
        for url in image_urls:
            image = Image(url=url, listing_id=listing.id)
            db.session.add(image)

        db.session.commit()
        return redirect(url_for('dashboard'))
    return render_template('create_listing.html', property_types=Listing.PROPERTY_TYPES)

@app.route('/booking/<int:listing_id>', methods=['GET', 'POST'])
@login_required
def booking(listing_id):
    listing = Listing.query.get_or_404(listing_id)
    booked_dates = Booking.query.filter_by(listing_id=listing_id).all()
    unavailable_dates = [
        (booking.check_in.strftime('%Y-%m-%d'), booking.check_out.strftime('%Y-%m-%d'))
        for booking in booked_dates
    ]

    if request.method == 'POST':
        check_in = datetime.strptime(request.form['check_in'], '%Y-%m-%d')
        check_out = datetime.strptime(request.form['check_out'], '%Y-%m-%d')
        guests = request.form['guests']
        booking = Booking(user_id=current_user.id, listing_id=listing_id, check_in=check_in, check_out=check_out, guests=guests)
        db.session.add(booking)
        db.session.commit()
        return redirect(url_for('home'))
    return render_template('booking.html', listing=listing, unavailable_dates=unavailable_dates)