[Vmdebootstrap-devel] [PATCH] Add option to install ssh key for root user

[Jan Gerber]

---
 vmdebootstrap | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/vmdebootstrap b/vmdebootstrap
index 126475f..8f32b95 100755
--- a/vmdebootstrap
+++ b/vmdebootstrap
@@ -79,6 +79,8 @@ class VmDebootstrap(cliapp.Application):
                              metavar='PASSWORD')
         self.settings.boolean(['lock-root-password'],
                               'lock root account so they cannot login?')
+        self.settings.string(['root-ssh-key'], 'Add PATH to root\'s ~/.ssh/authorized_keys',
+                             metavar='PATH')
         self.settings.string(['customize'],
                              'run SCRIPT after setting up system',
                              metavar='SCRIPT')
@@ -416,6 +418,13 @@ class VmDebootstrap(cliapp.Application):
         else:
             self.message('Give root an empty password')
             self.delete_password(rootdir, 'root')
+        if self.settings['root-ssh-key']:
+            self.message('Copy root ssh key')
+            authorized_keys = os.path.join(rootdir, 'root', '.ssh', 'authorized_keys')
+            dotssh = os.path.dirname(authorized_keys)
+            if not os.path.exists(dotssh):
+                os.mkdir(dotssh, 0700)
+            shutil.copy(self.settings['root-ssh-key'], authorized_keys)
 
     def create_users(self, rootdir):
         def create_user(user):
-- 
2.1.3