save restored secret key to gpg or file

1 files changed, 26 insertions, 0 deletions

diff --git a/SecretKey.hs b/SecretKey.hs
new file mode 100644
index 0000000..45d9680
--- /dev/null
+++ b/SecretKey.hs
@@ -0,0 +1,26 @@
+{- Copyright 2016 Joey Hess <id@joeyh.name>
+ -
+ - Licensed under the GNU AGPL version 3 or higher.
+ -}
+
+module SecretKey where
+
+import Types
+import qualified Gpg
+import qualified Data.ByteString as B
+import System.IO
+import System.Posix.IO
+
+getSecretKey :: SecretKeySource -> IO SecretKey
+getSecretKey (GpgKey kid) = Gpg.getSecretKey kid
+getSecretKey (KeyFile f) = SecretKey <$> B.readFile f
+
+-- | Can throw exception if the secret key already exists.
+writeSecretKey :: SecretKeySource -> SecretKey -> IO ()
+writeSecretKey (GpgKey _) secretkey = Gpg.writeSecretKey secretkey
+writeSecretKey (KeyFile f) (SecretKey b) = do
+	fd <- openFd f WriteOnly (Just 0o666)
+		(defaultFileFlags { exclusive = True } )
+	h <- fdToHandle fd
+	B.hPut h b
+	hClose h