feat(analyzer): add YAML support for ZAP Sites Tree specification pro…

…cessing and testing

Signed-off-by: HAHWUL <hahwul@gmail.com>

spec/functional_test/testers/specification/zap_sites_tree_spec.cr

@@ -0,0 +1,13 @@
+require "../../func_spec.cr"
+
+extected_endpoints = [
+  Endpoint.new("/v1/pets", "GET"),
+  Endpoint.new("/v1/pets", "POST", [Param.new("pet", "", "json")]),
+  Endpoint.new("/v1/pets/{petId}", "GET", [Param.new("petId", "", "path")]),
+  Endpoint.new("/v1/pets/{petId}", "PUT", [Param.new("petId", "", "path"), Param.new("pet", "", "json")]),
+]
+
+FunctionalTester.new("fixtures/specification/zap/", {
+  :techs     => 1,
+  :endpoints => extected_endpoints.size,
+}, extected_endpoints).test_all

src/analyzer/analyzers/specification/zap_sites_tree.cr

@@ -11,12 +11,12 @@ module Analyzer::Specification
           if File.exists?(sites_tree)
             details = Details.new(PathInfo.new(sites_tree))
             content = File.read(sites_tree, encoding: "utf-8", invalid: :skip)
-            json_obj = JSON.parse(content)
+            yaml_obj = YAML.parse(content)
 
             begin
-              children = json_obj["children"].as_h
-              if children.size > 0
-                process_node(children)
+              children = yaml_obj.as_a
+              children.each do |child|
+                process_node(child, details)
               end
             rescue e
               @logger.debug "Exception of #{sites_tree}/paths"
@@ -29,8 +29,26 @@ module Analyzer::Specification
       @result
     end
 
-    def process_node(children)
-      # TODO..
+    def process_node(node, details)
+      puts 1
+      if node.is_a?(Hash) && node.has_key?("url") && node.has_key?("method")
+        path = node["url"].as_s
+        method = node["method"].as_s.upcase || "GET"
+        if path != ""
+          puts path, method
+        end
+      end
+
+      puts 2
+
+      if node.is_a?(Hash) && node.has_key?("children")
+        children = node["children"].as_a
+        if children.size > 0
+          children.each do |child|
+            process_node(child, details)
+          end
+        end
+      end
     end
   end
 end