Initial implementation of 'bundle pristine'

2 files changed, 45 insertions, 0 deletions

diff --git a/lib/bundler/cli.rb b/lib/bundler/cli.rb
index 61c5427c90..5eeb1f8018 100644
--- a/lib/bundler/cli.rb
+++ b/lib/bundler/cli.rb
@@ -540,6 +540,12 @@ module Bundler
       Issue.new.run
     end
 
+    desc "pristine", "Restores installed gems to pristine condition from files located in the gem cache. Gem installed from a git repository will be issued `git checkout --force`."
+    def pristine
+      require "bundler/cli/pristine"
+      Pristine.new.run
+    end
+
     if Bundler.feature_flag.plugins?
       require "bundler/cli/plugin"
       desc "plugin SUBCOMMAND ...ARGS", "manage the bundler plugins"
diff --git a/lib/bundler/cli/pristine.rb b/lib/bundler/cli/pristine.rb
new file mode 100644
index 0000000000..539e38f2ff
--- /dev/null
+++ b/lib/bundler/cli/pristine.rb
@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+require "bundler/cli/common"
+
+module Bundler
+  class CLI::Pristine
+    def run
+
+      ::Bundler.load.specs.each do |spec|
+
+        gem_name = "#{spec.name} (#{spec.version}#{spec.git_version})"
+
+        if spec.source.is_a?(Source::Path)
+          ::Bundler.ui.warn("Cannot pristine #{gem_name} Gem is sourced from path.")
+          next
+        end
+
+        if spec.source.is_a?(Source::Rubygems)
+          cached_gem = spec.cache_file
+          unless File.exists?(cached_gem)
+            ::Bundler.ui.error("Failed to pristine #{gem_name}. Cached gem #{cached_gem} does not exist.")
+            next
+          end
+
+          installer = Gem::Installer.at(cached_gem,
+                                        :wrappers => true,
+                                        :force => true,
+                                        :install_dir => spec.base_dir,
+                                        :build_args => spec.build_args)
+          installer.install
+
+        elsif spec.source.is_a?(Source::Git)
+
+        end
+
+
+      end
+    end
+  end
+end