add listMethods

core.go

@@ -34,6 +34,7 @@ func ListServices(source DescriptorSource) ([]string, error) {
 	if err != nil {
 		return nil, err
 	}
+
 	sort.Strings(svcs)
 	return svcs, nil
 }

desc_source.go

@@ -44,11 +44,13 @@ func DescriptorSourceFromProtoSets(fileNames ...string) (DescriptorSource, error
 		if err != nil {
 			return nil, fmt.Errorf("could not load protoset file %q: %v", fileName, err)
 		}
+
 		var fs descpb.FileDescriptorSet
 		err = proto.Unmarshal(b, &fs)
 		if err != nil {
 			return nil, fmt.Errorf("could not parse contents of protoset file %q: %v", fileName, err)
 		}
+
 		files.File = append(files.File, fs.File...)
 	}
 	return DescriptorSourceFromFileDescriptorSet(files)
@@ -66,6 +68,7 @@ func DescriptorSourceFromProtoFiles(importPaths []string, fileNames ...string) (
 	if err != nil {
 		return nil, fmt.Errorf("could not parse given files: %v", err)
 	}
+
 	return DescriptorSourceFromFileDescriptors(fds...)
 }
 
@@ -75,24 +78,28 @@ func DescriptorSourceFromFileDescriptorSet(files *descpb.FileDescriptorSet) (Des
 	for _, fd := range files.File {
 		unresolved[fd.GetName()] = fd
 	}
+
 	resolved := map[string]*desc.FileDescriptor{}
 	for _, fd := range files.File {
 		_, err := resolveFileDescriptor(unresolved, resolved, fd.GetName())
 		if err != nil {
 			return nil, err
 		}
 	}
+
 	return &fileSource{files: resolved}, nil
 }
 
 func resolveFileDescriptor(unresolved map[string]*descpb.FileDescriptorProto, resolved map[string]*desc.FileDescriptor, filename string) (*desc.FileDescriptor, error) {
 	if r, ok := resolved[filename]; ok {
 		return r, nil
 	}
+
 	fd, ok := unresolved[filename]
 	if !ok {
 		return nil, fmt.Errorf("no descriptor found for %q", filename)
 	}
+
 	deps := make([]*desc.FileDescriptor, 0, len(fd.GetDependency()))
 	for _, dep := range fd.GetDependency() {
 		depFd, err := resolveFileDescriptor(unresolved, resolved, dep)
@@ -101,10 +108,12 @@ func resolveFileDescriptor(unresolved map[string]*descpb.FileDescriptorProto, re
 		}
 		deps = append(deps, depFd)
 	}
+
 	result, err := desc.CreateFileDescriptor(fd, deps...)
 	if err != nil {
 		return nil, err
 	}
+
 	resolved[filename] = result
 	return result, nil
 }
@@ -118,6 +127,7 @@ func DescriptorSourceFromFileDescriptors(files ...*desc.FileDescriptor) (Descrip
 			return nil, err
 		}
 	}
+
 	return &fileSource{files: fds}, nil
 }
 
@@ -131,12 +141,14 @@ func addFile(fd *desc.FileDescriptor, fds map[string]*desc.FileDescriptor) error
 		}
 		return nil
 	}
+
 	fds[name] = fd
 	for _, dep := range fd.GetDependencies() {
 		if err := addFile(dep, fds); err != nil {
 			return err
 		}
 	}
+
 	return nil
 }
 
@@ -157,9 +169,30 @@ func (fs *fileSource) ListServices() ([]string, error) {
 	for svc := range set {
 		sl = append(sl, svc)
 	}
+
 	return sl, nil
 }
 
+func (fs *fileSource) ListMethods() (map[string][]string, error) {
+	set := map[string][]string{}
+	for _, fd := range fs.files {
+		for _, svc := range fd.GetServices() {
+			m, ok := set[svc.GetFullyQualifiedName()]
+			if !ok {
+				m = []string{}
+			}
+			for _, v := range svc.GetMethods() {
+				m = append(m, v.GetName())
+				// m = append(.GetFullyQualifiedName()], v.GetName())
+			}
+
+			set[svc.GetFullyQualifiedName()] = m
+		}
+	}
+
+	return set, nil
+}
+
 // GetAllFiles returns all of the underlying file descriptors. This is
 // more thorough and more efficient than the fallback strategy used by
 // the GetAllFiles package method, for enumerating all files from a
@@ -171,6 +204,7 @@ func (fs *fileSource) GetAllFiles() ([]*desc.FileDescriptor, error) {
 		files[i] = fd
 		i++
 	}
+
 	return files, nil
 }
 
@@ -180,6 +214,7 @@ func (fs *fileSource) FindSymbol(fullyQualifiedName string) (desc.Descriptor, er
 			return dsc, nil
 		}
 	}
+
 	return nil, notFound("Symbol", fullyQualifiedName)
 }
 
@@ -214,10 +249,12 @@ func (ss serverSource) FindSymbol(fullyQualifiedName string) (desc.Descriptor, e
 	if err != nil {
 		return nil, reflectionSupport(err)
 	}
+
 	d := file.FindSymbol(fullyQualifiedName)
 	if d == nil {
 		return nil, notFound("Symbol", fullyQualifiedName)
 	}
+
 	return d, nil
 }
 
@@ -227,22 +264,26 @@ func (ss serverSource) AllExtensionsForType(typeName string) ([]*desc.FieldDescr
 	if err != nil {
 		return nil, reflectionSupport(err)
 	}
+
 	for _, fieldNum := range nums {
 		ext, err := ss.client.ResolveExtension(typeName, fieldNum)
 		if err != nil {
 			return nil, reflectionSupport(err)
 		}
 		exts = append(exts, ext)
 	}
+
 	return exts, nil
 }
 
 func reflectionSupport(err error) error {
 	if err == nil {
 		return nil
 	}
+
 	if stat, ok := status.FromError(err); ok && stat.Code() == codes.Unimplemented {
 		return ErrReflectionNotSupported
 	}
+
 	return err
 }