jit 0.0.0 → 1.0.0

data/bin/jit

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+
+require_relative "../lib/command"
+
+begin
+  cmd = Command.execute(Dir.getwd, ENV, ARGV, $stdin, $stdout, $stderr)
+  exit cmd.status
+
+rescue Command::Unknown => error
+  $stderr.puts "jit: #{ error.message }"
+  exit 1
+
+rescue => error
+  $stderr.puts "fatal: #{ error.message }"
+  if ENV["DEBUG"]
+    error.backtrace.each do |line|
+      $stderr.puts "        from #{ line }"
+    end
+  end
+  exit 1
+end

data/lib/color.rb

@@ -0,0 +1,32 @@
+module Color
+  SGR_CODES = {
+    "normal"  =>  0,
+    "bold"    =>  1,
+    "dim"     =>  2,
+    "italic"  =>  3,
+    "ul"      =>  4,
+    "reverse" =>  7,
+    "strike"  =>  9,
+    "black"   => 30,
+    "red"     => 31,
+    "green"   => 32,
+    "yellow"  => 33,
+    "blue"    => 34,
+    "magenta" => 35,
+    "cyan"    => 36,
+    "white"   => 37
+  }
+
+  def self.format(style, string)
+    codes = [*style].map { |name| SGR_CODES.fetch(name.to_s) }
+    color = false
+
+    codes.each_with_index do |code, i|
+      next unless code >= 30
+      codes[i] += 10 if color
+      color = true
+    end
+
+    "\e[#{ codes.join(";") }m#{ string }\e[m"
+  end
+end

data/lib/command.rb

@@ -0,0 +1,62 @@
+require_relative "./command/add"
+require_relative "./command/branch"
+require_relative "./command/checkout"
+require_relative "./command/cherry_pick"
+require_relative "./command/commit"
+require_relative "./command/config"
+require_relative "./command/diff"
+require_relative "./command/fetch"
+require_relative "./command/init"
+require_relative "./command/log"
+require_relative "./command/merge"
+require_relative "./command/push"
+require_relative "./command/remote"
+require_relative "./command/receive_pack"
+require_relative "./command/reset"
+require_relative "./command/rev_list"
+require_relative "./command/revert"
+require_relative "./command/rm"
+require_relative "./command/status"
+require_relative "./command/upload_pack"
+
+module Command
+  Unknown = Class.new(StandardError)
+
+  COMMANDS = {
+    "init"         => Init,
+    "config"       => Config,
+    "add"          => Add,
+    "rm"           => Rm,
+    "commit"       => Commit,
+    "status"       => Status,
+    "diff"         => Diff,
+    "branch"       => Branch,
+    "checkout"     => Checkout,
+    "reset"        => Reset,
+    "rev-list"     => RevList,
+    "log"          => Log,
+    "merge"        => Merge,
+    "cherry-pick"  => CherryPick,
+    "revert"       => Revert,
+    "remote"       => Remote,
+    "fetch"        => Fetch,
+    "push"         => Push,
+    "upload-pack"  => UploadPack,
+    "receive-pack" => ReceivePack
+  }
+
+  def self.execute(dir, env, argv, stdin, stdout, stderr)
+    name = argv.first
+    args = argv.drop(1)
+
+    unless COMMANDS.has_key?(name)
+      raise Unknown, "'#{ name }' is not a jit command."
+    end
+
+    command_class = COMMANDS[name]
+    command = command_class.new(dir, env, args, stdin, stdout, stderr)
+
+    command.execute
+    command
+  end
+end

data/lib/command/add.rb

@@ -0,0 +1,65 @@
+require "pathname"
+require_relative "./base"
+
+module Command
+  class Add < Base
+
+    LOCKED_INDEX_MESSAGE = <<~MSG
+      Another jit process seems to be running in this repository.
+      Please make sure all processes are terminated then try again.
+      If it still fails, a jit process may have crashed in this
+      repository earlier: remove the file manually to continue.
+    MSG
+
+    def run
+      repo.index.load_for_update
+      expanded_paths.each { |path| add_to_index(path) }
+      repo.index.write_updates
+      exit 0
+    rescue Lockfile::LockDenied => error
+      handle_locked_index(error)
+    rescue Workspace::MissingFile => error
+      handle_missing_file(error)
+    rescue Workspace::NoPermission => error
+      handle_unreadable_file(error)
+    end
+
+    private
+
+    def expanded_paths
+      @args.flat_map do |path|
+        repo.workspace.list_files(expanded_pathname(path))
+      end
+    end
+
+    def add_to_index(path)
+      data = repo.workspace.read_file(path)
+      stat = repo.workspace.stat_file(path)
+
+      blob = Database::Blob.new(data)
+      repo.database.store(blob)
+      repo.index.add(path, blob.oid, stat)
+    end
+
+    def handle_locked_index(error)
+      @stderr.puts "fatal: #{ error.message }"
+      @stderr.puts
+      @stderr.puts LOCKED_INDEX_MESSAGE
+      exit 128
+    end
+
+    def handle_missing_file(error)
+      @stderr.puts "fatal: #{ error.message }"
+      repo.index.release_lock
+      exit 128
+    end
+
+    def handle_unreadable_file(error)
+      @stderr.puts "error: #{ error.message }"
+      @stderr.puts "fatal: adding files failed"
+      repo.index.release_lock
+      exit 128
+    end
+
+  end
+end

