git-duet 0.1.1

data/lib/git/duet/cli.rb

@@ -0,0 +1,119 @@
+require 'optparse'
+require 'git/duet'
+require 'git/duet/script_die_error'
+
+class Git::Duet::Cli
+  class << self
+    def run(prog, argv)
+      case prog
+      when /solo$/
+        solo(parse_solo_options(argv.clone))
+        return 0
+      when /duet$/
+        duet(parse_duet_options(argv.clone))
+        return 0
+      when /pre-commit$/
+        pre_commit(parse_generic_options(argv.clone))
+        return 0
+      when /install-hook$/
+        install_hook(parse_generic_options(argv.clone))
+        return 0
+      when /commit$/
+        commit(parse_commit_options(argv.clone))
+        return 0
+      else
+        raise ScriptError.new('How did you get here???')
+      end
+    rescue Git::Duet::ScriptDieError => e
+      return e.exit_code
+    end
+
+    private
+    def with_common_opts(argv, banner)
+      options = {}
+      leftover_argv = OptionParser.new do |opts|
+        opts.banner = banner.gsub(/__PROG__/, opts.program_name)
+        opts.on('-q', 'Silence output') do |q|
+          options[:quiet] = true
+        end
+        if block_given?
+          yield opts, options
+        end
+      end.parse!(argv)
+      return leftover_argv, options
+    end
+
+    def parse_solo_options(argv)
+      leftover_argv, options = with_common_opts(
+        argv, 'Usage: __PROG__ [options] <soloist-initials>'
+      ) do |opts,options_hash|
+        opts.on('-g', '--global', 'Change global git config') do |g|
+          options_hash[:global] = true
+        end
+      end
+      options[:soloist] = leftover_argv.first
+      options
+    end
+
+    def parse_duet_options(argv)
+      leftover_argv, options = with_common_opts(
+        argv, 'Usage: __PROG__ [options] <alpha-initials> <omega-initials>'
+      ) do |opts,options_hash|
+        opts.on('-g', '--global', 'Change global git config') do |g|
+          options_hash[:global] = true
+        end
+      end
+      options[:alpha], options[:omega] = leftover_argv[0..1]
+      options
+    end
+
+    def parse_generic_options(argv)
+      with_common_opts(argv, 'Usage: __PROG__').last
+    end
+
+    def parse_commit_options(argv)
+      opts_argv = []
+      opts_argv << '-q' if argv.delete('-q')
+      options = with_common_opts(opts_argv, 'Usage: __PROG__').last
+      options[:passthrough_args] = argv
+      options
+    end
+
+    def solo(options)
+      require 'git/duet/solo_command'
+      Git::Duet::SoloCommand.new(
+        options.fetch(:soloist),
+        options[:quiet],
+        options[:global]
+      ).execute!
+    end
+
+    def duet(options)
+      require 'git/duet/duet_command'
+      Git::Duet::DuetCommand.new(
+        options.fetch(:alpha),
+        options.fetch(:omega),
+        options[:quiet],
+        options[:global]
+      ).execute!
+    end
+
+    def pre_commit(options)
+      require 'git/duet/pre_commit_command'
+      Git::Duet::PreCommitCommand.new(options[:quiet]).execute!
+    end
+
+    def install_hook(options)
+      require 'git/duet/install_hook_command'
+      Git::Duet::InstallHookCommand.new(options[:quiet]).execute!
+    end
+
+    def commit(options)
+      require 'git/duet/commit_command'
+      Git::Duet::CommitCommand.new(
+        options[:passthrough_args],
+        options[:quiet]
+      ).execute!
+    end
+  end
+end

data/lib/git/duet/command_methods.rb

