From d1aa76c6139aa634496ad066f0a194e53ed1c501 Mon Sep 17 00:00:00 2001
From: John Mulligan <jmulligan@redhat.com>
Date: Tue, 21 Sep 2021 14:41:41 -0400
Subject: [PATCH] implements: track all public functions, including stable ones

The go source (ast) visitor will now keep track of all public functions it
finds putting them into lists by category: deprecated, preview, stable.

Signed-off-by: John Mulligan <jmulligan@redhat.com>
---
 .../implements/internal/implements/gosrc.go   | 21 ++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/contrib/implements/internal/implements/gosrc.go b/contrib/implements/internal/implements/gosrc.go
index b20a5a0..6c8230b 100644
--- a/contrib/implements/internal/implements/gosrc.go
+++ b/contrib/implements/internal/implements/gosrc.go
@@ -51,6 +51,7 @@ type visitor struct {
 	docMap     map[string]*goFunction
 	deprecated []*goFunction
 	preview    []*goFunction
+	stable     []*goFunction
 }
 
 func newVisitor() *visitor {
@@ -59,6 +60,7 @@ func newVisitor() *visitor {
 		docMap:     map[string]*goFunction{},
 		deprecated: []*goFunction{},
 		preview:    []*goFunction{},
+		stable:     []*goFunction{},
 	}
 }
 
@@ -103,6 +105,10 @@ func readDocComment(fdec *ast.FuncDecl, gfunc *goFunction) {
 	}
 }
 
+func isPublic(gfunc *goFunction) bool {
+	return ast.IsExported(gfunc.shortName)
+}
+
 func (v *visitor) checkCalled(s *ast.SelectorExpr) {
 	ident, ok := s.X.(*ast.Ident)
 	if !ok {
@@ -141,13 +147,14 @@ func (v *visitor) Visit(node ast.Node) ast.Visitor {
 		}
 		readDocComment(n, gfunc)
 		v.currentFunc = gfunc
-		if gfunc.isDeprecated {
-			v.deprecated = append(v.deprecated, gfunc)
-			logger.Printf("rem1 %v\n", v.deprecated)
-		}
-		if gfunc.isPreview {
-			v.preview = append(v.preview, gfunc)
-			logger.Printf("rem2 %v\n", v.preview)
+		if isPublic(gfunc) {
+			if gfunc.isDeprecated {
+				v.deprecated = append(v.deprecated, gfunc)
+			} else if gfunc.isPreview {
+				v.preview = append(v.preview, gfunc)
+			} else {
+				v.stable = append(v.stable, gfunc)
+			}
 		}
 		return v
 	case *ast.CallExpr: