Implement container downloads

Author: sampsapenna

swift_upload_runner/api.py

@@ -5,7 +5,7 @@
 # import asyncio
 
 from .common import get_auth_instance
-from .download import FileDownloadProxy
+from .download import FileDownloadProxy, ContainerArchiveDownloadProxy
 
 
 async def handle_get_object(
@@ -23,9 +23,49 @@ async def handle_get_object(
     )
 
     resp = aiohttp.web.StreamResponse()
+
+    # Create headers
+    resp.headers["Content-Type"] = download.get_type()
+    resp.headers["Content-Length"] = str(download.get_size())
+
     await resp.prepare(request)
 
     # Create a task for writing the output into the StreamResponse
     await download.a_write_to_response(resp)
 
     return resp
+
+
+async def handle_get_container(
+        request: aiohttp.web.Request
+) -> aiohttp.web.StreamResponse:
+    """Handle a request for getting container contents as an archive."""
+    auth = get_auth_instance(request)
+
+    resp = aiohttp.web.StreamResponse()
+
+    project = request.match_info["project"]
+    container = request.match_info["container"]
+
+    # Create headers
+    resp.headers["Content-Type"] = "binary/octet-stream"
+    # Don't give content length, as the content length depends on
+    # compressibility
+    # Suggest {project_name}-{container}.tar as file name
+    disp_header = f'attachment; filename="{project}-{container}.tar"'
+    resp.headers["Content-Disposition"] = disp_header
+
+    await resp.prepare(request)
+
+    download = ContainerArchiveDownloadProxy(
+        auth,
+        project,
+        container
+    )
+
+    await download.a_begin_container_download()
+
+    # Create a task for writing the tarball into the StreamResponse
+    await download.a_write_to_response(resp)
+
+    return resp

swift_upload_runner/common.py

@@ -1,6 +1,9 @@
 """Common resources for swift-upload-runner."""
 
 
+import typing
+
+
 import aiohttp.web
 
 import keystoneauth1.session
@@ -35,6 +38,19 @@ def get_auth_instance(
             )
 
 
+def get_path_from_list(
+        to_parse: typing.List[str],
+        path_prefix: str
+) -> str:
+    """Parse a path from a list of path parts."""
+    ret = path_prefix
+
+    for i in to_parse:
+        ret += f"/{i}"
+
+    return ret.lstrip("/").rstrip("/")
+
+
 async def handle_delete_preflight(_) -> aiohttp.web.Response:
     """Serve correct response headers to allowed DELETE preflight query."""
     resp = aiohttp.web.Response(