pEpForiOS/UI/Settings/Setting/AccountSettings/ViewModel/AccountSettingsViewModel.swift
author Dirk Zimmermann <dz@pep.security>
Wed, 24 Apr 2019 12:43:18 +0200
branchIOS-1542
changeset 8364 0f5636c78961
parent 8363 15ce086afa27
child 8365 3ee18a0d2630
permissions -rw-r--r--
IOS-1542 Implementing VerifiableAccountDelegate.
andreas@3697
     1
//
andreas@3697
     2
//  AccountSettingsViewModel.swift
andreas@3697
     3
//  pEpForiOS
andreas@3697
     4
//
andreas@3697
     5
//  Created by Xavier Algarra on 13/06/2017.
andreas@3697
     6
//  Copyright © 2017 p≡p Security S.A. All rights reserved.
andreas@3697
     7
//
andreas@3697
     8
andreas@3697
     9
import Foundation
andreas@3697
    10
import MessageModel
xavier@7644
    11
import pEpIOSToolbox
andreas@3697
    12
dz@8358
    13
import PantomimeFramework
dz@8358
    14
andreas@3697
    15
public class AccountSettingsViewModel {
andreas@3697
    16
    public struct ServerViewModel {
andreas@3697
    17
        var address: String?
andreas@3697
    18
        var port: String?
andreas@3697
    19
        var transport: String?
andreas@3697
    20
    }
andreas@3697
    21
andreas@3697
    22
    public struct SecurityViewModel {
andreas@3697
    23
        var options = Server.Transport.toArray()
andreas@3697
    24
        var size : Int {
andreas@3697
    25
            get {
andreas@3697
    26
                return options.count
andreas@3697
    27
            }
andreas@3697
    28
        }
andreas@3697
    29
andreas@3697
    30
        subscript(option: Int) -> String {
andreas@3697
    31
            get {
andreas@3697
    32
                return options[option].asString()
andreas@3697
    33
            }
andreas@3697
    34
        }
andreas@3697
    35
    }
andreas@3697
    36
andreas@3697
    37
    private let headers = [
andreas@3697
    38
        NSLocalizedString("Account", comment: "Account settings"),
andreas@3697
    39
        NSLocalizedString("IMAP Settings", comment: "Account settings title IMAP"),
andreas@3697
    40
        NSLocalizedString("SMTP Settings", comment: "Account settings title SMTP")
andreas@3697
    41
    ]
andreas@3697
    42
    private var controlWord = "noRealPassword"
andreas@3697
    43
andreas@3697
    44
    public let svm = SecurityViewModel()
dirk@4019
    45
    public let isOAuth2: Bool
andreas@3697
    46
andreas@3697
    47
    public init(account: Account) {
dz@8354
    48
        // We are using a copy of the data here.
dz@8354
    49
        // The outside world must not know changed settings until they have been verified.
dirk@4019
    50
        isOAuth2 = account.server(with: .imap)?.authMethod == AuthMethod.saslXoauth2.rawValue
dz@8354
    51
        self.email = account.user.address
dz@8354
    52
        self.loginName = account.server(with: .imap)?.credentials.loginName ?? ""
dz@8354
    53
        self.name = account.user.userName ?? ""
andreas@3697
    54
dz@8354
    55
        if let server = account.smtpServer {
dz@8354
    56
            self.smtpServer = ServerViewModel(
dz@8354
    57
                address: server.address,
dz@8354
    58
                port: "\(server.port)",
dz@8354
    59
                transport: server.transport?.asString())
dz@8354
    60
        } else {
dz@8354
    61
            self.smtpServer = ServerViewModel()
dz@8354
    62
        }
dz@8354
    63
dz@8354
    64
        if let server = account.imapServer {
dz@8354
    65
            self.imapServer = ServerViewModel(
dz@8354
    66
                address: server.address,
dz@8354
    67
                port: "\(server.port)",
dz@8354
    68
                transport: server.transport?.asString())
dz@8354
    69
        } else {
dz@8354
    70
            self.imapServer = ServerViewModel()
andreas@3697
    71
        }
andreas@3697
    72
    }
andreas@3697
    73
dz@8356
    74
    private(set) var email: String
dz@8354
    75
dz@8353
    76
    /// - Note: The email model is based on the assumption that imap.loginName == smtp.loginName
dz@8356
    77
    private(set) var loginName: String
andreas@3697
    78
dz@8356
    79
    private(set) var name: String
andreas@3697
    80
dz@8356
    81
    private(set) var smtpServer: ServerViewModel
andreas@3697
    82
dz@8356
    83
    private(set) var imapServer: ServerViewModel
andreas@3697
    84
agp@8142
    85
    var verificationService: VerificationService?
andreas@3697
    86
    weak var delegate: AccountVerificationResultDelegate?
andreas@3697
    87
dz@8357
    88
    /// Holding both the data of the current account in verification,
dz@8357
    89
    /// and also the implementation of the verification.
dz@8357
    90
    private var verifiableAccount: VerifiableAccountProtocol?
dz@8357
    91
dz@8354
    92
    // Currently we assume imap and smtp servers exist already (update).
dz@8354
    93
    // If we run into problems here modify to updateOrCreate.
andreas@3697
    94
    func update(loginName: String, name: String, password: String? = nil, imap: ServerViewModel,
andreas@3697
    95
                smtp: ServerViewModel) {
dz@8358
    96
        var theVerifier = verifiableAccount ?? VerifiableAccount()
dz@8358
    97
        verifiableAccount = theVerifier
dz@8358
    98
dz@8358
    99
        theVerifier.address = email
dz@8358
   100
        theVerifier.userName = name
dz@8361
   101
dz@8361
   102
        // TODO: How to handle if the password got changed or not?
dz@8358
   103
        theVerifier.password = password
dz@8361
   104
dz@8358
   105
        if loginName != email {
dz@8358
   106
            theVerifier.loginName = loginName
dz@8358
   107
        }
dz@8358
   108
dz@8358
   109
        if isOAuth2 {
dz@8358
   110
            // TODO: Set correct auth method, etc.
dz@8358
   111
        }
dz@8358
   112
dz@8360
   113
        // IMAP
dz@8358
   114
        theVerifier.serverIMAP = imap.address
dz@8358
   115
        if let portString = imap.port, let port = UInt16(portString) {
dz@8358
   116
            theVerifier.portIMAP = port
dz@8358
   117
        }
dz@8360
   118
        if let transport = Server.Transport(fromString: imap.transport) {
dz@8360
   119
            theVerifier.transportIMAP = ConnectionTransport.init(transport: transport)
dz@8360
   120
        }
dz@8360
   121
dz@8360
   122
        // SMTP
dz@8360
   123
        theVerifier.serverSMTP = smtp.address
dz@8360
   124
        if let portString = smtp.port, let port = UInt16(portString) {
dz@8360
   125
            theVerifier.portSMTP = port
dz@8360
   126
        }
dz@8360
   127
        if let transport = Server.Transport(fromString: smtp.transport) {
dz@8360
   128
            theVerifier.transportSMTP = ConnectionTransport.init(transport: transport)
dz@8360
   129
        }
dz@8358
   130
dz@8363
   131
        theVerifier.verifiableAccountDelegate = self
dz@8361
   132
dz@8361
   133
        do {
dz@8361
   134
            try theVerifier.verify()
dz@8361
   135
        } catch {
dz@8361
   136
            delegate?.didVerify(result: .noImapConnectData, accountInput: theVerifier)
dz@8361
   137
        }
andreas@3697
   138
    }
andreas@3697
   139
andreas@3697
   140
    func sectionIsValid(section: Int) -> Bool {
borja@6875
   141
        return section >= 0 && section < headers.count
andreas@3697
   142
    }
andreas@3697
   143
andreas@3697
   144
    var count: Int {
andreas@3697
   145
        get {
andreas@3697
   146
            return headers.count
andreas@3697
   147
        }
andreas@3697
   148
    }
andreas@3697
   149
andreas@3697
   150
    subscript(section: Int) -> String {
andreas@3697
   151
        get {
andreas@3697
   152
            assert(sectionIsValid(section: section), "Section out of range")
andreas@3697
   153
            return headers[section]
andreas@3697
   154
        }
andreas@3697
   155
    }
andreas@3697
   156
andreas@3697
   157
    private func server(from viewModel:ServerViewModel, serverType:Server.ServerType,
andreas@3697
   158
                        loginName: String, password: String?, key: String? = nil) -> Server? {
andreas@3697
   159
        guard let viewModelPort = viewModel.port,
andreas@3697
   160
            let port = UInt16(viewModelPort),
andreas@3697
   161
            let address = viewModel.address else {
dz@7517
   162
                Logger.frontendLogger.errorAndCrash("viewModel misses required data.")
andreas@3697
   163
                return nil
andreas@3697
   164
        }
andreas@5601
   165
        let transport = Server.Transport(fromString: viewModel.transport)
andreas@3697
   166
andreas@3697
   167
        let credentials = ServerCredentials.create(loginName: loginName, key: key)
andreas@3697
   168
        if password != nil && password != "" {
andreas@3697
   169
            credentials.password = password
andreas@3697
   170
        }
andreas@3697
   171
andreas@3697
   172
        let server = Server.create(serverType: serverType, port: port, address: address,
andreas@3697
   173
                                   transport: transport, credentials: credentials)
andreas@3697
   174
andreas@3697
   175
        return server
andreas@3697
   176
    }
andreas@3697
   177
dirk@4052
   178
    func updateToken(accessToken: OAuth2AccessTokenProtocol) {
dz@8354
   179
        // TODO: What to do here? When does this get called?
dz@8354
   180
        /*
dirk@4053
   181
        guard let imapServer = account.imapServer,
dirk@4053
   182
            let smtpServer = account.smtpServer else {
dirk@4053
   183
                return
dirk@4052
   184
        }
dirk@4053
   185
        let password = accessToken.persistBase64Encoded()
dirk@4053
   186
        imapServer.credentials.password = password
dirk@4053
   187
        smtpServer.credentials.password = password
dz@8354
   188
         */
dirk@4052
   189
    }
andreas@3697
   190
}
andreas@3697
   191
