export logs by function name and args

.gitignore

@@ -17,6 +17,8 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+.vscode
+bin/
 
 # Go workspace file
 go.work

Makefile

@@ -0,0 +1,5 @@
+#!/usr/bin/make -f
+SHELL = bash
+
+build:
+	CGO_ENABLED=0 go build -v -trimpath -o ./bin/log-export ./cmd/main.go

cmd/main.go

@@ -0,0 +1,103 @@
+package main
+
+import (
+	"fmt"
+	"go/ast"
+	"go/parser"
+	"go/token"
+	"log"
+	"os"
+	"path/filepath"
+)
+
+func main() {
+	if len(os.Args) != 2 {
+		log.Fatalln("directory must be provided")
+	}
+
+	directories, err := getAllSubdirectories(os.Args[1])
+	if err != nil {
+		log.Fatalf("failed to list subdirs: %v", err)
+	}
+
+	for _, dir := range directories {
+		fset := token.NewFileSet()
+		packages, err := parser.ParseDir(fset, dir, nil, 0)
+		if err != nil {
+			log.Fatalf("failed to parse directory %s: %v", dir, err)
+		}
+		for _, p := range packages {
+			for _, f := range p.Files {
+				ast.Inspect(f, func(n ast.Node) bool {
+					expr, ok := n.(*ast.CallExpr)
+					if !ok {
+						return true
+					}
+					if isLogDot(expr.Fun) && len(expr.Args) > 0 && isStringValue(expr.Args[0]) {
+						position := fset.Position(expr.Pos())
+						fmt.Printf("%v %s\n", position, stringValue(expr.Args[0]))
+						return false
+					}
+					return true
+				})
+			}
+		}
+
+	}
+}
+
+func getAllSubdirectories(dir string) ([]string, error) {
+	result := make([]string, 0)
+
+	stack := make([]string, 0)
+	stack = append(stack, dir)
+
+	for len(stack) > 0 {
+		current := stack[len(stack)-1]
+		stack = stack[:len(stack)-1]
+
+		entities, err := os.ReadDir(current)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, e := range entities {
+			if e.IsDir() {
+				path := filepath.Join(current, e.Name())
+				result = append(result, path)
+				stack = append(stack, path)
+			}
+		}
+	}
+
+	return result, nil
+}
+
+func isLogDot(expr ast.Expr) bool {
+	sel, ok := expr.(*ast.SelectorExpr)
+	return ok && (isIdent(sel.Sel, "Debug") ||
+		isIdent(sel.Sel, "Info") ||
+		isIdent(sel.Sel, "Warn") ||
+		isIdent(sel.Sel, "Error"))
+}
+
+func isIdent(expr ast.Expr, ident string) bool {
+	id, ok := expr.(*ast.Ident)
+	if !ok {
+		return false
+	}
+	if id.Name == ident {
+		return true
+	}
+	return false
+}
+
+func isStringValue(expr ast.Expr) bool {
+	basicLit, ok := expr.(*ast.BasicLit)
+	return ok && basicLit.Kind == token.STRING
+}
+
+func stringValue(expr ast.Expr) string {
+	basicLit, _ := expr.(*ast.BasicLit)
+	return basicLit.Value
+}

go.mod

@@ -0,0 +1,3 @@
+module git.frostfs.info/dstepanov-yadro/log-export
+
+go 1.20