diff --git a/build/lex.go b/build/lex.go index 6126d5715..7486605c7 100644 --- a/build/lex.go +++ b/build/lex.go @@ -42,7 +42,7 @@ const ( TypeWorkspace // TypeBzl represents .bzl files TypeBzl - //TypeModule represents MODULE.bazel files + //TypeModule represents MODULE.bazel and *.MODULE.bazel files TypeModule ) @@ -130,7 +130,7 @@ func getFileType(filename string) FileType { if strings.HasSuffix(basename, ".oss") { basename = basename[:len(basename)-4] } - if basename == "module.bazel" { + if basename == "module.bazel" || strings.HasSuffix(basename, ".module.bazel") { return TypeModule } ext := filepath.Ext(basename) diff --git a/buildozer/buildozer_test.sh b/buildozer/buildozer_test.sh index 666d4b940..1aba85c92 100755 --- a/buildozer/buildozer_test.sh +++ b/buildozer/buildozer_test.sh @@ -2038,6 +2038,36 @@ EOF diff -u MODULE.bazel.expected MODULE.bazel || fail "Output didn't match" } +function test_module_bazel_segment() { + cat > go.MODULE.bazel < go.MODULE.bazel.expected < MODULE.bazel <