import pandas as pd
import streamlit as st

from config import DEFAULT_ICON
from shared_page import common_page_config

from data_storage import add_new_user, is_admin, migrate_players_from_week


def admin_add_new_user():
    add_new_user(
        st.session_state["admin_new_user_email"], st.session_state["admin_new_user_name"], st.session_state["db_client"]
    )


def admin_add_new_user_form():
    if st.button("Add new user"):
        with st.form("Add new user"):
            st.header("Admin add new user")
            st.write("Adds new user to sqlite db")
            st.text_input("Email Address", key="admin_new_user_email")
            st.text_input("Name", key="admin_new_user_name")
            st.form_submit_button("Submit", on_click=admin_add_new_user)


def migrate_players_week():
    with st.container():
        st.header("Utility for migrating players to the next week")
        week_migrate_from = st.selectbox("week to migrate from", options=[1, 2, 3])
        st.text_input("Enter week to confirm", key="week_migrate_from")
        if st.button("Migrate Week"):
            if st.session_state.get("week_migrate_from") == str(week_migrate_from):
                migrate_players_from_week(week_migrate_from, st.session_state["db_client"])
                st.warning("Week migrated")
            else:
                st.warning("Must confirm migration by entering matching week")


def get_page():
    page_title = "Admin"
    st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
    common_page_config()
    if not is_admin(st.session_state.get("logged_in_user"), st.session_state["db_client"]):
        st.write("Not authorized")
        st.stop()

    st.title(page_title)
    date_compare = pd.Timestamp.now(tz="America/New_York")
    st.write(f"Current datetime for use in roster lock: {date_compare}")

    admin_add_new_user_form()
    migrate_players_week()


if __name__ == "__main__":
    get_page()