summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/mgechev/revive/rule/empty-block.go
blob: 7861394b32dfeafb32fb684138f18521bbe1c033 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package rule

import (
	"go/ast"

	"github.com/mgechev/revive/lint"
)

// EmptyBlockRule lints given else constructs.
type EmptyBlockRule struct{}

// Apply applies the rule to given file.
func (r *EmptyBlockRule) Apply(file *lint.File, _ lint.Arguments) []lint.Failure {
	var failures []lint.Failure

	onFailure := func(failure lint.Failure) {
		failures = append(failures, failure)
	}

	w := lintEmptyBlock{make([]*ast.BlockStmt, 0), onFailure}
	ast.Walk(w, file.AST)
	return failures
}

// Name returns the rule name.
func (r *EmptyBlockRule) Name() string {
	return "empty-block"
}

type lintEmptyBlock struct {
	ignore    []*ast.BlockStmt
	onFailure func(lint.Failure)
}

func (w lintEmptyBlock) Visit(node ast.Node) ast.Visitor {
	fd, ok := node.(*ast.FuncDecl)
	if ok {
		w.ignore = append(w.ignore, fd.Body)
		return w
	}

	fl, ok := node.(*ast.FuncLit)
	if ok {
		w.ignore = append(w.ignore, fl.Body)
		return w
	}

	block, ok := node.(*ast.BlockStmt)
	if !ok {
		return w
	}

	if mustIgnore(block, w.ignore) {
		return w
	}

	if len(block.List) == 0 {
		w.onFailure(lint.Failure{
			Confidence: 1,
			Node:       block,
			Category:   "logic",
			Failure:    "this block is empty, you can remove it",
		})
	}

	return w
}

func mustIgnore(block *ast.BlockStmt, blackList []*ast.BlockStmt) bool {
	for _, b := range blackList {
		if b == block {
			return true
		}
	}
	return false
}