wikidata-diff-analyzer 0.1.1 → 2.0.1

data/lib/wikidata/diff/inside_claim_analyzer.rb

@@ -0,0 +1,84 @@
+class InsideClaimAnalyzer
+    def self.isolate_inside_claim_differences(current_content, parent_content)
+        # Initialize empty arrays to store the added, removed, and changed claims
+        added = []
+        removed = []
+        changed = []
+
+        if current_content.nil?
+            current_content_claims = {}
+        else
+            current_content_claims = current_content["claims"]
+            if !current_content_claims.is_a?(Hash)
+                current_content_claims = {}
+            end
+        end
+        
+
+        if parent_content.nil?
+            parent_content_claims = {}
+        else
+            parent_content_claims = parent_content["claims"]
+            if !parent_content_claims.is_a?(Hash)
+                parent_content_claims = {}
+            end
+        end
+        
+        # if parentid is 0, add all current claims as added claims and return it
+        if parent_content.nil?
+            current_content_claims.each do |claim_key, current_claims|
+                current_claims.each_with_index do |current_claim, index|
+                    added << { key: claim_key, index: index }
+                end
+            end
+        else
+            # Iterate over each claim key in the current content
+            current_content_claims.each do |claim_key, current_claims|
+            # Check if the claim key exists in the parent content
+            if parent_content_claims.key?(claim_key)
+                parent_claims = parent_content_claims[claim_key]
+                # Iterate over each claim in the current and parent content
+                current_claims.each_with_index do |current_claim, index|
+                    parent_claim = parent_claims[index]
+                    if parent_claim.nil?
+                        # Claim was added
+                        added << { key: claim_key, index: index }
+
+                    elsif current_claim != parent_claim
+                        # Claim was changed
+                        changed << { key: claim_key, index: index }
+                    end
+                end
+                # Check for removed claims
+                parent_claims.each_with_index do |parent_claim, index|
+                    current_claim = current_claims[index]
+                    if current_claim.nil?
+                        # Claim was removed
+                        removed << { key: claim_key, index: index }
+                    end
+                end
+            else
+                # All claims in current content with this key were added
+                current_claims.each_index do |index|
+                    added << { key: claim_key, index: index }
+                end
+            end
+            end
+
+            parent_content_claims.each do |claim_key, parent_claims|
+                # current content[claims] can be nil
+                parent_claims.each_index do |index|
+                    if current_content_claims.nil? || !current_content_claims.key?(claim_key)
+                        removed << { key: claim_key, index: index }
+                    end
+                end
+            end
+        end
+
+        {
+            added: added,
+            removed: removed,
+            changed: changed
+        }
+    end
+end

data/lib/wikidata/diff/label_analyzer.rb

@@ -1,53 +1,75 @@
-class LabelAnalyzer
+class LabelAnalyzer 
+      
     def self.isolate_labels_differences(current_content, parent_content)
         return {
-        changed: [],
-        removed: [],
-        added: []
+        changed_labels: [],
+        removed_labels: [],
+        added_labels: []
         } if current_content.nil? && parent_content.nil?
-    
-        current_labels = current_content['labels'] || {}
-        parent_labels = parent_content['labels'] || {}
-
-        changed_labels = []
-        removed_labels = []
-        added_labels = []
-
-        if current_labels.is_a?(Array) || parent_labels.is_a?(Array)
-        return {
-        changed: changed_labels,
-        removed: removed_labels,
-        added: added_labels
-        }
+        
+        if current_content
+            current_labels = current_content['labels']
+            if current_labels.nil? || current_labels.is_a?(Array)
+                current_labels = {}
+            end
+        else
+            current_labels = {}
         end
 
-    
-        # Iterate over each language in the current labels
-        (current_labels || {}).each do |lang, current_label|
-        parent_label = parent_labels[lang]
-    
-        if parent_label.nil?
-            added_labels << { lang: lang }
-        elsif current_label != parent_label
-            changed_labels << { lang: lang }
-        end
-        end
-    
-        # Iterate over each language in the parent labels to find removed labels
-        (parent_labels || {}).each do |lang, parent_label|
-        if current_labels[lang].nil?
-            removed_labels << { lang: lang }
-        end
+        if parent_content
+            parent_labels = parent_content['labels']
+            if parent_labels.nil? || parent_labels.is_a?(Array)
+                parent_labels = {}
+            end
+        else
+            parent_labels = {}
         end
 
