Add new hooks around fetching gems and git sources

Author: MrMarvin

bundler/lib/bundler/plugin/events.rb

@@ -45,6 +45,30 @@ def self.defined_event?(event)
       #   GEM_AFTER_INSTALL = "after-install"
       define :GEM_AFTER_INSTALL,  "after-install"
 
+      # @!parse
+      #   A hook called before each individual gem is fetched for installation.
+      #   Includes a Bundler::ParallelInstaller::SpecInstallation and a source reference.
+      #   GEM_BEFORE_FETCH = "before-fetch"
+      define :GEM_BEFORE_FETCH, "before-fetch"
+
+      # @!parse
+      #   A hook called after each individual gem is fetched for installation.
+      #   Includes a Bundler::ParallelInstaller::SpecInstallation and a source reference.
+      #   GEM_AFTER_FETCH = "after-fetch"
+      define :GEM_AFTER_FETCH, "after-fetch"
+
+      # @!parse
+      #   A hook called before a git source is fetched.
+      #   Includes a Bundler::Source::Git reference.
+      #   GIT_GEM_BEFORE_FETCH = "git-before-fetch"
+      define :GIT_BEFORE_FETCH, "git-before-fetch"
+
+      # @!parse
+      #   A hook called after a git source is fetched.
+      #   Includes a Bundler::Source::Git reference.
+      #   GEM_AFTER_FETCH = "git-after-fetch"
+      define :GIT_AFTER_FETCH, "git-after-fetch"
+
       # @!parse
       #   A hook called before any gems install
       #   Includes an Array of Bundler::Dependency objects

bundler/lib/bundler/source/git.rb

@@ -193,8 +193,10 @@ def specs(*)
         if requires_checkout? && !@copied
           FileUtils.rm_rf(app_cache_path) if use_app_cache? && git_proxy.not_a_repository?
 
+          Plugin.hook(Plugin::Events::GIT_BEFORE_FETCH, self)
           fetch
           checkout
+          Plugin.hook(Plugin::Events::GIT_AFTER_FETCH, self)
         end
 
         local_specs
@@ -207,7 +209,9 @@ def install(spec, options = {})
         print_using_message "Using #{version_message(spec, options[:previous_spec])} from #{self}"
 
         if (requires_checkout? && !@copied) || force
+          Plugin.hook(Plugin::Events::GEM_BEFORE_FETCH, spec, self)
           checkout
+          Plugin.hook(Plugin::Events::GEM_AFTER_FETCH, spec, self)
         end
 
         generate_bin_options = { disable_extensions: !spec.missing_extensions?, build_args: options[:build_args] }

bundler/lib/bundler/source/rubygems.rb

@@ -416,11 +416,14 @@ def fetch_names(fetchers, dependency_names, index)
       end
 
       def fetch_gem_if_possible(spec, previous_spec = nil)
-        if spec.remote
+        Plugin.hook(Plugin::Events::GEM_BEFORE_FETCH, spec, self)
+        gem_path = if spec.remote
           fetch_gem(spec, previous_spec)
         else
           cached_gem(spec)
         end
+        Plugin.hook(Plugin::Events::GEM_AFTER_FETCH, spec, self)
+        gem_path
       end
 
       def fetch_gem(spec, previous_spec = nil)