import os
# from PyQt5.QtCore import Qt
# from PyQt5.QtWidgets import (
#     QWidget, QGridLayout, QHBoxLayout, QStackedWidget, QVBoxLayout, QFrame
# )
from PySide6.QtCore import Qt
from PySide6.QtWidgets import (
    QWidget, QGridLayout, QHBoxLayout, QStackedWidget, QVBoxLayout, QFrame
)

from app.commons.image import Image
from .commons.custom_button import CustomButton
from .tools import get_icon
from .constants import FILE_BANNER

__all__ = ['ButtonsFunction', 'ButtonsStacked', 'StartButtons']

DIR_SHARE = os.path.abspath(os.path.normpath(
    os.path.join(__file__, '..', '..', 'share'))
)


class StartButtons(QVBoxLayout):

    def __init__(self, parent):
        super(StartButtons, self).__init__()
        grid = QGridLayout()
        frame = QFrame()
        pixmap_pos = Image(parent=parent, name='pixmap_pos', default_img=FILE_BANNER, scaled_rate=None)
        self.addWidget(pixmap_pos, 0)
        self.addLayout(grid, 0)
        self.addWidget(frame, 1)
        grid.setHorizontalSpacing(2)
        grid.setVerticalSpacing(2)
        grid.rowStretch(1)
        grid.parent = parent
        columns = 4
        rows = 4

        values = []
        values_extend = values.extend
        if parent.type_pos_user != 'cashier':
            values_extend([
                ['button_start_take_away', 'PARA LLEVAR', 'action_start_take_away', 'take_away'],
                ['button_start_delivery', 'DOMICILIO', 'action_start_delivery', 'delivery'],
            ])
            if parent.environment == 'restaurant':
                values_extend([
                    ['button_start_table', 'A LA MESA', 'action_start_table', 'table'],
                    ['button_start_catering', 'CATERING', 'action_start_catering', 'catering'],
                ])
        values_extend([
            ['button_start_search_order', 'BUSCAR ORDEN', 'action_search_sale', 'search_sale'],
            ['button_print_sale', 'IMPRIMIR', 'action_print_sale', 'print_sale'],
        ])

        if parent.environment == 'restaurant':
            values_extend([
                ['button_tables', 'VER MESAS', 'action_tables', 'tables']
            ])

        if parent.type_pos_user in ('cashier', 'frontend', 'frontend_admin'):
            values_extend([
                ['button_control_panel', 'PANEL DE CONTROL', 'action_control_panel', 'settings'],
                ['button_reports', 'REPORTES', 'action_reports', 'reports'],
            ])
            if parent.environment == 'retail':
                values_extend([
                    ['button_collection', 'RECAUDO', 'action_collection', 'collection']
                ])
            else:
                values_extend([
                    ['button_historic_sales', 'HISTORIAL', 'action_historic_sales', 'sales_history'],
                ])

        values_extend([
            ['button_help', 'AYUDA', 'action_help', 'help'],
            ['button_product_info', 'INFO. PRODUCTO', 'action_info_product', 'product_info'],
            ['button_test_print', 'TEST IMPRESION', 'action_test_print', 'test_print']
        ])

        positions = [(i, j) for i in range(rows) for j in range(columns)]
        for position, value in zip(positions, values):
            button = CustomButton(
                parent,
                id=value[0],
                size=parent.screen_size,
                icon=get_icon(value[3]),
                title=value[1],
                method=value[2],
                name_style='toolbar'
            )
            grid.addWidget(button, *position)


class ButtonsFunction(QGridLayout):

    def __init__(self, parent, values=[]):
        super(ButtonsFunction, self).__init__()

        rows = 4
        columns = 3
        self.setHorizontalSpacing(2)
        self.setVerticalSpacing(2)
        self.parent = parent
        self.set_values(values)
        if self.parent.environment == 'restaurant':
            rows = 3
            columns = 5
        name_style = 'toolbar'

        positions = [(i, j) for i in range(rows) for j in range(columns)]
        for position, value in zip(positions, self.values):
            name_icon = value[0][7:]
            button = CustomButton(
                parent,
                id=value[0],
                size=parent.screen_size,
                icon=get_icon(name_icon),
                title=value[1],
                method=value[2],
                name_style=name_style
            )
            setattr(self, value[0], button)
            self.addWidget(button, *position)
            last_row = position[0]
        self.setRowStretch(last_row + 1, 1)

    def set_values(self, values):
        if values:
            self.values = values
            return
        self.values = [
            ['button_search_product', 'PRODUCTOS', 'action_search_product'],
        ]

        self.values.extend([
            ['button_party', 'CLIENTE', 'action_party'],
            ['button_print_sale', 'IMPRIMIR', 'action_print_sale'],
            ['button_delivery_party', 'DOMICILIARIO', 'action_delivery_party'],
            ['button_position', 'POSICION', 'action_position'],
            ['button_print_order', 'ENVIAR ORDEN', 'action_send_order'],
            ['button_channel', 'CANALES', 'action_source'],
            ['button_advance', 'ANTICIPO', 'action_add_advance'],
        ])

        if self.parent.environment == 'restaurant':
            rest_options = [
                ['button_delivery', 'AGREGAR DOMICILIO', 'action_delivery'],
                ['button_comment', 'NOTA', 'action_comment'],
                ['button_tip', 'AGREGAR PROPINA', 'action_tip'],
                ['button_tables', 'MESAS', 'action_tables'],
                ['button_consumer', 'CONSUMIDOR', 'action_consumer'],
                ['button_split_sale', 'DIVIDIR CUENTA', 'action_split_sale'],
            ]
            self.values.extend(rest_options)
        else:
            self.values.extend([
                ['button_product_info', 'INFO. PRODUCTO', 'action_info_product'],
                ['button_change_salesman', 'CAMBIO VENDEDOR', 'action_change_salesman'],
            ])
            pos_user = self.parent.type_pos_user
            if pos_user in ('order', 'salesman'):
                self.values.extend([
                    ['button_to_draft', 'BORRADOR', 'button_to_draft_pressed'],
                    ['button_to_quote', 'COTIZAR', 'button_to_quote_pressed'],
                ])
            #    self.button_to_quote = CustomButton(
            #         id='button_to_draft',
            #         parent=parent,
            #         title='COTIZAR',
            #         name_style='toolbar',
            #         method='button_to_quote_pressed'
            #     )
            #     self.button_to_quote.hide()
            #     hbox.addWidget(self.button_to_quote, 0)
            #     self.button_to_draft = CustomButton(
            #         id='button_to_draft',
            #         parent=parent,
            #         title='BORRADOR',
            #         icon=get_icon('back'),
            #         name_style='toolbar',
            #         method='button_to_draft_pressed'
            #     )
            #     self.button_to_draft.hide()
            #     hbox.addWidget(self.button_to_draft, 0)

    def hide_buttons(self):
        for button in self.values:
            getattr(self, button[0]).hide()

    def show_buttons(self):
        for button in self.values:
            if button[0] in ('button_to_draft', 'button_to_quote'):
                continue
            getattr(self, button[0]).show()


