import logging
import streamlit as st
from sqlalchemy.sql import text
from queries.connection import connection
from enums import CounterType

logger = logging.getLogger(__name__)

def create_counter(title:str, counter_type:CounterType, counter_color) -> None:
    user_id = int(st.session_state.user_id)
    logger.info("Adding counter %s for user %d", counter_type, user_id)
    with connection().session as session:
        try:
            query = text('INSERT INTO counters (user_id, name, type, color) VALUES (:user, :title, :type, :color)')
            session.execute(query, {
                'user': user_id,
                'title': title,
                'type': counter_type,
                'color': counter_color
            })
        except Exception as e:
            logger.error(e)
            session.rollback()

def get_counters():
    user_id = int(st.session_state.user_id)
    try:
        return connection().query("""
              SELECT id, name, type, color 
              FROM counters 
              WHERE user_id = :user
              """, params={'user': user_id })
    except Exception as e:
        logger.error(e)
        return st.dataframe()

def increment_counter(counter_id:int) -> None:
    user_id = int(st.session_state.user_id)
    logger.info("Incrementing counter %d for user %d", counter_id, user_id)
    with connection().session as session:
        try:
            query = text('INSERT INTO entries (counter_id, user_id) VALUES (:id, :user)')
            session.execute(query, {
                'id': counter_id,
                'user': user_id
            })
        except Exception as e:
            logger.error(e)
            session.rollback()

def remove_counter(counter_id:int) -> None:
    user_id = int(st.session_state.user_id)
    logger.info("Removing counter %d from user %d", counter_id, user_id)
    with connection().session as session:
        try:
            query = text('DELETE FROM counters WHERE id = :id AND user_id = :user')
            session.execute(query, {
                'id': counter_id,
                'user': user_id
            })
        except Exception as e:
            logger.error(e)
            session.rollback()

def get_counter(counter_id:int):
    user_id = int(st.session_state.user_id)
    try:
        counters = connection().query("""
                   SELECT * FROM counters 
                   WHERE id = :id AND user_id = :user
                   """, params={ 'id': counter_id, 'user': user_id}
                )
        if counters.empty:
            return None

        return counters.iloc[0]
    except Exception as e:
        logger.error(e)
        return None


def get_colors(palette_id:int):
    try:
        return connection().query('''SELECT color1,color2,color3,color4,color5 FROM color_palettes WHERE id = :id''', params={'id': palette_id})
    except Exception as e:
        logger.error(e)
        return None