require 'YAML'
require 'rainbow'
require 'fileutils'

require_relative './alfred_git/version'

module AlfredGit
  class AlfredGit
    include AlfredGitVersion

    def initialize
      set_app_directory
      # If there's just one, it's the current version.
      restore_settings unless Dir.glob("#{@app_directory.chomp('/alfred_git-#{VERSION}')}/alfred_git*").length > 1
      config unless File.exists?("#{@app_directory}/lib/config.yaml")
      config_yaml = YAML.load_file("#{@app_directory}/lib/config.yaml")

      @repo_locations = config_yaml['repos']
      @name = config_yaml['name'].nil? ? 'Wayne' : config_yaml['name']
      @gender = config_yaml['gender'].nil? ? 'sir' : config_yaml['gender']
    end

    def alfred(arguments)
      @arguments = arguments
      alfred_command
      command = command_to_git_command
      repos = repos_string_to_array

      repos.each do |repo|
        lines_pretty_print Rainbow("Repo #{repo}:").yellow
        bash(@repo_locations[repo], command)

        single_space
      end
    end

    def alfred_command
      case @arguments[0]
      when nil, ''
        lines_pretty_print Rainbow("I need a command to run, Master #{@name}.").red

        abort
      when 'list_repo', 'list_repos'
        lines_pretty_print "Here are your repos and their locations, Master #{@name}:"

        single_space

        list_repos_and_locations

        single_space

        abort
      when 'add_repo'
        add_repo

        abort
      when 'delete_repo'
        delete_repo

        abort
      end
    end

    def list_repos_and_locations
      @repo_locations.each do |repo, location|
        lines_pretty_print Rainbow("#{repo}").yellow + ": #{location}"
      end
    end

    def add_repo
      config_yaml = YAML.load_file("#{@app_directory}/lib/config.yaml")
      config_file = File.open("#{@app_directory}/lib/config.yaml", 'w')

      lines_pretty_print "What is the 'friendly' name you'd like to give your repository? This is the "\
                         'name you will type when sending me commands.'

      repo_name = STDIN.gets.strip!

      single_space

      lines_pretty_print "Thank you, #{@gender}. Now, where is that repository? Please paste the full path."

      repo_path = STDIN.gets.strip!

      single_space

      config_yaml['repos'][repo_name] = repo_path
      YAML.dump(config_yaml, config_file)

      lines_pretty_print Rainbow("I've added that repository successfully, #{@gender}!").green

      single_space
    end

    def delete_repo
      config_yaml = YAML.load_file("#{@app_directory}/lib/config.yaml")
      config_file = File.open("#{@app_directory}/lib/config.yaml", 'w')

      lines_pretty_print "What repository would you like to delete from my list of repositories, #{@gender}?"

      repo_name = STDIN.gets.strip!

      single_space

      if config_yaml['repos'].keys.include?(repo_name)
        config_yaml['repos'].delete(repo_name)

        lines_pretty_print Rainbow("I've deleted that repository successfully, #{@gender}!").green
      else
        lines_pretty_print Rainbow("Sorry, #{@gender}, that is not a repository that is currently in my list. "\
                                   'If you\'d *really* like to delete it, please add it first using the '\
                                   '\'add_repo\' command.').red
      end

      YAML.dump(config_yaml, config_file)

      single_space
    end

    def command_to_git_command
      command = ''

      case @arguments[0]
      when 'pull'
        command = 'git pull'
        @arguments.delete_at(0)
      when 'push'
        command = 'git push'
        @arguments.delete_at(0)
      when 'checkout'
        if second_argument_missing?
          lines_pretty_print Rainbow('I need a branch name to execute the \'checkout\' command, Master '\
                                     "#{@name}.").red

          abort
        end

        command = "git checkout #{@arguments[1]}"

        @arguments.delete_at(0)
        @arguments.delete_at(0)
      when 'commit'
        if second_argument_missing?
          lines_pretty_print Rainbow('I need a commit message to execute the \'commit\' command, Master '\
                                     "#{@name}.").red

          abort
        end

        command = %Q[git commit -m "#{@arguments[1]}"]
      when 'status'
        command = 'git status'
        @arguments.delete_at(0)
      when 'branches', 'branch'
        command = 'git rev-parse --abbrev-ref HEAD'
        @arguments.delete_at(0)
      else
        command = @arguments[0] # Allow users to send any command to all repos.
        @arguments.delete_at(0)
      end

      command
    end

    # All other arguments are deleted before we get here, so this should be just the repo list.
    def repos_string_to_array
      if @arguments[0].nil?|| @arguments[0] == ''
        lines_pretty_print Rainbow("I need at least one repository to work with, Master #{@name}.").red

        abort
      elsif @arguments[0] == 'all'
        return @repo_locations.keys
      else
        return @arguments
      end
    end

    def config
      config_file = File.open("#{@app_directory}/lib/config.yaml", 'w')

      lines_pretty_print 'Good evening, Master Wayne. Thank you for utilizing my abilities. Please enter your '\
                         'last name if you\'d like me to address you as something other than Wayne.'
      lines_pretty_print Rainbow('(Just hit enter to stick with \'Master Wayne\'.)').yellow

      name = STDIN.gets.strip!

      name = name == '' ? 'Wayne' : name

      single_space

      lines_pretty_print 'Thank you, sir. You...are a sir, correct?'

      gender_set = false

      until gender_set
        gender = STDIN.gets.strip!

        single_space

        if gender == 'yes' || gender == 'y'
          gender = 'sir'
          gender_set = true
        elsif gender == 'no' || gender == 'n'
          gender = 'madam'
          gender_set = true
        else
          lines_pretty_print "Very funny, Master #{name}."
          lines_pretty_print Rainbow('Please input either \'yes\' or \'no\'.').yellow
        end
      end

      lines_pretty_print "Yes, of course. My apologies. Please do not take offense, #{gender}. I am, after "\
                         'all, a simple computer program. Don\'t have eyes, you see. Now, let\'s gather a '\
                         'list of the code repositories you work with, shall we?'

      single_space

      repos = {}
      done = false
      repo_count = 0

      until done do
        first = repo_count > 0 ? 'next' : 'first'

        lines_pretty_print "What is the 'friendly' name you'd like to give your #{first} repository? This is the "\
                           'name you will type when sending me commands. If you are done adding them, please '\
                           'enter \'x211\' as your input instead.'

        repo_name = STDIN.gets.strip!

        if repo_name == 'x211'
          done = true
          single_space
          next
        end

        single_space

        lines_pretty_print "Thank you, #{gender}. Now, where is that repository? Please paste the full path."

        repo_path = STDIN.gets.strip!

        single_space

        repos[repo_name] = repo_path
        repo_count += 1

        lines_pretty_print Rainbow("I've added that repository successfully, #{gender}!").green

        single_space
      end

      YAML.dump({ 'name' => name, 'gender' => gender, 'repos' => repos }, config_file)
      config_file.close

      lines_pretty_print "Thank you for helping me set things up, Master #{name}. Feel free to run me whenever "\
                         'you need.'

      abort
    end

    # Attempt to restore settings from previous version.
    def restore_settings
      lines_pretty_print 'I see that I\'ve been recently updated.'
      lines_pretty_print Rainbow('Would you like to restore the settings from the previous installation?').yellow

      answered = false

      until answered
        answer = STDIN.gets.strip!

        single_space

        if answer == 'yes' || answer == 'y' || answer == 'no' || answer == 'n'
          answered = true
        else
          lines_pretty_print 'You\'re hilarious. Really.'
          lines_pretty_print Rainbow('Please input either \'yes\' or \'no\'.').yellow
        end
      end

      return if answer == 'no' || answer == 'n'

      lines_pretty_print 'One moment, please.'

      single_space

      all_gems = Dir.glob("#{@app_directory.chomp("/alfred_git-#{VERSION}")}/alfred_git*")

      # glob orders things in the array alphabetically, so the second-to-last one in the array is the
      # most recent version that is not the current version.
      previous_config_file = "#{all_gems[-2]}/lib/config.yaml"
      FileUtils.copy_file(previous_config_file, "#{@app_directory}/lib/config.yaml")

      lines_pretty_print 'Done! Please run me again when you\'re ready.'

      abort
    end

    def second_argument_missing?
      @arguments[1].nil? || @arguments[1] == ''
    end

    def bash(directory, command)
      # Dir.chdir ensures all bash commands are being run from the correct
      # directory.
      Dir.chdir(directory) { system "#{command}" }
    end

    def set_app_directory
      @app_directory = File.expand_path(File.dirname(__FILE__)).chomp('/lib')
    end

    def lines_pretty_print(string)
      lines = string.scan(/\S.{0,70}\S(?=\s|$)|\S+/)

      lines.each { |line| puts line }
    end

    def single_space
      puts ''
    end

    def double_space
      puts "\n\n"
    end

    def surround_by_double_space(string)
      double_space
      lines_pretty_print(string)
      double_space
    end
  end
end