data/lib/command/base.rb

@@ -0,0 +1,92 @@
+require "optparse"
+require "pathname"
+
+require_relative "../color"
+require_relative "../editor"
+require_relative "../pager"
+require_relative "../repository"
+
+module Command
+  class Base
+
+    attr_reader :status
+
+    def initialize(dir, env, args, stdin, stdout, stderr)
+      @dir    = dir
+      @env    = env
+      @args   = args
+      @stdin  = stdin
+      @stdout = stdout
+      @stderr = stderr
+
+      @isatty = @stdout.isatty
+    end
+
+    def execute
+      parse_options
+      catch(:exit) { run }
+
+      if defined? @pager
+        @stdout.close_write
+        @pager.wait
+      end
+    end
+
+    private
+
+    def repo
+      @repo ||= Repository.new(Pathname.new(@dir).join(".git"))
+    end
+
+    def expanded_pathname(path)
+      Pathname.new(File.expand_path(path, @dir))
+    end
+
+    def parse_options
+      @options = {}
+      @parser  = OptionParser.new
+
+      define_options
+      @parser.parse!(@args)
+    end
+
+    def define_options
+    end
+
+    def setup_pager
+      return if defined? @pager
+      return unless @isatty
+
+      @pager  = Pager.new(@env, @stdout, @stderr)
+      @stdout = @pager.input
+    end
+
+    def edit_file(path)
+      Editor.edit(path, editor_command) do |editor|
+        yield editor
+        editor.close unless @isatty
+      end
+    end
+
+    def editor_command
+      core_editor = repo.config.get(["core", "editor"])
+      @env["GIT_EDITOR"] || core_editor || @env["VISUAL"] || @env["EDITOR"]
+    end
+
+    def fmt(style, string)
+      @isatty ? Color.format(style, string) : string
+    end
+
+    def puts(string)
+      @stdout.puts(string)
+    rescue Errno::EPIPE
+      exit 0
+    end
+
+    def exit(status = 0)
+      @status = status
+      throw :exit
+    end
+
+  end
+end

data/lib/command/branch.rb