-        # puts "Changed labels: #{changed_labels}"
-        # puts "Removed labels: #{removed_labels}"
-        # puts "Added labels: #{added_labels}"
+
+
+        changed_labels_labels = []
+        removed_labels_labels = []
+        added_labels_labels = []
+
+        # if parentid is 0, then add all labels as added_labels and return it
+        if parent_content.nil?  
+            if !current_labels.empty?
+                current_labels.each do |lang, label|
+                    added_labels_labels << { lang: lang }
+                end
+            end
+            return {
+                changed_labels: changed_labels_labels,
+                removed_labels: removed_labels_labels,
+                added_labels: added_labels_labels
+            }
+        else
+            # Iterate over each language in the current labels
+            (current_labels).each do |lang, current_label|
+                if parent_labels.empty?
+                    added_labels_labels << { lang: lang }
+                else
+                    parent_label = parent_labels[lang]
+                    if parent_label.nil?
+                        added_labels_labels << { lang: lang }
+                    elsif current_label != parent_label
+                        changed_labels_labels << { lang: lang }
+                    end
+                end
+            end
+            
+                # Iterate over each language in the parent labels to find removed_labels labels
+                (parent_labels).each do |lang, parent_label|
+                if current_labels.empty?
+                    removed_labels_labels << { lang: lang }
+                end
+            end
+        end
     
         {
-        changed: changed_labels,
-        removed: removed_labels,
-        added: added_labels
+        changed_labels: changed_labels_labels,
+        removed_labels: removed_labels_labels,
+        added_labels: added_labels_labels
         }
     end
 end

data/lib/wikidata/diff/large_batches_analyzer.rb

@@ -5,34 +5,49 @@ class LargeBatchesAnalyzer
     def self.handle_large_batches(revision_ids, batch_size)
         revision_contents = {}
         parent_contents = {}
-    
-    
-        revision_ids_batches = revision_ids.each_slice(batch_size).to_a
-        revision_ids_batches.each do |batch|
-            parsed_contents = Api.get_revision_contents(batch)
-            if parsed_contents
-                parent_ids = []
-                revision_contents.merge!(parsed_contents) if parsed_contents
-                parsed_contents.values.each do |data|
-                    parent_id = data[:parentid]
-                    
-                    if parent_id != 0 && !parent_id.nil?
-                        parent_ids << parent_id
-                    end
-                end
-                parent_contents_batch = Api.get_revision_contents(parent_ids)
-                parent_contents.merge!(parent_contents_batch) if parent_contents_batch
+        first_revisions = []
+      
+        revision_ids.each_slice(batch_size) do |batch|
+          parent_ids = []
+          parsed_contents = Api.get_revision_contents(batch)
+          next unless parsed_contents
+
+          # I have to check if any of the revision ids in the parsed content has parentid == 0
+          parsed_contents.each do |revid, data|
+            if data[:parentid] == 0
+              first_revisions << revid
+            else
+              parent_ids << data[:parentid]
             end
+          end
+          revision_contents.merge!(parsed_contents)
+          parent_contents_batch = Api.get_revision_contents(parent_ids)
+          parent_contents.merge!(parent_contents_batch) if parent_contents_batch
         end
-    
+
+        build_result(revision_contents, parent_contents, first_revisions)
+      end
+      
+      def self.build_result(revision_contents, parent_contents, first_revisions)
         result = {}
         revision_contents.each do |revid, data|
-            parentid = data[:parentid]
-            parent_content = parent_contents[parentid] if parentid
-            current = data ? data[:content] : nil
-            parent = parent_content ? parent_content[:content] : nil
-            result[revid] = { current_content: current, parent_content: parent }    
+          parent_content = parent_contents[data[:parentid]]
+          result[revid] = {
+            current_content: data&.fetch(:content, nil),
+            parent_content: parent_content&.fetch(:content, nil),
+            comment: data&.fetch(:comment, nil),
+            model: data&.fetch(:model, nil)
+          }
+        end
+        first_revisions.each do |revid|
+          result[revid] = {
+            current_content: revision_contents[revid]&.fetch(:content, nil),
+            parent_content: nil,
+            comment: revision_contents[revid]&.fetch(:comment, nil),
+            model: revision_contents[revid]&.fetch(:model, nil)
+          }
         end
         result
-    end
+      end
+      
 end

data/lib/wikidata/diff/lemma_analyzer.rb

@@ -0,0 +1,70 @@
+class LemmaAnalyzer
+    def self.isolate_lemmas_differences(current_content, parent_content)
+        return {
+        changed_lemmas: [],
+        removed_lemmas: [],
+        added_lemmas: []
+        } if current_content.nil? && parent_content.nil?
+    
+        
+        if current_content
+            current_labels = current_content['lemmas']
+            if current_labels.nil? || current_labels.is_a?(Array)
+                current_labels = {}
+            end
+        else
+            current_labels = {}
+        end
+        if parent_content
+            parent_labels = parent_content['lemmas'] 
+            if parent_labels.nil? || parent_labels.is_a?(Array)
+                parent_labels = {}
+            end
+        else
+            parent_labels = {}
+        end
+
+        changed_labels = []
+        removed_labels = []
+        added_labels = []
+
+
+        # if parentid is 0, then add all labels as added and return it
+        if parent_content.nil?
+            current_labels.each do |lang, label|
+                added_labels << { lang: lang }
+            end
+            return {
+                changed_lemmas: changed_labels,
+                removed_lemmas: removed_labels,
+                added_lemmas: added_labels
+            }
+        else
+
+        
+            # Iterate over each language in the current labels
+            (current_labels || {}).each do |lang, current_label|
+                parent_label = parent_labels[lang]
+            
+                if parent_label.nil?
+                    added_labels << { lang: lang }
+                elsif current_label != parent_label
+                    changed_labels << { lang: lang }
+                end
+                end
+            
+                # Iterate over each language in the parent labels to find removed labels
+                (parent_labels || {}).each do |lang, parent_label|
+                if current_labels[lang].nil?
+                    removed_labels << { lang: lang }
+                end
+            end
+        end
+    
+        {
+        changed_lemmas: changed_labels,
+        removed_lemmas: removed_labels,
+        added_lemmas: added_labels
+        }
+    end
+end

data/lib/wikidata/diff/qualifier_analyzer.rb

@@ -0,0 +1,83 @@
+class QualifierAnalyzer
+    # helper method for adding qualifiers
+    # handles added and removed qualifiers
+    def self.qualifier_updates(claim, updated_qualifiers, claim_key, claim_index)
+        if claim["qualifiers"]
+            qualifiers = claim["qualifiers"]
+            qualifiers.each do |qualifier_key, qualifier_values|
+                qualifier_values.each_with_index do |qualifier_value, qualifier_index|
+                updated_qualifiers << {
+                claim_key: claim_key,
+                claim_index: claim_index,
+                qualifier_key: qualifier_key,
+                qualifier_index: qualifier_index
+                }
+                end
+            end
+        end
+        updated_qualifiers
+    end
+
+    # helper method for changed qualifiers
+    def self.handle_changed_qualifiers(current_claim, parent_claim, changed_qualifiers, added_qualifiers, removed_qualifiers, claim_key, claim_index)
+        current_qualifiers = current_claim["qualifiers"] ? current_claim["qualifiers"] : {}
+        parent_qualifiers = parent_claim["qualifiers"] ? parent_claim["qualifiers"] : {}
+
+        current_qualifiers.each do |qualifier_key, qualifier_values|
+            qualifier_values.each_with_index do |qualifier_value, qualifier_index|
+                if parent_qualifiers.key?(qualifier_key)
+                parent = parent_qualifiers[qualifier_key]
+                end
+                # Check if the qualifier index exists in the parent content
+                if !parent.nil?
+                parent = parent[qualifier_index]
+                # check if the parent claim was changed by comparing the objects first
+                if parent != qualifier_value
+                    # Claim was changed
+                    changed_qualifiers << {
+                    claim_key: claim_key,
+                    claim_index: claim_index,
+                    qualifier_key: qualifier_key,
+                    qualifier_index: qualifier_index
+                    }
+                end
+                else
+                # Claim was added
+                added_qualifiers << {
+                    claim_key: claim_key,
+                    claim_index: claim_index,
+                    qualifier_key: qualifier_key,
+                    qualifier_index: qualifier_index
+                }
+                end
+            end
+        end
+        # Check for removed claims
+        parent_qualifiers.each do |qualifier_key, qualifier_values|
+            qualifier_values.each_with_index do |qualifier_value, qualifier_index|
+                if current_qualifiers.key?(qualifier_key)
+                current = current_qualifiers[qualifier_key]
+                end
+                # Check if the qualifier index exists in the current content
+                if !current.nil?
+                current = current[qualifier_index]
+                end
+                if current.nil?
+                # Claim was removed
+                removed_qualifiers << {
+                    claim_key: claim_key,
+                    claim_index: claim_index,
+                    qualifier_key: qualifier_key,
+                    qualifier_index: qualifier_index
+                }
+                end
+            end
+        end
+
+        {
+            added_qualifiers: added_qualifiers,
+            removed_qualifiers: removed_qualifiers,
+            changed_qualifiers: changed_qualifiers
+        }
+    end
+end

