From ee21f62da38bb6ba59fdc55c6dd8d5764b8cc820 Mon Sep 17 00:00:00 2001
From: Tim Wilhoit <twilhoi@amazon.com>
Date: Wed, 6 Sep 2023 13:45:30 -0500
Subject: [PATCH] Add script to install private CA certs to a kernel

---
 .../on-kernel-start.sh                        | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 scripts/install-private-ca-on-kernel/on-kernel-start.sh

diff --git a/scripts/install-private-ca-on-kernel/on-kernel-start.sh b/scripts/install-private-ca-on-kernel/on-kernel-start.sh
new file mode 100644
index 0000000..f56a168
--- /dev/null
+++ b/scripts/install-private-ca-on-kernel/on-kernel-start.sh
@@ -0,0 +1,39 @@
+#!/bin/bash
+
+set -eux
+
+cat > ~/.download_ca_certs.py <<EOL
+#!/opt/conda/bin/python3
+
+import sys
+import json
+import boto3
+import botocore
+import os
+
+# must be in same region as SM Studio
+CA_CERT_BUCKET = '[s3_bucket_name]'
+
+s3 = boto3.client('s3')
+objects = s3.list_objects(Bucket=CA_CERT_BUCKET)
+
+conda_cert_file = open('/opt/conda/ssl/cacert.pem', 'a+')
+
+# Download all cert files from bucket and append to conda ca cert file
+for object in objects['Contents']:
+    if '/' not in object['Key']:
+        s3.download_file(CA_CERT_BUCKET, object['Key'], object['Key'])
+        current_file = open(object['Key'], 'r')
+        contents = current_file.read()
+        if "BEGIN CERTIFICATE" in contents and "END CERTIFICATE" in contents:
+            conda_cert_file.write(contents + "\n")
+        current_file.close()
+        os.remove(object['Key'])
+
+conda_cert_file.close()
+
+EOL
+
+chmod +x ~/.download_ca_certs.py
+~/.download_ca_certs.py
+