%w(rubygems git fileutils yaml).each {|f| require f}

module  Yac
  extend self

  YACRC = File.join("#{ENV['HOME']}",".yacrc")
  FileUtils.cp(File.join(File.dirname(__FILE__), "..","resources","yacrc"), YACRC) unless File.exist?(YACRC)
  CONFIG = YAML.load_file(File.join(ENV['HOME'],".yacrc"))
  CONFIG["root"] ||= File.join(ENV['HOME'],".yac")

  @main_path, @pri_path = File.join(CONFIG["root"],"/main/"), File.join(CONFIG["root"],"/private/")
  @main_git = Git.open(@main_path) if File.exist?(@main_path)
  @pri_git = Git.open(@pri_path)if File.exist?(@pri_path)

  def new(args)
    unless File.exist?(@main_path) && File.exist?(@pri_path)
      return unless init
    end
    @all_result = []
    (help && exit) if args.empty?
    case args.first
    when "show" then show(args[1,args.size])
    when "name" then search(args[1,args.size],"name")
    when "content" then search(args[1,args.size],"content")
    when "update" then update(args[1,args.size])
    when /^(add|edit)$/ then edit(args[1,args.size])
    when "help" then help
    when "shell" then shell(args[1,args.size])
    when "rm" then rm(args[1,args.size])
    when "init" then init
    else show(args)
    end
    show_possible_result
  rescue
  end

  def show(args)
    args.each {|x| show_single(x)}
  end

  def search(args,type = "name")
    args.each {|x| search_single(x,type)}
  end

  def update(args)
    begin
      unless args.empty?
        @pri_git.pull if args.to_s =~ /pri/
        @main_git.pull if args.to_s =~ /main/
      else
        @main_git.pull
        @pri_git.pull
      end
    rescue
      puts "ERROR: can not update the repository,"
      puts $!
    end
  end

  def edit(args)
    args.each {|x| edit_single(x)}
  end

  def rm(args)
    args.each {|x| rm_single(x)}
  end

  def help
    format_file(File.dirname(__FILE__)+"/../README.rdoc")
  end

  def shell(args)
    case args.to_s
    when /main/
      puts "\033[31m Welcome To The Main Yac Repository \033[0m"
      system "cd #{@main_path}; sh"
    else
      puts "\033[31m Welcome To The Private Yac Repository \033[0m"
      system "cd #{@pri_path}; sh"
    end
  end

  protected
  def format_file(file)
    @level = 0
    File.new(file).each do |x|
      format_section(x)
    end
  end

  def format_section(section)
     case section
     when /^(=+)\s+(.*)/
       @level = $1.size
       puts "\033[" + CONFIG["level#{@level}"].to_s + "m" + "\s"*2*(@level-1) + $2 +"\033[0m"
     when /^(\s*)#/
     else
       puts section.sub(/^\#/,"#").sub(/^\s*/, "\s" * @level * 2 ).gsub(/@@@(.*)@@@/,"\033[" + CONFIG["empha"].to_s + "m" + '\1' + "\033[0m")
     end
  end

  def full_path(args)
    if args =~ /^@/
      @file_path = @main_path + args.sub(/^@/,"") + ".ch"
      @working_git = @main_git
    else
      @file_path = @pri_path + args + ".ch"
      @working_git = @pri_git
    end
  end

  def rm_single(args)
    full_path(args)
    begin
      @working_git.remove(@file_path)
      @working_git.commit_all("#{args.sub(/^@/,"")}.ch removed")
    rescue Git::GitExecuteError
      FileUtils.rm_rf(@file_path)
    end
  end

  def edit_single(args)
    full_path(args)
    prepare_dir
    system("#{editor} #{@file_path}")
    @working_git.add
    @working_git.commit_all(" #{args.sub(/^@/,"")}.ch Updated")
  end

  def editor
    ENV['VISUAL'] || ENV['EDITOR'] || "vim"
  end

  def show_single(args)
    result = search_single(args)
    if result.size == 1
      puts "\n\033[#{CONFIG["filename"]}m#{result.first}\033[0m\n"
      format_file(result.first)
    else
      result.map do |x|
        if x =~ /\/#{args}\.ch/
          puts "\n\033[#{CONFIG["filename"]}m#{x}\033[0m\n"
          format_file(x)
        end
      end
    end
  end

  def search_single(args,type="name")
    if type =~ /name/
      search_name(args)
    else
      search_content(args)
    end
  end

  def search_name(args)
    if args =~ /^@/ && main = args.sub(/^@/,"")
      @private_result = []
      @main_result = @main_git.ls_files.keys.grep(/#{main}/)
    else
      @private_result = @pri_git.ls_files.keys.grep(/#{args}/)
        @main_result = @main_git.ls_files.keys.grep(/#{args}/)
    end
    #ADD to all possible result
    @all_result << @main_result.collect {|x| "@" + x}
    @all_result << @private_result

    #Remove duplicate files
    @private_result.map do |x|
      @main_result.delete(x)
    end
    #Return full path
    return (@main_result.collect {|x| @main_path +x}).concat(@private_result.collect {|x| @pri_path +x})
  end

  def search_content(args)
    @main_git.grep(args).each do |file, lines|
      title = title_of_file(file.split(':')[1])
      lines.each do |l|
        puts "@#{title}:#{l[0]}:  #{l[1]}"
      end
    end
    @pri_git.grep(args).each do |file, lines|
      title = title_of_file(file.split(':')[1])
      lines.each do |l|
        puts "#{title}:#{l[0]}:  #{l[1]}"
      end
    end
  end

  def title_of_file(f)
    f[0..((f.rindex('.')||0) - 1)]
  end

  def show_possible_result
    puts "\n\033[" + CONFIG["possible_result_title"].to_s + "mALL POSSIBLE RESULT:\033[0m\n \033[" +CONFIG["possible_result_content"].to_s + "m" + @all_result.join("\s"*2) + "\033[0m" unless @all_result.to_s.empty?
  end

  def init
    FileUtils.mkdir_p(CONFIG['root'])
    if CONFIG['main']['clone-from']
      if File.exist?(@main_path)
        puts "Main repository has already been initialized."
      else
        puts "Initialize main repository from #{CONFIG['main']['clone-from']} to #{CONFIG['root']}/main"
        Git.clone(CONFIG['main']['clone-from'], 'main', :path => CONFIG['root'])
        puts "Main repository initialized."
      end
    else
      puts "ERROR: configuration for main repository repository is missing!"
      return
    end
    
    unless File.exist?(@pri_path)
      if CONFIG['private'] && CONFIG['private']['clone-from']
        puts "Initialize private repository from #{CONFIG['private']['clone-from']} to #{CONFIG['root']}/private"
        Git.clone(CONFIG['private']['clone-from'], 'private', :path => CONFIG['root'])
        puts "Private repository initialized."
      else
        puts "Initialize private repository from scratch to #{CONFIG['root']}/private"
        git = Git.init(@pri_path)
        git.add
        git.commit_all("init private repository")
        puts "Private repository initialized."
      end
    else
      puts "Private repository has already been initialized."
    end
    @main_git = Git.open(@main_path) if File.exist?(@main_path)
    @pri_git = Git.open(@pri_path)if File.exist?(@pri_path)
    puts "Repository init done."
  end

  def prepare_dir
    dirseparator = @file_path.rindex(File::Separator)+1
    FileUtils.mkdir_p(@file_path[0,dirseparator])
  end
end