# -*- coding: utf-8; mode: ruby; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- vim:fenc=utf-8:filetype=ruby:et:sw=2:ts=2:sts=2

require 'set'

# Git methods
class GitCommitNotifier::Git
  class << self
    # Runs specified command.
    # @return (String) Shell command STDOUT (forced to UTF-8)
    # @raise [ArgumentError] when command exits with nonzero status.
    def from_shell(cmd)
      r = `#{cmd}`
      raise ArgumentError.new("#{cmd} failed") unless $?.exitstatus.zero?
      r.force_encoding(Encoding::UTF_8) if r.respond_to?(:force_encoding)
      r
    end

    # runs `git show`
    # @note uses "--pretty=fuller" option.
    # @return [String] Its output
    # @see from_shell
    # @param [String] rev Revision
    # @param [Hash] opts Options
    # @option opts [Boolean] :ignore_whitespaces Ignore whitespaces or not
    def show(rev, opts = {})
      gitopt = ""
      gitopt += " --pretty=fuller"
      gitopt += " -w" if opts[:ignore_whitespaces]
      data = from_shell("git show #{rev.strip}#{gitopt}")
      data
    end

    # runs `git log`
    # @note uses "--pretty=fuller" option.
    # @return [String] Its output
    # @see from_shell
    # @param [String] rev1 First revision
    # @param [String] rev2 Second revision
    def log(rev1, rev2)
      data = from_shell("git log --pretty=fuller #{rev1}..#{rev2}").strip
      data
    end

    # runs `git log` and extract filenames only
    # @note uses "--pretty=fuller" and "--name-status" options.
    # @return [Array(String)] File names
    # @see from_shell
    # @param [String] rev1 First revision
    # @param [String] rev2 Second revision
    def changed_files(rev1, rev2)
      output = ""
      lines = from_shell("git log #{rev1}..#{rev2} --name-status --pretty=oneline")
      lines = lines.lines if lines.respond_to?(:lines)
      lines = lines.select {|line| line =~ /^\w{1}\s+\w+/} # grep out only filenames
      lines.uniq
    end

    def branch_commits(treeish)
      args = branch_heads - [ branch_head(treeish) ]
      args.map! { |tree| "^#{tree}" }
      args << treeish
      lines = from_shell("git rev-list #{args.join(' ')}")
      lines = lines.lines if lines.respond_to?(:lines)
      lines.to_a.map { |commit| commit.chomp }
    end

    def branch_heads
      lines = from_shell("git rev-parse --branches")
      lines = lines.lines if lines.respond_to?(:lines)
      lines.to_a.map { |head| head.chomp }
    end

    def git_dir()
      from_shell("git rev-parse --git-dir").strip
    end

    def rev_parse(param)
      from_shell("git rev-parse '#{param}'").strip
    end

    def branch_head(treeish)
      from_shell("git rev-parse #{treeish}").strip
    end
    
    def new_commits(oldrev, newrev, refname, unique_to_current_branch)
      # We want to get the set of commits (^B1 ^B2 ... ^oldrev newrev)
      # Where B1, B2, ..., are any other branch

      s = Set.new
      
      # If we want to include only those commits that are
      # unique to this branch, then exclude commits that occur on
      # other branches
      if unique_to_current_branch
        # Make a set of all branches, not'd (^BCURRENT ^B1 ^B2...)
        not_branches = from_shell("git rev-parse --not --branches")
        s.merge(not_branches.lines.map {|l| l.chomp}.to_set)
      
        # Remove the current branch (^BCURRENT) from the set
        current_branch = rev_parse(refname)
        s.delete("^#{current_branch}")
      end
      
      # Add not'd oldrev (^oldrev)
      s.add("^#{oldrev}") unless oldrev =~ /^0+$/

      # Add newrev
      s.add(newrev)
      
      # We should now have ^B1... ^oldrev newrev

      # Get all the commits that match that specification
      lines = from_shell("git rev-list #{s.to_a.join(' ')}")
      commits = lines.lines.map {|l| l.chomp}
    end

    def rev_type(rev)
      from_shell("git cat-file -t '#{rev}' 2> /dev/null").strip
    rescue ArgumentError
      nil
    end
    
    def tag_info(refname)
      fields = [
        ':tagobject => %(*objectname)',
        ':tagtype => %(*objecttype)',
        ':taggername => %(taggername)',
        ':taggeremail => %(taggeremail)',
        ':subject => %(subject)',
        ':contents => %(contents)'
      ]
      joined_fields = fields.join(",")
      hash_script = from_shell("git for-each-ref --shell --format='{ #{joined_fields} }' #{refname}")
      eval(hash_script)
    end

    def repo_name
      git_prefix = begin
        from_shell("git config hooks.emailprefix").strip
      rescue ArgumentError
        ''
      end
      return git_prefix unless git_prefix.empty?
      File.expand_path(git_dir).split("/").last.sub(/\.git$/, '')
    end

    def mailing_list_address
      from_shell("git config hooks.mailinglist").strip
    rescue ArgumentError
      nil
    end
  end
end