data/lib/wikidata/diff/reference_analyzer.rb

@@ -0,0 +1,49 @@
+class ReferenceAnalyzer
+    # helper method for adding and removing references
+    def self.reference_updates(claim, updated_references, claim_key, claim_index)
+        if claim["references"]
+            claim["references"].each_with_index do |current_ref, ref_index|
+                updated_references << { claim_key: claim_key, claim_index: claim_index, reference_index: ref_index }
+            end
+        end
+        updated_references
+    end
+
+    # helper method for changed references
+    def self.handle_changed_references(current_claim, parent_claim, changed_references, added_references, removed_references, claim_key, claim_index)
+        current_references = current_claim["references"] ? current_claim["references"] : []
+        parent_references = parent_claim["references"] ? parent_claim["references"] : []
+    
+        current_references.each_with_index do |current_ref, ref_index|
+            if parent_references.empty?
+                added_references << { claim_key: claim_key, claim_index: claim_index, reference_index: ref_index }
+            elsif !parent_references.include?(current_ref)
+                added_references << { claim_key: claim_key, claim_index: claim_index, reference_index: ref_index }
+            elsif ref_modified?(current_ref, parent_references)
+                changed_references << { claim_key: claim_key, claim_index: claim_index, reference_index: ref_index }
+            end
+        end
+    
+        parent_references.each_with_index do |parent_ref, ref_index|
+            if !current_references.include?(parent_ref)
+                removed_references << { claim_key: claim_key, claim_index: claim_index, reference_index: ref_index }
+            end
+        end
+
+        {
+            added_references: added_references,
+            removed_references: removed_references,
+            changed_references: changed_references
+        }
+    end
+
+    # helper method for checking if a reference has been modified
+    def self.ref_modified?(current_reference, parent_references)
+        parent_references.each do |parent_reference|
+            if current_reference["snaks"] != parent_reference["snaks"]
+                return true
+            end
+        end
+        false
+    end
+end

data/lib/wikidata/diff/representation_analyzer.rb

@@ -0,0 +1,71 @@
+class RepresentationAnalyzer
+    def self.isolate_representation_differences(current_content, parent_content)
+        return {
+        changed: [],
+        removed: [],
+        added: []
+        } if current_content.nil? && parent_content.nil?
+    
+        if current_content
+            current_representations = current_content['representations']
+            if current_representations.nil? || current_representations.is_a?(Array)
+                current_representations = {}
+            end
+        else
+            current_representations = {}
+        end
+        
+        if parent_content
+            parent_representations = parent_content['representations'] 
+            if parent_representations.nil? || parent_representations.is_a?(Array)
+                parent_representations = {}
+            end
+        else
+            parent_representations = {}
+        end
+
+
+        changed = []
+        removed = []
+        added = []
+
+
+        # if parentid is 0, then add all labels as added and return it
+        if parent_content.nil?
+            current_representations.each do |lang, label|
+                added << { lang: lang }
+            end
+            return {
+                changed: changed,
+                removed: removed,
+                added: added
+            }
+        else
+
+        
+            # Iterate over each language in the current labels
+            (current_representations || {}).each do |lang, current_representation|
+                parent_representation =  parent_representations[lang]
+            
+                if parent_representation.nil?
+                    added << { lang: lang }
+                elsif current_representation != parent_representation
+                    changed << { lang: lang }
+                end
+                end
+            
+                # Iterate over each language in the parent labels to find removed labels
+            (parent_representations || {}).each do |lang, parent_representation|
+                if current_representations[lang].nil?
+                    removed << { lang: lang }
+                end
+            end
+        end
+    
+        {
+        changed: changed,
+        removed: removed,
+        added: added
+        }
+    end
+end