Add Config CLI parser

and store config file.

Author: maxmynter

client/src/cli/commands/config.py

@@ -0,0 +1,39 @@
+import argparse
+from typing import Any
+
+import yaml
+
+from .util import Config
+
+
+def handle_config(args: argparse.Namespace) -> None:
+    config = Config()
+    if args.config_action == "view":
+        if args.key:
+            value = config.get(args.key)
+            print(f"{args.key}: {value}")
+        else:
+            print(yaml.dump(config.data))
+    elif args.config_action == "set":
+        config.set(args.key, args.value)
+
+
+def add_parser(subparsers: Any, parent: argparse.ArgumentParser) -> None:
+    parser: argparse.ArgumentParser = subparsers.add_parser(
+        "config", parents=[parent], description="Configure Jobq"
+    )
+
+    config_subparsers = parser.add_subparsers(
+        dest="config_action", help="Config actions"
+    )
+
+    view_parser = config_subparsers.add_parser(
+        "view", help="View current configuration"
+    )
+    view_parser.add_argument("--key", help="Specific config key to view")
+
+    set_parser = config_subparsers.add_parser("set", help="Set a configuration value")
+    set_parser.add_argument("key", help="Configuration key to set")
+    set_parser.add_argument("value", help="Value to set for the key")
+
+    parser.set_defaults(func=handle_config, config_action="view")

client/src/cli/commands/util.py

@@ -1,19 +1,68 @@
+from __future__ import annotations
+
 from collections.abc import Callable
 from functools import wraps
-from typing import Concatenate, ParamSpec, TypeVar
+from pathlib import Path
+from typing import Any, Concatenate, ParamSpec, TypeVar, cast
+
+import yaml
 
 import openapi_client
 from openapi_client.exceptions import ApiException
 
-# TODO: Factor backend host url into some kind of config file
-BACKEND_HOST = "http://localhost:8000"
+DEFAULT_CONFIG = {"backend": {"host": "http://localhost:8000"}}
+CONFIG_PATH = Path.home() / ".config" / "jobq" / "config.yaml"
+
+
+C = TypeVar("C", bound="Config")
+
+
+class Config:
+    _instance: Config | None = None
+    data: dict[str, Any]
+
+    def __new__(cls: type[C]) -> C:
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+            cls._instance.data = cls._instance.load()
+        return cast(C, cls._instance)
+
+    def load(self) -> dict[str, Any]:
+        if CONFIG_PATH.exists():
+            return yaml.safe_load(CONFIG_PATH.read_text())
+        return DEFAULT_CONFIG.copy()
+
+    def save(self) -> None:
+        CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
+        CONFIG_PATH.write_text(yaml.dump(self.data))
+
+    def get(self, key: str) -> Any:
+        keys = key.split(".")
+        value = self.data
+        for k in keys:
+            if isinstance(value, dict) and k in value:
+                value = value[k]
+            else:
+                raise ValueError(f"Didnot find {key} in config")
+        return value
+
+    def set(self, key: str, value: str) -> None:
+        keys = key.split(".")
+        data = self.data
+        for k in keys[:-1]:
+            if k not in data:
+                data[k] = {}
+            data = data[k]
+        data[keys[-1]] = value
+        self.save()
+
 
 T = TypeVar("T")
 P = ParamSpec("P")
 
 
 def _make_api_client() -> openapi_client.ApiClient:
-    api_config = openapi_client.Configuration(host=BACKEND_HOST)
+    api_config = openapi_client.Configuration(host=Config().get("backend.host"))
     return openapi_client.ApiClient(api_config)