12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- package rule
-
- import (
- "fmt"
-
- "github.com/mgechev/revive/lint"
- )
-
- // ImportsBlacklistRule lints given else constructs.
- type ImportsBlacklistRule struct{}
-
- // Apply applies the rule to given file.
- func (r *ImportsBlacklistRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure {
- var failures []lint.Failure
-
- if file.IsTest() {
- return failures // skip, test file
- }
-
- blacklist := make(map[string]bool, len(arguments))
-
- for _, arg := range arguments {
- argStr, ok := arg.(string)
- if !ok {
- panic(fmt.Sprintf("Invalid argument to the imports-blacklist rule. Expecting a string, got %T", arg))
- }
- // we add quotes if not present, because when parsed, the value of the AST node, will be quoted
- if len(argStr) > 2 && argStr[0] != '"' && argStr[len(argStr)-1] != '"' {
- argStr = fmt.Sprintf(`"%s"`, argStr)
- }
- blacklist[argStr] = true
- }
-
- for _, is := range file.AST.Imports {
- path := is.Path
- if path != nil && blacklist[path.Value] {
- failures = append(failures, lint.Failure{
- Confidence: 1,
- Failure: "should not use the following blacklisted import: " + path.Value,
- Node: is,
- Category: "imports",
- })
- }
- }
-
- return failures
- }
-
- // Name returns the rule name.
- func (r *ImportsBlacklistRule) Name() string {
- return "imports-blacklist"
- }
|