import base64
import threading

import requests
from flask import jsonify, request, Flask

from models.exceptions.ServiceException import ServiceException
from services.game_service import GameService
from services.mqtt_service import MQTTService


class GameController:

    _game_service : GameService
    _api_url : str
    _broker_service : MQTTService
    _has_started : bool
    _auth_token : str

    def __init__(self, app : Flask, api_url : str, broker_service : MQTTService):
        self._game_service = GameService()
        self._api_url = api_url
        self._game_service.set_on_terminated(self._stop_event)
        self._broker_service = broker_service
        self._register_routes(app)
        self._auth_token = "0eed89e8-7625-4f8d-bf2a-0872aede0efb"

    def _register_routes(self, app):
        app.add_url_rule("/command/party/start", view_func=self.start_game, methods=['POST'])
        app.add_url_rule("/command/party/stop", view_func=self.stop_game, methods=['POST'])
        app.add_url_rule("/command/party/play", view_func=self.make_move, methods=['POST'])

    def start_game(self):
        try:
            data = request.get_json()
            if data is None:
                raise Exception("Data must be provided")

            white_name = data["white_name"]
            black_name = data["black_name"]
            time_control = int(data["time_control"])
            increment = int(data["increment"])
            timestamp = int(data["timestamp"])

            self._game_service.start(white_name, black_name, time_control, increment, timestamp)
            self._has_started = True
            return jsonify({"status": "ok", "message": "Game started"}), 200
        except ServiceException as ex:
            return jsonify({"status": "error", "message": f"An error occurred : {ex}"}), 400
        except Exception as ex:
            print(ex)
            return jsonify({"status": "error", "message": f"An error occurred : {ex}"}), 500

    def stop_game(self):
        try :
            self._game_service.stop()
        except Exception as e:
            print(e)

    def make_move(self):
        try:
            auth_token = request.headers.get("Authorization")
            if auth_token != "Bearer " + self._auth_token:
                return jsonify({"status": "error", "message": "Invalid authorization token"}), 401

            img = self._game_service.make_move()
            b64_img = base64.b64encode(img).decode('utf-8')
            payload = {
                "image": f"data:image/jpeg;base64,{b64_img}"
            }
            response = requests.post(self._api_url, json=payload, verify=False)
            print(response.status_code)

            data = response.json()
            fen = data.get("fen")
            self._game_service.add_move(fen)
            return jsonify({"status": "ok"}), 200

        except ServiceException as ex:
            return jsonify({"status": "error", "message": f"An error occurred : {ex}"}), 400
        except Exception as ex:
            print(ex)
            return jsonify({"status": "error", "message": f"An error occurred : {ex}"}), 500

    def _stop_event(self, game_data : str):
        try :
            print(f"Exporting game data : {game_data}")
            self._broker_service.publish("/customer/game/data", game_data, 2)
        except Exception as ex:
            print(ex)