add import handler

Author: wzshiming

importer.go

@@ -17,6 +17,7 @@ type Importer struct {
 	bufType          map[string]Type
 	bufBuild         map[string]*build.Package
 	errorHandler     func(error)
+	importHandler    func(path, src, dir string)
 	isCommentLocator bool
 	ctx              build.Context
 }
@@ -133,6 +134,9 @@ func (i *Importer) Import(path string, src string) (Type, error) {
 	if ok {
 		return tt, nil
 	}
+	if i.importHandler != nil {
+		i.importHandler(path, src, dir)
+	}
 
 	m := map[string]bool{}
 	for _, v := range imp.GoFiles {

option.go

@@ -16,3 +16,10 @@ func WithCommentLocator() Option {
 		i.isCommentLocator = true
 	}
 }
+
+// ImportHandler returns the import handler option
+func ImportHandler(f func(path, src, dir string)) Option {
+	return func(i *Importer) {
+		i.importHandler = f
+	}
+}