#!/usr/bin/env ruby

# Copyright (c) 2022 Andy Maleh
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

version = File.read(File.expand_path('../../VERSION', __FILE__))
puts "== git-tui-setup version #{version.strip} =="

ruby = `which ruby`
ruby = `which jruby` if ruby.to_s.strip.empty?
ruby = ruby.strip
rvm = `rvm current`.strip
command = "rvm #{rvm} do #{ruby} #{File.expand_path('../../lib/git-tui.rb', __FILE__)}"

git_tui_script = <<~SHELL_SCRIPT
  function git-tui()
  {
    # Load RVM into a shell session *as a function*
    if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
      # First try to load from a user install
      source "$HOME/.rvm/scripts/rvm"
    elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
      # Then try to load from a root install
      source "/usr/local/rvm/scripts/rvm"
    else
      printf "ERROR: An RVM installation was not found.\\n"
    fi
    
    current_rvm="$(rvm current)"
    current_ruby="$(echo $current_rvm | sed -E 's/@.*$//')"
    current_gemset="default"
    if [[ $current_rvm == *"@"* ]] ; then
      current_gemset="$(echo $current_rvm | sed -E 's/^[^@]+@//')"
    fi
    #{command} $current_ruby $current_gemset
    if [ -f "$HOME/.git_command" ] ; then
      source $HOME/.git_command > /dev/null
      rm $HOME/.git_command
    fi
  }
  alias gitui=git-tui
  alias git-ui=git-tui
  alias g=git
SHELL_SCRIPT

require 'fileutils'
home_dir = `echo ~`.strip
FileUtils.mkdir_p("#{home_dir}/bin")
git_tui_script_file = "#{home_dir}/.git_tui_source"
git_tui_script_file_bash = "$HOME/.git_tui_source"
File.write(git_tui_script_file, git_tui_script)

[
  ['.bash_profile', '.bashrc'],
  ['.zprofile', '.zshrc']
].each do |profile_and_fallback|
  shell_profile_file_name = profile_and_fallback.first
  shell_profile_file_name = profile_and_fallback.last if !File.exist?("#{home_dir}/#{shell_profile_file_name}")
  shell_profile_file = "#{home_dir}/#{shell_profile_file_name}"
  
  FileUtils.touch(shell_profile_file)
  
  bash_profile = File.read(shell_profile_file)
  git_tui_source_statement = "source \"#{git_tui_script_file_bash}\""
  unless bash_profile.split("\n").detect {|line| line.include?(git_tui_source_statement) }
    File.write(shell_profile_file, "#{bash_profile}\n#{git_tui_source_statement}")
  end
  puts "~/#{shell_profile_file_name} has been modified to source the `git-tui` function from ~/.git_tui_source"
end

{
  'aa'  => 'add -A',
  'am'  => 'commit -am',
  'ap'  => 'add -p',
  'df'  => 'diff',
  'dh'  => 'diff HEAD',
  'br'  => 'branch',
  'co'  => 'checkout',
  'lp'  => 'log -p',
  'sa'  => 'stash apply',
  'sd'  => 'stash drop',
  'sh'  => 'stash -u',
  'sl'  => 'stash list',
  'sp'  => 'stash pop',
  'ss'  => 'stash save -u',
  'st'  => 'status',
  'cam' => 'commit -am',
  'cl'  => 'clone',
  'ci'  => 'commit',
  'cia' => 'commit --amend',
  'cm'  => 'commit -m',
  'm'   => 'commit -m',
  'mg'  => 'merge',
  'pk'  => 'cherry-pick',
  'ps'  => 'push',
  'pf'  => 'push --force-with-lease',
  'pl'  => 'pull',
  'pr'  => 'pull --rebase',
  'rb'  => 'rebase',
  'rc'  => 'rebase --continue',
  'rs'  => 'rebase --skip',
  'ra'  => 'rebase --abort',
  'ri'  => 'rebase --interactive',
  'rh'  => 'reset HEAD',
}.each do |git_alias, git_command|
  system "git config --global alias.#{git_alias} '#{git_command}'"
end

system "git config --global branch.autoSetupRebase 'always'"

puts <<~OUTPUT
  
  Before using `git-tui`, start a new shell session (open a new terminal tab) or run:
  
  source ~/.git_tui_source
  
  Afterwards, you may run `git-tui` via one of these commands (aliases):
  
  git-tui
  git-ui
  gitui
OUTPUT