module Hbc
  module Scopes
    def self.included(base)
      base.extend(ClassMethods)
    end

    module ClassMethods
      def all
        @all_casks ||= {}
        all_tokens.map { |t| @all_casks[t] ||= load(t) }
      end

      def all_tapped_cask_dirs
        Tap.map(&:cask_dir).select(&:directory?)
      end

      def all_tokens
        Tap.flat_map do |t|
          t.cask_files.map do |p|
            "#{t.name}/#{File.basename(p, ".rb")}"
          end
        end
      end

      def installed
        # CaskLoader.load has some DWIM which is slow.  Optimize here
        # by spoon-feeding CaskLoader.load fully-qualified paths.
        # TODO: speed up Hbc::Source::Tapped (main perf drag is calling Hbc.all_tokens repeatedly)
        # TODO: ability to specify expected source when calling CaskLoader.load (minor perf benefit)
        Pathname.glob(caskroom.join("*"))
                .map do |caskroom_path|
                  token = caskroom_path.basename.to_s

                  path_to_cask = all_tapped_cask_dirs.find do |tap_dir|
                    tap_dir.join("#{token}.rb").exist?
                  end

                  if path_to_cask
                    CaskLoader.load(path_to_cask.join("#{token}.rb"))
                  else
                    CaskLoader.load(token)
                  end
                end
      end
    end
  end
end

module Hbc
  module Scopes
    def self.included(base)
      base.extend(ClassMethods)
    end

    module ClassMethods
      def all
        all_tokens.map(&CaskLoader.public_method(:load))
      end

      def all_tapped_cask_dirs
        Tap.map(&:cask_dir).select(&:directory?)
      end

      def all_tokens
        Tap.flat_map do |t|
          t.cask_files.map do |p|
            "#{t.name}/#{File.basename(p, ".rb")}"
          end
        end
      end

      def installed
        # CaskLoader.load has some DWIM which is slow.  Optimize here
        # by spoon-feeding CaskLoader.load fully-qualified paths.
        # TODO: speed up Hbc::Source::Tapped (main perf drag is calling Hbc.all_tokens repeatedly)
        # TODO: ability to specify expected source when calling CaskLoader.load (minor perf benefit)
        Pathname.glob(caskroom.join("*"))
                .map do |caskroom_path|
                  token = caskroom_path.basename.to_s

                  path_to_cask = all_tapped_cask_dirs.find do |tap_dir|
                    tap_dir.join("#{token}.rb").exist?
                  end

                  if path_to_cask
                    CaskLoader.load(path_to_cask.join("#{token}.rb"))
                  else
                    CaskLoader.load(token)
                  end
                end
      end
    end
  end
end

Fix `brew cask audit` not working without argument.
Fix `brew cask audit` not working without argument.
