From b77ccbc67b1f258f9d66421f38782a1f94ae3b9c Mon Sep 17 00:00:00 2001
From: bashonly <bashonly@protonmail.com>
Date: Tue, 23 Apr 2024 18:03:44 -0500
Subject: [PATCH] new impl

Authored by: bashonly
---
 test/test_networking.py          |  2 +-
 yt_dlp/networking/_curlcffi.py   | 37 ++++++++++++++++++++------------
 yt_dlp/networking/common.py      |  1 +
 yt_dlp/networking/impersonate.py | 14 +-----------
 4 files changed, 26 insertions(+), 28 deletions(-)

diff --git a/test/test_networking.py b/test/test_networking.py
index 8056d4f5d..b50f70d08 100644
--- a/test/test_networking.py
+++ b/test/test_networking.py
@@ -1013,7 +1013,7 @@ class TestCurlCFFIRequestHandler(TestRequestHandlerBase):
 
         from yt_dlp.networking._curlcffi import CurlCFFIResponseAdapter
         curl_res = curl_cffi.requests.Response()
-        res = CurlCFFIResponseAdapter(curl_res, None)
+        res = CurlCFFIResponseAdapter(curl_res)
 
         def mock_read(*args, **kwargs):
             try:
diff --git a/yt_dlp/networking/_curlcffi.py b/yt_dlp/networking/_curlcffi.py
index cb1f6dbda..c30db8dea 100644
--- a/yt_dlp/networking/_curlcffi.py
+++ b/yt_dlp/networking/_curlcffi.py
@@ -5,7 +5,13 @@ import math
 import urllib.parse
 
 from ._helper import InstanceStoreMixin, select_proxy
-from .common import Features, Request, register_preference, register_rh
+from .common import (
+    Features,
+    Request,
+    Response,
+    register_preference,
+    register_rh,
+)
 from .exceptions import (
     CertificateVerifyError,
     HTTPError,
@@ -14,11 +20,7 @@ from .exceptions import (
     SSLError,
     TransportError,
 )
-from .impersonate import (
-    ImpersonateRequestHandler,
-    ImpersonateResponse,
-    ImpersonateTarget,
-)
+from .impersonate import ImpersonateRequestHandler, ImpersonateTarget
 from ..dependencies import curl_cffi
 from ..utils import int_or_none
 
@@ -78,16 +80,15 @@ class CurlCFFIResponseReader(io.IOBase):
         super().close()
 
 
-class CurlCFFIResponseAdapter(ImpersonateResponse):
+class CurlCFFIResponseAdapter(Response):
     fp: CurlCFFIResponseReader
 
-    def __init__(self, response: curl_cffi.requests.Response, impersonate: ImpersonateTarget | None):
+    def __init__(self, response: curl_cffi.requests.Response):
         super().__init__(
             fp=CurlCFFIResponseReader(response),
             headers=response.headers,
             url=response.url,
-            status=response.status_code,
-            impersonate=impersonate)
+            status=response.status_code)
 
     def read(self, amt=None):
         try:
@@ -131,6 +132,15 @@ class CurlCFFIRH(ImpersonateRequestHandler, InstanceStoreMixin):
         extensions.pop('cookiejar', None)
         extensions.pop('timeout', None)
 
+    def send(self, request: Request) -> Response:
+        try:
+            response = super().send(request)
+        except HTTPError as e:
+            e.response.extensions['impersonate'] = self._get_request_target(request)
+            raise
+        response.extensions['impersonate'] = self._get_request_target(request)
+        return response
+
     def _send(self, request: Request):
         max_redirects_exceeded = False
         session: curl_cffi.requests.Session = self._get_instance(
@@ -177,8 +187,6 @@ class CurlCFFIRH(ImpersonateRequestHandler, InstanceStoreMixin):
         session.curl.setopt(CurlOpt.LOW_SPEED_LIMIT, 1)  # 1 byte per second
         session.curl.setopt(CurlOpt.LOW_SPEED_TIME, math.ceil(timeout))
 
-        impersonate_target = self._get_request_target(request)
-
         try:
             curl_response = session.request(
                 method=request.method,
@@ -188,7 +196,8 @@ class CurlCFFIRH(ImpersonateRequestHandler, InstanceStoreMixin):
                 verify=self.verify,
                 max_redirects=5,
                 timeout=timeout,
-                impersonate=self._SUPPORTED_IMPERSONATE_TARGET_MAP.get(impersonate_target),
+                impersonate=self._SUPPORTED_IMPERSONATE_TARGET_MAP.get(
+                    self._get_request_target(request)),
                 interface=self.source_address,
                 stream=True
             )
@@ -208,7 +217,7 @@ class CurlCFFIRH(ImpersonateRequestHandler, InstanceStoreMixin):
             else:
                 raise TransportError(cause=e) from e
 
-        response = CurlCFFIResponseAdapter(curl_response, impersonate_target)
+        response = CurlCFFIResponseAdapter(curl_response)
 
         if not 200 <= response.status < 300:
             raise HTTPError(response, redirect_loop=max_redirects_exceeded)
diff --git a/yt_dlp/networking/common.py b/yt_dlp/networking/common.py
index 4c66ba66a..d8328aa1f 100644
--- a/yt_dlp/networking/common.py
+++ b/yt_dlp/networking/common.py
@@ -517,6 +517,7 @@ class Response(io.IOBase):
             self.reason = reason or HTTPStatus(status).phrase
         except ValueError:
             self.reason = None
+        self.extensions = {}
 
     def readable(self):
         return self.fp.readable()
diff --git a/yt_dlp/networking/impersonate.py b/yt_dlp/networking/impersonate.py
index 9d56fd1c0..ca66180c7 100644
--- a/yt_dlp/networking/impersonate.py
+++ b/yt_dlp/networking/impersonate.py
@@ -5,7 +5,7 @@ from abc import ABC
 from dataclasses import dataclass
 from typing import Any
 
-from .common import RequestHandler, Response, register_preference
+from .common import RequestHandler, register_preference
 from .exceptions import UnsupportedRequest
 from ..compat.types import NoneType
 from ..utils import classproperty, join_nonempty
@@ -58,18 +58,6 @@ class ImpersonateTarget:
         return cls(**mobj.groupdict())
 
 
-class ImpersonateResponse(Response):
-    """
-    Wrapper class for a Response to an impersonated Request.
-
-    Parameters:
-    @param impersonate: the ImpersonateTarget used in the originating Request.
-    """
-    def __init__(self, *args, impersonate: ImpersonateTarget | None, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.impersonate = impersonate
-
-
 class ImpersonateRequestHandler(RequestHandler, ABC):
     """
     Base class for request handlers that support browser impersonation.