safe-ios/Multisig/UI/Settings/OwnerKeyManagement/AddKeyAsNewOwner/AddOwnerFlowFromSettings.swift

//
//  AddOwnerFlowFromSettingsFactory.swift
//  Multisig
//
//  Created by Moaaz on 5/17/22.
//  Copyright © 2022 Gnosis Ltd. All rights reserved.
//

import Foundation
import UIKit

/// Flow for adding new owner to a safe.
///
/// Expected to be opened modally.
/// Expects that the new owner address is not known.
///
/// Screen sequence:
///
/// 1. Enter Address
/// 1.1. If address's name is unknown, then Enter Name
/// 2. Edit Confirmations (in superclass)
/// 3. Review (in superclass)
/// 4. Success (in superclass)
class AddOwnerFlowFromSettings: AddOwnerFlow {
    var newAddressName: String?

    init(safe: Safe, completion: @escaping (_ success: Bool) -> Void) {
        super.init(newOwner: nil, safe: safe, completion: completion)
    }

    override func start() {
        enterAddressViewController()
    }

    func enterAddressViewController() {
        let viewController = addOwnerFactory.enterOwnerAddress(
            safe: safe,
            stepNumber: 1,
            maxSteps: 3,
            title: "Add owner",
            trackingEvent: .addOwnerSelectAddress
        ) { [unowned self] address, resolvedName  in
            newOwner = address

            if let resolvedName = resolvedName {
                self.newAddressName = resolvedName
                confirmations()
            } else {
                enterOwnerNameViewController()
            }
        }

        show(viewController)
    }

    func enterOwnerNameViewController() {
        assert(newOwner != nil)
        let viewController = addOwnerFactory.enterOwnerName(
            safe: safe,
            address: newOwner!,
            stepNumber: 1,
            maxSteps: 3,
            title: "Add owner",
            trackingEvent: .addOwnerSpecifyName
        ) { [unowned self] name in
            newAddressName = name
            confirmations()
        }

        show(viewController)
    }

    func confirmations() {
        confirmations(stepNumber: 2, maxSteps: 3)
    }

    override func review(stepNumber: Int, maxSteps: Int) {
        assert(newConfirmations != nil)
        assert(newOwner != nil)
        let reviewVC = addOwnerFactory.review(
                safe: safe,
                newOwner: newOwner!,
                newThreshold: newConfirmations!,
                stepNumber: stepNumber,
                maxSteps: maxSteps,
                newAddressName: newAddressName
        ) { [unowned self] txDetails in
            transaction = txDetails
            success()
        }
        show(reviewVC)
    }

    override func success() {
        assert(transaction != nil)
        AddressBookEntry.addOrUpdate(newOwner!.checksummed, chain: safe.chain!, name: newAddressName!)
        super.success()
    }
}