From e735f79feb4df975728d3cb23a90a59291ac5908 Mon Sep 17 00:00:00 2001
From: Fabian Meumertzheim <meumertzheim@code-intelligence.com>
Date: Tue, 12 Dec 2023 20:59:18 +0100
Subject: [PATCH] Fix buildifier formatting for `MODULE.bazel` and `WORKSPACE`
 (#5829)

---
 .../idea/blaze/base/buildmodifier/BuildFileFormatter.java | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java b/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java
index d680f603ffc..8803597cbd8 100644
--- a/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java
+++ b/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java
@@ -93,7 +93,13 @@ static ImmutableList<String> getCommandLineArgs(String binary, BuildFile buildFi
   }
 
   private static String fileTypeArg(BlazeFileType fileType) {
-    return fileType == BlazeFileType.SkylarkExtension ? "--type=bzl" : "--type=build";
+    return "--type="
+        + switch (fileType) {
+          case SkylarkExtension -> "bzl";
+          case BuildPackage -> "build";
+          case Workspace -> "workspace";
+          case MODULE -> "module";
+        };
   }
 
   private static Iterable<String> pathArg(@Nullable BuildFile buildFile) {