Refactor hierarchy printers

spec/compiler/crystal/tools/hierarchy_spec.cr

@@ -0,0 +1,67 @@
+require "../../../spec_helper"
+
+describe Crystal::TextHierarchyPrinter do
+  it "works" do
+    program = semantic(%(
+      class Foo
+      end
+
+      class Bar < Foo
+      end
+      ), inject_primitives: false).program
+
+    output = String.build { |io| Crystal.print_hierarchy(program, io, "ar$", "text") }
+    output.should eq(<<-EOS)
+    - class Object (4 bytes)
+      |
+      +- class Reference (4 bytes)
+         |
+         +- class Foo (4 bytes)
+            |
+            +- class Bar (4 bytes)\n
+    EOS
+  end
+end
+
+describe Crystal::JSONHierarchyPrinter do
+  it "works" do
+    program = semantic(%(
+      class Foo
+      end
+
+      class Bar < Foo
+      end
+      ), inject_primitives: false).program
+
+    output = String.build { |io| Crystal.print_hierarchy(program, io, "ar$", "json") }
+    JSON.parse(output).should eq(JSON.parse(<<-EOS))
+    {
+      "name": "Object",
+      "kind": "class",
+      "size_in_bytes": 4,
+      "sub_types": [
+        {
+          "name": "Reference",
+          "kind": "class",
+          "size_in_bytes": 4,
+          "sub_types": [
+            {
+              "name": "Foo",
+              "kind": "class",
+              "size_in_bytes": 4,
+              "sub_types": [
+                {
+                  "name": "Bar",
+                  "kind": "class",
+                  "size_in_bytes": 4,
+                  "sub_types": []
+                }
+              ]
+            }
+          ]
+        }
+      ]
+    }
+    EOS
+  end
+end

src/compiler/crystal/command.cr

@@ -190,7 +190,7 @@ class Crystal::Command
   private def hierarchy
     config, result = compile_no_codegen "tool hierarchy", hierarchy: true, top_level: true
     @progress_tracker.stage("Tool (hierarchy)") do
-      Crystal.print_hierarchy result.program, config.hierarchy_exp, config.output_format
+      Crystal.print_hierarchy result.program, STDOUT, config.hierarchy_exp, config.output_format
     end
   end