// prepare the s3 bucket for the lambda logs
const bucket = new s3.Bucket(this, 'DemoBucket');

// prepare the Function
const fn = new Function(this, 'Handler', {
  code: lambda.Code.fromAsset(path.join(__dirname,
    '../aws-lambda-extensions/s3-logs-extension-demo/functionsrc')),
  runtime: lambda.Runtime.PYTHON_3_8,
  handler: 'lambda_function.lambda_handler',
  memorySize: 128,
  environment: {
    S3_BUCKET_NAME: bucket.bucketName,
  },
});
bucket.grantWrite(fn);

// plug the `s3-logs-extension` in the lambda function
fn.addExtension(new S3LogsExtension(this, 'S3BucketExtention').extension);