sst 2.1.35 → 2.2.1

package/support/nodejs-runtime/index.mjs

@@ -75,7 +75,7 @@ while (true) {
   }, 1e3 * 60 * 15);
   let request;
   let response;
-  let context = {};
+  let context;
   try {
     const result = await fetch({
       path: `/runtime/invocation/next`,
@@ -84,7 +84,43 @@ while (true) {
     });
     context = {
       awsRequestId: result.headers["lambda-runtime-aws-request-id"],
-      invokedFunctionArn: result.headers["lambda-runtime-invoked-function-arn"]
+      invokedFunctionArn: result.headers["lambda-runtime-invoked-function-arn"],
+      getRemainingTimeInMillis: () => Math.max(
+        Number(result.headers["lambda-runtime-deadline-ms"]) - Date.now(),
+        0
+      ),
+      identity: JSON.parse(result.headers["lambda-runtime-cognito-identity"]) ?? void 0,
+      clientContext: JSON.parse(result.headers["lambda-runtime-client-context"]) ?? void 0,
+      functionName: process.env.AWS_LAMBDA_FUNCTION_NAME,
+      functionVersion: process.env.AWS_LAMBDA_FUNCTION_VERSION,
+      memoryLimitInMB: process.env.AWS_LAMBDA_FUNCTION_MEMORY_SIZE,
+      logGroupName: result.headers["lambda-runtime-log-group-name"],
+      logStreamName: result.headers["lambda-runtime-log-stream-name"],
+      callbackWaitsForEmptyEventLoop: {
+        set value(_value) {
+          throw new Error(
+            "`callbackWaitsForEmptyEventLoop` on lambda Context is not implemented by SST Live Lambda Development."
+          );
+        },
+        get value() {
+          return true;
+        }
+      }.value,
+      done() {
+        throw new Error(
+          "`done` on lambda Context is not implemented by SST Live Lambda Development."
+        );
+      },
+      fail() {
+        throw new Error(
+          "`fail` on lambda Context is not implemented by SST Live Lambda Development."
+        );
+      },
+      succeed() {
+        throw new Error(
+          "`succeed` on lambda Context is not implemented by SST Live Lambda Development."
+        );
+      }
     };
     request = JSON.parse(result.body);
   } catch {

package/support/python-runtime/runtime.py

@@ -21,14 +21,18 @@ class ClientContext(object):
     #__slots__ = ['custom', 'env', 'client']
 
 class Context(object):
-    def __init__(self, invoked_function_arn, aws_request_id, deadline_ms, identity, client_context):
+    def __init__(self, invoked_function_arn, aws_request_id, deadline_ms, identity, client_context, log_group_name, log_stream_name):
         self.function_name = os.environ['AWS_LAMBDA_FUNCTION_NAME']
         self.invoked_function_arn = invoked_function_arn
         self.aws_request_id = aws_request_id
         self.memory_limit_in_mb = os.environ['AWS_LAMBDA_FUNCTION_MEMORY_SIZE']
         self.deadline_ms = deadline_ms
-        self.identity = Identity(**json.loads(identity))
-        self.client_context = ClientContext(**json.loads(client_context))
+        # If identity is null, we want to mimick AWS behavior and return an object with None values
+        self.identity = Identity(**json.loads(identity)) if identity != 'null' else Identity(cognito_identity_id = None, cognito_identity_pool_id = None)
+        # If client_context is null, we want to mimick AWS behavior and return None
+        self.client_context = ClientContext(**json.loads(client_context)) if client_context != 'null' else None
+        self.log_group_name = log_group_name
+        self.log_stream_name = log_stream_name
 
     def get_remaining_time_in_millis(self):
         return int(max(self.deadline_ms - int(round(time() * 1000)), 0))
@@ -75,7 +79,9 @@ if __name__ == '__main__':
         r.getheader('Lambda-Runtime-Aws-Request-Id'),
         r.getheader('Lambda-Runtime-Deadline-Ms'),
         r.getheader('Lambda-Runtime-Cognito-Identity'),
-        r.getheader('Lambda-Runtime-Client-Context')
+        r.getheader('Lambda-Runtime-Client-Context'),
+        r.getheader('Lambda-Runtime-Log-Group-Name'),
+        r.getheader('Lambda-Runtime-Log-Stream-Name')
     )
 
     # invoke handler