@@ -0,0 +1,199 @@
+require_relative "./base"
+require_relative "./shared/fast_forward"
+require_relative "../revision"
+
+module Command
+  class Branch < Base
+
+    include FastForward
+
+    def define_options
+      @parser.on("-a", "--all")     { @options[:all]     = true }
+      @parser.on("-r", "--remotes") { @options[:remotes] = true }
+
+      @options[:verbose] = 0
+      @parser.on("-v", "--verbose") { @options[:verbose] += 1 }
+
+      @parser.on("-d", "--delete") { @options[:delete] = true }
+      @parser.on("-f", "--force")  { @options[:force]  = true }
+
+      @parser.on "-D" do
+        @options[:delete] = @options[:force] = true
+      end
+
+      @parser.on "-u <upstream>", "--set-upstream-to=<upstream>" do |upstream|
+        @options[:upstream] = upstream
+      end
+
+      @parser.on("-t", "--track")    { @options[:track]    = true   }
+      @parser.on("--unset-upstream") { @options[:upstream] = :unset }
+    end
+
+    def run
+      if @options[:upstream]
+        set_upstream_branch
+      elsif @options[:delete]
+        delete_branches
+      elsif @args.empty?
+        list_branches
+      else
+        create_branch
+      end
+
+      exit 0
+    end
+
+    private
+
+    def list_branches
+      current   = repo.refs.current_ref
+      branches  = branch_refs.sort_by(&:path)
+      max_width = branches.map { |b| b.short_name.size }.max
+
+      setup_pager
+
+      branches.each do |ref|
+        info = format_ref(ref, current)
+        info.concat(extended_branch_info(ref, max_width))
+        puts info
+      end
+    end
+
+    def branch_refs
+      branches = repo.refs.list_branches
+      remotes  = repo.refs.list_remotes
+
+      return branches + remotes if @options[:all]
+      return remotes if @options[:remotes]
+
+      branches
+    end
+
+    def format_ref(ref, current)
+      if ref == current
+        "* #{ fmt :green, ref.short_name }"
+      elsif ref.remote?
+        "  #{ fmt :red, ref.short_name }"
+      else
+        "  #{ ref.short_name }"
+      end
+    end
+
+    def extended_branch_info(ref, max_width)
+      return "" unless @options[:verbose] > 0
+
+      commit   = repo.database.load(ref.read_oid)
+      short    = repo.database.short_oid(commit.oid)
+      space    = " " * (max_width - ref.short_name.size)
+      upstream = upstream_info(ref)
+
+      "#{ space } #{ short }#{ upstream } #{ commit.title_line }"
+    end
+
+    def upstream_info(ref)
+      divergence = repo.divergence(ref)
+      return "" unless divergence.upstream
+
+      ahead  = divergence.ahead
+      behind = divergence.behind
+
+      return "" if ahead == 0 and behind == 0
+
+      info = []
+
+      if @options[:verbose] > 1
+        info.push(fmt(:blue, repo.refs.short_name(divergence.upstream)))
+      end
+      info.push("ahead #{ ahead }") if ahead > 0
+      info.push("behind #{ behind }") if behind > 0
+
+      " [#{ info.join(", ") }]"
+    end
+
+    def create_branch
+      branch_name = @args[0]
+      start_point = @args[1]
+
+      if start_point
+        revision  = Revision.new(repo, start_point)
+        start_oid = revision.resolve(Revision::COMMIT)
+      else
+        start_oid = repo.refs.read_head
+      end
+
+      repo.refs.create_branch(branch_name, start_oid)
+      set_upstream(branch_name, start_point) if @options[:track]
+
+    rescue Refs::InvalidBranch => error
+      @stderr.puts "fatal: #{ error.message }"
+      exit 128
+
+    rescue Revision::InvalidObject => error
+      revision.errors.each do |err|
+        @stderr.puts "error: #{ err.message }"
+        err.hint.each { |line| @stderr.puts "hint: #{ line }" }
+      end
+      @stderr.puts "fatal: #{ error.message }"
+      exit 128
+    end
+
+    def delete_branches
+      @args.each { |branch_name| delete_branch(branch_name) }
+    end
+
+    def delete_branch(branch_name)
+      check_merge_status(branch_name) unless @options[:force]
+
+      oid   = repo.refs.delete_branch(branch_name)
+      short = repo.database.short_oid(oid)
+
+      repo.remotes.unset_upstream(branch_name)
+
+      puts "Deleted branch #{ branch_name } (was #{ short })."
+
+    rescue Refs::InvalidBranch => error
+      @stderr.puts "error: #{ error }"
+      exit 1
+    end
+
+    def check_merge_status(branch_name)
+      upstream   = repo.remotes.get_upstream(branch_name)
+      head_oid   = upstream ? repo.refs.read_ref(upstream) : repo.refs.read_head
+      branch_oid = repo.refs.read_ref(branch_name)
+
+      if fast_forward_error(branch_oid, head_oid)
+        @stderr.puts "error: The branch '#{ branch_name }' is not fully merged."
+        exit 1
+      end
+    end
+
+    def set_upstream_branch
+      branch_name = @args.first || repo.refs.current_ref.short_name
+
+      if @options[:upstream] == :unset
+        repo.remotes.unset_upstream(branch_name)
+      else
+        set_upstream(branch_name, @options[:upstream])
+      end
+    end
+
+    def set_upstream(branch_name, upstream)
+      upstream    = repo.refs.long_name(upstream)
+      remote, ref = repo.remotes.set_upstream(branch_name, upstream)
+
+      base = repo.refs.short_name(ref)
+
+      puts "Branch '#{ branch_name }' set up to track remote " +
+           "branch '#{ base }' from '#{ remote }'."
+
+    rescue Refs::InvalidBranch => error
+      @stderr.puts "error: #{ error.message }"
+      exit 1
+
+    rescue Remotes::InvalidBranch => error
+      @stderr.puts "fatal: #{ error.message }"
+      exit 128
+    end
+
+  end
+end