Add Cache-Control to headers.

Author: paul-1

adafruit_httpserver/response.py

@@ -29,7 +29,7 @@ class HTTPResponse:
     status: HTTPStatus
     headers: Dict[str, str]
     content_type: str
-
+    cache: Optional[int]
     filename: Optional[str]
     root_path: str
 
@@ -41,6 +41,7 @@ def __init__(  # pylint: disable=too-many-arguments
         body: str = "",
         headers: Dict[str, str] = None,
         content_type: str = MIMEType.TYPE_TXT,
+        cache: Optional[int] = 0,
         filename: Optional[str] = None,
         root_path: str = "",
         http_version: str = "HTTP/1.1",
@@ -54,6 +55,7 @@ def __init__(  # pylint: disable=too-many-arguments
         self.body = body
         self.headers = headers or {}
         self.content_type = content_type
+        self.cache = cache
         self.filename = filename
         self.root_path = root_path
         self.http_version = http_version
@@ -64,6 +66,7 @@ def _construct_response_bytes(  # pylint: disable=too-many-arguments
         status: HTTPStatus = CommonHTTPStatus.OK_200,
         content_type: str = MIMEType.TYPE_TXT,
         content_length: Union[int, None] = None,
+        cache: int = 0,
         headers: Dict[str, str] = None,
         body: str = "",
     ) -> bytes:
@@ -77,6 +80,8 @@ def _construct_response_bytes(  # pylint: disable=too-many-arguments
         headers.setdefault("Content-Length", content_length or len(body))
         headers.setdefault("Connection", "close")
 
+        response += f"Cache-Control: max-age={cache}\r\n"
+
         for header, value in headers.items():
             response += f"{header}: {value}\r\n"
 
@@ -128,6 +133,7 @@ def _send_response(  # pylint: disable=too-many-arguments
             self._construct_response_bytes(
                 status=status,
                 content_type=content_type,
+                cache=self.cache,
                 headers=headers,
                 body=body,
             ),
@@ -147,6 +153,7 @@ def _send_file_response(  # pylint: disable=too-many-arguments
                 status=self.status,
                 content_type=MIMEType.from_file_name(filename),
                 content_length=file_length,
+                cache=self.cache,
                 headers=headers,
             ),
         )

adafruit_httpserver/server.py

@@ -167,7 +167,7 @@ def poll(self):
                 # If no handler exists and request method is GET, try to serve a file.
                 elif request.method == HTTPMethod.GET:
                     response = HTTPResponse(
-                        filename=request.path, root_path=self.root_path
+                        filename=request.path, root_path=self.root_path, cache=604800
                     )
 
                 # If no handler exists and request method is not GET, return 400 Bad Request.