1 files changed, 8 insertions, 4 deletions

diff --git a/lib/KeyRing.hs b/lib/KeyRing.hs
index 44c3dce..0637044 100644
--- a/lib/KeyRing.hs
+++ b/lib/KeyRing.hs
@@ -841,8 +841,12 @@ subkeysForExport subspec (KeyData key _ _ subkeys) = do
                           sigtrusts
         in fmap fst v==Just True
 
-writePEM :: String -> String -> String
-writePEM typ dta = pem
+data PemType = PemPublicKey | PemPrivateKey | PemCertificate
+pemTypeString PemPublicKey  = "PUBLIC KEY"
+pemTypeString PemPrivateKey = "PRIVATE KEY"
+pemTypeString PemCertificate = "CERTIFICATE"
+writePEM :: PemType -> String -> String
+writePEM (pemTypeString -> typ) dta = pem
  where
     pem = unlines . concat $
             [ ["-----BEGIN " <> typ <> "-----"]
@@ -898,7 +902,7 @@ pemFromPacket Sec packet =
 #elif defined(VERSION_dataenc)
             dta = Base64.encode (L.unpack bs)
 #endif
-            output = writePEM "RSA PRIVATE KEY" dta
+            output = writePEM PemPrivateKey dta
         Just output
      algo -> Nothing
 pemFromPacket Pub packet =
@@ -912,7 +916,7 @@ pemFromPacket Pub packet =
 #elif defined(VERSION_dataenc)
             dta = Base64.encode (L.unpack bs)
 #endif
-            output = writePEM "PUBLIC KEY" dta
+            output = writePEM PemPublicKey dta
         Just output
      algo -> Nothing
 pemFromPacket AutoAccess p@(PublicKeyPacket {}) = pemFromPacket Pub p