from flask import Blueprint, request, session, redirect, url_for

from atst.domain.users import Users

bp = Blueprint("dev", __name__)

_DEV_USERS = {
    "sam": {
        "dod_id": "1234567890",
        "first_name": "Sam",
        "last_name": "Seeceepio",
        "atat_role": "ccpo",
    },
    "amanda": {
        "dod_id": "2345678901",
        "first_name": "Amanda",
        "last_name": "Adamson",
        "atat_role": "default",
    },
    "brandon": {
        "dod_id": "3456789012",
        "first_name": "Brandon",
        "last_name": "Buchannan",
        "atat_role": "default",
    },
    "christina": {
        "dod_id": "4567890123",
        "first_name": "Christina",
        "last_name": "Collins",
        "atat_role": "default",
    },
    "dominick": {
        "dod_id": "5678901234",
        "first_name": "Dominick",
        "last_name": "Domingo",
        "atat_role": "default",
    },
    "erica": {
        "dod_id": "6789012345",
        "first_name": "Erica",
        "last_name": "Eichner",
        "atat_role": "default",
    },
}


@bp.route("/login-dev")
def login_dev():
    role = request.args.get("username", "amanda")
    user_data = _DEV_USERS[role]
    basic_data = {k:v for k,v in user_data.items() if k not in ["dod_id", "atat_role"]}
    user = _set_user_permissions(user_data["dod_id"], user_data["atat_role"], basic_data)
    session["user_id"] = user.id
    return redirect(url_for("atst.home"))


def _set_user_permissions(dod_id, role, user_data):
    return Users.get_or_create_by_dod_id(dod_id, atat_role_name=role, **user_data)