summaryrefslogtreecommitdiff
path: root/searx/favicons/config.py
blob: bf53b0821f05f33b1a9133d32deace383a30453d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# SPDX-License-Identifier: AGPL-3.0-or-later
# pylint: disable=missing-module-docstring

from __future__ import annotations

import pathlib
import msgspec

from .cache import FaviconCacheConfig
from .proxy import FaviconProxyConfig

CONFIG_SCHEMA: int = 1
"""Version of the configuration schema."""

TOML_CACHE_CFG: dict[str, "FaviconConfig"] = {}
"""Cache config objects by TOML's filename."""

DEFAULT_CFG_TOML_PATH = pathlib.Path(__file__).parent / "favicons.toml"


class FaviconConfig(msgspec.Struct):  # pylint: disable=too-few-public-methods
    """The class aggregates configurations of the favicon tools"""

    cfg_schema: int
    """Config's schema version.  The specification of the version of the schema
    is mandatory, currently only version :py:obj:`CONFIG_SCHEMA` is supported.
    By specifying a version, it is possible to ensure downward compatibility in
    the event of future changes to the configuration schema"""

    cache: FaviconCacheConfig = FaviconCacheConfig
    """Setup of the :py:obj:`.cache.FaviconCacheConfig`."""

    proxy: FaviconProxyConfig = FaviconCacheConfig
    """Setup of the :py:obj:`.proxy.FaviconProxyConfig`."""

    @classmethod
    def from_toml_file(cls, cfg_file: pathlib.Path, use_cache: bool) -> "FaviconConfig":
        """Create a config object from a TOML file, the ``use_cache`` argument
        specifies whether a cache should be used.
        """

        cached = TOML_CACHE_CFG.get(str(cfg_file))
        if use_cache and cached:
            return cached

        with cfg_file.open("rb") as f:
            data = f.read()

        cfg = msgspec.toml.decode(data, type=_FaviconConfig)
        schema = cfg.favicons.cfg_schema
        if schema != CONFIG_SCHEMA:
            raise ValueError(
                f"config schema version {CONFIG_SCHEMA} is needed, version {schema} is given in {cfg_file}"
            )

        cfg = cfg.favicons
        if use_cache and cached:
            TOML_CACHE_CFG[str(cfg_file.resolve())] = cfg

        return cfg


class _FaviconConfig(msgspec.Struct):  # pylint: disable=too-few-public-methods
    # wrapper struct for root object "favicons."
    favicons: FaviconConfig