Merge pull request #5182 from clintoncwolfe/add-ssh-key-replacement-for-solaris

guests/solaris: Add insert/remove public key guest capability for solaris

plugins/guests/solaris/cap/insert_public_key.rb

@@ -0,0 +1,22 @@
+require "vagrant/util/shell_quote"
+
+module VagrantPlugins
+  module GuestSolaris
+    module Cap
+      class InsertPublicKey
+        def self.insert_public_key(machine, contents)
+          # TODO: Code is identical to linux/cap/insert_public_key
+          contents = contents.chomp
+          contents = Vagrant::Util::ShellQuote.escape(contents, "'")
+
+          machine.communicate.tap do |comm|
+            comm.execute("mkdir -p ~/.ssh")
+            comm.execute("chmod 0700 ~/.ssh")
+            comm.execute("printf '#{contents}\\n' >> ~/.ssh/authorized_keys")
+            comm.execute("chmod 0600 ~/.ssh/authorized_keys")
+          end
+        end
+      end
+    end
+  end
+end

plugins/guests/solaris/cap/remove_public_key.rb

@@ -0,0 +1,22 @@
+require "vagrant/util/shell_quote"
+
+module VagrantPlugins
+  module GuestSolaris
+    module Cap
+      class RemovePublicKey
+        def self.remove_public_key(machine, contents)
+          # TODO: code is identical to linux/cap/remove_public_key
+          contents = contents.chomp
+          contents = Vagrant::Util::ShellQuote.escape(contents, "'")
+
+          machine.communicate.tap do |comm|
+            if comm.test("test -f ~/.ssh/authorized_keys")
+              comm.execute(
+                "sed -i '/^.*#{contents}.*$/d' ~/.ssh/authorized_keys")
+            end
+          end
+        end
+      end
+    end
+  end
+end

plugins/guests/solaris/plugin.rb

@@ -26,6 +26,11 @@ module VagrantPlugins
         Cap::ConfigureNetworks
       end
 
+      guest_capability("solaris", "insert_public_key") do
+        require_relative "cap/insert_public_key"
+        Cap::InsertPublicKey
+      end
+
       guest_capability("solaris", "halt") do
         require_relative "cap/halt"
         Cap::Halt
@@ -36,6 +41,11 @@ module VagrantPlugins
         Cap::MountVirtualBoxSharedFolder
       end
 
+      guest_capability("solaris", "remove_public_key") do
+        require_relative "cap/remove_public_key"
+        Cap::RemovePublicKey
+      end
+
       guest_capability("solaris", "rsync_installed") do
         require_relative "cap/rsync"
         Cap::RSync