From c13cf13ae3e301040dbaae23021e0e705d11de10 Mon Sep 17 00:00:00 2001
From: Sergey Fedoseev <fedoseev.sergey@gmail.com>
Date: Thu, 22 Apr 2021 17:49:13 +0500
Subject: [PATCH] WIP: package creation lambda large requests

---
 catalog/config.json.tmpl |   1 +
 lambdas/pkgpush/index.py | 106 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 107 insertions(+)

diff --git a/catalog/config.json.tmpl b/catalog/config.json.tmpl
index 05052dff096..4356d05f959 100644
--- a/catalog/config.json.tmpl
+++ b/catalog/config.json.tmpl
@@ -19,5 +19,6 @@
     "sentryDSN": "${SENTRY_DSN}",
     "mixpanelToken": "${MIXPANEL_TOKEN}",
     "analyticsBucket": "${ANALYTICS_BUCKET}",
+    "serviceBucket": "${SERVICE_BUCKET}",
     "mode": "${CATALOG_MODE}"
 }
diff --git a/lambdas/pkgpush/index.py b/lambdas/pkgpush/index.py
index b3153d09d28..e8a4b57e886 100644
--- a/lambdas/pkgpush/index.py
+++ b/lambdas/pkgpush/index.py
@@ -25,6 +25,8 @@
 PKG_FROM_FOLDER_MAX_PKG_SIZE = int(os.environ['PKG_FROM_FOLDER_MAX_PKG_SIZE'])
 PKG_FROM_FOLDER_MAX_FILES = int(os.environ['PKG_FROM_FOLDER_MAX_FILES'])
 
+SERVICE_BUCKET = os.environ['SERVICE_BUCKET']
+
 PACKAGE_ID_PROPS = {
     'registry': {
         'type': 'string',
@@ -120,6 +122,17 @@
 }
 
 
+PACKAGE_CREATE_SCHEMA = {
+    'type': 'object',
+    'properties': {
+        **PACKAGE_ID_PROPS,
+        **PACKAGE_BUILD_META_PROPS,
+    },
+    'required': PACKAGE_ID_PROPS,
+    'additionalProperties': False
+}
+
+
 # Monkey patch quilt3 S3ClientProvider, so it builds a client using user credentials.
 user_boto_session = None
 quilt3.data_transfer.S3ClientProvider.get_boto_session = staticmethod(lambda: user_boto_session)
@@ -360,3 +373,96 @@ def get_pkg(src_registry, data):
         pkg_max_size=PKG_FROM_FOLDER_MAX_PKG_SIZE,
         pkg_max_files=PKG_FROM_FOLDER_MAX_FILES,
     )
+
+
+@api(cors_origins=get_default_origins(), request_class=ELBRequest)
+@api_exception_handler
+@auth
+@json_api({'type': 'string', 'minLength': 1, 'maxLength': 1024})
+@setup_telemetry
+def create_package(request):
+    version_id = request.data
+
+    with tempfile.NamedTemporaryFile('r', encoding='utf-8') as tmp_file:
+        # FIXME: check file size
+        quilt3.data_transfer.copy_file(
+            PhysicalKey(SERVICE_BUCKET, 'user-requests/create-package', version_id),
+            PhysicalKey.from_path(tmp_file.name),
+        )
+
+        json_iterator = map(json.JSONDecoder().decode, tmp_file)
+
+        # FIXME: validate with schema.
+        data = next(json_iterator)
+
+        handle = data['name']
+        registry = data['registry']
+
+        try:
+            package_registry = get_registry(registry)
+
+            meta = data.get('meta')
+            message = data.get('message')
+            quilt3.util.validate_package_name(handle)
+            pkg = quilt3.Package()
+            if meta is not None:
+                pkg.set_meta(meta)
+            pkg._validate_with_workflow(
+                registry=package_registry,
+                workflow=data.get('workflow', ...),
+                message=message,
+            )
+
+            size_to_hash = 0
+            files_to_hash = 0
+            for entry in json_iterator:
+                try:
+                    physical_key = PhysicalKey.from_url(entry['physical_key'])
+                except ValueError:
+                    raise ApiException(HTTPStatus.BAD_REQUEST, f"{entry['physical_key']} is not a valid s3 URL.")
+                if physical_key.is_local():
+                    raise ApiException(HTTPStatus.BAD_REQUEST, f"{str(physical_key)} is not in S3.")
+                logical_key = entry['logical_key']
+
+                hash_ = entry.get('hash')
+                obj_size = entry.get('size')
+                meta = entry.get('meta')
+
+                if hash_ and obj_size:
+                    hash_obj = dict(type='SHA256', value=hash_)
+                    pkg.set(
+                        logical_key,
+                        quilt3.packages.PackageEntry(
+                            physical_key,
+                            obj_size,
+                            hash_obj,
+                            meta,
+                        )
+                    )
+                else:
+                    pkg.set(logical_key, str(physical_key), meta)
+                    size_to_hash += pkg[logical_key].size
+                    files_to_hash += 1
+                    # FIXME: raise error if these are beyond our limits.
+
+        except quilt3.util.QuiltException as qe:
+            raise ApiException(HTTPStatus.BAD_REQUEST, qe.message)
+
+        try:
+            top_hash = pkg._build(
+                name=handle,
+                registry=registry,
+                message=message,
+            )
+        except ClientError as boto_error:
+            boto_response = boto_error.response
+            status_code = boto_response['ResponseMetadata']['HTTPStatusCode']
+            message = "{0}: {1}".format(
+                boto_response['Error']['Code'],
+                boto_response['Error']['Message']
+            )
+            raise ApiException(status_code, message)
+
+        return {
+            'top_hash': top_hash,
+        }