feat: update runtime versions, use uv in docker builds (#208)

Author: hrodmn

.gitignore

@@ -6,7 +6,7 @@ lib/**/*.d.ts
 dist
 docs
 __pycache__
-.venv
+/.venv
 .tox
 tests/*.egg*
 tests/*venv*

integration_tests/cdk/package-lock.json

@@ -41,7 +41,7 @@ export class PgStacApiLambdaRuntime extends Construct {
   constructor(
     scope: Construct,
     id: string,
-    props: PgStacApiLambdaRuntimeProps
+    props: PgStacApiLambdaRuntimeProps,
   ) {
     super(scope, id);
 
@@ -60,8 +60,8 @@ export class PgStacApiLambdaRuntime extends Construct {
         if (!isValidExtension(ext)) {
           throw new Error(
             `Invalid extension: "${ext}". Must be one of: ${Object.values(
-              EXTENSIONS
-            ).join(", ")}`
+              EXTENSIONS,
+            ).join(", ")}`,
           );
         }
       }
@@ -75,7 +75,7 @@ export class PgStacApiLambdaRuntime extends Construct {
     this.lambdaFunction = new lambda.Function(this, "lambda", {
       // defaults
       runtime: lambda.Runtime.PYTHON_3_12,
-      handler: "handler.handler",
+      handler: "stac_api.handler.handler",
       memorySize: 8192,
       logRetention: aws_logs.RetentionDays.ONE_WEEK,
       timeout: Duration.seconds(30),
@@ -106,7 +106,7 @@ export class PgStacApiLambdaRuntime extends Construct {
       this.lambdaFunction.connections.allowTo(
         props.db,
         ec2.Port.tcp(5432),
-        "allow connections from stac-fastapi-pgstac"
+        "allow connections from stac-fastapi-pgstac",
       );
     }
   }

lib/stac-api/index.ts

@@ -1,14 +1,33 @@
 ARG PYTHON_VERSION
 FROM --platform=linux/amd64 public.ecr.aws/lambda/python:${PYTHON_VERSION}
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 
 WORKDIR /tmp
-RUN python -m pip install pip -U
+COPY stac-api/runtime/uv.lock stac-api/runtime/pyproject.toml ./
+COPY stac-api/runtime/src/ ./src/
+COPY utils/utils.py /asset/
 
-COPY stac-api/runtime/requirements.txt requirements.txt
-RUN python -m pip install -r requirements.txt -t /asset --no-binary pydantic
+RUN <<EOF
+uv export --locked --no-editable --no-dev --format requirements.txt -o requirements.txt
+uv pip install \
+  --compile-bytecode \
+  --no-binary pydantic \
+  --target /asset \
+  --no-cache-dir \
+  --disable-pip-version-check \
+  -r requirements.txt
+EOF
 
-RUN mkdir -p /asset/src
-COPY stac-api/runtime/src/*.py /asset/
-COPY utils/utils.py /asset/
+# Reduce package size and remove useless files
+WORKDIR /asset
+RUN <<EOF
+dnf install -y findutils && \
+dnf clean all && \
+rm -rf /var/cache/dnf
+find . -type f -name '*.pyc' | while read f; do n=$(echo $f | sed 's/__pycache__\///' | sed 's/.cpython-[0-9]*//'); cp $f $n; done;
+find . -type d -a -name '__pycache__' -print0 | xargs -0 rm -rf
+find . -type f -a -name '*.py' -print0 | xargs -0 rm -f
+find . -type d -a -name 'tests' -print0 | xargs -0 rm -rf
+EOF
 
 CMD ["echo", "hello world"]

lib/stac-api/runtime/Dockerfile

@@ -0,0 +1,14 @@
+[project]
+name = "stac-api"
+version = "0.0.0"
+description = "stac-api runtime"
+authors = [{ name = "hrodmn", email = "[email protected]" }]
+requires-python = ">=3.12"
+dependencies = [
+    "stac-fastapi-pgstac[awslambda]>=6.0,<6.1",
+    "starlette-cramjam>=0.4,<0.5",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"

lib/stac-api/runtime/pyproject.toml

@@ -13,7 +13,7 @@ export class StacAuthProxyLambdaRuntime extends Construct {
   constructor(
     scope: Construct,
     id: string,
-    props: StacAuthProxyLambdaRuntimeProps
+    props: StacAuthProxyLambdaRuntimeProps,
   ) {
     super(scope, id);
 
@@ -22,7 +22,7 @@ export class StacAuthProxyLambdaRuntime extends Construct {
 
     this.lambdaFunction = new lambda.Function(this, "lambda", {
       runtime: lambda.Runtime.PYTHON_3_13,
-      handler: "handler.handler",
+      handler: "stac_auth_proxy_api.handler.handler",
       memorySize: 8192,
       logRetention: cdk.aws_logs.RetentionDays.ONE_WEEK,
       timeout: cdk.Duration.seconds(30),
@@ -114,7 +114,7 @@ export class StacAuthProxyLambda extends Construct {
     const runtime = new StacAuthProxyLambdaRuntime(
       this,
       "runtime",
-      runtimeProps
+      runtimeProps,
     );
     this.lambdaFunction = runtime.lambdaFunction;