class ButtonsStacked(QWidget):

    def __init__(self, parent):
        super(ButtonsStacked, self).__init__()
        self.stacked = QStackedWidget()
        _size = 100
        hbox = QHBoxLayout()
        pos_user = parent.type_pos_user
        if parent.screen_size == 'small':
            _size = 70
        self.stacked.setMaximumHeight(_size)

        self.button_accept = CustomButton(
            id='button_accept',
            parent=parent,
            icon=get_icon('accept'),
            title='FINALIZAR',
            name_style='toolbar',
            method='button_accept_pressed'
        )
        self.button_checkout = CustomButton(
            id='button_checkout',
            parent=parent,
            icon=get_icon('cash'),
            title='FACTURAR',
            name_style='toolbar',
            method='button_checkout_pressed'
        )
        self.button_payment_method = CustomButton(
            id='button_payment',
            parent=parent,
            title='MEDIO DE PAGO',
            icon=get_icon('money'),
            name_style='toolbar',
            method='button_payment_pressed'
        )

        self.button_payment_term = CustomButton(
            id='button_payment_term',
            parent=parent,
            title='PLAZO DE PAGO',
            icon=get_icon('terminal_journal'),
            name_style='toolbar',
            method='action_payment_term'
        )

        self.button_transfer_to_folio = CustomButton(
            id='button_transfer_to_folio',
            parent=parent,
            title='TRANSFERIR A FOLIO',
            icon=get_icon('transfer_to_folio'),
            name_style='toolbar',
            method='action_transfer_to_folio'
        )

        if pos_user not in ('order', 'salesman'):
            self.stacked.addWidget(self.button_accept)
            self.stacked.addWidget(self.button_checkout)
            self.stacked.addWidget(self.button_payment_method)

            hbox.addWidget(self.stacked, 0)
        if pos_user in ('cashier', 'frontend_admin'):
            if parent._hotel_activated:
                hbox.addWidget(self.button_transfer_to_folio, 0)
            hbox.addWidget(self.button_payment_term, 0)
        if parent.environment == 'retail':
            if pos_user == 'cashier' and getattr(parent, 'button_to_draft_active', None):
                self.button_to_draft = CustomButton(
                    id='button_to_draft',
                    parent=parent,
                    title='VOLVER A BORRADOR',
                    icon=get_icon('draft'),
                    name_style='toolbar',
                    method='button_to_force_draft_pressed'
                )
                hbox.addWidget(self.button_to_draft, 0)
            elif pos_user in ('order', 'salesman'):
                self.button_send_order = CustomButton(
                    id='button_send_to_pay',
                    icon=get_icon('draft'),
                    parent=parent,
                    title='IR A PAGAR',
                    method='button_send_to_pay_pressed',
                    name_style='toolbar'
                )
                hbox.addWidget(self.button_send_order, 0)
        self.setLayout(hbox)


class ButtonsNumber(QGridLayout):

    def __init__(self, parent):
        # Numpad for Numbers
        super(ButtonsNumber, self).__init__()
        self.setHorizontalSpacing(1)
        self.setVerticalSpacing(1)


class Buttonpad(QWidget):

    def __init__(self, parent):
        super(Buttonpad, self).__init__()
        self._text = ''
        self._keyStates = {}
        self.set_keys()

    def set_keys(self):
        q = Qt
        self.keys_numbers = list(range(q.Key_0, q.Key_9 + 1))
        self.keys_alpha = list(range(q.Key_A, q.Key_Z + 1))
        self.keys_special = [
            q.Key_Asterisk, q.Key_Comma, q.Key_Period,
            q.Key_Minus, q.Key_Slash]
        self.show_keys = self.keys_numbers + self.keys_alpha + self.keys_special