lib/tomograph/tomogram.rb in tomograph-0.3.0 vs lib/tomograph/tomogram.rb in tomograph-0.4.0

@@ -61,10 +61,14 @@
             resource['meta']['classes'][0] == 'resourceGroup' # skip Data Structures
         end
       end
 
       def documentation
-        YAML.load(File.read("#{Rails.root}/#{Tomograph.configuration.documentation}"))
+        if Tomograph.configuration.drafter_yaml
+          YAML.load(Tomograph.configuration.drafter_yaml)
+        else
+          YAML.load(File.read("#{Rails.root}/#{Tomograph.configuration.documentation}"))
+        end
       end
 
       def delete_till_the_end(path, beginning_with)
         path_index = path.index(beginning_with)
         path_index ||= 0