feat: check for mainbranch

01-essentials/08-stash/setup.ps1

@@ -25,6 +25,9 @@ git init | Out-Null
 git config user.name "Workshop User" | Out-Null
 git config user.email "user@workshop.local" | Out-Null
 
+# Detect the default branch name after first commit (created below)
+# Will be detected after the initial commit
+
 # Create initial application on main
 $app = @"
 class Application:
@@ -45,6 +48,14 @@ Set-Content -Path "app.py" -Value $app
 git add app.py
 git commit -m "Initial application" | Out-Null
 
+# Detect the main branch name after first commit
+$mainBranch = git branch --show-current
+if (-not $mainBranch) {
+    $mainBranch = git config --get init.defaultBranch
+    if (-not $mainBranch) { $mainBranch = "main" }
+}
+Write-Host "Default branch detected: $mainBranch" -ForegroundColor Yellow
+
 $readme = @"
 # MyApp
 
@@ -78,7 +89,7 @@ git add login.py
 git commit -m "Start login service implementation" | Out-Null
 
 # Add a critical bug to main branch (simulating a bug that was introduced)
-git checkout main | Out-Null
+git checkout $mainBranch | Out-Null
 
 $appWithBug = @"
 class Application:
@@ -140,13 +151,14 @@ Write-Host "========================================" -ForegroundColor Green
 Write-Host "`nSituation:" -ForegroundColor Cyan
 Write-Host "You're working on the login feature (feature-login branch)" -ForegroundColor White
 Write-Host "You have uncommitted changes - the feature is NOT complete yet" -ForegroundColor Yellow
-Write-Host "`nUrgent: A critical security bug was found in production (main branch)!" -ForegroundColor Red
+Write-Host "`nUrgent: A critical security bug was found in production ($mainBranch branch)!" -ForegroundColor Red
 Write-Host "You need to fix it immediately, but your current work isn't ready to commit." -ForegroundColor Red
+Write-Host "`nDetected main branch: $mainBranch" -ForegroundColor Cyan
 Write-Host "`nYour task:" -ForegroundColor Yellow
 Write-Host "1. Navigate to the challenge directory: cd challenge" -ForegroundColor White
 Write-Host "2. Check your status: git status (see uncommitted changes)" -ForegroundColor White
 Write-Host "3. Stash your work: git stash save 'WIP: login feature'" -ForegroundColor White
-Write-Host "4. Switch to main: git checkout main" -ForegroundColor White
+Write-Host "4. Switch to $mainBranch: git checkout $mainBranch" -ForegroundColor White
 Write-Host "5. Fix the security bug in app.py (remove the comment and fix the auth)" -ForegroundColor White
 Write-Host "6. Commit the fix: git add app.py && git commit -m 'Fix critical security bug'" -ForegroundColor White
 Write-Host "7. Switch back: git checkout feature-login" -ForegroundColor White