From 3e35c79072651882ecf1548befae3139c14e5c26 Mon Sep 17 00:00:00 2001
From: Lucas Vieira <vieiralucas4@gmail.com>
Date: Mon, 9 Sep 2024 20:17:08 -0300
Subject: [PATCH] Forward briefer related environment variables to container

---
 pypi/src/briefer/main.py | 31 ++++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/pypi/src/briefer/main.py b/pypi/src/briefer/main.py
index 31fda27a..0108f8e3 100644
--- a/pypi/src/briefer/main.py
+++ b/pypi/src/briefer/main.py
@@ -1,9 +1,33 @@
+import os
 import docker
 import sys
 import socket
 import argparse
 import signal
 
+ENV_VARS = [
+    "LOG_LEVEL",
+    "API_URL",
+    "FRONTEND_URL",
+    "TLD",
+    "LOGIN_JWT_SECRET",
+    "AUTH_JWT_SECRET",
+    "AI_API_URL",
+    "AI_API_USERNAME",
+    "AI_API_PASSWORD",
+    "PYTHON_ALLOWED_LIBRARIES",
+    "POSTGRES_USERNAME",
+    "POSTGRES_PASSWORD",
+    "POSTGRES_HOSTNAME",
+    "POSTGRES_PORT",
+    "POSTGRES_DATABASE",
+    "ENVIRONMENT_VARIABLES_ENCRYPTION_KEY",
+    "WORKSPACE_SECRETS_ENCRYPTION_KEY",
+    "DATASOURCES_ENCRYPTION_KEY",
+    "JUPYTER_HOST",
+    "JUPYTER_PORT",
+    "JUPYTER_TOKEN"
+]
 
 def check_docker_running():
     client = docker.from_env()
@@ -94,12 +118,17 @@ def start_or_run_container(client, container_name, image, detach):
             'briefer_briefer_data': {'bind': '/home/briefer', 'mode': 'rw'}
         }
         print('Starting Briefer...')
+        env = {}
+        for var in ENV_VARS:
+            if var in os.environ:
+                env[var] = os.environ[var]
         container = client.containers.run(
             image,
             detach=True,
             ports={'3000/tcp': 3000, '8080/tcp': 8080},
             name=container_name,
-            volumes=volumes
+            volumes=volumes,
+            environment=env
         )
 
     if not detach: