joelmoss-grit 1.1.4

data/lib/grit/commit_stats.rb

@@ -0,0 +1,128 @@
+module Grit
+  
+  class CommitStats
+    
+    attr_reader :id, :files, :additions, :deletions, :total
+    
+    # Instantiate a new CommitStats
+    #   +id+ is the id of the commit
+    #   +files+ is an array of :
+    #     [ [filename, adds, deletes, total],
+    #       [filename, adds, deletes, total],
+    #       [filename, adds, deletes, total] ]
+    #
+    # Returns Grit::CommitStats (baked)
+    def initialize(repo, id, files)
+      @repo = repo
+      @id = id
+      @files = files
+      @additions  = files.inject(0) { |total, a| total += a[1] } 
+      @deletions  = files.inject(0) { |total, a| total += a[2] } 
+      @total  = files.inject(0) { |total, a| total += a[3] } 
+    end
+    
+    # Find all commit stats matching the given criteria.
+    #   +repo+ is the Repo
+    #   +ref+ is the ref from which to begin (SHA1 or name) or nil for --all
+    #   +options+ is a Hash of optional arguments to git
+    #     :max_count is the maximum number of commits to fetch
+    #     :skip is the number of commits to skip
+    #
+    # Returns assoc array [sha, Grit::Commit[] (baked)]
+    def self.find_all(repo, ref, options = {})
+      allowed_options = [:max_count, :skip, :since]
+            
+      default_options = {:numstat => true}
+      actual_options = default_options.merge(options)
+      
+      if ref
+        output = repo.git.log(actual_options, ref)
+      else
+        output = repo.git.log(actual_options.merge(:all => true))
+      end
+      
+      self.list_from_string(repo, output)
+    end
+    
+    # Parse out commit information into an array of baked Commit objects
+    #   +repo+ is the Repo
+    #   +text+ is the text output from the git command (raw format)
+    #
+    # Returns assoc array [sha, Grit::Commit[] (baked)]
+    def self.list_from_string(repo, text)
+      lines = text.split("\n")
+      
+      commits = []
+
+      while !lines.empty?
+        id = lines.shift.split.last
+        
+        lines.shift
+        lines.shift
+        lines.shift
+        
+        message_lines = []
+        message_lines << lines.shift[4..-1] while lines.first =~ /^ {4}/ || lines.first == ''
+        
+        lines.shift while lines.first && lines.first.empty?
+
+        files = []
+        while lines.first =~ /^([-\d]+)\s+([-\d]+)\s+(.+)/
+          (additions, deletions, filename) = lines.shift.split
+          additions = additions.to_i
+          deletions = deletions.to_i
+          total = additions + deletions
+          files << [filename, additions, deletions, total]
+        end
+        
+        lines.shift while lines.first && lines.first.empty?
+        
+        commits << [id, CommitStats.new(repo, id, files)]
+      end
+      
+      commits
+    end
+    
+    # Pretty object inspection
+    def inspect
+      %Q{#<Grit::CommitStats "#{@id}">}
+    end
+    
+    # Convert to an easy-to-traverse structure
+    def to_diffstat
+      files.map do |metadata|
+        DiffStat.new(*metadata)
+      end
+    end
+    
+    # private
+    
+    def to_hash
+      {
+        'id'        => id,
+        'files'     => files,
+        'additions' => additions,
+        'deletions' => deletions,
+        'total'     => total
+      }
+    end
+    
+  end # CommitStats
+  
+  class DiffStat
+    attr_reader :filename, :additions, :deletions
+
+    def initialize(filename, additions, deletions, total=nil)
+      @filename, @additions, @deletions = filename, additions, deletions
+    end
+    
+    def net
+      additions - deletions
+    end
+
+    def inspect
+      "#{filename}: +#{additions} -#{deletions}"
+    end
+  end
+  
+end # Grit

data/lib/grit/config.rb

@@ -0,0 +1,44 @@
+module Grit
+  
+  class Config
+    def initialize(repo)
+      @repo = repo
+    end
+    
+    def []=(key, value)
+      @repo.git.config({}, key, value)
+      @data = nil
+    end
+    
+    def [](key)
+      data[key]
+    end
+    
+    def fetch(key, default = nil)
+      data[key] || default || raise(IndexError.new("key not found"))
+    end
+    
+    def keys
+      data.keys
+    end
+    
+    protected
+      def data
+        @data ||= load_config
+      end
+      
+      def load_config
+        hash = {}
+        config_lines.map do |line|
+          key, value = line.split(/=/, 2)
+          hash[key] = value
+        end
+        hash
+      end
+      
+      def config_lines
+        @repo.git.config(:list => true).split(/\n/)
+      end
+  end # Config
+  
+end # Grit

data/lib/grit/diff.rb

@@ -0,0 +1,70 @@
+module Grit
+  
+  class Diff
+    attr_reader :a_path, :b_path
+    attr_reader :a_blob, :b_blob
+    attr_reader :a_mode, :b_mode
+    attr_reader :new_file, :deleted_file
+    attr_reader :diff
+    
+    def initialize(repo, a_path, b_path, a_blob, b_blob, a_mode, b_mode, new_file, deleted_file, diff)
+      @repo = repo
+      @a_path = a_path
+      @b_path = b_path
+      @a_blob = a_blob =~ /^0{40}$/ ? nil : Blob.create(repo, :id => a_blob)
+      @b_blob = b_blob =~ /^0{40}$/ ? nil : Blob.create(repo, :id => b_blob)
+      @a_mode = a_mode
+      @b_mode = b_mode
+      @new_file = new_file
+      @deleted_file = deleted_file
+      @diff = diff
+    end
+    
+    def self.list_from_string(repo, text)
+      lines = text.split("\n")
+      
+      diffs = []
+      
+      while !lines.empty?
+        m, a_path, b_path = *lines.shift.match(%r{^diff --git a/(.+?) b/(.+)$})
+        
+        if lines.first =~ /^old mode/
+          m, a_mode = *lines.shift.match(/^old mode (\d+)/)
+          m, b_mode = *lines.shift.match(/^new mode (\d+)/)
+        end
+        
+        if lines.empty? || lines.first =~ /^diff --git/
+          diffs << Diff.new(repo, a_path, b_path, nil, nil, a_mode, b_mode, false, false, nil)
+          next
+        end
+        
+        new_file = false
+        deleted_file = false
+        
+        if lines.first =~ /^new file/
+          m, b_mode = lines.shift.match(/^new file mode (.+)$/)
+          a_mode = nil
+          new_file = true
+        elsif lines.first =~ /^deleted file/
+          m, a_mode = lines.shift.match(/^deleted file mode (.+)$/)
+          b_mode = nil
+          deleted_file = true
+        end
+        
+        m, a_blob, b_blob, b_mode = *lines.shift.match(%r{^index ([0-9A-Fa-f]+)\.\.([0-9A-Fa-f]+) ?(.+)?$})
+        b_mode.strip! if b_mode
+        
+        diff_lines = []
+        while lines.first && lines.first !~ /^diff/
+          diff_lines << lines.shift
+        end
+        diff = diff_lines.join("\n")
+        
+        diffs << Diff.new(repo, a_path, b_path, a_blob, b_blob, a_mode, b_mode, new_file, deleted_file, diff)
+      end
+      
+      diffs
+    end
+  end # Diff
+  
+end # Grit

data/lib/grit/errors.rb

@@ -0,0 +1,7 @@
+module Grit
+  class InvalidGitRepositoryError < StandardError
+  end
+  
+  class NoSuchPathError < StandardError
+  end
+end

data/lib/grit/git-ruby.rb

@@ -0,0 +1,186 @@
+require 'grit/git-ruby/repository'
+require 'grit/git-ruby/file_index'
+
+module Grit
+  
+  # the functions in this module intercept the calls to git binary
+  # made buy the grit objects and attempts to run them in pure ruby
+  # if it will be faster, or if the git binary is not available (!!TODO!!)
+  module GitRuby
+    
+    attr_accessor :ruby_git_repo, :git_file_index
+    
+    def init(options = {})
+      if options.size == 0
+        Grit::GitRuby::Repository.init(@git_dir)
+      else
+        method_missing('init', options) 
+      end
+    end
+    
+    def cat_file(options, ref)
+      if options[:t]
+        file_type(ref)
+      elsif options[:s]
+        file_size(ref)
+      elsif options[:p]
+        try_run { ruby_git.cat_file(ref) }
+      end
+    rescue Grit::GitRuby::Repository::NoSuchShaFound
+      ''
+    end
+    
+    # lib/grit/tree.rb:16:      output = repo.git.ls_tree({}, treeish, *paths)
+    def ls_tree(options, treeish, *paths)
+      sha = rev_parse({}, treeish)
+      ruby_git.ls_tree(sha, paths.flatten)
+    rescue Grit::GitRuby::Repository::NoSuchShaFound
+      ''
+    end
+
+    # git diff --full-index 'ec037431382e83c3e95d4f2b3d145afbac8ea55d' 'f1ec1aea10986159456846b8a05615b87828d6c6'
+    def diff(options, sha1, sha2)
+      try_run { ruby_git.diff(sha1, sha2, options) }
+    end
+    
+    def rev_list(options, ref = 'master')
+      options.delete(:skip) if options[:skip].to_i == 0
+      allowed_options = [:max_count, :since, :until, :pretty]  # this is all I can do right now
+      if ((options.keys - allowed_options).size > 0)
+        return method_missing('rev-list', options, ref)
+      elsif (options.size == 0)
+        # pure rev-list
+        begin
+          return file_index.commits_from(rev_parse({}, ref)).join("\n") + "\n"
+        rescue
+          return method_missing('rev-list', options, ref) 
+        end
+      else
+        aref = rev_parse({}, ref)
+        if aref.is_a? Array
+          return method_missing('rev-list', options, ref) 
+        else
+          return try_run { ruby_git.rev_list(aref, options) }
+        end
+      end
+    end
+    
+    def rev_parse(options, string)
+      raise RuntimeError, "invalid string: #{string}" unless string.is_a?(String)
+
+      if string =~ /\.\./
+        (sha1, sha2) = string.split('..')
+        return [rev_parse({}, sha1), rev_parse({}, sha2)]
+      end
+
+      if /^[0-9a-f]{40}$/.match(string)  # passing in a sha - just no-op it
+        return string.chomp
+      end
+
+      head = File.join(@git_dir, 'refs', 'heads', string)
+      return File.read(head).chomp if File.file?(head)
+
+      head = File.join(@git_dir, 'refs', 'remotes', string)
+      return File.read(head).chomp if File.file?(head)
+      
+      head = File.join(@git_dir, 'refs', 'tags', string)
+      return File.read(head).chomp if File.file?(head)
+      
+      ## check packed-refs file, too 
+      packref = File.join(@git_dir, 'packed-refs')
+      if File.file?(packref)
+        File.readlines(packref).each do |line|
+          if m = /^(\w{40}) refs\/.+?\/(.*?)$/.match(line)
+            next if !Regexp.new(Regexp.escape(string) + '$').match(m[3])
+            return m[1].chomp
+          end
+        end
+      end
+      
+      ## !! more - partials and such !!
+      
+      # revert to calling git - grr
+      return method_missing('rev-parse', {}, string).chomp
+    end
+    
+    def file_size(ref)
+      try_run { ruby_git.cat_file_size(ref).to_s }
+    end
+    
+    def file_type(ref)
+      try_run { ruby_git.cat_file_type(ref).to_s }
+    end
+    
+    def blame_tree(commit, path = nil)
+      begin
+        path = [path].join('/').to_s + '/' if (path && path != '')
+        path = '' if !path.is_a? String
+        commits = file_index.last_commits(rev_parse({}, commit), looking_for(commit, path))
+        clean_paths(commits)
+      rescue FileIndex::IndexFileNotFound
+        {}
+      end
+    end
+    
+    def file_index
+      @git_file_index ||= FileIndex.new(@git_dir)
+    end
+    
+    def ruby_git
+      @ruby_git_repo ||= Repository.new(@git_dir)
+    end
+    
+    private
+    
+      def try_run
+        ret = ''
+        Timeout.timeout(self.class.git_timeout) do
+          ret = yield
+        end
+        @bytes_read += ret.size
+
+        #if @bytes_read > 5242880 # 5.megabytes
+        #  bytes = @bytes_read
+        #  @bytes_read = 0
+        #  raise Grit::Git::GitTimeout.new(command, bytes) 
+        #end
+
+        ret
+      rescue Timeout::Error => e
+        bytes = @bytes_read
+        @bytes_read = 0
+        raise Grit::Git::GitTimeout.new(command, bytes)
+      end
+    
+      def looking_for(commit, path = nil)
+        tree_sha = ruby_git.get_subtree(rev_parse({}, commit), path)
+
+        looking_for = []
+        ruby_git.get_object_by_sha1(tree_sha).entry.each do |e|
+          if path && !(path == '' || path == '.' || path == './')
+            file = File.join(path, e.name)
+          else
+            file = e.name
+          end
+          file += '/' if e.type == :directory
+          looking_for << file
+        end
+        looking_for
+      end
+    
+      def clean_paths(commit_array)
+        new_commits = {}
+        commit_array.each do |file, sha|
+          file = file.chop if file[file.size - 1 , 1] == '/'
+          new_commits[file] = sha
+        end
+        new_commits
+      end
+
+    # TODO     
+    # git grep -n 'foo' 'master'
+    # git log --pretty='raw' --max-count='1' 'master' -- 'LICENSE'
+    # git log --pretty='raw' --max-count='1' 'master' -- 'test'
+    
+  end
+end

data/lib/grit/git-ruby/commit_db.rb

@@ -0,0 +1,52 @@
+begin
+  require 'sequel'
+  
+  module Grit
+  
+    class CommitDb
+    
+      SCHEMA_VERSION = 1
+
+      attr_accessor :db, :git
+        
+      def initialize(git_obj, index_location = nil)
+        @git = git_obj
+        db_file = File.join(index_location || @git.git_dir, 'commit_db')
+        if !File.exists?(db_file)
+          @db = Sequel.open "sqlite:///#{db_file}"
+          setup_tables
+        else
+          @db = Sequel.open "sqlite:///#{db_file}"
+        end
+      end
+      
+      def rev_list(branch, options)
+      end
+      
+      def update_db(branch = nil)
+        # find all refs/heads, for each
+        # add branch if not there
+        # go though all commits in branch
+          # add new commit_branches a
+          # and commit_nodes for each new one
+          # stop if reach commit that already has branch and node links
+      end
+
+      def setup_tables
+        @db << "create table meta (meta_key text, meta_value text)"
+        @db[:meta] << {:meta_key => 'schema', :meta_value => SCHEMA_VERSION}
+        
+        @db << "create table commits (id integer, sha text, author_date integer)"
+        @db << "create table nodes (id integer, path text, type text)"
+        @db << "create table branches (id integer, ref text, commit_id integer)"
+        
+        @db << "create table commit_branches (commit_id integer, branch_id integer)"
+        @db << "create table commit_nodes (commit_id integer, node_id integer, node_sha string)"        
+      end
+    
+    end
+  end
+        
+rescue LoadError
+  # no commit db
+end