import streamlit as st
from utils.login import get_login
import os


def initialize_level():
    if "level" not in st.session_state:
        if get_login()["status"]:
            if not os.path.exists(f".sessions/{get_login()['username']}/level.txt"):
                with open(f".sessions/{get_login()['username']}/level.txt", "w") as f:
                    f.write("0")
                st.session_state["level"] = 0
            else:
                with open(f".sessions/{get_login()['username']}/level.txt", "r") as f:
                    st.session_state["level"] = int(f.read())


def get_level():
    return st.session_state["level"]


def render_page(page, level):
    if get_login()["status"]:
        if st.session_state["level"] < level:
            st.error(f"You need to complete Level {st.session_state['level']} first!")
        else:
            page()
    else:
        st.error("You need to login first!")


def complete_level(level):
    if st.session_state["level"] > level:
        st.info(f"You have Already completed Level {level}!")
    else:
        st.session_state["level"] = level + 1
        with open(f".sessions/{get_login()['username']}/level.txt", "w") as f:
            f.write(str(st.session_state["level"]))
        st.balloons()
        st.success(
            f"You have completed Level {level}! You can now move on to the next level."
        )