@@ -0,0 +1,109 @@
+require 'git/duet'
+require 'git/duet/script_die_error'
+
+module Git::Duet::CommandMethods
+  private
+  def report_env_vars
+    var_map.each do |key,value|
+      info("#{key}='#{value}'")
+    end
+  end
+
+  def write_env_vars
+    in_repo_root do
+      var_map.each do |key,value|
+        exec_check("git config #{@global ? '--global ' : ''}duet.env.#{key.downcase.gsub(/_/, '-')} '#{value}'")
+      end
+      exec_check("git config #{@global ? '--global ' : ''}duet.env.mtime #{Time.now.to_i}")
+    end
+  end
+
+  def dump_env_vars
+    extract_env_vars_from_git_config.each do |k,v|
+      puts "#{k}='#{v}'"
+    end
+  end
+
+  def extract_env_vars_from_git_config
+    dest = {}
+    env_vars.each do |env_var,config_key|
+      begin
+        value = exec_check("git config duet.env.#{config_key}").chomp
+        dest[env_var] = value if !value.empty?
+      rescue StandardError => e
+        error("#{e.message}")
+      end
+    end
+    dest
+  end
+
+  def exec_git_commit
+    exec 'git commit --signoff ' << quoted_passthrough_args
+  end
+
+  def env_vars
+    @env_vars ||= Hash[env_var_pairs]
+  end
+
+  def env_var_pairs
+    %w(
+      GIT_AUTHOR_NAME
+      GIT_AUTHOR_EMAIL
+      GIT_COMMITTER_NAME
+      GIT_COMMITTER_EMAIL
+    ).map do |env_var|
+      [env_var, env_var.downcase.gsub(/_/, '-')]
+    end
+  end
+
+  def in_repo_root
+    Dir.chdir(exec_check('git rev-parse --show-toplevel').chomp) do
+      yield
+    end
+  end
+
+  def exec_check(command, okay_statuses = [0].freeze)
+    output = `#{command}`
+    if !okay_statuses.include?($?.exitstatus)
+      error("Command #{command.inspect} exited with #{$?.to_i}")
+      raise Git::Duet::ScriptDieError.new(1)
+    end
+    output
+  end
+
+  def with_output_unquieted(&block)
+    @old_quiet = @quiet
+    @quiet = false
+    block.call
+    @quiet = @old_quiet
+  rescue StandardError => e
+    @quiet = @old_quiet
+    raise e
+  end
+
+  def with_output_quieted(&block)
+    @old_quiet = @quiet
+    @quiet = true
+    block.call
+  rescue StandardError => e
+    raise e
+  ensure
+    @quiet = @old_quiet
+  end
+
+  def info(msg)
+    STDOUT.puts(msg) unless quiet?
+  end
+
+  def error(msg)
+    STDERR.puts(msg) unless quiet?
+  end
+
+  def prompt
+    STDOUT.print '> '
+  end
+
+  def quiet?
+    ENV['GIT_DUET_QUIET'] == '1' || @quiet
+  end
+end

data/lib/git/duet/commit_command.rb

@@ -0,0 +1,76 @@
+require 'git/duet'
+require 'git/duet/command_methods'
+
+class Git::Duet::CommitCommand
+  include Git::Duet::CommandMethods
+
+  def initialize(passthrough_argv, quiet = false)
+    @passthrough_argv = passthrough_argv
+    @quiet = quiet
+  end
+
+  def execute!
+    in_repo_root do
+      add_env_vars_to_env
+      exec_git_commit
+    end
+  end
+
+  private
+  def add_env_vars_to_env
+    extract_env_vars_from_git_config.each do |k,v|
+      ENV[k] = v
+    end
+  end
+
+  def exec_git_commit
+    exec 'git commit ' << signoff_arg << quoted_passthrough_args
+  end
+
+  def env_vars
+    @env_vars ||= Hash[env_var_pairs]
+  end
+
+  def env_var_pairs
+    env_var_names.map do |env_var|
+      [env_var, env_var.downcase.gsub(/_/, '-')]
+    end
+  end
+
+  def quoted_passthrough_args
+    @passthrough_argv.map do |arg|
+      "'#{arg}'"
+    end.join(' ')
+  end
+
+  def signoff_arg
+    soloing? ? '' : '--signoff '
+  end
+
+  def env_var_names
+    if soloing?
+      %w(
+        GIT_AUTHOR_NAME
+        GIT_AUTHOR_EMAIL
+      )
+    else
+      %w(
+        GIT_AUTHOR_NAME
+        GIT_AUTHOR_EMAIL
+        GIT_COMMITTER_NAME
+        GIT_COMMITTER_EMAIL
+      )
+    end
+  end
+
+  def soloing?
+    @soloing ||= begin
+      with_output_quieted do
+        exec_check('git config duet.env.git-committer-name').chomp
+      end
+      false
+    rescue StandardError
+      true
+    end
+  end
+end

data/lib/git/duet/duet_command.rb

