require 'pathname'
require 'minigit'
require 'awesome_print'
require 'vendorificator/config'

module Vendorificator
  class Environment
    attr_reader :config
    attr_accessor :shell, :vendor_instances

    def initialize(vendorfile=nil, &block)
      @vendor_instances = []

      @config = Vendorificator::Config.new
      @config.environment = self
      if vendorfile || !block_given?
        @config.read_file(find_vendorfile(vendorfile).to_s)
      end
      @config.instance_eval(&block) if block_given?

      self.each_vendor_instance{ |mod| mod.compute_dependencies! }
    end

    def say_status(*args)
      shell.say_status(*args) if shell
    end

    # Main MiniGit instance
    def git
      @git ||= MiniGit::new(config[:vendorfile_path])
    end

    # Git helpers
    def remotes
      @remotes ||= git.capturing.remote.lines.map(&:strip)
    end

    def current_branch
      git.capturing.rev_parse({:abbrev_ref => true}, 'HEAD').strip
    end

    def fast_forwardable?(to, from)
      git.capturing.merge_base(to, from).strip == from
    end

    # Public: Pulls all the remotes specified in options[:remote] or the config.
    #
    # options - The Hash of options.
    #
    # Returns nothing.
    def pull_all(options = {})
      ensure_clean!
      remotes = options[:remote] ? options[:remote].split(',') : config[:remotes]
      remotes.each do |remote|
        indent 'remote', remote do
          pull(remote, options)
        end
      end
    end

    # Public: Pulls a single remote and updates the branches.
    #
    # options - The Hash of options.
    #
    # Returns nothing.
    def pull(remote, options={})
      raise RuntimeError, "Unknown remote #{remote}" unless remotes.include?(remote)

      git.fetch(remote)
      git.fetch({:tags => true}, remote)
      git.fetch(remote, 'refs/notes/vendor:refs/notes/vendor')

      ref_rx = /^refs\/remotes\/#{Regexp.quote(remote)}\//
      remote_branches = Hash[ git.capturing.show_ref.
        lines.
        map(&:split).
        map { |sha, name| name =~ ref_rx ? [$', sha] : nil }.
        compact ]

      each_vendor_instance do |mod|
        ours = mod.head
        theirs = remote_branches[mod.branch_name]
        if theirs
          if not ours
            say_status 'new', mod.branch_name, :yellow
            git.branch({:track => true}, mod.branch_name, theirs) unless options[:dry_run]
          elsif ours == theirs
            say_status 'unchanged', mod.branch_name
          elsif fast_forwardable?(theirs, ours)
            say_status 'updated', mod.name, :yellow
            mod.in_branch { git.merge({:ff_only => true}, theirs) } unless options[:dry_run]
          elsif fast_forwardable?(ours, theirs)
            say_status 'older', mod.branch_name
          else
            say_status 'complicated', mod.branch_name, :red
          end
        else
          say_status 'unknown', mod.branch_name
        end
      end
    end

    # Public: Displays info about the last merged version of module.
    #
    # mod - String with the module name
    # options - Hash containing options
    #
    # Returns nothing.
    def info(mod_name, options = {})
      if vendor = find_vendor_instance_by_name(mod_name)
        shell.say "Module name: #{vendor.name}\n"
        shell.say "Module category: #{vendor.category}\n"
        shell.say "Module merged version: #{vendor.merged_version}\n"
        shell.say "Module merged notes: #{vendor.merged_notes.ai}\n"
      elsif (commit = Commit.new(mod_name, git)).exists?
        shell.say "Branches that contain this commit: #{commit.branches.join(', ')}\n"
        shell.say "Vendorificator notes on this commit: #{commit.notes.ai}\n"
      else
        shell.say "Module or ref #{mod_name.inspect} not found."
      end
    end

    # Public: Push changes on module branches.
    #
    # options - The Hash containing options
    #
    # Returns nothing.
    def push(options = {})
      ensure_clean!

      pushable = []
      each_vendor_instance{ |mod| pushable += mod.pushable_refs }

      remotes = options[:remote] ? options[:remote].split(',') : config[:remotes]
      remotes.each do |remote|
        git.push remote, pushable
        git.push remote, :tags => true
        git.push remote, 'refs/notes/vendor'
      end
    end

    # Public: Runs all the vendor modules.
    #
    # options - The Hash of options.
    #
    # Returns nothing.
    def sync(options = {})
      ensure_clean!
      config[:use_upstream_version] = options[:update]
      metadata = metadata_snapshot

      each_vendor_instance(*options[:modules]) do |mod|
        say_status :module, mod.name
        indent do
          mod.run!(:metadata => metadata)
        end
      end
    end

    # Public: Goes through all the Vendor instances and runs the block
    #
    # modules - ?
    #
    # Returns nothing.
    def each_vendor_instance(*modules)
      modpaths = modules.map { |m| File.expand_path(m) }

      # We don't use @vendor_instances.each here, because Vendor#run! is
      # explicitly allowed to append to instantiate new dependencies, and #each
      # fails to catch up on some Ruby implementations.
      i = 0
      while true
        break if i >= @vendor_instances.length
        mod = @vendor_instances[i]
        yield mod if modules.empty? ||
          modules.include?(mod.name) ||
          modpaths.include?(mod.work_dir)
        i += 1
      end
    end

    # Public: Checks if the repository is clean.
    #
    # Returns boolean answer to the question.
    def clean?
      # copy code from http://stackoverflow.com/a/3879077/16390
      git.update_index '-q', '--ignore-submodules', '--refresh'
      git.diff_files '--quiet', '--ignore-submodules', '--'
      git.diff_index '--cached', '--quiet', 'HEAD', '--ignore-submodules', '--'
      true
    rescue MiniGit::GitError
      false
    end

    def metadata_snapshot
      {
        :vendorificator_version => ::Vendorificator::VERSION,
        :current_branch => git.capturing.rev_parse({:abbrev_ref => true}, 'HEAD').strip,
        :current_sha => git.capturing.rev_parse('HEAD').strip,
        :git_describe => (git.capturing.describe.strip rescue '')
      }
    end

    # Public: returns `config[:root_dir]` relative to Git repository root
    def relative_root_dir
      @relative_root_dir ||= config[:root_dir].relative_path_from(
        Pathname.new(git.git_work_tree))
    end

    # Public: Returns module with given name
    def [](name)
      vendor_instances.find { |v| v.name == name }
    end

    private

    # Private: Finds a vendor instance by module name.
    #
    # mod_name - The String containing the module name.
    #
    # Returns Vendor instance.
    def find_vendor_instance_by_name(mod_name)
      each_vendor_instance do |mod|
        return mod if mod.name == mod_name
      end
      nil
    end

    # Private: Finds the vendorfile to use.
    #
    # given - the optional String containing vendorfile path.
    #
    # Returns a String containing the vendorfile path.
    def find_vendorfile(given=nil)
      given = [ given, ENV['VENDORFILE'] ].find do |candidate|
        candidate && !(candidate.respond_to?(:empty?) && candidate.empty?)
      end
      return given if given

      Pathname.pwd.ascend do |dir|
        vf = dir.join('Vendorfile')
        return vf if vf.exist?

        vf = dir.join('config/vendor.rb')
        return vf if vf.exist?

        # avoid stepping above the tmp directory when testing
        if ENV['VENDORIFICATOR_SPEC_RUN'] &&
            dir.join('vendorificator.gemspec').exist?
          raise ArgumentError, "Vendorfile not found"
        end
      end

      raise ArgumentError, "Vendorfile not found"
    end

    # Private: Aborts on a dirty repository.
    #
    # Returns nothing.
    def ensure_clean!
      raise DirtyRepoError unless clean?
    end

    # Private: Indents the output.
    #
    # Returns nothing.
    def indent(*args, &block)
      say_status *args unless args.empty?
      shell.padding += 1 if shell
      yield
    ensure
      shell.padding -= 1 if shell
    end
  end
end