# Cleans a newly installed keg.
# By default:
# * removes info files
# * removes .la files
# * removes empty directories
# * sets permissions on executables
class Cleaner

  # Create a cleaner for the given formula name, and clean the keg
  def initialize f
    @f = Formula.factory f
    [f.bin, f.sbin, f.lib].select{ |d| d.exist? }.each{ |d| clean_dir d }

    if ENV['HOMEBREW_KEEP_INFO']
      # Get rid of the directory file, so it no longer bother us at link stage.
      info_dir_file = f.info + 'dir'
      if info_dir_file.file? and not f.skip_clean? info_dir_file
        puts "rm #{info_dir_file}" if ARGV.verbose?
        info_dir_file.unlink
      end
    else
      f.info.rmtree if f.info.directory? and not f.skip_clean? f.info
    end

    # Remove empty folders.
    # We want post-order traversal, so use a stack.
    paths = []
    f.prefix.find do |path|
      if path.directory?
        if f.skip_clean? path
          Find.prune
        else
          paths << path
        end
      end
    end

    paths.each do |d|
      if d.children.empty? and not f.skip_clean? d
        puts "rmdir: #{d} (empty)" if ARGV.verbose?
        d.rmdir
      end
    end
  end

  private

  # Set permissions for executables and non-executables
  def clean_file_permissions path
    perms = if path.mach_o_executable? || path.text_executable?
      0555
    else
      0444
    end
    # Uncomment this block to show permission changes using brew install -v
    # if ARGV.verbose?
    #   old_perms = path.stat.mode
    #   if perms != old_perms
    #     puts "Fixing #{path} permissions from #{old_perms.to_s(8)} to #{perms.to_s(8)}"
    #   end
    # end
    path.chmod perms
  end

  # Clean a single folder (non-recursively)
  def clean_dir d
    d.find do |path|
      path.extend(NoiseyPathname) if ARGV.verbose?

      if path.directory?
        # Stop cleaning this subtree if protected
        Find.prune if @f.skip_clean? path
      elsif not path.file?
        # Sanity?
        next
      elsif path.extname == '.la'
        # *.la files are stupid
        path.unlink unless @f.skip_clean? path
      elsif path == @f.lib+'charset.alias'
        # Many formulae symlink this file, but it is not strictly needed
        path.unlink unless @f.skip_clean? path
      elsif not path.symlink?
        # Fix permissions
        clean_file_permissions path
      end
    end
  end

end


class Pathname
  alias_method :orig_unlink, :unlink
end

module NoiseyPathname
  def unlink
    puts "rm: #{self}"
    orig_unlink
  end
end