import os
import sys
import threading
import time

from termcolor import colored

from cpl.console.background_color_enum import BackgroundColorEnum
from cpl.console.foreground_color_enum import ForegroundColorEnum


class SpinnerThread(threading.Thread):
    r"""Thread to show spinner in terminal

    Parameter
    ---------
        msg_len: :class:`int`
            Length of the message
        foreground_color: :class:`cpl.console.foreground_color.ForegroundColorEnum`
            Foreground color of the spinner
        background_color: :class:`cpl.console.background_color.BackgroundColorEnum`
            Background color of the spinner
    """

    def __init__(self, msg_len: int, foreground_color: ForegroundColorEnum, background_color: BackgroundColorEnum):
        threading.Thread.__init__(self)

        self._msg_len = msg_len
        self._foreground_color = foreground_color
        self._background_color = background_color

        self._is_spinning = True
        self._exit = False

    @staticmethod
    def _spinner():
        r"""Selects active spinner char"""
        while True:
            for cursor in '|/-\\':
                yield cursor

    def _get_color_args(self) -> list[str]:
        r"""Creates color arguments"""
        color_args = []
        if self._foreground_color is not None:
            color_args.append(str(self._foreground_color.value))

        if self._background_color is not None:
            color_args.append(str(self._background_color.value))

        return color_args

    def run(self) -> None:
        r"""Entry point of thread, shows the spinner"""
        columns = 0
        if sys.platform == 'win32':
            columns = os.get_terminal_size().columns
        else:
            term_rows, term_columns = os.popen('stty size', 'r').read().split()
            columns = int(term_columns)

        end_msg = 'done'
        end_msg_pos = columns - self._msg_len - len(end_msg)
        if end_msg_pos > 0:
            print(f'{"" : >{end_msg_pos}}', end='')
        else:
            print('', end='')

        first = True
        spinner = self._spinner()
        while self._is_spinning:
            if first:
                first = False
                print(colored(f'{next(spinner): >{len(end_msg) - 1}}', *self._get_color_args()), end='')
            else:
                print(colored(f'{next(spinner): >{len(end_msg)}}', *self._get_color_args()), end='')
            time.sleep(0.1)
            back = ''
            for i in range(0, len(end_msg)):
                back += '\b'

            print(back, end='')
            sys.stdout.flush()

        if not self._exit:
            print(colored(end_msg, *self._get_color_args()), end='')

    def stop_spinning(self):
        r"""Stops the spinner"""
        self._is_spinning = False
        time.sleep(0.1)

    def exit(self):
        r"""Stops the spinner"""
        self._is_spinning = False
        self._exit = True
        time.sleep(0.1)