From 62902ca80b2a0883d810518de8b1838f29a682eb Mon Sep 17 00:00:00 2001
From: Harsh Shandilya <me@msfjarvis.dev>
Date: Sun, 24 Apr 2022 21:25:34 +0530
Subject: Reorganize crypto-common code and fix a couple minor bugs (#1868)

---
 .../dev/msfjarvis/aps/crypto/PGPKeyManager.kt      | 28 +++++++++++++---------
 1 file changed, 17 insertions(+), 11 deletions(-)

(limited to 'crypto-pgpainless/src/main')

diff --git a/crypto-pgpainless/src/main/kotlin/dev/msfjarvis/aps/crypto/PGPKeyManager.kt b/crypto-pgpainless/src/main/kotlin/dev/msfjarvis/aps/crypto/PGPKeyManager.kt
index 8c54d516..3cc68ab3 100644
--- a/crypto-pgpainless/src/main/kotlin/dev/msfjarvis/aps/crypto/PGPKeyManager.kt
+++ b/crypto-pgpainless/src/main/kotlin/dev/msfjarvis/aps/crypto/PGPKeyManager.kt
@@ -10,6 +10,12 @@ import androidx.annotation.VisibleForTesting
 import com.github.michaelbull.result.Result
 import dev.msfjarvis.aps.crypto.KeyUtils.tryGetId
 import dev.msfjarvis.aps.crypto.KeyUtils.tryParseKeyring
+import dev.msfjarvis.aps.crypto.errors.InvalidKeyException
+import dev.msfjarvis.aps.crypto.errors.KeyAlreadyExistsException
+import dev.msfjarvis.aps.crypto.errors.KeyDeletionFailedException
+import dev.msfjarvis.aps.crypto.errors.KeyDirectoryUnavailableException
+import dev.msfjarvis.aps.crypto.errors.KeyNotFoundException
+import dev.msfjarvis.aps.crypto.errors.NoKeysAvailableException
 import dev.msfjarvis.aps.util.coroutines.runSuspendCatching
 import java.io.File
 import javax.inject.Inject
@@ -29,16 +35,16 @@ constructor(
   override suspend fun addKey(key: PGPKey, replace: Boolean): Result<PGPKey, Throwable> =
     withContext(dispatcher) {
       runSuspendCatching {
-        if (!keyDirExists()) throw KeyManagerException.KeyDirectoryUnavailableException
-        if (tryParseKeyring(key) == null) throw KeyManagerException.InvalidKeyException
+        if (!keyDirExists()) throw KeyDirectoryUnavailableException
+        if (tryParseKeyring(key) == null) throw InvalidKeyException
         val keyFile = File(keyDir, "${tryGetId(key)}.$KEY_EXTENSION")
         if (keyFile.exists()) {
           // Check for replace flag first and if it is false, throw an error
           if (!replace)
-            throw KeyManagerException.KeyAlreadyExistsException(
+            throw KeyAlreadyExistsException(
               tryGetId(key)?.toString() ?: "Failed to retrieve key ID"
             )
-          if (!keyFile.delete()) throw KeyManagerException.KeyDeletionFailedException
+          if (!keyFile.delete()) throw KeyDeletionFailedException
         }
 
         keyFile.writeBytes(key.contents)
@@ -50,11 +56,11 @@ constructor(
   override suspend fun removeKey(key: PGPKey): Result<PGPKey, Throwable> =
     withContext(dispatcher) {
       runSuspendCatching {
-        if (!keyDirExists()) throw KeyManagerException.KeyDirectoryUnavailableException
-        if (tryParseKeyring(key) == null) throw KeyManagerException.InvalidKeyException
+        if (!keyDirExists()) throw KeyDirectoryUnavailableException
+        if (tryParseKeyring(key) == null) throw InvalidKeyException
         val keyFile = File(keyDir, "${tryGetId(key)}.$KEY_EXTENSION")
         if (keyFile.exists()) {
-          if (!keyFile.delete()) throw KeyManagerException.KeyDeletionFailedException
+          if (!keyFile.delete()) throw KeyDeletionFailedException
         }
 
         key
@@ -64,9 +70,9 @@ constructor(
   override suspend fun getKeyById(id: GpgIdentifier): Result<PGPKey, Throwable> =
     withContext(dispatcher) {
       runSuspendCatching {
-        if (!keyDirExists()) throw KeyManagerException.KeyDirectoryUnavailableException
+        if (!keyDirExists()) throw KeyDirectoryUnavailableException
         val keyFiles = keyDir.listFiles()
-        if (keyFiles.isNullOrEmpty()) throw KeyManagerException.NoKeysAvailableException
+        if (keyFiles.isNullOrEmpty()) throw NoKeysAvailableException
         val keys = keyFiles.map { file -> PGPKey(file.readBytes()) }
 
         val matchResult =
@@ -92,14 +98,14 @@ constructor(
           return@runSuspendCatching matchResult
         }
 
-        throw KeyManagerException.KeyNotFoundException("$id")
+        throw KeyNotFoundException("$id")
       }
     }
 
   override suspend fun getAllKeys(): Result<List<PGPKey>, Throwable> =
     withContext(dispatcher) {
       runSuspendCatching {
-        if (!keyDirExists()) throw KeyManagerException.KeyDirectoryUnavailableException
+        if (!keyDirExists()) throw KeyDirectoryUnavailableException
         val keyFiles = keyDir.listFiles()
         if (keyFiles.isNullOrEmpty()) return@runSuspendCatching emptyList()
         keyFiles.map { keyFile -> PGPKey(keyFile.readBytes()) }.toList()
-- 
cgit v1.2.3