Rework key deletion to accept an identifier
This commit is contained in:
parent
15f2489550
commit
c0f04bec77
3 changed files with 11 additions and 12 deletions
|
@ -20,8 +20,8 @@ public interface KeyManager<Key, KeyIdentifier> {
|
|||
*/
|
||||
public suspend fun addKey(key: Key, replace: Boolean = false): Result<Key, Throwable>
|
||||
|
||||
/** Removes [key] from the store. */
|
||||
public suspend fun removeKey(key: Key): Result<Key, Throwable>
|
||||
/** Finds a key for [identifier] in the store and deletes it. */
|
||||
public suspend fun removeKey(identifier: KeyIdentifier): Result<Unit, Throwable>
|
||||
|
||||
/**
|
||||
* Get a [Key] for the given [id]. The actual semantics of what [id] is are left to individual
|
||||
|
|
|
@ -17,6 +17,7 @@ import app.passwordstore.crypto.errors.KeyNotFoundException
|
|||
import app.passwordstore.crypto.errors.NoKeysAvailableException
|
||||
import app.passwordstore.util.coroutines.runSuspendCatching
|
||||
import com.github.michaelbull.result.Result
|
||||
import com.github.michaelbull.result.unwrap
|
||||
import java.io.File
|
||||
import javax.inject.Inject
|
||||
import kotlinx.coroutines.CoroutineDispatcher
|
||||
|
@ -71,17 +72,15 @@ constructor(
|
|||
}
|
||||
}
|
||||
|
||||
override suspend fun removeKey(key: PGPKey): Result<PGPKey, Throwable> =
|
||||
override suspend fun removeKey(identifier: GpgIdentifier): Result<Unit, Throwable> =
|
||||
withContext(dispatcher) {
|
||||
runSuspendCatching {
|
||||
if (!keyDirExists()) throw KeyDirectoryUnavailableException
|
||||
if (tryParseKeyring(key) == null) throw InvalidKeyException
|
||||
val key = getKeyById(identifier).unwrap()
|
||||
val keyFile = File(keyDir, "${tryGetId(key)}.$KEY_EXTENSION")
|
||||
if (keyFile.exists()) {
|
||||
if (!keyFile.delete()) throw KeyDeletionFailedException
|
||||
}
|
||||
|
||||
key
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
@ -2,6 +2,7 @@ package app.passwordstore.crypto
|
|||
|
||||
import app.passwordstore.crypto.GpgIdentifier.KeyId
|
||||
import app.passwordstore.crypto.GpgIdentifier.UserId
|
||||
import app.passwordstore.crypto.KeyUtils.tryGetId
|
||||
import app.passwordstore.crypto.errors.KeyAlreadyExistsException
|
||||
import app.passwordstore.crypto.errors.KeyNotFoundException
|
||||
import app.passwordstore.crypto.errors.NoKeysAvailableException
|
||||
|
@ -93,13 +94,12 @@ class PGPKeyManagerTest {
|
|||
// Add key using KeyManager
|
||||
keyManager.addKey(secretKey).unwrap()
|
||||
|
||||
// Check if the key id returned is correct
|
||||
val keyId = keyManager.getKeyId(keyManager.removeKey(secretKey).unwrap())
|
||||
assertEquals(KeyId(CryptoConstants.KEY_ID), keyId)
|
||||
// Remove key
|
||||
keyManager.removeKey(tryGetId(secretKey)!!).unwrap()
|
||||
|
||||
// Check if the keys directory have 0 files
|
||||
val keysDir = File(filesDir, PGPKeyManager.KEY_DIR_NAME)
|
||||
assertEquals(0, keysDir.list()?.size)
|
||||
// Check that no keys remain
|
||||
val keys = keyManager.getAllKeys().unwrap()
|
||||
assertEquals(0, keys.size)
|
||||
}
|
||||
|
||||
@Test
|
||||
|
|
Loading…
Reference in a new issue