Add documents as JSON, NDJSON and CSV arrays

- Fix the aux method that merge request options.
- Skip to_json transformation for these methods.
- Add related tests
- Doesn't add in batches.
- Doesn't support key transformations from camelCase to snake_case

Author: thicolares

lib/meilisearch/http_request.rb

@@ -66,16 +66,17 @@ def build_default_options_headers(api_key = nil)
       }.compact
     end
 
-    def merge_options(defaults, added = {})
-      merged = defaults.merge(added)
-      merged[:headers].merge(added[:headers]) if added.key?(:headers)
-      merged
+    def merge_options(default_options, added_options = {})
+      default_cloned_headers = default_options[:headers].clone
+      merged_options = default_options.merge(added_options)
+      merged_options[:headers] = default_cloned_headers.merge(added_options[:headers]) if added_options.key?(:headers)
+      merged_options
     end
 
     def remove_options_header(options, key)
-      new_options = clone_options(options)
-      new_options[:headers].tap { |headers| headers.delete(key) }
-      new_options
+      cloned_options = clone_options(options)
+      cloned_options[:headers].tap { |headers| headers.delete(key) }
+      cloned_options
     end
 
     def clone_options(options)
@@ -97,7 +98,7 @@ def send_request(http_method, relative_path, query_params: nil, body: nil, optio
     end
 
     def http_config(query_params, body, options)
-      body = transform_attributes(body).to_json
+      body = transform_attributes(body).to_json if options[:transform_body?] == true
       {
         headers: options[:headers],
         query: query_params,

lib/meilisearch/index.rb

@@ -83,6 +83,27 @@ def add_documents!(documents, primary_key = nil)
     alias replace_documents! add_documents!
     alias add_or_replace_documents! add_documents!
 
+    def add_documents_json(documents, primary_key = nil)
+      options = { headers: { 'Content-Type' => 'application/json' }, transform_body?: false }
+      http_post "/indexes/#{@uid}/documents", documents, { primaryKey: primary_key }.compact, options
+    end
+    alias replace_documents_json add_documents_json
+    alias add_or_replace_documents_json add_documents_json
+
+    def add_documents_ndjson(documents, primary_key = nil)
+      options = { headers: { 'Content-Type' => 'application/x-ndjson' }, transform_body?: false }
+      http_post "/indexes/#{@uid}/documents", documents, { primaryKey: primary_key }.compact, options
+    end
+    alias replace_documents_ndjson add_documents_ndjson
+    alias add_or_replace_documents_ndjson add_documents_ndjson
+
+    def add_documents_csv(documents, primary_key = nil)
+      options = { headers: { 'Content-Type' => 'text/csv' }, transform_body?: false }
+      http_post "/indexes/#{@uid}/documents", documents, { primaryKey: primary_key }.compact, options
+    end
+    alias replace_documents_csv add_documents_csv
+    alias add_or_replace_documents_csv add_documents_csv
+
     def update_documents(documents, primary_key = nil)
       documents = [documents] if documents.is_a?(Hash)
       http_put "/indexes/#{@uid}/documents", documents, { primaryKey: primary_key }.compact

spec/meilisearch/index/documents_spec.rb

@@ -25,6 +25,52 @@
         expect(index.documents.count).to eq(documents.count)
       end
 
+      it 'adds JSON documents (as a array of documents)' do
+        documents = <<JSON
+        [
+          { "objectId": 123,  "title": "Pride and Prejudice",                    "comment": "A great book" },
+          { "objectId": 456,  "title": "Le Petit Prince",                        "comment": "A french book" },
+          { "objectId": 1,    "title": "Alice In Wonderland",                    "comment": "A weird book" },
+          { "objectId": 1344, "title": "The Hobbit",                             "comment": "An awesome book" },
+          { "objectId": 4,    "title": "Harry Potter and the Half-Blood Prince", "comment": "The best book" }
+        ]
+JSON
+        response = index.add_documents_json(documents)
+
+        expect(response).to be_a(Hash)
+        expect(response).to have_key('updateId')
+        index.wait_for_pending_update(response['updateId'])
+        expect(index.documents.count).to eq(5)
+      end
+
+      it 'adds NDJSON documents (as a array of documents)' do
+        documents = <<NDJSON
+        { "objectId": 123,  "title": "Pride and Prejudice",                    "comment": "A great book" }
+        { "objectId": 456,  "title": "Le Petit Prince",                        "comment": "A french book" }
+        { "objectId": 1,    "title": "Alice In Wonderland",                    "comment": "A weird book" }
+        { "objectId": 4,    "title": "Harry Potter and the Half-Blood Prince", "comment": "The best book" }
+NDJSON
+        response = index.add_documents_ndjson(documents)
+        expect(response).to be_a(Hash)
+        expect(response).to have_key('updateId')
+        index.wait_for_pending_update(response['updateId'])
+        expect(index.documents.count).to eq(4)
+      end
+
+      it 'adds CSV documents (as a array of documents)' do
+        documents = <<CSV
+"objectId:number","title:string","comment:string"
+"1239","Pride and Prejudice","A great book"
+"4569","Le Petit Prince","A french book"
+"49","Harry Potter and the Half-Blood Prince","The best book"
+CSV
+        response = index.add_documents_csv(documents)
+        expect(response).to be_a(Hash)
+        expect(response).to have_key('updateId')
+        index.wait_for_pending_update(response['updateId'])
+        expect(index.documents.count).to eq(3)
+      end
+
       it 'adds documents in a batch (as a array of documents)' do
         response = index.add_documents_in_batches(documents, 5)
         expect(response).to be_a(Array)