import streamlit as st

import sql
from enums import CounterType

@st.dialog("Add New Counter", icon=":material/add_box:")
def _add_counter():
    colors = sql.get_colors(1)
    with st.form(key="add_counter", border=False, clear_on_submit=True):
        title = st.text_input("Title:")
        counter_type_name = st.selectbox("Type", options=[e.name for e in CounterType])
        color = st.radio("Color",
                     key="color-selector",
                     width="stretch",
                     options=[colors[key][0] for key in colors],
                     format_func=lambda c: f"#{c}")
        with st.container(horizontal=True, width="stretch", horizontal_alignment="center"):
            if st.form_submit_button(label="Create", icon=":material/save:"):
                sql.create_counter(title, CounterType[counter_type_name], color)
                st.rerun()


@st.dialog("Remove Counter", icon=":material/delete:")
def _remove_counter(counter_id:int):
    with st.form(key="remove_counter", border=False, clear_on_submit=True):
        st.subheader("Are you sure?")
        with st.container(horizontal=True, width="stretch", horizontal_alignment="center"):
            if st.form_submit_button("Confirm", icon=":material/delete:"):
                sql.remove_counter(counter_id)
                st.rerun()

df = sql.get_counters()

with st.container(key="counter-table"):
    for counter_id, name, counter_type_str, color in zip(df['id'], df['name'], df['type'], df['color']):
        with st.container(width="stretch", key=f"counter_{counter_id}"):
            with st.container(horizontal=True, width="stretch"):
                st.header(f":material/calendar_clock: {name}", width="stretch")

                if st.button("", icon=":material/exposure_plus_1:", key=f"increment_counter_{counter_id}"):
                    sql.increment_counter(counter_id)
                    st.rerun()

                if st.button("", icon=":material/delete_forever:", key=f"remove_counter_{counter_id}"):
                    _remove_counter(counter_id)

            with st.container(horizontal=True, width="stretch"):

                counter_type = CounterType(counter_type_str)
                stats_current_unit = counter_type.current_unit_text()
                stats_prev_unit = counter_type.previous_unit_text()
                match counter_type:
                    case CounterType.DAILY.value | CounterType.SIMPLE.value:
                        stats = sql.get_daily_analytics(counter_id)
                        stats_current = stats.iloc[0]["count"]
                        stats_prev = stats.iloc[1]["count"]

                    case CounterType.WEEKLY.value:
                        stats = sql.get_weekly_analytics(counter_id)
                        stats_current = stats.iloc[-1]["count"]
                        stats_prev = stats.iloc[-2]["count"]

                    case CounterType.MONTHLY.value:
                        stats = sql.get_monthly_analytics(counter_id)
                        stats_current = stats.iloc[-1]["count"]
                        stats_prev = stats.iloc[-2]["count"]

                    case CounterType.YEARLY.value:
                        stats = sql.get_yearly_analytics(counter_id)
                        stats_current = stats.iloc[-1]["count"]
                        stats_prev = stats.iloc[-2]["count"]

                if counter_type is CounterType.SIMPLE.value:
                     st.markdown(f"**{stats_current} {stats_current_unit}**")
                else:
                     st.markdown(f"""
                         **{stats_current} {stats_current_unit}**
                         *{stats_prev} {stats_prev_unit}*
                     """)

                with st.container(horizontal=True, width="stretch", horizontal_alignment="right"):
                    st.page_link("pages/stats.py", icon=":material/bar_chart:", icon_position="right", label="", query_params={"counter_id": str(counter_id)})

            st.html(f"""
                <style>
                    div:has(> .st-key-counter_{counter_id}) {{
                        background-color: {color};
                    }}
                </style>
            """)

if st.button("Add Counter", width="stretch", icon=":material/add_box:"):
    _add_counter()