require 'mkalias/version'
require 'set'

module Mkalias

  BASHRC_PATH = "#{File.expand_path('~')}/.bashrc"

  def self.new_alias(alias_name, commands, file_path=BASHRC_PATH)
    alias_names = Mkalias.list_alias(file_path)
    return false if alias_names.include?(alias_name)

    commands = commands.join('; ') if commands.kind_of?(Array)
    commands = commands.gsub('#', '$')

    function_name = "mkalias_#{alias_name}"
    bash_function = "function #{function_name}(){ #{commands}; }"
    bash_alias = "alias #{alias_name}='#{function_name}'"

    open(file_path, 'a') do |file|
      file.puts("\n")
      file.puts(bash_alias)
      file.puts(bash_function)
    end

    true
  end

  def self.list_alias(file_path=BASHRC_PATH)
    alias_names = Set.new

    alias_regex = /\bmkalias_(.*)[(]/

    alias_functions = File.foreach(file_path).grep(alias_regex)
    alias_functions.each do |function|
      result = function.match(alias_regex)
      alias_names << result.captures.first if result
    end

    alias_names.to_a
  end

  def self.show_alias(alias_names, file_path=BASHRC_PATH)
    alias_names = [alias_names] unless alias_names.kind_of?(Array)

    alias_functions = {}
    alias_names.each do |alias_name|
      alias_functions[alias_name] = Mkalias.get_alias_function(alias_name,
                                                               file_path)
    end

    alias_functions.select!{ |key, value| !value.nil? }
    return alias_functions
  end

  def self.remove_alias(alias_names, file_path=BASHRC_PATH)
    alias_names = [alias_names] unless alias_names.kind_of?(Array)

    removed_alias = []
    alias_names.each do |alias_name|
      removed = Mkalias.remove_one_alias(alias_name, file_path)
      removed_alias << alias_name if removed
    end

    return removed_alias
  end

  private

  def self.get_alias_function(alias_name, file_path=BASHRC_PATH)
    alias_names = Mkalias.list_alias(file_path)
    return nil unless alias_names.include?(alias_name)

    alias_regex = /\bmkalias_#{alias_name}[(]/
    function_regex = /[{](.*)[;]/

    alias_functions = File.foreach(file_path).grep(alias_regex)

    alias_functions.each do |function|
      result = function.match(function_regex)
        return result.captures.first.split(';').each{ |c| c.strip! } if result
    end

    nil
  end

  def self.remove_one_alias(alias_name, file_path=BASHRC_PATH)
    alias_names = Mkalias.list_alias(file_path)
    return false unless alias_names.include?(alias_name)

    alias_regex = /\bmkalias_#{alias_name}[(']/

    lines = File.readlines(file_path).reject{ |line| line =~ alias_regex }

    File.open(file_path, "w"){ |f| lines.each { |line| f.puts line } }

    return true
  end
end