@@ -0,0 +1,49 @@
+require 'git/duet'
+require 'git/duet/author_mapper'
+require 'git/duet/command_methods'
+
+class Git::Duet::DuetCommand
+  include Git::Duet::CommandMethods
+
+  def initialize(alpha, omega, quiet = false, global = false)
+    @alpha, @omega = alpha, omega
+    @quiet = !!quiet
+    @global = !!global
+    @author_mapper = Git::Duet::AuthorMapper.new
+  end
+
+  def execute!
+    set_alpha_as_git_config_user
+    report_env_vars
+    write_env_vars
+  end
+
+  private
+  attr_accessor :alpha, :omega, :author_mapper
+
+  def set_alpha_as_git_config_user
+    exec_check("git config #{@global ? '--global ' : ''}user.name '#{alpha_info[:name]}'")
+    exec_check("git config #{@global ? '--global ' : ''}user.email '#{alpha_info[:email]}'")
+  end
+
+  def var_map
+    {
+      'GIT_AUTHOR_NAME' => alpha_info[:name],
+      'GIT_AUTHOR_EMAIL' => alpha_info[:email],
+      'GIT_COMMITTER_NAME' => omega_info[:name],
+      'GIT_COMMITTER_EMAIL' => omega_info[:email]
+    }
+  end
+
+  def alpha_info
+    alpha_omega_info[@alpha]
+  end
+
+  def omega_info
+    alpha_omega_info[@omega]
+  end
+
+  def alpha_omega_info
+    @alpha_omega_info ||= author_mapper.map(@alpha, @omega)
+  end
+end

data/lib/git/duet/install_hook_command.rb

@@ -0,0 +1,33 @@
+require 'git/duet'
+require 'fileutils'
+require 'git/duet/command_methods'
+
+class Git::Duet::InstallHookCommand
+  include Git::Duet::CommandMethods
+
+  HOOK = <<-EOF.gsub(/^  /, '')
+  #!/bin/bash
+  exec git duet-pre-commit "$@"
+  EOF
+
+  def initialize(quiet = false)
+    @quiet = quiet
+  end
+
+  def execute!
+    Dir.chdir(`git rev-parse --show-toplevel`.chomp) do
+      dest = File.join(Dir.pwd, '.git', 'hooks', 'pre-commit')
+      if File.exist?(dest)
+        error("git-duet-install-hook: A pre-commit hook already exists at #{dest}!")
+        error("git-duet-install-hook: Move it out of the way first, mkay?")
+        return 1
+      end
+      File.open(dest, 'w') do |f|
+        f.puts HOOK
+      end
+      FileUtils.chmod(0755, dest)
+      info("git-duet-install-hook: Installed hook to #{dest}")
+    end
+    return 0
+  end
+end

data/lib/git/duet/key_error.rb

@@ -0,0 +1,3 @@
+unless defined?(KeyError)
+  KeyError = Class.new(IndexError)
+end

data/lib/git/duet/pre_commit_command.rb

@@ -0,0 +1,43 @@
+require 'git/duet'
+require 'git/duet/command_methods'
+require 'git/duet/script_die_error'
+
+class Git::Duet::PreCommitCommand
+  include Git::Duet::CommandMethods
+
+  def initialize(quiet = false)
+    @quiet = !!quiet
+  end
+
+  def execute!
+    in_repo_root do
+      if !env_cache_exists? || env_cache_stale?
+        explode!
+      end
+    end
+  end
+
+  private
+  def explode!
+    error("Your git duet settings are stale, human!")
+    error("Update them with `git duet` or `git solo`.")
+    raise Git::Duet::ScriptDieError.new(1)
+  end
+
+  def env_cache_exists?
+    with_output_quieted do
+      exec_check('git config duet.env.mtime')
+    end
+    true
+  rescue
+    false
+  end
+
+  def env_cache_stale?
+    Integer(exec_check('git config duet.env.mtime')) < stale_cutoff
+  end
+
+  def stale_cutoff
+    Integer(Time.now - Integer(ENV.fetch('GIT_DUET_SECONDS_AGO_STALE', '1200')))
+  end
+end

data/lib/git/duet/script_die_error.rb

@@ -0,0 +1,11 @@
+module Git
+  module Duet
+    class ScriptDieError < StandardError
+      attr_reader :exit_code
+
+      def initialize(exit_code)
+        @exit_code = exit_code
+      end
+    end
+  end
+end