tfs_graph 0.1.1 → 0.1.2

data/lib/tfs_graph/extensions.rb

@@ -0,0 +1,27 @@
+require 'active_support/inflector'
+require 'active_support/concern'
+
+module TFSGraph
+  module Extensions
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+      def base_class_name
+        name.demodulize
+      end
+    end
+
+    def base_class_name
+      self.class.base_class_name
+    end
+
+    private
+    def add_behavior(repo, additions)
+      repo.extend additions
+    end
+
+    def constantize(string)
+      string.constantize
+    end
+  end
+end

data/lib/tfs_graph/graph_populator.rb

@@ -3,7 +3,7 @@ require 'tfs_graph/branch/branch_store'
 require 'tfs_graph/changeset/changeset_store'
 require 'tfs_graph/changeset_merge/changeset_merge_store'
 
-require 'tfs_graph/associators/changeset_tree_creator'
+require 'tfs_graph/associators/changeset_tree_builder'
 require 'tfs_graph/associators/branch_associator'
 require 'tfs_graph/branch/branch_archive_handler'
 
@@ -30,6 +30,14 @@ module TFSGraph
       def populate_all_from_time(time)
         populate_graph(Populators::SinceDate, time)
       end
+
+      def update_project(project)
+        populate_graph(Populators::ForProject, project)
+      end
+
+      def update_branch(project, branch)
+        populate_graph(Populators::ForBranch, project, branch)
+      end
     end
   end
 end

data/lib/tfs_graph/persistable_entity.rb

@@ -0,0 +1,60 @@
+require 'tfs_graph/entity'
+require 'tfs_graph/extensions'
+
+module TFSGraph
+  class PersistableEntity < Entity
+    include Comparable
+    include Extensions
+
+    NotPersisted = Class.new(RuntimeError)
+
+    attr_accessor :db_object, :id
+
+    def self.repository
+      TFSGraph::RepositoryRegistry.instance.send "#{base_class_name.downcase}_repository"
+    end
+
+    def initialize(repo, args)
+      @repo = repo
+      super args
+    end
+
+    def persisted?
+      !db_object.nil?
+    end
+
+    def save!
+      @repo.save(self)
+    end
+
+    def delete!
+      @repo.delete(self)
+
+      @db_object = nil
+      @id = nil
+    end
+
+    def persist(id, db_object)
+      @id ||= id
+      @db_object = db_object
+
+      self
+    end
+
+    def db_object
+      return nil unless id
+
+      begin
+        @db_object ||= @repo.find_native(id)
+      rescue Repository::NotFound
+      end
+    end
+
+    def to_hash
+      hash = super
+      hash[:id] = id
+
+      hash
+    end
+  end
+end

data/lib/tfs_graph/populators/everything.rb

@@ -6,13 +6,25 @@ module TFSGraph
       def populate
         clean
 
-        collect_projects.map do |project|
-          branches = collect_branches(project)
-          branches.map {|branch| collect_changesets(branch) }
+        projects = ProjectStore.fetch_and_cache
 
-          branches.each {|branch| collect_merges(branch) }
+        projects.each do |project|
+          branches = BranchStore.new(project).fetch_and_cache
+
+          changesets = branches.select(&:active?).map do |branch|
+            changesets = ChangesetStore.new(branch).fetch_and_cache
+            ChangesetTreeBuilder.to_tree(branch, changesets)
+
+            branch.updated!
+            changesets
+          end
+
+          ChangesetTreeBuilder.set_branch_merges(changesets)
+
+          project.updated!
         end
 
+        BranchArchiveHandler.hide_moved_archives_for_project(projects)
         finalize
       end
     end

data/lib/tfs_graph/populators/for_archived_branch.rb

@@ -0,0 +1,28 @@
+module TFSGraph
+  module Populators
+
+    # incremental updates for a branch and its changesets
+    # does not update merges. have to do that once all changesets
+    # for a project are fetched
+    class ForArchivedBranch
+      include Utilities
+
+      def initialize(project, branch)
+        @project = project
+        @branch = branch
+
+        @changeset_store = ChangesetStore.new(branch)
+      end
+
+      def populate
+        return @branch.changesets unless @branch.changesets.empty?
+
+        all_changesets = @changeset_store.fetch_and_cache
+        ChangesetTreeBuilder.to_tree(@branch, all_changesets.sort)
+
+        @branch.updated!
+        all_changesets
+      end
+    end
+  end
+end

