schleyfox-grit 2.3.0

data/lib/grit/git.rb

@@ -0,0 +1,324 @@
+require 'tempfile'
+module Grit
+
+  class Git
+    class GitTimeout < RuntimeError
+      attr_reader :command, :bytes_read
+
+      def initialize(command = nil, bytes_read = nil)
+        @command = command
+        @bytes_read = bytes_read
+      end
+    end
+
+    undef_method :clone
+
+    include GitRuby
+
+    def exist?
+      File.exist?(self.git_dir)
+    end
+
+    def put_raw_object(content, type)
+      ruby_git.put_raw_object(content, type)
+    end
+
+    def object_exists?(object_id)
+      ruby_git.object_exists?(object_id)
+    end
+
+    def select_existing_objects(object_ids)
+      object_ids.select do |object_id|
+        object_exists?(object_id)
+      end
+    end
+
+    class << self
+      attr_accessor :git_binary, :git_timeout, :git_max_size
+    end
+
+    if RUBY_PLATFORM.downcase =~ /mswin(?!ce)|mingw|bccwin/
+      self.git_binary   = "git" # using search path
+    else
+      self.git_binary   = "/usr/bin/env git"
+    end
+    self.git_timeout  = 10
+    self.git_max_size = 5242880 # 5.megabytes
+
+    def self.with_timeout(timeout = 10.seconds)
+      old_timeout = Grit::Git.git_timeout
+      Grit::Git.git_timeout = timeout
+      yield
+      Grit::Git.git_timeout = old_timeout
+    end
+
+    attr_accessor :git_dir, :bytes_read, :work_tree
+
+    def initialize(git_dir)
+      self.git_dir    = git_dir
+      self.work_tree  = git_dir.gsub(/\/\.git$/,'')
+      self.bytes_read = 0
+    end
+
+    def shell_escape(str)
+      str.to_s.gsub("'", "\\\\'").gsub(";", '\\;')
+    end
+    alias_method :e, :shell_escape
+
+    # Check if a normal file exists on the filesystem
+    #   +file+ is the relative path from the Git dir
+    #
+    # Returns Boolean
+    def fs_exist?(file)
+      File.exist?(File.join(self.git_dir, file))
+    end
+
+    # Read a normal file from the filesystem.
+    #   +file+ is the relative path from the Git dir
+    #
+    # Returns the String contents of the file
+    def fs_read(file)
+      File.read(File.join(self.git_dir, file))
+    end
+
+    # Write a normal file to the filesystem.
+    #   +file+ is the relative path from the Git dir
+    #   +contents+ is the String content to be written
+    #
+    # Returns nothing
+    def fs_write(file, contents)
+      path = File.join(self.git_dir, file)
+      FileUtils.mkdir_p(File.dirname(path))
+      File.open(path, 'w') do |f|
+        f.write(contents)
+      end
+    end
+
+    # Delete a normal file from the filesystem
+    #   +file+ is the relative path from the Git dir
+    #
+    # Returns nothing
+    def fs_delete(file)
+      FileUtils.rm_rf(File.join(self.git_dir, file))
+    end
+
+    # Move a normal file
+    #   +from+ is the relative path to the current file
+    #   +to+ is the relative path to the destination file
+    #
+    # Returns nothing
+    def fs_move(from, to)
+      FileUtils.mv(File.join(self.git_dir, from), File.join(self.git_dir, to))
+    end
+
+    # Make a directory
+    #   +dir+ is the relative path to the directory to create
+    #
+    # Returns nothing
+    def fs_mkdir(dir)
+      FileUtils.mkdir_p(File.join(self.git_dir, dir))
+    end
+
+    # Chmod the the file or dir and everything beneath
+    #   +file+ is the relative path from the Git dir
+    #
+    # Returns nothing
+    def fs_chmod(mode, file = '/')
+      FileUtils.chmod_R(mode, File.join(self.git_dir, file))
+    end
+
+    def list_remotes
+      remotes = []
+      Dir.chdir(File.join(self.git_dir, 'refs/remotes')) do
+        remotes = Dir.glob('*')
+      end
+      remotes
+    rescue
+      []
+    end
+
+    def create_tempfile(seed, unlink = false)
+      path = Tempfile.new(seed).path
+      File.unlink(path) if unlink
+      return path
+    end
+
+    def commit_from_sha(id)
+      git_ruby_repo = GitRuby::Repository.new(self.git_dir)
+      object = git_ruby_repo.get_object_by_sha1(id)
+
+      if object.type == :commit
+        id
+      elsif object.type == :tag
+        object.object
+      else
+        ''
+      end
+    end
+
+    def check_applies(head_sha, applies_sha)
+      git_index = create_tempfile('index', true)
+      (o1, exit1) = raw_git("git read-tree #{head_sha} 2>/dev/null", git_index)
+      (o2, exit2) = raw_git("git diff #{applies_sha}^ #{applies_sha} | git apply --check --cached >/dev/null 2>/dev/null", git_index)
+      return (exit1 + exit2)
+    end
+
+    def get_patch(applies_sha)
+      git_index = create_tempfile('index', true)
+      (patch, exit2) = raw_git("git diff #{applies_sha}^ #{applies_sha}", git_index)
+      patch
+    end
+
+    def apply_patch(head_sha, patch)
+      git_index = create_tempfile('index', true)
+
+      git_patch = create_tempfile('patch')
+      File.open(git_patch, 'w+') { |f| f.print patch }
+
+      raw_git("git read-tree #{head_sha} 2>/dev/null", git_index)
+      (op, exit) = raw_git("git apply --cached < #{git_patch}", git_index)
+      if exit == 0
+        return raw_git("git write-tree", git_index).first.chomp
+      end
+      false
+    end
+
+    # RAW CALLS WITH ENV SETTINGS
+    def raw_git_call(command, index)
+      tmp = ENV['GIT_INDEX_FILE']
+      ENV['GIT_INDEX_FILE'] = index
+      out = `#{command}`
+      after = ENV['GIT_INDEX_FILE'] # someone fucking with us ??
+      ENV['GIT_INDEX_FILE'] = tmp
+      if after != index
+        raise 'environment was changed for the git call'
+      end
+      [out, $?.exitstatus]
+    end
+
+    def raw_git(command, index)
+      output = nil
+      Dir.chdir(self.git_dir) do
+        output = raw_git_call(command, index)
+      end
+      output
+    end
+    # RAW CALLS WITH ENV SETTINGS END
+
+
+
+    # Run the given git command with the specified arguments and return
+    # the result as a String
+    #   +cmd+ is the command
+    #   +options+ is a hash of Ruby style options
+    #   +args+ is the list of arguments (to be joined by spaces)
+    #
+    # Examples
+    #   git.rev_list({:max_count => 10, :header => true}, "master")
+    #
+    # Returns String
+    def method_missing(cmd, options = {}, *args, &block)
+      run('', cmd, '', options, args, &block)
+    end
+
+    # Bypass any pure Ruby implementations and go straight to the native Git command
+    #
+    # Returns String
+    def native(cmd, options = {}, *args, &block)
+      method_missing(cmd, options, *args, &block)
+    end
+
+    def run(prefix, cmd, postfix, options, args, &block)
+      timeout  = options.delete(:timeout) rescue nil
+      timeout  = true if timeout.nil?
+
+      base     = options.delete(:base) rescue nil
+      base     = true if base.nil?
+
+      opt_args = transform_options(options)
+
+      if RUBY_PLATFORM.downcase =~ /mswin(?!ce)|mingw|bccwin/
+        ext_args = args.reject { |a| a.empty? }.map { |a| (a == '--' || a[0].chr == '|' || Grit.no_quote) ? a : "\"#{e(a)}\"" }
+        gitdir = base ? "--git-dir=\"#{self.git_dir}\"" : ""
+        call = "#{prefix}#{Git.git_binary} #{gitdir} #{cmd.to_s.gsub(/_/, '-')} #{(opt_args + ext_args).join(' ')}#{e(postfix)}"
+      else
+        ext_args = args.reject { |a| a.empty? }.map { |a| (a == '--' || a[0].chr == '|' || Grit.no_quote) ? a : "'#{e(a)}'" }
+        gitdir = base ? "--git-dir='#{self.git_dir}'" : ""
+        call = "#{prefix}#{Git.git_binary} #{gitdir} #{cmd.to_s.gsub(/_/, '-')} #{(opt_args + ext_args).join(' ')}#{e(postfix)}"
+      end
+      Grit.log(call) if Grit.debug
+      response, err = timeout ? sh(call, &block) : wild_sh(call, &block)
+      Grit.log(response) if Grit.debug
+      Grit.log(err) if Grit.debug
+      response
+    end
+
+    def sh(command, &block)
+      ret, err = '', ''
+      max = self.class.git_max_size
+      Open3.popen3(command) do |stdin, stdout, stderr|
+        block.call(stdin) if block
+        Timeout.timeout(self.class.git_timeout) do
+          while tmp = stdout.read(8192)
+            ret << tmp
+            raise GitTimeout.new(command, ret.size) if ret.size > max
+          end
+        end
+
+        while tmp = stderr.read(8192)
+          err << tmp
+        end
+      end
+      [ret, err]
+    rescue Timeout::Error, Grit::Git::GitTimeout
+      raise GitTimeout.new(command, ret.size)
+    end
+
+    def wild_sh(command, &block)
+      ret, err = '', ''
+      Open3.popen3(command) do |stdin, stdout, stderr|
+        block.call(stdin) if block
+        while tmp = stdout.read(8192)
+          ret << tmp
+        end
+
+        while tmp = stderr.read(8192)
+          err << tmp
+        end
+      end
+      [ret, err]
+    end
+
+    # Transform Ruby style options into git command line options
+    #   +options+ is a hash of Ruby style options
+    #
+    # Returns String[]
+    #   e.g. ["--max-count=10", "--header"]
+    def transform_options(options)
+      args = []
+      options.keys.each do |opt|
+        if opt.to_s.size == 1
+          if options[opt] == true
+            args << "-#{opt}"
+          elsif options[opt] == false
+            # ignore
+          else
+            val = options.delete(opt)
+            args << "-#{opt.to_s} '#{e(val)}'"
+          end
+        else
+          if options[opt] == true
+            args << "--#{opt.to_s.gsub(/_/, '-')}"
+          elsif options[opt] == false
+            # ignore
+          else
+            val = options.delete(opt)
+            args << "--#{opt.to_s.gsub(/_/, '-')}='#{e(val)}'"
+          end
+        end
+      end
+      args
+    end
+  end # Git
+
+end # Grit

