Handle invalid function

Author: styd

.gitignore

@@ -9,3 +9,4 @@
 
 # rspec failure tracking
 .rspec_status
+Gemfile.lock

Gemfile.lock

@@ -8,8 +8,8 @@ require 'functionable/json'
 # with your gem easier. You can also use a different console, if you like.
 
 # (If you use this, don't forget to add pry to your Gemfile!)
-# require "pry"
-# Pry.start
+require "pry"
+Pry.start
 
-require 'irb'
-IRB.start(__FILE__)
+#require 'irb'
+#IRB.start(__FILE__)

bin/console

@@ -17,6 +17,14 @@ def to_json(*)
     end
 
     def scrape_function!
+      @file_name ? valid_function : invalid_function
+    end
+
+    def cleanup_function!
+      function.tr("\n", '').squeeze(' ')
+    end
+
+    def valid_function
       @function = +'function'
       open_braces = 0
 
@@ -43,8 +51,8 @@ def scrape_function!
       end
     end
 
-    def cleanup_function!
-      function.tr("\n", '').squeeze(' ')
+    def invalid_function
+      @function = 'undefined'
     end
   end
 end

lib/jsonable/function.rb

@@ -21,6 +21,7 @@
       return 0
     }
   }
+  INVALID_FUNCTION = function(val, &method(:to_s))
 
   it 'is a kind of Numeric' do
     expect(FUNCTION1).to be_a Numeric
@@ -40,6 +41,7 @@
         "}"
       )
       expect({a: FUNCTION2}.to_json).to eq "{\"a\":function(arg) { return arg + 'km' }}"
+      expect({a: INVALID_FUNCTION}.to_json).to eq "{\"a\":undefined}"
     end
   end
 end