andreas@5601
   192
// MARK: - AccountVerificationServiceDelegate
andreas@5601
   193
andreas@3697
   194
extension AccountSettingsViewModel: AccountVerificationServiceDelegate {
dz@7999
   195
    public func verified(account: Account,
andreas@5601
   196
                  service: AccountVerificationServiceProtocol,
andreas@3697
   197
                  result: AccountVerificationResult) {
andreas@5601
   198
        if result == .ok {
dz@7914
   199
            MessageModelUtil.performAndWait {
andreas@5601
   200
                account.save()
andreas@5601
   201
            }
andreas@5601
   202
        }
andreas@5601
   203
        GCD.onMainWait { [weak self] in
andreas@5601
   204
            guard let me = self else {
dz@7517
   205
                Logger.frontendLogger.lostMySelf()
andreas@5601
   206
                return
andreas@5601
   207
            }
andreas@5601
   208
            me.delegate?.didVerify(result: result, accountInput: nil)
andreas@3697
   209
        }
andreas@3697
   210
    }
andreas@3697
   211
}
dz@8363
   212
dz@8363
   213
// MARK: - VerifiableAccountDelegate
dz@8363
   214
dz@8363
   215
extension AccountSettingsViewModel: VerifiableAccountDelegate {
dz@8363
   216
    public func didEndVerification(result: Result<Void, Error>) {
dz@8364
   217
        switch result {
dz@8364
   218
        case .success(()):
dz@8364
   219
            do {
dz@8364
   220
                try verifiableAccount?.save()
dz@8364
   221
            } catch {
dz@8364
   222
                Logger.frontendLogger.log(error: error)
dz@8364
   223
                Logger.frontendLogger.errorAndCrash("Unexpected error on saving the account")
dz@8364
   224
            }
dz@8364
   225
        case .failure(let error):
dz@8364
   226
            if let imapError = error as? ImapSyncError {
dz@8364
   227
                delegate?.didVerify(
dz@8364
   228
                    result: .imapError(imapError), accountInput: verifiableAccount)
dz@8364
   229
            } else if let smtpError = error as? SmtpSendError {
dz@8364
   230
                delegate?.didVerify(
dz@8364
   231
                    result: .smtpError(smtpError), accountInput: verifiableAccount)
dz@8364
   232
            } else {
dz@8364
   233
                Logger.frontendLogger.log(error: error)
dz@8364
   234
                Logger.frontendLogger.errorAndCrash("Unexpected error")
dz@8364
   235
            }
dz@8364
   236
        }
dz@8363
   237
    }
dz@8363
   238
}