Improve API usability; minor fixes

src/pyrobusta/bindings/http_connection.py

@@ -98,7 +98,7 @@ async def _run_state_machine(self):
                 await self._response_handler(self._engine.resp_handler)
 
     async def _response_handler(self, resp_handler):
-        if "closure" == type(resp_handler).__name__:
+        if callable(resp_handler):
             if self._engine.get_response_header(b"transfer-encoding") == b"chunked":
                 for is_finished in resp_handler(self._send_buf):
                     await self.write(b"%x\r\n" % self._send_buf.size())

src/pyrobusta/protocol/http.py

@@ -387,9 +387,7 @@ def _get_mp_boundary(headers: dict) -> str:
         and return the boundary value.
         """
         content_type = headers.get("content-type")
-        if not content_type or not content_type.lower().startswith(
-            "multipart/form-data"
-        ):
+        if not content_type or not content_type.lower().startswith("multipart/"):
             return None
 
         parts = content_type.split(";")
@@ -440,6 +438,7 @@ def set_response_header(self, key: bytes, value: bytes):
         :param key: HTTP header key
         :param value: HTTP header value
         """
+        key = key.lower()
         if (
             key in self.resp_headers
             and (index := self.resp_headers.index(key)) % 2 == 0
@@ -810,7 +809,8 @@ def _app_endpoint_st(self, rx):
             return
 
         dtype, data = callback_response
-        if dtype.startswith("multipart/"):
+        if dtype.startswith("multipart/") and callable(data):
+            self.set_response_header(b"transfer-encoding", b"chunked")
             self.state = lambda _rx: self._generate_multipart_response(_rx, data, dtype)
             return
 

src/pyrobusta/protocol/http_file_server.py

@@ -254,7 +254,7 @@ def _traverse_dir(tx):
                 tx.write(b",")
 
             file_stat = stat(it)
-            obj = dumps(
+            data = dumps(
                 {
                     "path": it,
                     "size": str(file_stat[6]),
@@ -263,11 +263,11 @@ def _traverse_dir(tx):
             ).encode("ascii")
 
             written = 0
-            while written < len(obj):
+            while written < len(data):
                 to_write = tx.capacity - tx.size()
                 if not to_write:
                     raise BufferError()
-                tx.write(obj[written : written + to_write])
+                tx.write(data[written : written + to_write])
                 written += to_write
                 yield False
         tx.write(b"]\r\n")