mirror of
https://codeberg.org/forgejo/forgejo
synced 2024-12-05 02:54:46 +01:00
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
|
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
|
||
|
}
|