session-ios/SignalServiceKit/src/Account/CreatePreKeysOperation.swift

//
//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
//

import Foundation
import PromiseKit

@objc(SSKCreatePreKeysOperation)
public class CreatePreKeysOperation: OWSOperation {
    private var accountManager: AccountManager {
        return AccountManager.shared
    }
    private var primaryStorage: OWSPrimaryStorage {
        return OWSPrimaryStorage.shared()
    }

    private var identityKeyManager: OWSIdentityManager {
        return OWSIdentityManager.shared()
    }

    public override func run() {
        Logger.debug("")

        if self.identityKeyManager.identityKeyPair() == nil {
            self.identityKeyManager.generateNewIdentityKey()
        }
        let identityKey: Data = self.identityKeyManager.identityKeyPair()!.publicKey
        let signedPreKeyRecord: SignedPreKeyRecord = self.primaryStorage.generateRandomSignedRecord()
        let preKeyRecords: [PreKeyRecord] = self.primaryStorage.generatePreKeyRecords()

        firstly {
            self.primaryStorage.storeSignedPreKey(signedPreKeyRecord.id, signedPreKeyRecord: signedPreKeyRecord)
            self.primaryStorage.storePreKeyRecords(preKeyRecords)

            return self.accountManager.setPreKeys(identityKey: identityKey, signedPreKeyRecord: signedPreKeyRecord, preKeyRecords: preKeyRecords)
            }.then { () -> Void in
                signedPreKeyRecord.markAsAcceptedByService()
                self.primaryStorage.setCurrentSignedPrekeyId(signedPreKeyRecord.id)
            }.then { () -> Void in
                Logger.debug("done")
                self.reportSuccess()
            }.catch { error in
                self.reportError(error)
            }.retainUntilComplete()
    }
}