prefer JSON, but fall back to YAML if JSON isn't available

elofgren · Dec 3, 2013 · 27c9774 · 27c9774
1 parent 10cadb8
commit 27c9774
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/linguist/samples.rb b/lib/linguist/samples.rb
@@ -1,4 +1,8 @@
-require 'json'
+begin
+  require 'json'
+rescue LoadError
+  require 'yaml'
+end
 
 require 'linguist/md5'
 require 'linguist/classifier'
@@ -14,7 +18,8 @@ module Samples
 
     # Hash of serialized samples object
     if File.exist?(PATH)
-      DATA = JSON.load(File.read(PATH))
+      serializer = defined?(JSON) ? JSON : YAML
+      DATA = serializer.load(File.read(PATH))
     end
 
     # Public: Iterate over each sample.