json_extractor 1.0.0 → 1.0.1

data/README.md

@@ -3,8 +3,8 @@
 Pull out specific objects from a large JSON document by key without having to
 deserialize the whole object in to Ruby.
 
-**NOTE:** This gem only supports extracting objects from JSON documents.
-Strings, numbers, and arrays are not supported. Yet.
+**NOTE:** This gem only supports extracting objects and arrays from JSON
+documents. Strings and numbers are not supported. Yet.
 
 ## Example
 

data/ext/json_extractor/json_extractor.c

@@ -7,6 +7,8 @@
 
 PARSER_FUNC(object);
 PARSER_FUNC(object_body);
+PARSER_FUNC(array);
+PARSER_FUNC(array_body);
 PARSER_FUNC(start);
 PARSER_FUNC(whitespace);
 
@@ -115,6 +117,52 @@ unsigned int object(char * str, long len, unsigned int pos) {
   return(-1);
 }
 
+/**
+ * Finds the end of the current array body that is under pos within str.
+ *
+ * @param [char *] str the string to search.
+ * @param [long] len the overall length of the string.
+ * @param [unsigned int] pos the initial position (i.e. where to start)
+ * @return [unsigned int] the end of the array body.
+ */
+unsigned int array_body(char * str, long len, unsigned int pos) {
+  while(pos < len) {
+    if(str[pos+1] == '[') {
+      pos = array(str, len, pos+1);
+    }
+
+    if(str[pos+1] == ']') {
+      return pos;
+    }
+
+    pos++;
+  }
+
+  return -1;
+}
+
+/**
+ * Finds the end of the current array that is under pos within str.
+ *
+ * @param [char *] str the string to search.
+ * @param [long] len the overall length of the string.
+ * @param [unsigned int] pos the initial position (i.e. where to start)
+ * @return [unsigned int] the end of the array.
+ */
+unsigned int array(char * str, long len, unsigned int pos) {
+  if(str[pos] != '[') {
+    return -1;
+  }
+
+  pos = array_body(str, len, pos);
+
+  if(str[pos+1] == ']') {
+    return pos+1;
+  }
+
+  return -1;
+}
+
 /**
  * Starts parsing.
  *
@@ -128,6 +176,8 @@ unsigned int start(char * str, long len, unsigned int pos) {
 
   if(str[pos] == '{') {
     return object(str, len, pos);
+  } else if(str[pos] == '[') {
+    return array(str, len, pos);
   }
 
   return -1;
@@ -178,6 +228,17 @@ char * extract_subdocument(char * data, const char * key) {
   return final;
 }
 
+int file_exists(char * filename) {
+  FILE * fp;
+
+  if(fp = fopen(filename, "r")) {
+    fclose(fp);
+    return 1;
+  }
+
+  return 0;
+}
+
 static VALUE rb_extract_subdocument(VALUE self, VALUE str, VALUE key) {
   char *data, *substr;
   VALUE result;
@@ -192,9 +253,14 @@ static VALUE rb_extract_subdocument(VALUE self, VALUE str, VALUE key) {
     return Qnil;
   }
 
-  data = read_all(RSTRING_PTR(str));
-  substr = extract_subdocument(data, RSTRING_PTR(key));
-  free(data);
+  // If str points to a file, let's read it and use that. Otherwise...
+  if(file_exists(RSTRING_PTR(str))) {
+    data = read_all(RSTRING_PTR(str));
+    substr = extract_subdocument(data, RSTRING_PTR(key));
+    free(data);
+  } else {
+    substr = extract_subdocument(RSTRING_PTR(str), RSTRING_PTR(key));
+  }
 
   if(substr == NULL) {
     return Qnil;

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: json_extractor
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
   prerelease: 
 platform: ruby
 authors:
@@ -89,7 +89,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2562154935140798148
+      hash: 243038373631746394
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: