1 files changed, 26 insertions, 0 deletions

diff --git a/lib/Kiki.hs b/lib/Kiki.hs
index 575cf26..be99ed8 100644
--- a/lib/Kiki.hs
+++ b/lib/Kiki.hs
@@ -39,6 +39,15 @@ refresh root homepass = do
 
 data CommonArgsParsed = CommonArgsParsed { cap_homespec :: Maybe String, cap_passfd :: Maybe InputFile }
 
+streaminfo :: StreamInfo
+streaminfo = StreamInfo
+    { fill = KF_None
+    , spill = KF_None
+    , typ = KeyRingFile
+    , initializer = NoCreate
+    , access = AutoAccess
+    , transforms = []
+    }
 
 minimalOp :: CommonArgsParsed -> KeyRingOperation
 minimalOp cap = op
@@ -167,3 +176,20 @@ sshblobFromPacket k = blob
                 <$> optional (arg "--homedir")
                 <*> optional (FileDesc <$> read <$> arg "--passphrase-fd")
 
+replaceSshServerKeys root cmn = do
+    let homepass' = cmn { cap_homespec = fmap root (cap_homespec cmn) }
+        replaceSSH op = op { opFiles = files }
+            where
+                files = Map.adjust delssh HomeSec
+                        $ Map.adjust delssh HomePub
+                        $ Map.insert (ArgFile $ root "/etc/ssh/ssh_host_rsa_key") strm $ opFiles op
+                strm = streaminfo { typ = PEMFile, spill = KF_Match "ssh-server", access = Sec }
+                delssh strm = strm { transforms = DeleteSubkeyByUsage "ssh-server" : transforms strm
+                                   , fill = KF_All }
+    KikiResult r report <- runKeyRing $ minimalOp homepass'
+    case r of
+        KikiSuccess rt -> Kiki.refreshCache rt $ case root "" of
+                                                    "/" -> Nothing
+                                                    ""  -> Nothing
+                                                    pth -> Just pth
+        err            -> hPutStrLn stderr $ errorString err