vfeskov_edu_gem 0.1.11 → 0.1.12

data/lib/vfeskov_edu_gem.rb

@@ -1,6 +1,59 @@
-require 'C:\Users\vfeskov\RubymineProjects\vfeskov_edu_gem\lib\vfeskov_edu_gem\object_extension.rb'
-require 'C:\Users\vfeskov\RubymineProjects\vfeskov_edu_gem\lib\vfeskov_edu_gem\version.rb'
+require 'vfeskov_edu_gem/object_extension'
+require 'vfeskov_edu_gem/version'
 
 module VfeskovEduGem
+  def self.extend_to_json(object)
 
+    def object.to_be_serialized(*fields)
+      @to_be_serialized = []
+      fields.each do |field|
+        @to_be_serialized << field.to_s.gsub(/@/,'')
+      end
+    end
+
+    def object.should_variable_be_serialized?(variable_symbol)
+      variable_name = variable_symbol.to_s.gsub(/@/,'')
+      if defined? @to_be_serialized
+        serialize = false
+        if @to_be_serialized.length > 0 && variable_name != 'to_be_serialized'
+          @to_be_serialized.each do |to_be_serialized|
+            if to_be_serialized == variable_name
+              serialize = true
+              break
+            end
+          end
+        end
+      else
+        serialize = true
+      end
+      return serialize
+    end
+
+    def object.method_missing(name, *args)
+      if name.to_s =~ /to_json_for_/
+        prev_state = @to_be_serialized
+
+        @to_be_serialized = []
+        variables = name.to_s[12..-1].split('_and_');
+        variables.each do |variable_name|
+          @to_be_serialized << variable_name
+        end
+
+        json = self.to_json
+
+        @to_be_serialized = prev_state
+
+        return json
+      else
+        super
+      end
+    end
+
+    def object.to_json
+      json = self.orig_to_json
+      json.gsub!(/([\"\'\&\/\\\@])/,'\\\\\1')
+      return json
+    end
+
+  end
 end

data/lib/vfeskov_edu_gem/object_extension.rb

@@ -1,4 +1,5 @@
 class Object
+
   def to_json
     if self.instance_variables.length == 0
       if self.class.name == 'Array'
@@ -16,15 +17,23 @@ class Object
     else
       result = "{";
       first = true
-      self.instance_variables.each do |property_symbol|
+      self.instance_variables.each do |variable_symbol|
+
+        if defined? self.should_variable_be_serialized?
+          next unless self.should_variable_be_serialized?(variable_symbol)
+        end
+
         result += ',' unless first
-        property_name = property_symbol.to_s.gsub(/@/,'')
-        property = self.instance_variable_get(property_symbol)
-        result += '"' + property_name + '":' + property.to_json
+        variable_name = variable_symbol.to_s.gsub(/@/,'')
+        variable = self.instance_variable_get(variable_symbol)
+        result += '"' + variable_name + '":' + variable.to_json
         first = false
+
       end
       result += "}"
     end
     return result
   end
+
+  alias_method :orig_to_json, :to_json
 end

data/lib/vfeskov_edu_gem/version.rb

@@ -1,3 +1,3 @@
 module VfeskovEduGem
-  VERSION = "0.1.11"
+  VERSION = "0.1.12"
 end

data/test/common/test_object.rb

@@ -0,0 +1,5 @@
+class TestObject
+attr_accessor :property1
+attr_accessor :property2
+attr_accessor :property3
+end

data/test/tc_further_object_extension.rb

@@ -0,0 +1,54 @@
+require "test/unit"
+require File.expand_path('../../lib/vfeskov_edu_gem', __FILE__)
+require File.expand_path('../common/test_object', __FILE__)
+class FurtherObjectExtensionTest < Test::Unit::TestCase
+
+  def test_to_be_serialized
+
+    t = TestObject.new
+    t.property1 = 123
+    t.property2 = 'test string'
+    t.property3 = [123, 'test string', 234]
+
+    VfeskovEduGem.extend_to_json(t)
+
+    t.to_be_serialized(:property1, :property2)
+    assert_equal('{\"property1\":\"123\",\"property2\":\"test string\"}', t.to_json)
+
+  end
+
+  def test_to_json_for
+
+    t = TestObject.new
+    t.property1 = 123
+    t.property2 = 'test string'
+    t.property3 = [123, 'test string', 234]
+
+    VfeskovEduGem.extend_to_json(t)
+
+    t.to_be_serialized(:property1, :property2)
+
+    assert_equal('{\"property1\":\"123\",\"property3\":[\"123\",\"test string\",\"234\"]}', t.to_json_for_property1_and_property3)
+
+    assert_equal('{\"property1\":\"123\",\"property2\":\"test string\",\"property3\":[\"123\",\"test string\",\"234\"]}', t.to_json_for_property1_and_property2_and_property3)
+
+    assert_equal('{\"property1\":\"123\",\"property2\":\"test string\"}', t.to_json)
+
+  end
+
+  def test_escape_special_characters
+
+    t = TestObject.new
+    t.property1 = 'bill@gates.cool'
+    t.property2 = 'he said "go"'
+    t.property3 = ['\'guys\' & co', '\--||--/', 234]
+
+    assert_equal('{"property1":"bill@gates.cool","property2":"he said "go"","property3":["\'guys\' & co","\--||--/","234"]}', t.to_json)
+
+    VfeskovEduGem.extend_to_json(t)
+
+    assert_equal('{\"property1\":\"bill\@gates.cool\",\"property2\":\"he said \"go\"\",\"property3\":[\"\\\'guys\\\' \& co\",\"\\\--||--\/\",\"234\"]}', t.to_json)
+
+  end
+
+end

data/test/tc_object_extension.rb

@@ -0,0 +1,32 @@
+require "test/unit"
+require File.expand_path('../../lib/vfeskov_edu_gem', __FILE__)
+require File.expand_path("../common/test_object", __FILE__)
+
+class TestObjectExtension < Test::Unit::TestCase
+
+  def test_array
+    assert_equal('["123456","test string","test string 2"]', [123456,"test string","test string 2"].to_json )
+  end
+
+  def test_object
+    t = TestObject.new()
+    t.property1 = 123
+    t.property2 = 'test string'
+    t.property3 = [123, 'test string', 234]
+    assert_equal('{"property1":"123","property2":"test string","property3":["123","test string","234"]}', t.to_json)
+  end
+
+  def test_object_in_object
+    t1 = TestObject.new
+    t1.property1 = 123
+    t1.property2 = 'test string'
+    t1.property3 = [123, 'test string', 234]
+
+    t2 = TestObject.new
+    t2.property1 = [t1,'asd']
+    t2.property2 = [213,342,564]
+
+    assert_equal('{"property1":[{"property1":"123","property2":"test string","property3":["123","test string","234"]},"asd"],"property2":["213","342","564"]}', t2.to_json)
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vfeskov_edu_gem
 version: !ruby/object:Gem::Version
-  version: 0.1.11
+  version: 0.1.12
   prerelease: 
 platform: ruby
 authors:
@@ -24,6 +24,9 @@ files:
 - lib/vfeskov_edu_gem.rb
 - lib/vfeskov_edu_gem/object_extension.rb
 - lib/vfeskov_edu_gem/version.rb
+- test/common/test_object.rb
+- test/tc_further_object_extension.rb
+- test/tc_object_extension.rb
 - vfeskov_edu_gem.gemspec
 homepage: ''
 licenses: []