data/lib/tfs_graph/populators/for_branch.rb

@@ -0,0 +1,35 @@
+module TFSGraph
+  module Populators
+
+    # incremental updates for a branch and its changesets
+    # does not update merges. have to do that once all changesets
+    # for a project are fetched
+    class ForBranch
+      include Utilities
+
+      def initialize(project, branch)
+        @project = project
+        @branch = branch
+
+        @changeset_store = ChangesetStore.new(branch)
+      end
+
+      def populate
+        populate_since(@branch.last_updated.iso8601)
+      end
+
+      def populate_since(time)
+        new_changesets = @changeset_store.fetch_since_date(time)
+        @changeset_store.cache_all new_changesets
+
+        # skip rebuilding tree or marking as updated if no new were found
+        return new_changesets if new_changesets.empty?
+
+        ChangesetTreeBuilder.to_tree(@branch, @branch.changesets.sort)
+
+        @branch.updated!
+        new_changesets
+      end
+    end
+  end
+end

data/lib/tfs_graph/populators/for_project.rb

@@ -1,18 +1,35 @@
+require 'tfs_graph/populators/for_branch'
+
 module TFSGraph
   module Populators
+
+    #incremental updates for a project, its branches, changesets and merges
     class ForProject
       include Utilities
 
-      def initialize(project)
+      def initialize(project, since=Time.at(0).utc)
         @project = project
+        @branch_store = BranchStore.new(@project)
       end
 
       def populate
-        branches = collect_branches(@project)
-        branches.map {|branch| collect_changesets(branch) }
+        new_branches = @branch_store.fetch_since_last_update
+        @branch_store.cache_all(new_branches)
+
+        active_branches = @project.active_branches
+
+        changesets = active_branches.map do |branch|
+          ForBranch.new(@project, branch).populate
+          branch.changesets
+        end
+
+        # Add merges
+        active_branches.each {|branch| ChangesetMergeStore.new(branch).fetch_and_cache }
+        ChangesetTreeBuilder.set_branch_merges(changesets.flatten)
 
-        branches.each {|branch| collect_merges(branch) }
+        BranchArchiveHandler.hide_moved_archives_for_project(@project)
+        @project.updated!
       end
     end
   end
-end
+end

data/lib/tfs_graph/populators/since_date.rb

@@ -7,16 +7,32 @@ module TFSGraph
         @since = since
       end
 
+      # could use the ForProject populator, except that we allow
+      # specification of a date rather than using last updated.
+      # also dumps all existing data
       def populate
         clean
 
-        collect_projects.map do |project|
-          branches = collect_branches(project)
-          branches.map do |branch|
-            collect_changesets(branch, :cache_since_date, @since)
+        projects = ProjectStore.fetch_and_cache
+
+        projects.each do |project|
+          ForProject.new(project).populate
+          branches = BranchStore.new(project).fetch_and_cache
+
+          changesets = branches.select(&:active?).map do |branch|
+            changesets = ChangesetStore.new(branch).fetch_since_date @since
+            ChangesetTreeBuilder.to_tree(branch, changesets)
+
+            branch.updated!
+            changesets
           end
 
-          branches.each {|branch| collect_merges(branch) }
+          # setup merges
+          branches.each {|branch| ChangesetMergeStore.new(branch).fetch_and_cache }
+          ChangesetTreeBuilder.set_branch_merges(changesets)
+
+          BranchArchiveHandler.hide_moved_archives_for_project(@project)
+          project.updated!
         end
 
         finalize

data/lib/tfs_graph/populators/since_last.rb

@@ -4,23 +4,35 @@ module TFSGraph
       include Utilities
 
       def populate
-        ProjectStore.all_cached.map do |project|
-          new_changesets = project.active_branches.map {|branch| collect_changesets branch, :cache_since_last_update}
-          new_branches = BranchStore.new(project).cache_since_last_update
+        projects = ProjectStore.new.fetch_all
+        existing_projects = RepositoryRegistry.project_repository.all
 
-          new_changesets.concat new_branches.map {|branch| collect_changesets branch }
+        names = existing_projects.map(&:name)
+        new_project_data = projects.reject {|project| names.include? project[:name]  }
+        new_projects = ProjectStore.cache_all(new_project_data)
 
