Set default for extra imports

Author: hermanschaaf

interfaces/generate.go

@@ -31,6 +31,9 @@ func (o *Options) SetDefaults() {
 	if o.ShouldInclude == nil {
 		o.ShouldInclude = func(reflect.Method) bool { return true }
 	}
+	if o.ExtraImports == nil {
+		o.ExtraImports = func(reflect.Method) []string { return nil }
+	}
 }
 
 type Option func(*Options)

interfaces/generate_test.go

@@ -41,9 +41,12 @@ type InterfacesClient interface {
 
 func TestGenerate(t *testing.T) {
 	dir := t.TempDir()
-	err := Generate([]any{&Client{}}, dir, WithIncludeFunc(func(m reflect.Method) bool {
-		return MethodHasAnyPrefix(m, []string{"List"}) && MethodHasAnySuffix(m, []string{"Tables"})
-	}))
+	err := Generate([]any{&Client{}}, dir,
+		WithIncludeFunc(func(m reflect.Method) bool {
+			return MethodHasAnyPrefix(m, []string{"List"}) && MethodHasAnySuffix(m, []string{"Tables"})
+		}),
+		WithExtraImports(func(m reflect.Method) []string { return []string{"net/http"} },
+		))
 	if err != nil {
 		t.Fatalf("unexpected error: %v", err)
 	}