feature: code action to generate a stub function from a "no function …

…f" type error

gopls/internal/golang/codeaction.go

@@ -309,13 +309,21 @@ func quickFix(ctx context.Context, req *codeActionsRequest) error {
 			strings.HasPrefix(msg, "cannot convert") ||
 			strings.Contains(msg, "not implement") {
 			path, _ := astutil.PathEnclosingInterval(req.pgf.File, start, end)
-			si := stubmethods.GetStubInfo(req.pkg.FileSet(), info, path, start)
+			si := stubmethods.GetIfaceStubInfo(req.pkg.FileSet(), info, path, start)
 			if si != nil {
 				qf := typesutil.FileQualifier(req.pgf.File, si.Concrete.Obj().Pkg(), info)
 				iface := types.TypeString(si.Interface.Type(), qf)
 				msg := fmt.Sprintf("Declare missing methods of %s", iface)
 				req.addApplyFixAction(msg, fixStubMethods, req.loc)
 			}
+		} else if strings.Contains(msg, "has no field or method") {
+			path, _ := astutil.PathEnclosingInterval(req.pgf.File, start, end)
+
+			si := stubmethods.GetCallStubInfo(req.pkg, info, path, start)
+			if si != nil {
+				msg := fmt.Sprintf("Declare missing methods of %s", si.Receiver.Obj().Name())
+				req.addApplyFixAction(msg, fixMissingMethods, req.loc)
+			}
 		}
 	}
 

gopls/internal/golang/fix.go

@@ -66,6 +66,7 @@ const (
 	fixSplitLines        = "split_lines"
 	fixJoinLines         = "join_lines"
 	fixStubMethods       = "stub_methods"
+	fixMissingMethods    = "missing_methods"
 )
 
 // ApplyFix applies the specified kind of suggested fix to the given
@@ -109,7 +110,8 @@ func ApplyFix(ctx context.Context, fix string, snapshot *cache.Snapshot, fh file
 		fixInvertIfCondition: singleFile(invertIfCondition),
 		fixSplitLines:        singleFile(splitLines),
 		fixJoinLines:         singleFile(joinLines),
-		fixStubMethods:       stubMethodsFixer,
+		fixStubMethods:       stubMethodsIfaceFixer,
+		fixMissingMethods:    stubMethodsCallFixer,
 	}
 	fixer, ok := fixers[fix]
 	if !ok {