from kivy.graphics.context_instructions import Color
from kivy.graphics.vertex_instructions import Line, SmoothLine
from kivy.uix.boxlayout import BoxLayout
from kivy.uix.popup import Popup

from gui.popups import ConfigAIPopup


class Controls(BoxLayout):
    def __init__(self, **kwargs):
        super(Controls, self).__init__(**kwargs)
        self.status = None
        self.status_node = None

    def set_status(self, msg, at_node=None):
        self.status = msg
        self.status_node = at_node or self.katrain.game and self.katrain.game.current_node
        self.info.text = msg
        self.update_evaluation()

    def select_mode(self, mode):
        if mode == "analyze":
            self.analyze_tab_button.trigger_action(duration=0)
        else:
            self.play_tab_button.trigger_action(duration=0)

    def switch_mode(self):
        if self.play_tab_button.state == "down":
            self.select_mode("analyze")
        else:
            self.select_mode("play")

    def player_mode(self, player):
        return self.player_mode_groups[player].value

    def ai_mode(self, player):
        return self.ai_mode_groups[player].text

    def on_size(self, *args):
        self.update_evaluation()

    # handles showing completed analysis and score graph
    def update_evaluation(self):
        katrain = self.katrain
        current_node = katrain.game and katrain.game.current_node

        info = ""
        if current_node is self.status_node or (self.status is not None and self.status_node is None and current_node.is_root):  # startup errors on root
            info += self.status + "\n"
        else:
            self.status_node = None

        if current_node:
            move = current_node.single_move
            next_player_is_human_or_both_robots = current_node.player and ("ai" not in self.player_mode(current_node.player) or "ai" in self.player_mode(current_node.next_player))
            current_player_is_ai_playing_human = current_node.player and "ai" in self.player_mode(current_node.player) and "ai" not in self.player_mode(current_node.next_player)
            if next_player_is_human_or_both_robots and not current_node.is_root and move:
                info += current_node.comment(teach="undo" in self.player_mode(current_node.player), hints=self.hints.active)
            elif current_player_is_ai_playing_human and current_node.parent:
                info += current_node.parent.comment(teach="undo" in self.player_mode(current_node.next_player), hints=self.hints.active)

            if current_node.analysis_ready:
                self.score.text = current_node.format_score()
                self.win_rate.text = current_node.format_win_rate()
                if move and next_player_is_human_or_both_robots:  # don't immediately hide this when an ai moves comes in
                    points_lost = current_node.points_lost
                    self.score_change.label = f"Points lost for {move.player}" if points_lost and points_lost > 0 else f"Points gained for {move.player}"
                    self.score_change.text = f"{abs(points_lost):.1f}" if points_lost else "..."
                elif not current_player_is_ai_playing_human:
                    self.score_change.label = f"Points lost"
                    self.score_change.text = ""

            self.graph.update_value(current_node)

        self.info.text = info

    def configure_ais(self):
        config_popup = Popup(title="Edit AI Settings", size_hint=(0.9, 0.9))
        popup_contents = ConfigAIPopup(self.katrain, config_popup, {self.ai_mode("B"), self.ai_mode("W")})
        config_popup.add_widget(popup_contents)
        config_popup.open()