Support for symbols

Author: x4d3

lib/json/canonicalization.rb

@@ -61,13 +61,21 @@ def to_json_c14n
   end
 end
 
+def encode_key(k)
+  case k
+  when String
+    return k.encode(Encoding::UTF_16)
+  end
+  k
+end
+
 class Hash
   # Output JSON with keys sorted lexicographically
   # @return [String]
   def to_json_c14n
     "{" + self.
       keys.
-      sort_by {|k| k.encode(Encoding::UTF_16)}.
+      sort_by {|k| encode_key(k)}.
       map {|k| k.to_json_c14n + ':' + self[k].to_json_c14n}
       .join(',') +
     '}'
@@ -81,3 +89,12 @@ def to_json_c14n
     self.to_json
   end
 end
+
+class Symbol
+  # Output JSON with control characters escaped
+  # @return [String]
+  def to_json_c14n
+    self.to_json
+  end
+end
+

spec/c14n_spec.rb

@@ -9,3 +9,10 @@
     end
   end
 end
+
+describe "special cases for hash keys" do
+    it "handles hash defined with symbols" do
+      data = { a: [{b:"b"}] }
+      expect(data.to_json_c14n).to eq "{\"a\":[{\"b\":\"b\"}]}"
+    end
+end