-          # recache and reassociate all merges for all branches.
-          # should not lead to dupes thanks to Related
-          new_branches.concat(project.branches).each do |branch|
-            collect_merges(branch)
-            BranchAssociator.associate(branch.changesets)
+        existing_projects.concat(new_projects).each do |project|
+          ForProject.new(project).populate
+          branches = BranchStore.new(project).fetch_and_cache
+
+          changesets = branches.select(&:active?).map do |branch|
+            changesets = ChangesetStore.new(branch).fetch_since_date @since
+            ChangesetTreeBuilder.to_tree(branch, changesets)
+
+            branch.updated!
+            changesets
           end
+
+          # setup merges
+          branches.each {|branch| ChangesetMergeStore.new(branch).fetch_and_cache }
+          ChangesetTreeBuilder.set_branch_merges(changesets)
+
+          BranchArchiveHandler.hide_moved_archives_for_project(@project)
+          project.updated!
         end
 
         finalize
       end
-
     end
   end
 end

data/lib/tfs_graph/populators/utilities.rb

@@ -8,30 +8,15 @@ module TFSGraph
       end
 
       def finalize
-        BranchArchiveHandler.hide_all_archives
         mark_as_updated
       end
 
-      def collect_projects
-        ProjectStore.cache
+      def find_project(name)
+        TFSGraph::RepositoryRegistry.project_repository.find_by_name name
       end
 
-      def collect_branches(project)
-        BranchStore.new(project).cache_all
-      end
-
-      def collect_changesets(branch, method=:cache_all, *args)
-        changesets = ChangesetStore.new(branch).send(method, *args)
-        generate_branch_tree(branch)
-        changesets.compact
-      end
-
-      def generate_branch_tree(branch)
-        ChangesetTreeCreator.to_tree branch
-      end
-
-      def collect_merges(branch)
-        ChangesetMergeStore.new(branch).cache
+      def find_branch(path)
+        TFSGraph::RepositoryRegistry.branch_repository.find_by_path path
       end
     end
   end

data/lib/tfs_graph/project.rb

@@ -1,33 +1,57 @@
-require 'tfs_graph/entity'
+require 'tfs_graph/persistable_entity'
 
 module TFSGraph
-  class Project < Entity
-    extend Comparable
+  class Project < PersistableEntity
     SCHEMA = {
-      name: {key: "Name"}
+      name: {key: "Name"},
+      last_updated: {type: Time, default: Time.at(0).utc},
+      hidden: {default: false, type: String}
     }
 
     act_as_entity
 
     def <=>(other)
-      name <=> other.name
+      id <=> other.id
     end
 
-    def last_change
-      branches.map {|b| b.last_changeset }
+    def updated!
+      @last_updated = Time.now.utc
+      save!
+    end
+
+    def hide!
+      self.hidden = true
+      save!
+    end
+
+    # force string
+    def hidden
+      @hidden.to_s
+    end
+
+    def hidden?
+      @hidden.to_s == "true"
+    end
+
+    def add_branch(branch)
+      branch.project = self.name
+      branch.save!
+
+      @repo.relate(:branches, db_object, branch.db_object)
+    end
+
+    def root_branches
+      @repo.root_branches(self)
     end
 
     def all_activity
-      branches.map {|b| b.changesets }.flatten
+      @repo.activity(self)
     end
 
     def all_activity_by_date(limiter=nil)
       raise InvalidArgument("parameter must be a Date") unless limiter.nil? || limiter.is_a?(Time)
 
-      activity = all_activity
-      activity = activity.select {|c| c.created > limiter } unless limiter.nil?
-
-      activity.group_by(&:formatted_created)
+      @repo.activity_by_date(self, limiter).group_by(&:formatted_created)
     end
 
     def active_branches
@@ -38,8 +62,16 @@ module TFSGraph
       branches_with_hidden.reject(&:hidden?)
     end
 
+    def branches_for_root(root)
+      @repo.branches_for_root(self, root)
+    end
+
+    def changesets_for_root(root)
+      @repo.changesets_for_root(self, root)
+    end
+
     def branches_with_hidden
-      outgoing(:branches).options(model: Branch).nodes.to_a
+      @repo.get_nodes(db_object, :outgoing, :branches, Branch)
     end
 
     %w(master release feature).each do |type|
@@ -51,6 +83,10 @@ module TFSGraph
         branches_with_hidden.select {|b| b.send "#{type}?" }
       end
 
+      define_method "active_#{type}s" do
+        active_branches.select {|b| b.send("#{type}?") }
+      end
+
       define_method "archived_#{type}s" do
         branches.select {|b| b.send("#{type}?") && b.archived? }
       end