Merge pull request #263 from qtzar/BugFix_BadMainBranch

Fix to halt site generation if the default branch is not valid.
avisi-cloud · Jul 7, 2023 · 1a5bc94 · 1a5bc94
2 parents 90e4f42 + 27ba480
commit 1a5bc94
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/kotlin/nl/avisi/structurizr/site/generatr/GenerateSiteCommand.kt b/src/main/kotlin/nl/avisi/structurizr/site/generatr/GenerateSiteCommand.kt
@@ -104,6 +104,10 @@ class GenerateSiteCommand : Subcommand(
 
         println("The following branches contain a valid Structurizr workspace: $branchesToGenerate")
 
+        if (!branchesToGenerate.contains(defaultBranch)) {
+            throw Exception("$defaultBranch does not contain a valid structurizr workspace. Site generation halted.")
+        }
+
         branchesToGenerate.forEach { branch ->
             println("Generating site for branch $branch")
             clonedRepository.checkoutBranch(branch)