steveh-grit 1.1.1

data/lib/grit/commit.rb

@@ -0,0 +1,242 @@
+module Grit
+
+  class Commit
+    attr_reader :id
+    lazy_reader :parents
+    lazy_reader :tree
+    lazy_reader :author
+    lazy_reader :authored_date
+    lazy_reader :committer
+    lazy_reader :committed_date
+    lazy_reader :message
+    lazy_reader :short_message
+    lazy_reader :author_string
+
+    # Instantiate a new Commit
+    #   +id+ is the id of the commit
+    #   +parents+ is an array of commit ids (will be converted into Commit instances)
+    #   +tree+ is the correspdonding tree id (will be converted into a Tree object)
+    #   +author+ is the author string
+    #   +authored_date+ is the authored Time
+    #   +committer+ is the committer string
+    #   +committed_date+ is the committed Time
+    #   +message+ is an array of commit message lines
+    #
+    # Returns Grit::Commit (baked)
+    def initialize(repo, id, parents, tree, author, authored_date, committer, committed_date, message)
+      @repo = repo
+      @id = id
+      @parents = parents.map { |p| Commit.create(repo, :id => p) }
+      @tree = Tree.create(repo, :id => tree)
+      @author = author
+      @authored_date = authored_date
+      @committer = committer
+      @committed_date = committed_date
+      @message = message.join("\n")
+      @short_message = message[0] || ''
+    end
+
+    def id_abbrev
+      @id_abbrev ||= @repo.git.rev_parse({}, self.id).chomp[0, 7]
+    end
+
+    # Create an unbaked Commit containing just the specified attributes
+    #   +repo+ is the Repo
+    #   +atts+ is a Hash of instance variable data
+    #
+    # Returns Grit::Commit (unbaked)
+    def self.create(repo, atts)
+      self.allocate.create_initialize(repo, atts)
+    end
+
+    # Initializer for Commit.create
+    #   +repo+ is the Repo
+    #   +atts+ is a Hash of instance variable data
+    #
+    # Returns Grit::Commit (unbaked)
+    def create_initialize(repo, atts)
+      @repo = repo
+      atts.each do |k, v|
+        instance_variable_set("@#{k}", v)
+      end
+      self
+    end
+
+    def lazy_source
+      self.class.find_all(@repo, @id, {:max_count => 1}).first
+    end
+
+    # Count the number of commits reachable from this ref
+    #   +repo+ is the Repo
+    #   +ref+ is the ref from which to begin (SHA1 or name)
+    #
+    # Returns Integer
+    def self.count(repo, ref)
+      repo.git.rev_list({}, ref).size / 41
+    end
+
+    # Find all commits 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 Grit::Commit[] (baked)
+    def self.find_all(repo, ref, options = {})
+      allowed_options = [:max_count, :skip, :since]
+
+      default_options = {:pretty => "raw"}
+      actual_options = default_options.merge(options)
+
+      if ref
+        output = repo.git.rev_list(actual_options, ref)
+      else
+        output = repo.git.rev_list(actual_options.merge(:all => true))
+      end
+
+      self.list_from_string(repo, output)
+    rescue Grit::GitRuby::Repository::NoSuchShaFound
+      []
+    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 Grit::Commit[] (baked)
+    #
+    # really should re-write this to be more accepting of non-standard commit messages
+    # - it broke when 'encoding' was introduced - not sure what else might show up
+    #
+    def self.list_from_string(repo, text)
+      lines = text.split("\n")
+
+      commits = []
+
+      while !lines.empty?
+        id = lines.shift.split.last
+        tree = lines.shift.split.last
+
+        parents = []
+        parents << lines.shift.split.last while lines.first =~ /^parent/
+
+        author, authored_date = self.actor(lines.shift)
+        committer, committed_date = self.actor(lines.shift)
+
+        # not doing anything with this yet, but it's sometimes there
+        encoding = lines.shift.split.last if lines.first =~ /^encoding/
+
+        lines.shift
+
+        message_lines = []
+        message_lines << lines.shift[4..-1] while lines.first =~ /^ {4}/
+
+        lines.shift while lines.first && lines.first.empty?
+
+        commits << Commit.new(repo, id, parents, tree, author, authored_date, committer, committed_date, message_lines)
+      end
+
+      commits
+    end
+
+    # Show diffs between two trees:
+    #   +repo+ is the Repo
+    #   +a+ is a named commit
+    #   +b+ is an optional named commit.  Passing an array assumes you
+    #     wish to omit the second named commit and limit the diff to the
+    #     given paths.
+    #   +paths* is an array of paths to limit the diff.
+    #
+    # Returns Grit::Diff[] (baked)
+    def self.diff(repo, a, b = nil, paths = [])
+      if b.is_a?(Array)
+        paths = b
+        b     = nil
+      end
+      paths.unshift("--") unless paths.empty?
+      paths.unshift(b)    unless b.nil?
+      paths.unshift(a)
+      text = repo.git.diff({:full_index => true}, *paths)
+      Diff.list_from_string(repo, text)
+    end
+
+    def show
+      diff = @repo.git.show({:full_index => true, :pretty => 'raw'}, @id)
+      if diff =~ /diff --git a/
+        diff = diff.sub(/.+?(diff --git a)/m, '\1')
+      else
+        diff = ''
+      end
+      Diff.list_from_string(@repo, diff)
+    end
+
+    def diffs
+      if parents.empty?
+        show
+      else
+        self.class.diff(@repo, parents.first.id, @id)
+      end
+    end
+    
+    def stats
+      stats = @repo.commit_stats(self.sha, 1)[0][-1]
+    end
+    
+    # Convert this Commit to a String which is just the SHA1 id
+    def to_s
+      @id
+    end
+
+    def sha
+      @id
+    end
+
+    def date
+      @committed_date
+    end
+
+    def to_patch
+      @repo.git.format_patch({'1' => true, :stdout => true}, to_s)
+    end
+
+    # Pretty object inspection
+    def inspect
+      %Q{#<Grit::Commit "#{@id}">}
+    end
+
+    # private
+
+    # Parse out the actor (author or committer) info
+    #
+    # Returns [String (actor name and email), Time (acted at time)]
+    def self.actor(line)
+      m, actor, epoch = *line.match(/^.+? (.*) (\d+) .*$/)
+      [Actor.from_string(actor), Time.at(epoch.to_i)]
+    end
+
+    def author_string
+      "%s <%s> %s %+05d" % [author.name, author.email, authored_date.to_i, 800]
+    end
+
+    def to_hash
+      {
+        'id'       => id,
+        'parents'  => parents.map { |p| { 'id' => p.id } },
+        'tree'     => tree.id,
+        'message'  => message,
+        'author'   => {
+          'name'  => author.name,
+          'email' => author.email
+        },
+        'committer' => {
+          'name'  => committer.name,
+          'email' => committer.email
+        },
+        'authored_date'  => authored_date.xmlschema,
+        'committed_date' => committed_date.xmlschema,
+      }
+    end
+  end # Commit
+
+end # Grit

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/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