Support updating documents with JSON, NDJSON, CSV

Previously, you could only *replace* documents with the add endpoints.

Author: ellnix

lib/meilisearch/index.rb

@@ -145,6 +145,34 @@ def update_documents(documents, primary_key = nil)
     end
     alias add_or_update_documents update_documents
 
+    def update_documents_json(documents, primary_key = nil)
+      options = { convert_body?: false }
+      response = http_put "/indexes/#{@uid}/documents", documents, { primaryKey: primary_key }.compact, options
+
+      Models::Task.new(response, task_endpoint)
+    end
+    alias add_or_update_documents_json update_documents_json
+
+    def update_documents_ndjson(documents, primary_key = nil)
+      options = { headers: { 'Content-Type' => 'application/x-ndjson' }, convert_body?: false }
+      response = http_put "/indexes/#{@uid}/documents", documents, { primaryKey: primary_key }.compact, options
+
+      Models::Task.new(response, task_endpoint)
+    end
+    alias add_or_update_documents_ndjson update_documents_ndjson
+
+    def update_documents_csv(documents, primary_key = nil, delimiter = nil)
+      options = { headers: { 'Content-Type' => 'text/csv' }, convert_body?: false }
+
+      response = http_put "/indexes/#{@uid}/documents", documents, {
+        primaryKey: primary_key,
+        csvDelimiter: delimiter
+      }.compact, options
+
+      Models::Task.new(response, task_endpoint)
+    end
+    alias add_or_update_documents_csv add_documents_csv
+
     def update_documents!(documents, primary_key = nil)
       Utils.soft_deprecate(
         'Index#update_documents!',

spec/meilisearch/index/documents_spec.rb

@@ -122,7 +122,7 @@
       let(:csv_docs) do
         <<~CSV
           "objectRef:number","title:string","comment:string"
-          "1239","Pride and Prejudice","A great book"
+          "123","Pride and Prejudice","A great book"
           "456","Le Petit Prince","A french book"
           "49","Harry Potter and the Half-Blood Prince","The best book"
           "55","The Three Body Problem","An interesting book"
@@ -133,7 +133,7 @@
       let(:csv_docs_custom_delim) do
         <<~CSV
           "objectRef:number"|"title:string"|"comment:string"
-          "1239"|"Pride and Prejudice"|"A great book"
+          "123"|"Pride and Prejudice"|"A great book"
           "456"|"Le Petit Prince"|"A french book"
           "49"|"Harry Potter and the Half-Blood Prince"|"The best book"
           "55"|"The Three Body Problem"|"An interesting book"
@@ -199,6 +199,66 @@
 
         expect(index.documents['results']).to include(batch1_doc, batch2_doc)
       end
+
+      it '#update_documents_json' do
+        index.add_documents_json(json_docs, 'objectRef').await
+        edit = '[{ "objectRef": 123, "comment": "AN OLD BOOK" }]'
+        index.update_documents_json(edit).await
+        expect(index.documents['results']).to include(
+          {
+            'objectRef' => 123,
+            'title' => 'Pride and Prejudice',
+            'comment' => 'AN OLD BOOK'
+          }
+        )
+      end
+
+      it '#update_documents_ndjson' do
+        index.add_documents_ndjson(ndjson_docs, 'objectRef').await
+        edit = '{ "objectRef": 123, "comment": "AN OLD BOOK" }'
+        index.update_documents_ndjson(edit).await
+        expect(index.documents['results']).to include(
+          {
+            'objectRef' => 123,
+            'title' => 'Pride and Prejudice',
+            'comment' => 'AN OLD BOOK'
+          }
+        )
+      end
+
+      it '#update_documents_csv' do
+        index.add_documents_csv(csv_docs, 'objectRef').await
+        edit = <<~CSV
+          "objectRef:number","comment:string"
+          "123","AN OLD BOOK"
+        CSV
+
+        index.update_documents_csv(edit).await
+        expect(index.documents['results']).to include(
+          {
+            'objectRef' => 123,
+            'title' => 'Pride and Prejudice',
+            'comment' => 'AN OLD BOOK'
+          }
+        )
+      end
+
+      it '#update_documents_csv with custom delimiter' do
+        index.add_documents_csv(csv_docs, 'objectRef').await
+        edit = <<~CSV
+          "objectRef:number"|"comment:string"
+          "123"|"AN OLD BOOK"
+        CSV
+
+        index.update_documents_csv(edit, 'objectRef', '|').await
+        expect(index.documents['results']).to include(
+          {
+            'objectRef' => 123,
+            'title' => 'Pride and Prejudice',
+            'comment' => 'AN OLD BOOK'
+          }
+        )
+      end
     end
 
     describe '#add_documents!' do