Fix to halt site generation if the default branch is not valid.

avisi-cloud · Jul 7, 2023 · 27ba480 · 27ba480
1 parent 0fde981
commit 27ba480
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)