Improve by using parsing library to find imports

Author: owen-mc

vendor.go

@@ -4,6 +4,8 @@ package main
 
 import (
 	"bytes"
+	"go/parser"
+	"go/token"
 	"io/ioutil"
 	"log"
 	"os"
@@ -157,35 +159,16 @@ func findPackagesInSourceCode(root string) map[string]bool {
 
 		// Only process Go files
 		if !info.IsDir() && strings.HasSuffix(path, ".go") {
-			content, err := os.ReadFile(path)
+			fset := token.NewFileSet()
+			file, err := parser.ParseFile(fset, path, nil, parser.ImportsOnly)
 			if err != nil {
 				return err
 			}
 
-			// Simple parsing to find import statements
-			// This is a basic implementation - a more robust solution would use go/parser
-			lines := strings.Split(string(content), "\n")
-			inImport := false
-			for _, line := range lines {
-				line = strings.TrimSpace(line)
-
-				if strings.HasPrefix(line, "import (") {
-					inImport = true
-					continue
-				}
-
-				if inImport && strings.HasPrefix(line, ")") {
-					inImport = false
-					continue
-				}
-
-				if strings.HasPrefix(line, "import ") || inImport {
-					// Extract package path from import line
-					pkgPath := extractPackagePath(line)
-					if pkgPath != "" {
-						packages[pkgPath] = true
-					}
-				}
+			// Extract import paths from the AST
+			for _, imp := range file.Imports {
+				pkgPath := strings.Trim(imp.Path.Value, "\"")
+				packages[pkgPath] = true
 			}
 		}
 		return nil
@@ -198,36 +181,6 @@ func findPackagesInSourceCode(root string) map[string]bool {
 	return packages
 }
 
-// extractPackagePath parses an import line to extract the package path
-func extractPackagePath(line string) string {
-	line = strings.TrimSpace(line)
-
-	// Skip if not an import line
-	if !(strings.HasPrefix(line, "import ") || strings.Contains(line, "\"") || strings.Contains(line, "`")) {
-		return ""
-	}
-
-	// Handle quoted imports
-	if strings.Contains(line, "\"") {
-		start := strings.Index(line, "\"")
-		end := strings.LastIndex(line, "\"")
-		if start >= 0 && end > start {
-			return line[start+1 : end]
-		}
-	}
-
-	// Handle backtick imports
-	if strings.Contains(line, "`") {
-		start := strings.Index(line, "`")
-		end := strings.LastIndex(line, "`")
-		if start >= 0 && end > start {
-			return line[start+1 : end]
-		}
-	}
-
-	return ""
-}
-
 // findSubmodulesForModule returns the submodules of a given module that are actually used in the source code
 func findSubmodulesForModule(modulePath string, usedPackages map[string]bool) []string {
 	var submodules []string