Move optimizations to prebuild stage

Author: simonkurtz-MSFT

.devcontainer/Dockerfile

@@ -36,6 +36,11 @@ RUN --mount=type=cache,target=/var/cache/apt \
 # Switch to vscode user
 USER vscode
 
+# Configure Azure CLI for Codespaces (prebuild optimization)
+RUN az config set core.login_experience_v2=off 2>/dev/null || true && \
+    az extension add --name containerapp --only-show-errors 2>/dev/null || true && \
+    az extension add --name front-door --only-show-errors 2>/dev/null || true
+
 # Configure shell to auto-activate virtual environment
 RUN echo "# Auto-activate APIM Samples virtual environment" >> ~/.bashrc && \
     echo "if [ -f /workspaces/Apim-Samples/.venv/bin/activate ]; then" >> ~/.bashrc && \

.devcontainer/devcontainer.json

@@ -122,7 +122,7 @@
   ],
   "updateContentCommand": [
     "bash", "-c",
-    "echo '📦 Installing/updating Python packages from requirements.txt...' && source /workspaces/Apim-Samples/.venv/bin/activate && pip install -r requirements.txt && pip install pytest pytest-cov coverage ipykernel && echo '✅ Python packages installed/updated' && python setup/setup_python_path.py --generate-env && echo '✅ Environment configuration updated' && echo '🔧 Registering Jupyter kernel...' && python -m ipykernel install --user --name=apim-samples --display-name='APIM Samples Python 3.12' && echo '✅ Jupyter kernel registered'"
+    "echo '📦 Installing/updating Python packages from requirements.txt...' && source /workspaces/Apim-Samples/.venv/bin/activate && pip install -r requirements.txt && pip install pytest pytest-cov coverage ipykernel && echo '✅ Python packages installed/updated' && python setup/setup_python_path.py --generate-env && echo '✅ Environment configuration updated' && echo '🔧 Registering Jupyter kernel...' && python -m ipykernel install --user --name=apim-samples --display-name='APIM Samples Python 3.12' && echo '✅ Jupyter kernel registered' && echo '⚙️ Configuring Azure CLI...' && az config set core.login_experience_v2=off 2>/dev/null || true && az extension add --name containerapp --only-show-errors 2>/dev/null || true && az extension add --name front-door --only-show-errors 2>/dev/null || true && echo '✅ Azure CLI configured for Codespaces'"
   ],
   "postStartCommand": [
     "bash", "-c", 

.devcontainer/post-start-setup.sh

@@ -1,21 +1,21 @@
 #!/bin/bash
 
 # ------------------------------
-#    APIM SAMPLES POST-START SETUP
+#    APIM SAMPLES POST-START VERIFICATION
 # ------------------------------
 
 start=$(date +%s.%N)
 
 # Make terminal output more prominent
 clear
 echo "============================================================================"
-echo "                       🚀 APIM SAMPLES CODESPACE SETUP                    "
+echo "                    🚀 APIM SAMPLES CODESPACE VERIFICATION                "
 echo "============================================================================"
 echo ""
-echo "🔧 This terminal shows the Codespace setup progress."
-echo "📋 Keep this panel open to monitor the environment configuration."
+echo "🔧 This terminal shows the Codespace verification progress."
+echo "📋 Keep this panel open to monitor the environment status."
 echo ""
-echo -e "🚀 APIM Samples environment starting...\n"
+echo -e "✅ Most setup completed during prebuild - verifying environment...\n"
 
 # ------------------------------
 #    CONFIGURATION
@@ -67,7 +67,7 @@ else
 fi
 
 # ------------------------------
-#    GENERATE .ENV FILE
+#    ENVIRONMENT FILE VERIFICATION
 # ------------------------------
 
 echo -e "\n3/5) Verifying .env file...\n"
@@ -76,39 +76,45 @@ cd "$WORKSPACE_ROOT"
 if [ -f ".env" ]; then
     echo "   ✅ .env file exists"
 else
-    echo "   ⚠️  .env file missing, generating..."
+    echo "   ⚠️  .env file missing, regenerating..."
     if [ -f "setup/setup_python_path.py" ]; then
         python setup/setup_python_path.py --generate-env
-        echo "   ✅ .env file generated"
+        echo "   ✅ .env file regenerated"
     else
         echo "   ⚠️  setup_python_path.py not found, creating basic .env"
         cat > .env << EOF
 # Auto-generated for APIM Samples dev container
 PROJECT_ROOT=$WORKSPACE_ROOT
 PYTHONPATH=$WORKSPACE_ROOT/shared/python:$WORKSPACE_ROOT
 EOF
+        echo "   ✅ Basic .env file created"
     fi
 fi
 
 # ------------------------------
-#    AZURE CLI SETUP
+#    AZURE CLI VERIFICATION
 # ------------------------------
 
-echo -e "\n4/5) Configuring Azure CLI...\n"
+echo -e "\n4/5) Verifying Azure CLI configuration...\n"
 
-# We need to have a device code-based login experience within Codespaces. Redirect error output to /dev/null to avoid cluttering the output and ensure
-# that the script continues (|| true) even if this command fails.
-echo -e "   Setting Azure CLI login experience to device code (needed for Codespaces)...\n"
-az config set core.login_experience_v2=off 2>/dev/null || true
+# Verify Azure CLI extensions are installed (they should be from prebuild)
+echo "   Checking Azure CLI extensions..."
+if az extension list --query "[?name=='containerapp']" -o tsv | grep -q "containerapp"; then
+    echo "   ✅ containerapp extension installed"
+else
+    echo "   ⚠️  containerapp extension missing, installing..."
+    az extension add --name containerapp --only-show-errors 2>/dev/null || true
+fi
+
+if az extension list --query "[?name=='front-door']" -o tsv | grep -q "front-door"; then
+    echo "   ✅ front-door extension installed"
+else
+    echo "   ⚠️  front-door extension missing, installing..."
+    az extension add --name front-door --only-show-errors 2>/dev/null || true
+fi
 
-# Install extensions used by infrastructure samples
-# - containerapp: Required for infrastructure/apim-aca and infrastructure/afd-apim
-# - front-door: Required for infrastructure/afd-apim and shared/python/utils.py
-echo "   1/2: Installing containerapp extension..."
-az extension add --name containerapp --only-show-errors 2>/dev/null || true
-echo "   2/2: Installing front-door extension..."
-az extension add --name front-door --only-show-errors 2>/dev/null || true
-echo -e "   \n✅ Azure CLI and extensions configured"
+# Verify Azure CLI configuration
+echo "   ✅ Azure CLI configured for device code authentication"
 
 # ------------------------------
 #    FINAL VERIFICATION
@@ -128,8 +134,8 @@ echo "   Jupyter Kernels     : $(jupyter kernelspec list --json | python -c "imp
 if jupyter kernelspec list | grep -q "apim-samples" 2>/dev/null; then
     echo "   APIM Samples Kernel : ✅"
 else
-    echo "   APIM Samples Kernel : ❌ (registering...)"
-    python -m ipykernel install --user --name=apim-samples --display-name="APIM Samples Python 3.12" 2>/dev/null && echo "   ✅ Kernel registered successfully" || echo "   ⚠️  Failed to register kernel"
+    echo "   APIM Samples Kernel : ⚠️ (missing, re-registering...)"
+    python -m ipykernel install --user --name=apim-samples --display-name="APIM Samples Python 3.12" 2>/dev/null && echo "   ✅ Kernel registered successfully" || echo "   ❌ Failed to register kernel"
 fi
 
 # Test core imports
@@ -148,13 +154,13 @@ duration=$(python3 -c "print(f'{float('$end') - float('$start'):.2f}')")
 
 echo ""
 echo "============================================================================"
-echo "                          🎉 SETUP COMPLETED!                             "
+echo "                          🎉 VERIFICATION COMPLETED!                      "
 echo "============================================================================"
 echo ""
-printf "⏱️ Total setup time: %s seconds\n" "$duration"
-echo "💡 All requirements are ready to use for virtual environment $VENV_PATH"
+printf "⏱️ Total verification time: %s seconds\n" "$duration"
+echo "💡 Environment prebuild optimizations have significantly reduced startup time!"
 echo ""
-echo "🔍 This terminal shows your Codespace setup progress and logs."
+echo "🔍 This terminal shows your Codespace verification progress and logs."
 echo "📋 You can minimize this panel or open a new terminal for your work."
 echo ""
 echo "🚀 Your APIM Samples environment is ready to use!"