add post-processing for method names

Author: burrbd

parser.go

@@ -116,10 +116,43 @@ func ParseFile(filePath string) (*APIDefinition, error) {
 		return nil, ramlError
 	}
 
+	postProcess(apiDefinition)
+
 	// Good.
 	return apiDefinition, nil
 }
 
+func postProcess(d *APIDefinition) {
+	for _, r := range d.Resources {
+		addMethodNames(&r)
+	}
+}
+
+func addMethodNames(r *Resource) {
+	if r.Get != nil {
+		r.Get.Name = "GET"
+	}
+	if r.Post != nil {
+		r.Post.Name = "POST"
+	}
+	if r.Put != nil {
+		r.Put.Name = "PUT"
+	}
+	if r.Patch != nil {
+		r.Patch.Name = "PATCH"
+	}
+	if r.Head != nil {
+		r.Head.Name = "HEAD"
+	}
+	if r.Delete != nil {
+		r.Delete.Name = "DELETE"
+	}
+
+	for _, n := range r.Nested {
+		addMethodNames(n)
+	}
+}
+
 // Reads the contents of a file, returns a bytes buffer
 func readFileContents(workingDirectory string, fileName string) ([]byte, error) {
 

raml_test.go

@@ -90,3 +90,21 @@ func TestParsing(t *testing.T) {
 		// 	pretty.Println(apiDefinition)
 	}
 }
+
+func TestMethodStringer(t *testing.T) {
+	def, _ := ParseFile("./samples/simple_example.raml")
+	r := def.Resources["/resources"]
+	if r.Get.Name != "GET" {
+		t.Errorf("Got %s, instead of GET", r.Get.Name)
+	}
+	n := r.Nested["/{resourceId}"]
+	if n.Get.Name != "GET" {
+		t.Errorf("Got %s, instead of GET", n.Get.Name)
+	}
+	if n.Put.Name != "PUT" {
+		t.Errorf("Got %s, instead of PUT", n.Put.Name)
+	}
+	if n.Delete.Name != "DELETE" {
+		t.Errorf("Got %s, instead of DELETE", n.Delete.Name)
+	}
+}

types.go

@@ -366,7 +366,6 @@ type Trait struct {
 // doesn't contain Usage, optional fields etc.
 type ResourceTypeMethod struct {
 	Name string
-	// TODO: Fill this during the post-processing phase
 
 	// Briefly describes what the method does to the resource
 	Description string
@@ -531,7 +530,6 @@ type SecurityScheme struct {
 // Methods are operations that are performed on a resource
 type Method struct {
 	Name string
-	// TODO: Fill this during the post-processing phase
 
 	// Briefly describes what the method does to the resource
 	Description string