refactor(plugin): Add typings

Author: tony

src/tmuxp/plugin.py

@@ -27,7 +27,7 @@
 
 
 if t.TYPE_CHECKING:
-    from typing_extensions import TypedDict
+    from typing_extensions import NotRequired, TypedDict, Unpack
 
     class VersionConstraints(TypedDict):
         version: t.Union[Version, str]
@@ -40,21 +40,65 @@ class TmuxpPluginVersionConstraints(TypedDict):
         tmuxp: VersionConstraints
         libtmux: VersionConstraints
 
+    class ConfigSchema(TypedDict):
+        plugin_name: NotRequired[str]
+        tmux_min_version: NotRequired[str]
+        tmux_max_version: NotRequired[str]
+        tmux_version_incompatible: NotRequired[t.List[str]]
+        libtmux_min_version: NotRequired[str]
+        libtmux_max_version: NotRequired[str]
+        libtmux_version_incompatible: NotRequired[t.List[str]]
+        tmuxp_min_version: NotRequired[str]
+        tmuxp_max_version: NotRequired[str]
+        tmuxp_version_incompatible: NotRequired[t.List[str]]
+
+
+class Config(t.TypedDict):
+    plugin_name: str
+    tmux_min_version: str
+    tmux_max_version: t.Optional[str]
+    tmux_version_incompatible: t.Optional[t.List[str]]
+    libtmux_min_version: str
+    libtmux_max_version: t.Optional[str]
+    libtmux_version_incompatible: t.Optional[t.List[str]]
+    tmuxp_min_version: str
+    tmuxp_max_version: t.Optional[str]
+    tmuxp_version_incompatible: t.Optional[t.List[str]]
+
+
+DEFAULT_CONFIG: "Config" = {
+    "plugin_name": "tmuxp-plugin",
+    "tmux_min_version": TMUX_MIN_VERSION,
+    "tmux_max_version": TMUX_MAX_VERSION,
+    "tmux_version_incompatible": None,
+    "libtmux_min_version": LIBTMUX_MIN_VERSION,
+    "libtmux_max_version": LIBTMUX_MAX_VERSION,
+    "libtmux_version_incompatible": None,
+    "tmuxp_min_version": TMUXP_MIN_VERSION,
+    "tmuxp_max_version": TMUXP_MAX_VERSION,
+    "tmuxp_version_incompatible": None,
+}
+
+
+def validate_plugin_config(config: "ConfigSchema") -> t.TypeGuard["Config"]:
+    return isinstance(config, dict)
+
+
+def setup_plugin_config(
+    config: "ConfigSchema", default_config: "Config" = DEFAULT_CONFIG
+) -> "Config":
+    new_config = config.copy()
+    for default_key, default_value in default_config.items():
+        if default_key not in new_config:
+            new_config[default_key] = default_value
+
+    assert validate_plugin_config(new_config)
+
+    return new_config
+
 
 class TmuxpPlugin:
-    def __init__(
-        self,
-        plugin_name: str = "tmuxp-plugin",
-        tmux_min_version: str = TMUX_MIN_VERSION,
-        tmux_max_version: t.Optional[str] = TMUX_MAX_VERSION,
-        tmux_version_incompatible: t.Optional[t.List[str]] = None,
-        libtmux_min_version: str = LIBTMUX_MIN_VERSION,
-        libtmux_max_version: t.Optional[str] = LIBTMUX_MAX_VERSION,
-        libtmux_version_incompatible: t.Optional[t.List[str]] = None,
-        tmuxp_min_version: str = TMUXP_MIN_VERSION,
-        tmuxp_max_version: t.Optional[str] = TMUXP_MAX_VERSION,
-        tmuxp_version_incompatible: t.Optional[t.List[str]] = None,
-    ) -> None:
+    def __init__(self, **kwargs: "Unpack[ConfigSchema]") -> None:
         """
         Initialize plugin.
 
@@ -95,7 +139,8 @@ def __init__(
             Versions of tmuxp that are incompatible with the plugin
 
         """
-        self.plugin_name = plugin_name
+        config = setup_plugin_config(config=kwargs)
+        self.plugin_name = config["plugin_name"]
 
         # Dependency versions
         self.tmux_version = get_version()
@@ -105,26 +150,26 @@ def __init__(
         self.version_constraints: "TmuxpPluginVersionConstraints" = {
             "tmux": {
                 "version": self.tmux_version,
-                "vmin": tmux_min_version,
-                "vmax": tmux_max_version,
-                "incompatible": tmux_version_incompatible
-                if tmux_version_incompatible
+                "vmin": config["tmux_min_version"],
+                "vmax": config["tmux_max_version"],
+                "incompatible": config["tmux_version_incompatible"]
+                if config["tmux_version_incompatible"]
                 else [],
             },
             "libtmux": {
                 "version": self.libtmux_version,
-                "vmin": libtmux_min_version,
-                "vmax": libtmux_max_version,
-                "incompatible": libtmux_version_incompatible
-                if libtmux_version_incompatible
+                "vmin": config["libtmux_min_version"],
+                "vmax": config["libtmux_max_version"],
+                "incompatible": config["libtmux_version_incompatible"]
+                if config["libtmux_version_incompatible"]
                 else [],
             },
             "tmuxp": {
                 "version": self.tmuxp_version,
-                "vmin": tmuxp_min_version,
-                "vmax": tmuxp_max_version,
-                "incompatible": tmuxp_version_incompatible
-                if tmuxp_version_incompatible
+                "vmin": config["tmuxp_min_version"],
+                "vmax": config["tmuxp_max_version"],
+                "incompatible": config["tmuxp_version_incompatible"]
+                if config["tmuxp_version_incompatible"]
                 else [],
             },
         }