WorkspaceAPIClient: Allows "main" to be used to refer to the main/def…

…ault branch.
structurizr · Oct 17, 2024 · cb8a407 · cb8a407
1 parent 1e6c2a3
commit cb8a407
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/structurizr-client/src/main/java/com/structurizr/api/WorkspaceApiClient.java b/structurizr-client/src/main/java/com/structurizr/api/WorkspaceApiClient.java
@@ -39,6 +39,7 @@
 public class WorkspaceApiClient extends AbstractApiClient {
 
     private static final Log log = LogFactory.getLog(WorkspaceApiClient.class);
+    private static final String MAIN_BRANCH = "main";
 
     private String user;
 
@@ -234,7 +235,7 @@ public Workspace getWorkspace(long workspaceId) throws StructurizrClientExceptio
             log.info("Getting workspace with ID " + workspaceId);
 
             HttpGet httpGet;
-            if (StringUtils.isNullOrEmpty(branch)) {
+            if (StringUtils.isNullOrEmpty(branch) || branch.equalsIgnoreCase(MAIN_BRANCH)) {
                 httpGet = new HttpGet(url + WORKSPACE_PATH + "/" + workspaceId);
             } else {
                 httpGet = new HttpGet(url + WORKSPACE_PATH + "/" + workspaceId + "/branch/" + branch);