import streamlit as st
import json
import pandas as pd
from tools import agstyler
from tools.agstyler import PINLEFT
from toolbar import component_toolbar_buttons
 

class Setup:
    class Model:
        header1 = "Labels"
        header2 = "Groups"
        labels_file = "docs/labels.json"
        groups_file = "docs/groups.json"

    def view(self, model):
        tab = st.radio("Select", ["Labels", "Groups"], horizontal=True, label_visibility="collapsed")

        if tab == "Labels":
            st.title(model.header1)
            self.setup_labels(model)
        elif tab == "Groups":
            st.title(model.header2)
            self.setup_groups(model)

    def setup_labels(self, model):
        self.action_event = False
        if 'action' not in st.session_state:
            st.session_state['action'] = None

        with open(model.labels_file, "r") as f:
            labels_json = json.load(f)

        labels = labels_json["labels"]

        data = []
        for label in labels:
            data.append({'id': label['id'], 'name': label['name'], 'description': label['description']})
        self.df = pd.DataFrame(data)

        formatter = {
            'id': ('ID', {'hide': True}),
            'name': ('Label', {**PINLEFT, 'editable': True}),
            'description': ('Description', {**PINLEFT, 'editable': True})
        }

        def run_component(props):
            value = component_toolbar_buttons(key='toolbar_buttons_labels', **props)
            return value

        def handle_event(value):
            if value is not None:
                if 'action_timestamp' not in st.session_state:
                    self.action_event = True
                    st.session_state['action_timestamp'] = value['timestamp']
                else:
                    if st.session_state['action_timestamp'] != value['timestamp']:
                        self.action_event = True
                        st.session_state['action_timestamp'] = value['timestamp']
                    else:
                        self.action_event = False

            if value is not None and value['action'] == 'create' and self.action_event:
                if st.session_state['action'] != 'delete':
                    max_id = self.df['id'].max()
                    self.df.loc[-1] = [max_id + 1, '', '']  # adding a row
                    self.df.index = self.df.index + 1  # shifting index
                    self.df.sort_index(inplace=True)
                    st.session_state['action'] = 'create'
            elif value is not None and value['action'] == 'delete' and self.action_event:
                if st.session_state['action'] != 'delete' and st.session_state['action'] != 'create':
                    rows = st.session_state['selected_rows']
                    if len(rows) > 0:
                        idx = rows[0]['_selectedRowNodeInfo']['nodeRowIndex']
                        self.df.drop(self.df.index[idx], inplace=True)
                        self.df.reset_index(drop=True, inplace=True)
                    st.session_state['action'] = 'delete'
            elif value is not None and value['action'] == 'save' and self.action_event:
                st.session_state['action'] = 'save'

        props = {
            'buttons': {
                'create': True,
                'delete': True,
                'save': True,
            }
        }

        handle_event(run_component(props))

        if st.session_state['action'] == 'save' and 'response' in st.session_state:
            if st.session_state['response'] is not None:
                self.df = st.session_state['response']
            st.session_state['response'] = None

        if st.session_state['action'] == 'create' and 'response' in st.session_state:
            if st.session_state['response'] is not None:
                self.df = st.session_state['response']

        if st.session_state['action'] == 'delete' and 'response' in st.session_state:
            if st.session_state['response'] is not None:
                self.df = st.session_state['response']

        response = agstyler.draw_grid(
            self.df,
            formatter=formatter,
            fit_columns=True,
            pagination_size=10,
            selection="single",
            use_checkbox=False
        )

        rows = response['selected_rows']
        st.session_state['selected_rows'] = rows

        if st.session_state['action'] == 'create' and self.action_event:
            st.session_state['response'] = response['data']
        elif st.session_state['action'] == 'delete' and self.action_event:
            st.session_state['response'] = response['data']
        elif st.session_state['action'] == 'save' and self.action_event:
            data = response['data'].values.tolist()
            rows = []
            for row in data:
                rows.append({'id': row[0], 'name': row[1], 'description': row[2]})

            labels_json['labels'] = rows
            with open(model.labels_file, "w") as f:
                json.dump(labels_json, f, indent=2)


    def setup_groups(self, model):
        self.action_event = False
        if 'action' not in st.session_state:
            st.session_state['action'] = None

        with open(model.groups_file, "r") as f:
            groups_json = json.load(f)

        groups = groups_json["groups"]

        data = []
        for group in groups:
            data.append({'id': group['id'], 'name': group['name'], 'description': group['description']})
        self.df = pd.DataFrame(data)

        formatter = {
            'id': ('ID', {'hide': True}),
            'name': ('Group', {**PINLEFT, 'editable': True}),
            'description': ('Description', {**PINLEFT, 'editable': True})
        }

        def run_component(props):
            value = component_toolbar_buttons(key='toolbar_buttons_groups', **props)
            return value

        def handle_event(value):
            if value is not None:
                if 'action_timestamp' not in st.session_state:
                    self.action_event = True
                    st.session_state['action_timestamp'] = value['timestamp']
                else:
                    if st.session_state['action_timestamp'] != value['timestamp']:
                        self.action_event = True
                        st.session_state['action_timestamp'] = value['timestamp']
                    else:
                        self.action_event = False

            if value is not None and value['action'] == 'create' and self.action_event:
                if st.session_state['action'] != 'delete':
                    max_id = self.df['id'].max()
                    self.df.loc[-1] = [max_id + 1, '', '']  # adding a row
                    self.df.index = self.df.index + 1  # shifting index
                    self.df.sort_index(inplace=True)
                    st.session_state['action'] = 'create'
            elif value is not None and value['action'] == 'delete' and self.action_event:
                if st.session_state['action'] != 'delete' and st.session_state['action'] != 'create':
                    rows = st.session_state['selected_rows']
                    if len(rows) > 0:
                        idx = rows[0]['_selectedRowNodeInfo']['nodeRowIndex']
                        self.df.drop(self.df.index[idx], inplace=True)
                        self.df.reset_index(drop=True, inplace=True)
                    st.session_state['action'] = 'delete'
            elif value is not None and value['action'] == 'save' and self.action_event:
                st.session_state['action'] = 'save'

        props = {
            'buttons': {
                'create': True,
                'delete': True,
                'save': True,
            }
        }

        handle_event(run_component(props))

        if st.session_state['action'] == 'save' and 'response' in st.session_state:
            if st.session_state['response'] is not None:
                self.df = st.session_state['response']
            st.session_state['response'] = None

        if st.session_state['action'] == 'create' and 'response' in st.session_state:
            if st.session_state['response'] is not None:
                self.df = st.session_state['response']

        if st.session_state['action'] == 'delete' and 'response' in st.session_state:
            if st.session_state['response'] is not None:
                self.df = st.session_state['response']

        response = agstyler.draw_grid(
            self.df,
            formatter=formatter,
            fit_columns=True,
            pagination_size=10,
            selection="single",
            use_checkbox=False
        )

        rows = response['selected_rows']
        st.session_state['selected_rows'] = rows

        if st.session_state['action'] == 'create' and self.action_event:
            st.session_state['response'] = response['data']
        elif st.session_state['action'] == 'delete' and self.action_event:
            st.session_state['response'] = response['data']
        elif st.session_state['action'] == 'save' and self.action_event:
            data = response['data'].values.tolist()
            rows = []
            for row in data:
                rows.append({'id': row[0], 'name': row[1], 'description': row[2]})

            groups_json['groups'] = rows
            with open(model.groups_file, "w") as f:
                json.dump(groups_json, f, indent=2)