1 files changed, 63 insertions, 0 deletions
diff --git a/vendor/github.com/yagipy/maintidx/maintidx.go b/vendor/github.com/yagipy/maintidx/maintidx.go
new file mode 100644
index 000000000..31ad9ca0c
--- /dev/null
+++ b/vendor/github.com/yagipy/maintidx/maintidx.go
@@ -0,0 +1,63 @@
+package maintidx
+
+import (
+	"go/ast"
+	"go/token"
+	"golang.org/x/tools/go/analysis"
+	"golang.org/x/tools/go/analysis/passes/inspect"
+	"golang.org/x/tools/go/ast/inspector"
+)
+
+const doc = "maintidx measures the maintainability index of each function."
+
+var Analyzer = &analysis.Analyzer{
+	Name: "maintidx",
+	Doc:  doc,
+	Run:  run,
+	Requires: []*analysis.Analyzer{
+		inspect.Analyzer,
+	},
+}
+
+var under int
+
+func init() {
+	Analyzer.Flags.IntVar(&under, "under", 20, "show functions with maintainability index < N only.")
+}
+
+func run(pass *analysis.Pass) (interface{}, error) {
+	i := pass.ResultOf[inspect.Analyzer].(*inspector.Inspector)
+
+	nodeFilter := []ast.Node{
+		(*ast.FuncDecl)(nil),
+	}
+
+	i.Preorder(nodeFilter, func(n ast.Node) {
+		switch n := n.(type) {
+		case *ast.FuncDecl:
+			v := analyze(n)
+
+			v.Coef.Cyc.Calc()
+			v.Coef.HalstVol.Calc()
+			v.calc(loc(pass.Fset, n))
+			if v.MaintIdx < under {
+				pass.Reportf(n.Pos(), "Function name: %v, Cyclomatic Complexity: %v, Halstead Volume: %0.2f, Maintainability Index: %v", n.Name, v.Coef.Cyc.Val, v.Coef.HalstVol.Val, v.MaintIdx)
+			}
+		}
+	})
+
+	return nil, nil
+}
+
+func analyze(n ast.Node) Visitor {
+	v := NewVisitor()
+	ast.Walk(v, n)
+	return *v
+}
+
+func loc(fs *token.FileSet, n *ast.FuncDecl) int {
+	f := fs.File(n.Pos())
+	startLine := f.Line(n.Pos())
+	endLine := f.Line(n.End())
+	return endLine - startLine + 1
+}