kedder/openvario-shell

import json
import os
from typing import Optional, TypeVar

from ovshell import api

JT = TypeVar("JT", bound=api.JsonType)


class StoredSettingsImpl(api.StoredSettings):
    _settings: dict[str, Optional[api.JsonType]]
    _filename: Optional[str]

    def __init__(
        self,
        settings: Optional[dict[str, Optional[api.JsonType]]] = None,
        filename: Optional[str] = None,
    ) -> None:
        self._settings = settings or {}
        self._filename = filename

    @classmethod
    def load(cls, filename: str) -> "StoredSettingsImpl":
        if not os.path.exists(filename):
            return StoredSettingsImpl(filename=filename)

        with open(filename) as f:
            settings = json.load(f)
        return StoredSettingsImpl(settings, filename)

    def save(self) -> None:
        assert self._filename is not None
        with open(self._filename, "w") as f:
            json.dump(self._settings, f, indent=4, sort_keys=True)

    def setdefault(self, key: str, value: api.JsonType) -> None:
        self._settings.setdefault(key, value)

    def set(self, key: str, value: Optional[api.JsonType], save: bool = False):
        self._settings[key] = value
        if save:
            self.save()

    def get(
        self, key: str, type: type[JT], default: Optional[JT] = None
    ) -> Optional[JT]:
        v = self._settings.get(key, default)
        return v if isinstance(v, type) else None

    def getstrict(self, key: str, type: type[JT]) -> JT:
        v = self.get(key, type)
        assert v is not None
        return v