import yaml


class Settings:
    def __init__(self, config_file):
        self.config_file = config_file
        self.load()

    def default_get(_, data, name, default):
        value = default
        try:
            value = data.get(name, default)
        except:
            pass
        return value

    def load(self):
        try:
            with open(self.config_file, "r") as f:
                data = yaml.load(f, Loader=yaml.FullLoader)
        except:
            data = None

        self.selected_theme = self.default_get(data, "selected_theme", "Default")
        self.server_name = self.default_get(data, "server_name", "")
        self.server_port = self.default_get(data, "server_port", 0)
        self.server_share = self.default_get(data, "server_share", False)
        self.output_image_format = self.default_get(data, "output_image_format", "png")
        self.output_video_format = self.default_get(data, "output_video_format", "mp4")
        self.output_video_codec = self.default_get(
            data, "output_video_codec", "libx264"
        )
        self.video_quality = self.default_get(data, "video_quality", 14)
        self.clear_output = self.default_get(data, "clear_output", True)
        self.max_threads = self.default_get(data, "max_threads", 2)
        self.memory_limit = self.default_get(data, "memory_limit", 0)
        self.provider = self.default_get(data, "provider", "cuda")
        self.force_cpu = self.default_get(data, "force_cpu", False)
        self.output_template = self.default_get(
            data, "output_template", "{file}_{time}"
        )
        self.use_os_temp_folder = self.default_get(data, "use_os_temp_folder", False)
        self.output_show_video = self.default_get(data, "output_show_video", True)
        self.launch_browser = self.default_get(data, "launch_browser", True)

    def save(self):
        data = {
            "selected_theme": self.selected_theme,
            "server_name": self.server_name,
            "server_port": self.server_port,
            "server_share": self.server_share,
            "output_image_format": self.output_image_format,
            "output_video_format": self.output_video_format,
            "output_video_codec": self.output_video_codec,
            "video_quality": self.video_quality,
            "clear_output": self.clear_output,
            "max_threads": self.max_threads,
            "memory_limit": self.memory_limit,
            "provider": self.provider,
            "force_cpu": self.force_cpu,
            "output_template": self.output_template,
            "use_os_temp_folder": self.use_os_temp_folder,
            "output_show_video": self.output_show_video,
        }
        with open(self.config_file, "w") as f:
            yaml.dump(data, f)