40 lines
1.5 KiB
Swift
40 lines
1.5 KiB
Swift
//
|
|
// Copyright 2020 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import LibSignalClient
|
|
|
|
extension SSKPreKeyStore: LibSignalClient.PreKeyStore {
|
|
enum Error: Swift.Error {
|
|
case noPreKeyWithId(UInt32)
|
|
}
|
|
|
|
public func loadPreKey(id: UInt32, context: StoreContext) throws -> LibSignalClient.PreKeyRecord {
|
|
guard let preKey = self.loadPreKey(Int32(bitPattern: id), transaction: context.asTransaction) else {
|
|
throw Error.noPreKeyWithId(id)
|
|
}
|
|
let keyPair = preKey.keyPair.identityKeyPair
|
|
return try .init(id: UInt32(bitPattern: preKey.id),
|
|
publicKey: keyPair.publicKey,
|
|
privateKey: keyPair.privateKey)
|
|
}
|
|
|
|
public func storePreKey(_ record: LibSignalClient.PreKeyRecord, id: UInt32, context: StoreContext) throws {
|
|
let keyPair = IdentityKeyPair(
|
|
publicKey: try record.publicKey(),
|
|
privateKey: try record.privateKey()
|
|
)
|
|
self.storePreKey(Int32(bitPattern: id),
|
|
preKeyRecord: SignalServiceKit.PreKeyRecord(id: Int32(bitPattern: id),
|
|
keyPair: ECKeyPair(keyPair),
|
|
createdAt: Date()),
|
|
transaction: context.asTransaction)
|
|
}
|
|
|
|
public func removePreKey(id: UInt32, context: StoreContext) throws {
|
|
self.removePreKey(Int32(bitPattern: id), transaction: context.asTransaction)
|
|
}
|
|
|
|
}
|