import os.path
import json
import sys
from random import randrange, randint


class MaidDataReader:
    PATH = os.path.join(os.path.dirname(__file__), "..", "..", "resources", "data.json")

    def __init__(self):
        self.data = None
        with open(MaidDataReader.PATH, "r") as f:
            content = f.read()
            self.data = json.loads(content)

        if self.data is None:
            sys.exit("Could not read JSON data.")

        rare_color_count = len(self.data["common"]["colors"]["rare"])
        self.color_random_max = rare_color_count + 2 * len(self.data["common"]["colors"]["double_chance"])
        self.rare_color_start_index = self.color_random_max - rare_color_count

    def get_random_color(self):
        index = randrange(self.color_random_max)
        if index >= self.rare_color_start_index:
            return self.data["common"]["colors"]["rare"][index - self.rare_color_start_index]

        return self.data["common"]["colors"]["double_chance"][int(index / 2)]

    def get_butler_random_suit_color(self):
        return self.__get_butler_random_color(self.data["butler"]["colors"]["suit"])

    def get_butler_random_hair_color(self):
        return self.__get_butler_random_color(self.data["butler"]["colors"]["hair"])

    @staticmethod
    def generate_master_age(type_index):
        if type_index == 0:
            return randint(1, 6) + randint(1, 6)
        elif type_index == 1:
            return randint(1, 6) + randint(1, 6) + 5
        elif type_index == 2:
            return randint(1, 6) + randint(1, 6) + 8
        elif type_index == 3:
            return randint(1, 6) + randint(1, 6) + 12
        elif type_index == 4:
            return int(str(randint(1, 6)) + str(randint(1, 6)))
        elif type_index == 5:
            return int(str(randint(1, 6)) + str(randint(1, 6))) + 10

        raise ValueError("Invalid master type index: {}".format(type_index))

    def __get_butler_random_color(self, color_list):
        index = randrange(len(color_list) + 1)
        if index >= len(color_list):
            return self.get_random_color()

        return color_list[index]