from flask import Blueprint, request
from db import cursor
from auth.login_token import token_login
from rating import get_rating
import json

# check if user_id exists in database
def valid_user_id(user_id):
    query = cursor.execute("select user_id from users where user_id = ?", [user_id]).fetchone()
    return bool(query)

# get relation to user_2_id from user_1_id's perspective
def get_relation_to(user_1_id, user_2_id):
    relation = cursor.execute("select * from social where " + \
            "(user_1_id = ? and user_2_id = ?) or " + \
            "(user_1_id = ? and user_2_id = ?)", [user_1_id, user_2_id, user_2_id, user_1_id]).fetchone()
    if not relation: return "none"
    if relation[2] == "friendship": return "friends"
    if relation[2] == "outgoing" and relation[0] == user_1_id: return "outgoing"
    if relation[2] == "outgoing" and relation[1] == user_1_id: return "incoming"
    if relation[2] == "block" and relation[0] == user_1_id: return "blocked"
    return "none"

# get users friend count
def count_friends(user_id):
    query = cursor.execute("select type from social where (user_1_id = ? or user_2_id = ?) and type = \"friendship\"", [user_id, user_id]).fetchall()
    return len(query) #FIXME: use SQL count() instead of python's len()

# get user/info of `user_id` as `viewer` (id)
def format_user(user_id, viewer = ''):
    user = cursor.execute("select " + ", ".join([
        "username",
        "user_id",
        "country",
        "registered",
        "status",
    ]) + " from users where user_id = ?", [user_id]).fetchone()
    formatted_user = {
        "username": user[0],
        "id": user[1],
        "country": user[2],
        "registered": user[3],
        "status": user[4],
        "friends": count_friends(user_id),
        "rating": get_rating(user_id), #TODO: calculate rating based on game analysis
    }
    if viewer:
        #FIXME: validate viewer id?
        formatted_user["relation"] = get_relation_to(viewer, user_id)
    return formatted_user

info = Blueprint('info', __name__)

# view own user/info if no user_id or username is provided and is logged in,
# else view user/info of user with user_id = `user_id` or username = `username`
@info.route('/info', methods = ['GET', 'POST'])
def index():
    data_string = request.data or "{}"
    data = json.loads(data_string)

    username = data.get("username") or ""
    user_id = data.get("id") or ""
    token = request.cookies.get("token") or ""
    viewer = ""

    if not username and \
       not user_id and \
       not token:
           return "", 400

    if username:
        temp_user_id = cursor.execute("select user_id from users where username = ?", [username]).fetchone()
        if len(temp_user_id) > 0: user_id = temp_user_id

    if token:
        self_id = token_login(token)
        if not (username or user_id):
            user_id = self_id
        if user_id:
            viewer = self_id

    if user_id and not valid_user_id(user_id): return "", 403
    user = format_user(user_id, viewer)

    return user, 200

dynamic_route = ["/user", info]