from flask import Flask, url_for, redirect, render_template, request
from pymongo import MongoClient
from datetime import datetime
from bson.objectid import ObjectId

app = Flask(__name__)

#initialize the database
client = MongoClient('localhost', 27017)
db = client.flask_db
posts_collection = db.posts_collection


#app routes
@app.route('/', methods=['GET'])
def index():
    return render_template('index.html')

@app.route('/board/<board_name>', methods=['GET'])
def board(board_name):
    links = [
        {'name': 'b', 'display_name': '/b/ - random'},
        {'name': 'g', 'display_name': '/g/ - technology'},
        {'name': 'a', 'display_name': '/a/ - anime'},
        {'name': 'v', 'display_name': '/v/ - video games'},
        {'name': 'w', 'display_name': '/w/ - wallpapers'},
        {'name': 'x', 'display_name': '/x/ - paranormal'},
        {'name': 't', 'display_name': '/t/ - test'},
        {'name': 's', 'display_name': '/s/ - soyjaks'},
        {'name': 'pol', 'display_name': '/pol/ - politically incorrect'}
    ]
    #global posts
    #posts = posts_collection.find({'board_name': board_name}).sort('timestamp', -1)
    display_name = next((link['display_name'] for link in links if link['name'] == board_name), board_name)
    return render_template('board.html', title=board_name, header=display_name, links=links)

@app.route('/post', methods=['POST'])
def post():
    board_name = request.form['board_name']
    content = request.form['content']
    image = request.files['image']
    timestamp = datetime.now()

    #insert the post into MongoDB
    posts.insert_one({
        'board_name': board_name,
        'content': content,
        'image': image.read(),
        'timestamp': timestamp
    })

    return redirect(url_for('board', board_name=board_name))


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