added ObjectIsInput and ObjectIsOutput helpers

Author: matryer

parser/parser.go

@@ -45,6 +45,36 @@ func (d *Definition) Object(name string) (*Object, error) {
 	return nil, ErrNotFound
 }
 
+// ObjectIsInput gets whether this object is a method
+// input (request) type or not.\
+// Returns true if any method.InputObject.ObjectName matches
+// name.
+func (d *Definition) ObjectIsInput(name string) bool {
+	for _, service := range d.Services {
+		for _, method := range service.Methods {
+			if method.InputObject.ObjectName == name {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+// ObjectIsOutput gets whether this object is a method
+// output (response) type or not.
+// Returns true if any method.OutputObject.ObjectName matches
+// name.
+func (d *Definition) ObjectIsOutput(name string) bool {
+	for _, service := range d.Services {
+		for _, method := range service.Methods {
+			if method.OutputObject.ObjectName == name {
+				return true
+			}
+		}
+	}
+	return false
+}
+
 // Service describes a service, akin to an interface in Go.
 type Service struct {
 	Name    string   `json:"name"`

parser/parser_test.go

@@ -208,3 +208,18 @@ func TestExtractCommentMetadata(t *testing.T) {
 	is.Equal(metadata["required"], true)
 	is.Equal(metadata["monkey"], float64(24))
 }
+
+func TestObjectIsInputOutput(t *testing.T) {
+	is := is.New(t)
+	patterns := []string{"./testdata/services/pleasantries"}
+	parser := New(patterns...)
+	parser.Verbose = testing.Verbose()
+	parser.ExcludeInterfaces = []string{"Ignorer"}
+	def, err := parser.Parse()
+	is.NoErr(err)
+
+	is.Equal(def.ObjectIsInput("GreetRequest"), true)
+	is.Equal(def.ObjectIsInput("GreetResponse"), false)
+	is.Equal(def.ObjectIsOutput("GreetRequest"), false)
+	is.Equal(def.ObjectIsOutput("GreetResponse"), true)
+}