Refactor: Removing the duplicated code.

app/models/network/commit.rb

@@ -26,5 +26,14 @@ module Network
         0
       end
     end
+
+    def parents(map)
+      @commit.parents.map do |p|
+        if map.include?(p.id)
+          map[p.id]
+        end
+      end
+      .compact
+    end
   end
 end

app/models/network/graph.rb

@@ -2,7 +2,7 @@ require "grit"
 
 module Network
   class Graph
-    attr_reader :days, :commits
+    attr_reader :days, :commits, :map
 
     def self.max_count
       @max_count ||= 650
@@ -61,9 +61,7 @@ module Network
       end
 
       commits_sort_by_ref.each do |commit|
-        if @map.include? commit.id then
+        place_chain(commit)
-          place_chain(commit)
-        end
       end
 
       # find parent spaces for not overlap lines
@@ -115,25 +113,21 @@ module Network
     def find_free_parent_spaces(commit)
       spaces = []
 
-      commit.parents.each do |p|
+      commit.parents(@map).each do |parent|
-        if @map.include?(p.id) then
+        range = if commit.time < parent.time then
-          parent = @map[p.id]
+                  commit.time..parent.time
+                else
+                  parent.time..commit.time
+                end
 
-          range = if commit.time < parent.time then
+        space = if commit.space >= parent.space then
-                    commit.time..parent.time
+                  find_free_parent_space(range, parent.space, -1, commit.space)
-                  else
+                else
-                    parent.time..commit.time
+                  find_free_parent_space(range, commit.space, -1, parent.space)
-                  end
+                end
 
-          space = if commit.space >= parent.space then
+        mark_reserved(range, space)
-                    find_free_parent_space(range, parent.space, -1, commit.space)
+        spaces << space
-                  else
-                    find_free_parent_space(range, commit.space, -1, parent.space)
-                  end
-
-          mark_reserved(range, space)
-          spaces << space
-        end
       end
 
       spaces
@@ -175,25 +169,18 @@ module Network
       leaves.each do |l|
         l.spaces << space
         # Also add space to parent
-        l.parents.each do |p|
+        l.parents(@map).each do |parent|
-          if @map.include?(p.id)
+          if parent.space > 0
-            parent = @map[p.id]
+            parent.spaces << space
-            if parent.space > 0
-              parent.spaces << space
-            end
           end
         end
       end
 
       # and mark it as reserved
       min_time = leaves.last.time
-      parents = leaves.last.parents.collect
+      leaves.last.parents(@map).each do |parent|
-      parents.each do |p|
+        if parent.time < min_time
-        if @map.include? p.id
+          min_time = parent.time
-          parent = @map[p.id]
-          if parent.time < min_time
-            min_time = parent.time
-          end
         end
       end
 
@@ -206,7 +193,7 @@ module Network
 
       # Visit branching chains
       leaves.each do |l|
-        parents = l.parents.collect.select{|p| @map.include? p.id and @map[p.id].space.zero?}
+        parents = l.parents(@map).select{|p| p.space.zero?}
         for p in parents
           place_chain(p, l.time)
         end
@@ -215,13 +202,10 @@ module Network
 
     def get_space_base(leaves)
       space_base = 1
-      if leaves.last.parents.size > 0
+      parents = leaves.last.parents(@map)
-        first_parent = leaves.last.parents.first
+      if parents.size > 0
-        if @map.include?(first_parent.id)
+        if parents.first.space > 0
-          first_p = @map[first_parent.id]
+          space_base = parents.first.space
-          if first_p.space > 0
-            space_base = first_p.space
-          end
         end
       end
       space_base
@@ -266,10 +250,9 @@ module Network
       leaves.push(commit) if commit.space.zero?
 
       while true
-        return leaves if commit.parents.count.zero?
+        return leaves if commit.parents(@map).count.zero?
-        return leaves unless @map.include? commit.parents.first.id
 
-        commit = @map[commit.parents.first.id]
+        commit = commit.parents(@map).first
 
         return leaves unless commit.space.zero?
 

app/views/graph/show.json.erb

@@ -5,7 +5,7 @@
     days: @graph.days.compact.map { |d| [d.day, d.strftime("%b")] },
     commits: @graph.commits.map do |c|
       {
-        parents: parents_zip_spaces(c.parents, c.parent_spaces),
+        parents: parents_zip_spaces(c.parents(@graph.map), c.parent_spaces),
         author: {
           name: c.author.name,
           email: c.author.email,