Project model refactored. Modularity gem

app/models/project/repository_trait.rb

@@ -0,0 +1,109 @@
+module Project::RepositoryTrait
+  as_trait do
+    def valid_repo?
+      repo
+    rescue
+      errors.add(:path, "Invalid repository path")
+      false
+    end
+
+    def commit(commit_id = nil)
+      Commit.find_or_first(repo, commit_id)
+    end
+
+    def fresh_commits(n = 10)
+      Commit.fresh_commits(repo, n)
+    end
+
+    def commits_with_refs(n = 20)
+      Commit.commits_with_refs(repo, n)
+    end
+
+    def commits_since(date)
+      Commit.commits_since(repo, date)
+    end
+
+    def commits(ref, path = nil, limit = nil, offset = nil)
+      Commit.commits(repo, ref, path, limit, offset)
+    end
+
+    def commits_between(from, to)
+      Commit.commits_between(repo, from, to)
+    end
+
+    def write_hooks
+      %w(post-receive).each do |hook|
+        write_hook(hook, File.read(File.join(Rails.root, 'lib', "#{hook}-hook")))
+      end
+    end
+
+    def write_hook(name, content)
+      hook_file = File.join(path_to_repo, 'hooks', name)
+
+      File.open(hook_file, 'w') do |f|
+        f.write(content)
+      end
+
+      File.chmod(0775, hook_file)
+    end
+
+    def tags
+      repo.tags.map(&:name).sort.reverse
+    end
+
+    def repo
+      @repo ||= Grit::Repo.new(path_to_repo)
+    end
+
+    def url_to_repo
+      Gitlabhq::GitHost.url_to_repo(path)
+    end
+
+    def path_to_repo
+      File.join(GIT_HOST["base_path"], "#{path}.git")
+    end
+
+    def update_repository
+      Gitlabhq::GitHost.system.update_project(path, self)
+
+      write_hooks if File.exists?(path_to_repo)
+    end
+
+    def destroy_repository
+      Gitlabhq::GitHost.system.destroy_project(self)
+    end
+
+    def repo_exists?
+      @repo_exists ||= (repo && !repo.branches.empty?)
+    rescue 
+      @repo_exists = false
+    end
+
+    def heads
+      @heads ||= repo.heads
+    end
+
+    def tree(fcommit, path = nil)
+      fcommit = commit if fcommit == :head
+      tree = fcommit.tree
+      path ? (tree / path) : tree
+    end
+
+    def open_branches
+      if protected_branches.empty?
+        self.repo.heads
+      else
+        pnames = protected_branches.map(&:name)
+        self.repo.heads.reject { |h| pnames.include?(h.name) }
+      end.sort_by(&:name)
+    end
+
+    def has_commits?
+      !!commit
+    end
+
+    def root_ref 
+      default_branch || "master"
+    end
+  end
+end