lib/tomograph/tomogram.rb in tomograph-2.1.0 vs lib/tomograph/tomogram.rb in tomograph-2.2.0

@@ -1,41 +1,53 @@
 require 'multi_json'
 require 'tomograph/path'
+require 'tomograph/api_blueprint/json_schema'
 require 'tomograph/api_blueprint/yaml'
 
 module Tomograph
   class Tomogram
-    def initialize(prefix: '', apib_path: nil, drafter_yaml_path: nil)
-      @documentation = Tomograph::ApiBlueprint::Yaml.new(prefix, apib_path, drafter_yaml_path)
+    extend Gem::Deprecate
+
+    def initialize(prefix: '', apib_path: nil, drafter_yaml_path: nil, tomogram_json_path: nil)
+      @documentation = if tomogram_json_path
+                         Tomograph::ApiBlueprint::JsonSchema.new(prefix, tomogram_json_path)
+                       else
+                         Tomograph::ApiBlueprint::Yaml.new(prefix, apib_path, drafter_yaml_path)
+                       end
       @prefix = prefix
     end
 
+    def to_a
+      @actions ||= @documentation.to_tomogram
+    end
+
     def to_hash
-      @documentation.to_tomogram.map(&:to_hash)
+      to_a.map(&:to_hash)
     end
+    deprecate :to_hash, 'to_a with method access', 2018, 8
 
     def to_json
-      MultiJson.dump(to_hash)
+      MultiJson.dump(to_a.map(&:to_hash), pretty: true)
     end
 
     def find_request(method:, path:)
       path = Tomograph::Path.new(path).to_s
 
-      @documentation.to_tomogram.find do |action|
+      to_a.find do |action|
         action.method == method && action.path.match(path)
       end
     end
 
     def find_request_with_content_type(method:, path:, content_type:)
       path = Tomograph::Path.new(path).to_s
 
-      @documentation.to_tomogram.find do |action|
+      to_a.find do |action|
         action.method == method && action.path.match(path) && action.content_type == content_type
       end
     end
 
     def to_resources
-      @documentation.to_resources
+      @resources ||= @documentation.to_resources
     end
 
     def prefix_match?(raw_path)
       raw_path.include?(@prefix)
     end