from pymongo import MongoClient
from bson.objectid import ObjectId
from backend import accounts
import datetime, pymongo

client = MongoClient()
db = client.tweeder
accounts_db = db.accounts
messages_db = db.messages


def send_message(msg_from, msg_to, msg_content):
    if type(msg_from) == str:
        from_id = accounts.account_details(msg_from.lower())['_id']
    elif type(msg_from) == ObjectId:
        from_id = msg_from
        msg_from = accounts_db.find_one({'_id': from_id})['displayname']

    if type(msg_to) == str:
        to_id = accounts.account_details(msg_to.lower())['_id']
    elif type(msg_to) == ObjectId:
        to_id = msg_to
        msg_to = accounts_db.find_one({'_id': to_id})['displayname']

    currentTimeDate = datetime.datetime.now()

    message = {
        'from': from_id,
        'to': to_id,
        'fromName': msg_from,
        'toName': msg_to,
        'content': msg_content,
        'timeSent': currentTimeDate
    }

    messages_db.insert_one(message)


def get_messages(user1, user2):
    user1_id = accounts_db.find_one({"username": user1.lower()})['_id']
    user2_id = accounts_db.find_one({"username": user2.lower()})['_id']

    messages = messages_db.find(
        {"$or": [
            {
                "from": user1_id,
                "to"  : user2_id
            },
            {
                "from": user2_id,
                "to"  : user1_id
            }
        ]}
    ).sort('timeSent', pymongo.ASCENDING)
    return messages