12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- package rule
-
- import (
- "go/ast"
-
- "strings"
-
- "github.com/mgechev/revive/lint"
- )
-
- // MaxPublicStructsRule lints given else constructs.
- type MaxPublicStructsRule struct{}
-
- // Apply applies the rule to given file.
- func (r *MaxPublicStructsRule) Apply(file *lint.File, arguments lint.Arguments) []lint.Failure {
- var failures []lint.Failure
-
- fileAst := file.AST
- walker := &lintMaxPublicStructs{
- fileAst: fileAst,
- onFailure: func(failure lint.Failure) {
- failures = append(failures, failure)
- },
- }
-
- ast.Walk(walker, fileAst)
-
- max, ok := arguments[0].(int64) // Alt. non panicking version
- if !ok {
- panic(`invalid value passed as argument number to the "max-public-structs" rule`)
- }
-
- if walker.current > max {
- walker.onFailure(lint.Failure{
- Failure: "you have exceeded the maximum number of public struct declarations",
- Confidence: 1,
- Node: fileAst,
- Category: "style",
- })
- }
-
- return failures
- }
-
- // Name returns the rule name.
- func (r *MaxPublicStructsRule) Name() string {
- return "max-public-structs"
- }
-
- type lintMaxPublicStructs struct {
- current int64
- fileAst *ast.File
- onFailure func(lint.Failure)
- }
-
- func (w *lintMaxPublicStructs) Visit(n ast.Node) ast.Visitor {
- switch v := n.(type) {
- case *ast.TypeSpec:
- name := v.Name.Name
- first := string(name[0])
- if strings.ToUpper(first) == first {
- w.current++
- }
- break
- }
- return w
- }
|