Add a buildifier warning to enforce rule load location

Author: styurin

WARNINGS.md

@@ -64,6 +64,7 @@ Warning categories supported by buildifier's linter:
   * [`repository-name`](#repository-name)
   * [`return-value`](#return-value)
   * [`rule-impl-return`](#rule-impl-return)
+  * [`rule-load-location`](#rule-load-location)
   * [`same-origin-load`](#same-origin-load)
   * [`skylark-comment`](#skylark-comment)
   * [`skylark-docstring`](#skylark-docstring)
@@ -1095,6 +1096,26 @@ consider using
 [providers](https://docs.bazel.build/versions/main/skylark/rules.html#providers)
 or lists of providers instead.
 
+--------------------------------------------------------------------------------
+
+## <a name="rule-load-location"></a>Rule must be loaded from a specific location
+
+  * Category name: `rule-load-location`
+  * Automatic fix: no
+  * [Suppress the warning](#suppress): `# buildifier: disable=rule-load-location`
+
+Warns when a rule is loaded from a location other than the expected one.
+Expected locations are specified in the tables file:
+
+```json
+{
+  "RuleLoadLocation": {
+    "genrule": "//:genrule.bzl"
+  }
+}
+```
+
+
 --------------------------------------------------------------------------------
 
 ## <a name="same-origin-load"></a>Same label is used for multiple loads

buildifier/config/config_test.go

@@ -102,6 +102,7 @@ func ExampleExample() {
 	//     "repository-name",
 	//     "return-value",
 	//     "rule-impl-return",
+	//     "rule-load-location",
 	//     "skylark-comment",
 	//     "skylark-docstring",
 	//     "string-iteration",
@@ -290,6 +291,7 @@ func TestValidate(t *testing.T) {
 			"repository-name",
 			"return-value",
 			"rule-impl-return",
+			"rule-load-location",
 			"skylark-comment",
 			"skylark-docstring",
 			"string-iteration",
@@ -359,6 +361,7 @@ func TestValidate(t *testing.T) {
 			"repository-name",
 			"return-value",
 			"rule-impl-return",
+			"rule-load-location",
 			"skylark-comment",
 			"skylark-docstring",
 			"string-iteration",
@@ -427,7 +430,7 @@ func TestValidate(t *testing.T) {
 			"repository-name",
 			"return-value",
 			"rule-impl-return",
-
+			"rule-load-location",
 			"skylark-comment",
 			"skylark-docstring",
 			"string-iteration",

buildifier/integration_test.sh

@@ -307,6 +307,7 @@ cat > golden/.buildifier.example.json <<EOF
     "repository-name",
     "return-value",
     "rule-impl-return",
+    "rule-load-location",
     "skylark-comment",
     "skylark-docstring",
     "string-iteration",

tables/jsonparser.go

@@ -31,6 +31,7 @@ type Definitions struct {
 	NamePriority                    map[string]int
 	StripLabelLeadingSlashes        bool
 	ShortenAbsoluteLabelsToRelative bool
+	RuleLoadLocation                map[string]string
 }
 
 // ParseJSONDefinitions reads and parses JSON table definitions from file.
@@ -55,9 +56,9 @@ func ParseAndUpdateJSONDefinitions(file string, merge bool) error {
 	}
 
 	if merge {
-		MergeTables(definitions.IsLabelArg, definitions.LabelDenylist, definitions.IsListArg, definitions.IsSortableListArg, definitions.SortableDenylist, definitions.SortableAllowlist, definitions.NamePriority, definitions.StripLabelLeadingSlashes, definitions.ShortenAbsoluteLabelsToRelative)
+		MergeTables(definitions.IsLabelArg, definitions.LabelDenylist, definitions.IsListArg, definitions.IsSortableListArg, definitions.SortableDenylist, definitions.SortableAllowlist, definitions.NamePriority, definitions.StripLabelLeadingSlashes, definitions.ShortenAbsoluteLabelsToRelative, definitions.RuleLoadLocation)
 	} else {
-		OverrideTables(definitions.IsLabelArg, definitions.LabelDenylist, definitions.IsListArg, definitions.IsSortableListArg, definitions.SortableDenylist, definitions.SortableAllowlist, definitions.NamePriority, definitions.StripLabelLeadingSlashes, definitions.ShortenAbsoluteLabelsToRelative)
+		OverrideTables(definitions.IsLabelArg, definitions.LabelDenylist, definitions.IsListArg, definitions.IsSortableListArg, definitions.SortableDenylist, definitions.SortableAllowlist, definitions.NamePriority, definitions.StripLabelLeadingSlashes, definitions.ShortenAbsoluteLabelsToRelative, definitions.RuleLoadLocation)
 	}
 	return nil
 }

tables/jsonparser_test.go

@@ -37,6 +37,7 @@ func TestParseJSONDefinitions(t *testing.T) {
 		SortableAllowlist:        map[string]bool{},
 		NamePriority:             map[string]int{"name": -1},
 		StripLabelLeadingSlashes: true,
+		RuleLoadLocation:         map[string]string{"genrule": "//:genrule.bzl"},
 	}
 	if !reflect.DeepEqual(expected, definitions) {
 		t.Errorf("ParseJSONDefinitions(simple_tables.json) = %v; want %v", definitions, expected)

tables/tables.go

@@ -283,8 +283,11 @@ var IsModuleOverride = map[string]bool{
 	"single_version_override":   true,
 }
 
+// RuleLoadLocation contains custom locations for loading rules.
+var RuleLoadLocation = map[string]string{}
+
 // OverrideTables allows a user of the build package to override the special-case rules. The user-provided tables replace the built-in tables.
-func OverrideTables(labelArg, denylist, listArg, sortableListArg, sortDenylist, sortAllowlist map[string]bool, namePriority map[string]int, stripLabelLeadingSlashes, shortenAbsoluteLabelsToRelative bool) {
+func OverrideTables(labelArg, denylist, listArg, sortableListArg, sortDenylist, sortAllowlist map[string]bool, namePriority map[string]int, stripLabelLeadingSlashes, shortenAbsoluteLabelsToRelative bool, ruleLoadLocation map[string]string) {
 	IsLabelArg = labelArg
 	LabelDenylist = denylist
 	IsListArg = listArg
@@ -294,10 +297,11 @@ func OverrideTables(labelArg, denylist, listArg, sortableListArg, sortDenylist,
 	NamePriority = namePriority
 	StripLabelLeadingSlashes = stripLabelLeadingSlashes
 	ShortenAbsoluteLabelsToRelative = shortenAbsoluteLabelsToRelative
+	RuleLoadLocation = ruleLoadLocation
 }
 
 // MergeTables allows a user of the build package to override the special-case rules. The user-provided tables are merged into the built-in tables.
-func MergeTables(labelArg, denylist, listArg, sortableListArg, sortDenylist, sortAllowlist map[string]bool, namePriority map[string]int, stripLabelLeadingSlashes, shortenAbsoluteLabelsToRelative bool) {
+func MergeTables(labelArg, denylist, listArg, sortableListArg, sortDenylist, sortAllowlist map[string]bool, namePriority map[string]int, stripLabelLeadingSlashes, shortenAbsoluteLabelsToRelative bool, ruleLoadLocation map[string]string) {
 	for k, v := range labelArg {
 		IsLabelArg[k] = v
 	}
@@ -321,4 +325,7 @@ func MergeTables(labelArg, denylist, listArg, sortableListArg, sortDenylist, sor
 	}
 	StripLabelLeadingSlashes = stripLabelLeadingSlashes || StripLabelLeadingSlashes
 	ShortenAbsoluteLabelsToRelative = shortenAbsoluteLabelsToRelative || ShortenAbsoluteLabelsToRelative
+	for k, v := range ruleLoadLocation {
+		RuleLoadLocation[k] = v
+	}
 }

tables/testdata/simple_tables.json

@@ -19,5 +19,8 @@
   "NamePriority": {
     "name": -1
   },
-  "StripLabelLeadingSlashes": true
+  "StripLabelLeadingSlashes": true,
+  "RuleLoadLocation": {
+    "genrule": "//:genrule.bzl"
+  }
 }

warn/BUILD.bazel

@@ -13,6 +13,7 @@ go_library(
         "warn_cosmetic.go",
         "warn_deprecated.go",
         "warn_docstring.go",
+        "warn_load.go",
         "warn_macro.go",
         "warn_naming.go",
         "warn_operation.go",
@@ -41,6 +42,7 @@ go_test(
         "warn_cosmetic_test.go",
         "warn_deprecated_test.go",
         "warn_docstring_test.go",
+        "warn_load_test.go",
         "warn_macro_test.go",
         "warn_naming_test.go",
         "warn_operation_test.go",

warn/docs/warnings.textproto

@@ -811,6 +811,21 @@ warnings: {
     "or lists of providers instead."
 }
 
+warnings: {
+  name: "rule-load-location"
+  header: "Rule must be loaded from a specific location"
+  description:
+      "Warns when a rule is loaded from a location other than the expected one.\n"
+      "Expected locations are specified in the tables file:\n\n"
+  "```json\n"
+  "{\n"
+  "  \"RuleLoadLocation\": {\n"
+  "    \"genrule\": \"//:genrule.bzl\"\n"
+  "  }\n"
+  "}\n"
+  "```\n"
+}
+
 warnings: {
   name: "same-origin-load"
   header: "Same label is used for multiple loads"

warn/warn.go

@@ -174,6 +174,7 @@ var FileWarningMap = map[string]func(f *build.File) []*LinterFinding{
 	"redefined-variable":               redefinedVariableWarning,
 	"repository-name":                  repositoryNameWarning,
 	"rule-impl-return":                 ruleImplReturnWarning,
+	"rule-load-location":               ruleLoadLocationWarning,
 	"return-value":                     missingReturnValueWarning,
 	"skylark-comment":                  skylarkCommentWarning,
 	"skylark-docstring":                skylarkDocstringWarning,

warn/warn_load.go

@@ -0,0 +1,32 @@
+package warn
+
+import (
+	"fmt"
+	"github.com/bazelbuild/buildtools/build"
+	"github.com/bazelbuild/buildtools/tables"
+)
+
+func ruleLoadLocationWarning(f *build.File) []*LinterFinding {
+	var findings []*LinterFinding
+
+	for stmtIndex := 0; stmtIndex < len(f.Stmt); stmtIndex++ {
+		load, ok := f.Stmt[stmtIndex].(*build.LoadStmt)
+		if !ok {
+			continue
+		}
+
+		for i := 0; i < len(load.To); i++ {
+			from := load.From[i]
+
+			expectedLocation, ok := tables.RuleLoadLocation[from.Name]
+			if !ok || expectedLocation == load.Module.Value {
+				continue
+			}
+
+			f := makeLinterFinding(load.From[i], fmt.Sprintf("Rule %q must be loaded from %v.", from.Name, expectedLocation))
+			findings = append(findings, f)
+		}
+
+	}
+	return findings
+}

warn/warn_load_test.go

@@ -0,0 +1,39 @@
+/*
+Copyright 2020 Google LLC
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    https://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package warn
+
+import (
+	"github.com/bazelbuild/buildtools/tables"
+	"testing"
+)
+
+func TestWarnLoadLocation(t *testing.T) {
+	tables.RuleLoadLocation["s1"] = ":z.bzl"
+	tables.RuleLoadLocation["s3"] = ":y.bzl"
+	checkFindingsAndFix(t, "rule-load-location", `
+load(":f.bzl", "s1", "s2")
+load(":a.bzl", "s3")
+`, `
+load(":f.bzl", "s1", "s2")
+load(":a.bzl", "s3")
+`,
+		[]string{
+			":1: Rule \"s1\" must be loaded from :z.bzl.",
+			":2: Rule \"s3\" must be loaded from :y.bzl.",
+		},
+		scopeEverywhere)
+}