from textual.app import App from pathlib import Path import os import configparser from typing import Any class ConfigHandler: def __init__(self, app: App): self.app: App = app self.config: configparser.ConfigParser = configparser.ConfigParser() self.plugin_defined_settings = {} self.config_dir: str = self.ensure_hidden_config_dir() self.load_settings() def define_plugin_setting(self, plugin_name: str, option_name: str, description: str, option_type: str, default_value: Any, on_changed_func = None): actual_type = None if option_type == "boolean": actual_type = bool elif option_type == "string": actual_type = str elif option_type == "float": actual_type = float elif option_type == "integer": actual_type = int else: raise Exception(f"Invalid type name \"{option_type}\".") if plugin_name in self.plugin_defined_settings: self.plugin_defined_settings[plugin_name][option_name] = { "option_name": option_name, "description": description, "type": actual_type, "default_value": default_value, "on_changed_func": on_changed_func } else: self.plugin_defined_settings[plugin_name] = {option_name: { "option_name": option_name, "description": description, "type": actual_type, "default_value": default_value, "on_changed_func": on_changed_func }} # user hasnt used this plugin before if self.get("plugin_" + plugin_name, option_name) == None: self.set("plugin_" + plugin_name, option_name, default_value) def ensure_hidden_config_dir(self): config_dir = Path.home() / ".berry" config_dir.mkdir(parents=True, exist_ok=True) return config_dir def get(self, section: str, option: str): if not self.config.has_section(section): self.config.add_section(section) if not self.config.has_option(section, option): self.set(section, option, "0") return "0" return self.config.get(section, option, fallback=None) def set(self, section: str, option: str, new_value: str): if not self.config.has_section(section): self.config.add_section(section) self.config.set(section, option, new_value) if section.startswith("plugin_"): section = section.removeprefix("plugin_") if self.plugin_defined_settings[section].get("on_changed_func", None) != None: self.plugin_defined_settings[section]["on_changed_func"](new_value) self.write_settings() def apply_settings(self): self.app.query_one("#code-editor").soft_wrap = bool(int(self.get("editor", "word_wrap"))) self.app.theme = self.get("appearance", "colour_theme") def write_settings(self): with open(self.config_dir / "config.ini", "w") as configfile: self.config.write(configfile) def load_settings(self): if os.path.isfile(self.config_dir / "config.ini"): self.config.read(self.config_dir / "config.ini") return self.config["config"] = { "version": "1" } self.config["editor"] = { "word_wrap": "0", "line_numbers": "1" } self.config["plugins"] = { "enabled": "1", "log": "1", "log_timeout": "10" } self.config["appearance"] = { "colour_theme": "textual-dark" } with open(self.config_dir / "config.ini", "w") as configfile: self.config.write(configfile)