53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
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"
|
|
}
|