depot/packages/servers/reflex-cache/reflex_cache/nix_cache.py

from functools import lru_cache

import requests

from reflex_cache.util import Uncached


class NixCacheFetcher:
    def __init__(self, caches):
        self.__caches = caches

    @lru_cache(maxsize=32768)
    def __try_all_cached(self, method, path, hint):
        fn = (
            requests.get
            if method == "get"
            else requests.head
            if method == "head"
            else Exception("invalid method")
        )

        bestState = 404

        print(f"  fetching [{method}] from any cache {path}")
        if hint != None:
            caches = []
            caches.append(hint)
            caches.extend(self.__caches)
        else:
            caches = self.__caches
        for cache in caches:
            try:
                rCache = fn(f"{cache}{path}", allow_redirects=True)
                if rCache.status_code < bestState:
                    bestState = rCache.status_code

                print(f"  {rCache.status_code} - [{method}] {cache}{path}")
                if bestState == 200:
                    r = (bestState, cache, rCache.content if method != "head" else False)
                    if path.endswith(".narinfo"):
                        return r
                    else:
                        raise Uncached(r)
            except requests.ConnectionError as e:
                print(e)

        # HACK: lru_cache does not cache results if an exception occurs
        # since we don't want to cache empty query results, we make use of this behavior
        raise Uncached((bestState, None, False))

    def try_all(self, method, path, hint=None):
        try:
            return self.__try_all_cached(method, path, hint)
        except Uncached as r:
            return r.args[0]