data/lib/grit/index.rb

@@ -0,0 +1,197 @@
+module Grit
+
+  class Index
+    # Public: Gets/Sets the Grit::Repo to which this index belongs.
+    attr_accessor :repo
+
+    # Public: Gets/Sets the Hash tree map that holds the changes to be made
+    # in the next commit.
+    attr_accessor :tree
+
+    # Public: Gets/Sets the Grit::Tree object representing the tree upon
+    # which the next commit will be based.
+    attr_accessor :current_tree
+
+    # Initialize a new Index object.
+    #
+    # repo - The Grit::Repo to which the index belongs.
+    #
+    # Returns the newly initialized Grit::Index.
+    def initialize(repo)
+      self.repo = repo
+      self.tree = {}
+      self.current_tree = nil
+    end
+
+    # Public: Add a file to the index.
+    #
+    # path - The String file path including filename (no slash prefix).
+    # data - The String binary contents of the file.
+    #
+    # Returns nothing.
+    def add(path, data)
+      path = path.split('/')
+      filename = path.pop
+
+      current = self.tree
+
+      path.each do |dir|
+        current[dir] ||= {}
+        node = current[dir]
+        current = node
+      end
+
+      current[filename] = data
+    end
+
+    # Public: Delete the given file from the index.
+    #
+    # path - The String file path including filename (no slash prefix).
+    #
+    # Returns nothing.
+    def delete(path)
+      add(path, false)
+    end
+
+    # Public: Read the contents of the given Tree into the index to use as a
+    # starting point for the index.
+    #
+    # tree - The String branch/tag/sha of the Git tree object.
+    #
+    # Returns nothing.
+    def read_tree(tree)
+      self.current_tree = self.repo.tree(tree)
+    end
+
+    # Public: Commit the contents of the index.  This method supports two
+    # formats for arguments:
+    #
+    # message - The String commit message.
+    # options - An optional Hash of index options.
+    #           :parents        - Array of String commit SHA1s or Grit::Commit
+    #                             objects to attach this commit to to form a 
+    #                             new head (default: nil).
+    #           :actor          - The Grit::Actor details of the user making 
+    #                             the commit (default: nil).
+    #           :last_tree      - The String SHA1 of a tree to compare with
+    #                             in order to avoid making empty commits 
+    #                             (default: nil).
+    #           :head           - The String branch name to write this head to
+    #                             (default: "master").
+    #           :committed_date - The Time that the commit was made.  
+    #                             (Default: Time.now)
+    #           :authored_date  - The Time that the commit was authored.  
+    #                             (Default: committed_date)
+    #
+    # The legacy argument style looks like:
+    #
+    # message   - The String commit message.
+    # parents   - Array of String commit SHA1s or Grit::Commit objects to
+    #             attach this commit to to form a new head (default: nil).
+    # actor     - The Grit::Actor details of the user making the commit
+    #             (default: nil).
+    # last_tree - The String SHA1 of a tree to compare with in order to avoid
+    #             making empty commits (default: nil).
+    # head      - The String branch name to write this head to
+    #             (default: "master").
+    #
+    # Returns a String of the SHA1 of the new commit.
+    def commit(message, parents = nil, actor = nil, last_tree = nil, head = 'master')
+      if parents.is_a?(Hash)
+        actor          = parents[:actor]
+        committer      = parents[:committer]
+        author         = parents[:author]
+        last_tree      = parents[:last_tree]
+        head           = parents[:head]
+        committed_date = parents[:committed_date]
+        authored_date  = parents[:authored_date]
+        parents        = parents[:parents]
+      end
+
+      committer ||= actor
+      author    ||= committer
+
+      tree_sha1 = write_tree(self.tree, self.current_tree)
+
+      # don't write identical commits
+      return false if tree_sha1 == last_tree
+
+      contents = []
+      contents << ['tree', tree_sha1].join(' ')
+      parents.each do |p|
+        contents << ['parent', p].join(' ')
+      end if parents
+
+      committer      ||= begin
+        config = Config.new(self.repo)
+        Actor.new(config['user.name'], config['user.email'])
+      end
+      author         ||= committer
+      committed_date ||= Time.now
+      authored_date  ||= committed_date
+
+      contents << ['author',    author.output(authored_date)].join(' ')
+      contents << ['committer', committer.output(committed_date)].join(' ')
+      contents << ''
+      contents << message
+
+      commit_sha1 = self.repo.git.put_raw_object(contents.join("\n"), 'commit')
+
+      self.repo.update_ref(head, commit_sha1)
+    end
+
+    # Recursively write a tree to the index.
+    #
+    # tree -     The Hash tree map:
+    #            key - The String directory or filename.
+    #            val - The Hash submap or the String contents of the file.
+    # now_tree - The Grit::Tree representing the a previous tree upon which
+    #            this tree will be based (default: nil).
+    #
+    # Returns the String SHA1 String of the tree.
+    def write_tree(tree, now_tree = nil)
+      tree_contents = {}
+
+      # fill in original tree
+      now_tree.contents.each do |obj|
+        sha = [obj.id].pack("H*")
+        k = obj.name
+        k += '/' if (obj.class == Grit::Tree)
+        tmode = obj.mode.to_i.to_s  ## remove zero-padding
+        tree_contents[k] = "%s %s\0%s" % [tmode, obj.name, sha]
+      end if now_tree
+
+      # overwrite with new tree contents
+      tree.each do |k, v|
+        case v
+          when String
+            sha = write_blob(v)
+            sha = [sha].pack("H*")
+            str = "%s %s\0%s" % ['100644', k, sha]
+            tree_contents[k] = str
+          when Hash
+            ctree = now_tree/k if now_tree
+            sha = write_tree(v, ctree)
+            sha = [sha].pack("H*")
+            str = "%s %s\0%s" % ['40000', k, sha]
+            tree_contents[k + '/'] = str
+          when false
+            tree_contents.delete(k)
+        end
+      end
+
+      tr = tree_contents.sort.map { |k, v| v }.join('')
+      self.repo.git.put_raw_object(tr, 'tree')
+    end
+
+    # Write a blob to the index.
+    #
+    # data - The String data to write.
+    #
+    # Returns the String SHA1 of the new blob.
+    def write_blob(data)
+      self.repo.git.put_raw_object(data, 'blob')
+    end
+  end # Index
+
+end # Grit