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

01-essentials/08-stash/verify.ps1

@@ -32,6 +32,21 @@ if (-not (Test-Path ".git")) {
     exit 1
 }
 
+# Detect the main branch name
+$allBranches = git branch --list 2>$null | ForEach-Object { $_.Trim('* ') }
+if ($allBranches -contains "main") {
+    $mainBranch = "main"
+} elseif ($allBranches -contains "master") {
+    $mainBranch = "master"
+} else {
+    $mainBranch = git config --get init.defaultBranch
+    if (-not $mainBranch) {
+        $mainBranch = $allBranches | Select-Object -First 1
+        if (-not $mainBranch) { $mainBranch = "main" }
+    }
+}
+Write-Host "Detected main branch: $mainBranch" -ForegroundColor Cyan
+
 # Check current branch
 $currentBranch = git branch --show-current 2>$null
 if ($currentBranch -ne "feature-login") {
@@ -53,14 +68,14 @@ if ($status) {
 }
 
 # Verify main branch has the security fix
-Write-Host "`nChecking main branch for bug fix..." -ForegroundColor Cyan
-git checkout main 2>$null | Out-Null
+Write-Host "`nChecking $mainBranch branch for bug fix..." -ForegroundColor Cyan
+git checkout $mainBranch 2>$null | Out-Null
 
 # Check for bug fix commit
-$mainCommits = git log --pretty=format:"%s" main 2>$null
+$mainCommits = git log --pretty=format:"%s" $mainBranch 2>$null
 if ($mainCommits -notmatch "security bug|Fix.*bug|security fix") {
-    Write-Host "[FAIL] No security bug fix commit found on main branch." -ForegroundColor Red
-    Write-Host "Hint: After stashing, switch to main and commit a bug fix" -ForegroundColor Yellow
+    Write-Host "[FAIL] No security bug fix commit found on $mainBranch branch." -ForegroundColor Red
+    Write-Host "Hint: After stashing, switch to $mainBranch and commit a bug fix" -ForegroundColor Yellow
     git checkout feature-login 2>$null | Out-Null
     Set-Location ..
     exit 1
@@ -68,7 +83,7 @@ if ($mainCommits -notmatch "security bug|Fix.*bug|security fix") {
 
 # Check that app.py has been fixed
 if (-not (Test-Path "app.py")) {
-    Write-Host "[FAIL] app.py not found on main branch." -ForegroundColor Red
+    Write-Host "[FAIL] app.py not found on $mainBranch branch." -ForegroundColor Red
     git checkout feature-login 2>$null | Out-Null
     Set-Location ..
     exit 1