Add digest auth hack

Temporary hack until issue upstream is fixed
encode/httpx#1467

haphilipsjs/__init__.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional, Tuple, TypeVar, Union, cast
+from typing import Any, Dict, Optional, Tuple, TypeVar, Union, cast, Generator
 import httpx
 import logging
 import warnings
@@ -92,6 +92,35 @@ def __init__(self, data):
 
 T = TypeVar('T') 
 
+
+class DigestAuthCached(httpx.DigestAuth):
+    # Temporary digest fix until https://github.com/encode/httpx/issues/1467 is resolved
+    _challenge = None
+
+    def auth_flow(self, request: httpx.Request) -> Generator[httpx.Request, httpx.Response, None]:
+        if self._challenge:
+            request.headers["Authorization"] = self._build_auth_header(request, self._challenge)
+
+        response = yield request
+
+        if response.status_code != 401 or "www-authenticate" not in response.headers:
+            # If the response is not a 401 then we don't
+            # need to build an authenticated request.
+            return
+
+        for auth_header in response.headers.get_list("www-authenticate"):
+            if auth_header.lower().startswith("digest "):
+                break
+        else:
+            # If the response does not include a 'WWW-Authenticate: Digest ...'
+            # header, then we don't need to build an authenticated request.
+            return
+
+        self._challenge = self._parse_challenge(request, response, auth_header)
+        request.headers["Authorization"] = self._build_auth_header(request, self._challenge)
+        yield request
+
+
 class PhilipsTV(object):
 
     def __init__(self, host=None, api_version=DEFAULT_API_VERSION, secured_transport=None, username=None, password=None, verify=False, auth_shared_key=None):
@@ -138,7 +167,7 @@ def __init__(self, host=None, api_version=DEFAULT_API_VERSION, secured_transport
         self.session.headers["Accept"] = "application/json"
 
         if username and password:
-            self.session.auth = httpx.DigestAuth(username, password)
+            self.session.auth = DigestAuthCached(username, password)
 
     @property
     def quirk_playpause_spacebar(self):
@@ -333,7 +362,7 @@ async def pairRequest(self, app_id: str, app_name: str, device_name: str, device
 
     async def pairGrant(self, state: Dict[str, Any], pin: str):
         """Finish a pairing sequence"""
-        auth_handler = httpx.DigestAuth(
+        auth_handler = DigestAuthCached(
             state["device"]["id"],
             state["auth_key"]
         )