pEpForiOS/UI/Settings/Setting/AccountSettings/ViewModel/AccountSettingsViewModel.swift
author Dirk Zimmermann <dz@pep.security>
Thu, 25 Apr 2019 16:25:40 +0200
branchIOS-1542
changeset 8374 0bfb3112a735
parent 8373 ecdd11870f06
child 8375 81ef9257d9b5
permissions -rw-r--r--
IOS-1542 Try do deduce original account password.
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@8373
    55
        if let server = account.imapServer {
dz@8374
    56
            self.originalPassword = server.credentials.password
dz@8373
    57
            self.imapServer = ServerViewModel(
dz@8373
    58
                address: server.address,
dz@8373
    59
                port: "\(server.port)",
dz@8373
    60
                transport: server.transport?.asString())
dz@8373
    61
        } else {
dz@8373
    62
            self.imapServer = ServerViewModel()
dz@8373
    63
        }
dz@8373
    64
dz@8354
    65
        if let server = account.smtpServer {
dz@8374
    66
            self.originalPassword = self.originalPassword ?? server.credentials.password
dz@8354
    67
            self.smtpServer = ServerViewModel(
dz@8354
    68
                address: server.address,
dz@8354
    69
                port: "\(server.port)",
dz@8354
    70
                transport: server.transport?.asString())
dz@8354
    71
        } else {
dz@8354
    72
            self.smtpServer = ServerViewModel()
dz@8354
    73
        }
andreas@3697
    74
    }
andreas@3697
    75
dz@8356
    76
    private(set) var email: String
dz@8354
    77
dz@8353
    78
    /// - Note: The email model is based on the assumption that imap.loginName == smtp.loginName
dz@8356
    79
    private(set) var loginName: String
andreas@3697
    80
dz@8356
    81
    private(set) var name: String
andreas@3697
    82
dz@8356
    83
    private(set) var smtpServer: ServerViewModel
andreas@3697
    84
dz@8356
    85
    private(set) var imapServer: ServerViewModel
andreas@3697
    86
andreas@3697
    87
    weak var delegate: AccountVerificationResultDelegate?
andreas@3697
    88
dz@8357
    89
    /// Holding both the data of the current account in verification,
dz@8357
    90
    /// and also the implementation of the verification.
dz@8357
    91
    private var verifiableAccount: VerifiableAccountProtocol?
dz@8357
    92
dz@8374
    93
    private var originalPassword: String?
dz@8374
    94
dz@8354
    95
    // Currently we assume imap and smtp servers exist already (update).
dz@8354
    96
    // If we run into problems here modify to updateOrCreate.
andreas@3697
    97
    func update(loginName: String, name: String, password: String? = nil, imap: ServerViewModel,
andreas@3697
    98
                smtp: ServerViewModel) {
dz@8358
    99
        var theVerifier = verifiableAccount ?? VerifiableAccount()
dz@8358
   100
        verifiableAccount = theVerifier
dz@8358
   101
dz@8358
   102
        theVerifier.address = email
dz@8358
   103
        theVerifier.userName = name
dz@8361
   104
dz@8374
   105
        theVerifier.password = originalPassword
dz@8374
   106
        if password != nil {
dz@8374
   107
            theVerifier.password = password
dz@8374
   108
        }
dz@8361
   109
dz@8358
   110
        if loginName != email {
dz@8358
   111
            theVerifier.loginName = loginName
dz@8358
   112
        }
dz@8358
   113
dz@8358
   114
        if isOAuth2 {
dz@8358
   115
            // TODO: Set correct auth method, etc.
dz@8358
   116
        }
dz@8358
   117
dz@8360
   118
        // IMAP
dz@8358
   119
        theVerifier.serverIMAP = imap.address
dz@8358
   120
        if let portString = imap.port, let port = UInt16(portString) {
dz@8358
   121
            theVerifier.portIMAP = port
dz@8358
   122
        }
dz@8360
   123
        if let transport = Server.Transport(fromString: imap.transport) {
dz@8360
   124
            theVerifier.transportIMAP = ConnectionTransport.init(transport: transport)
dz@8360
   125
        }
dz@8360
   126
dz@8360
   127
        // SMTP
dz@8360
   128
        theVerifier.serverSMTP = smtp.address
dz@8360
   129
        if let portString = smtp.port, let port = UInt16(portString) {
dz@8360
   130
            theVerifier.portSMTP = port
dz@8360
   131
        }
dz@8360
   132
        if let transport = Server.Transport(fromString: smtp.transport) {
dz@8360
   133
            theVerifier.transportSMTP = ConnectionTransport.init(transport: transport)
dz@8360
   134
        }
dz@8358
   135
dz@8363
   136
        theVerifier.verifiableAccountDelegate = self
dz@8361
   137
dz@8361
   138
        do {
dz@8361
   139
            try theVerifier.verify()
dz@8361
   140
        } catch {
dz@8361
   141
            delegate?.didVerify(result: .noImapConnectData, accountInput: theVerifier)
dz@8361
   142
        }
andreas@3697
   143
    }
andreas@3697
   144
andreas@3697
   145
    func sectionIsValid(section: Int) -> Bool {
borja@6875
   146
        return section >= 0 && section < headers.count
andreas@3697
   147
    }
andreas@3697
   148
andreas@3697
   149
    var count: Int {
andreas@3697
   150
        get {
andreas@3697
   151
            return headers.count
andreas@3697
   152
        }
andreas@3697
   153
    }
andreas@3697
   154
andreas@3697
   155
    subscript(section: Int) -> String {
andreas@3697
   156
        get {
andreas@3697
   157
            assert(sectionIsValid(section: section), "Section out of range")
andreas@3697
   158
            return headers[section]
andreas@3697
   159
        }
andreas@3697
   160
    }
andreas@3697
   161
andreas@3697
   162
    private func server(from viewModel:ServerViewModel, serverType:Server.ServerType,
andreas@3697
   163
                        loginName: String, password: String?, key: String? = nil) -> Server? {
andreas@3697
   164
        guard let viewModelPort = viewModel.port,
andreas@3697
   165
            let port = UInt16(viewModelPort),
andreas@3697
   166
            let address = viewModel.address else {
dz@7517
   167
                Logger.frontendLogger.errorAndCrash("viewModel misses required data.")
andreas@3697
   168
                return nil
andreas@3697
   169
        }
andreas@5601
   170
        let transport = Server.Transport(fromString: viewModel.transport)
andreas@3697
   171
andreas@3697
   172
        let credentials = ServerCredentials.create(loginName: loginName, key: key)
andreas@3697
   173
        if password != nil && password != "" {
andreas@3697
   174
            credentials.password = password
andreas@3697
   175
        }
andreas@3697
   176
andreas@3697
   177
        let server = Server.create(serverType: serverType, port: port, address: address,
andreas@3697
   178
                                   transport: transport, credentials: credentials)
andreas@3697
   179
andreas@3697
   180
        return server
andreas@3697
   181
    }
andreas@3697
   182
dirk@4052
   183
    func updateToken(accessToken: OAuth2AccessTokenProtocol) {
dz@8354
   184
        // TODO: What to do here? When does this get called?
dz@8354
   185
        /*
dirk@4053
   186
        guard let imapServer = account.imapServer,
dirk@4053
   187
            let smtpServer = account.smtpServer else {
dirk@4053
   188
                return
dirk@4052
   189
        }
dirk@4053
   190
        let password = accessToken.persistBase64Encoded()
dirk@4053
   191
        imapServer.credentials.password = password
dirk@4053
   192
        smtpServer.credentials.password = password
dz@8354
   193
         */
dirk@4052
   194
    }
andreas@3697
   195
}
andreas@3697
   196
andreas@5601
   197
// MARK: - AccountVerificationServiceDelegate
andreas@5601
   198
andreas@3697
   199
extension AccountSettingsViewModel: AccountVerificationServiceDelegate {
dz@7999
   200
    public func verified(account: Account,
andreas@5601
   201
                  service: AccountVerificationServiceProtocol,
andreas@3697
   202
                  result: AccountVerificationResult) {
andreas@5601
   203
        if result == .ok {
dz@7914
   204
            MessageModelUtil.performAndWait {
andreas@5601
   205
                account.save()
andreas@5601
   206
            }
andreas@5601
   207
        }
andreas@5601
   208
        GCD.onMainWait { [weak self] in
andreas@5601
   209
            guard let me = self else {
dz@7517
   210
                Logger.frontendLogger.lostMySelf()
andreas@5601
   211
                return
andreas@5601
   212
            }
andreas@5601
   213
            me.delegate?.didVerify(result: result, accountInput: nil)
andreas@3697
   214
        }
andreas@3697
   215
    }
andreas@3697
   216
}
dz@8363
   217
dz@8363
   218
// MARK: - VerifiableAccountDelegate
dz@8363
   219
dz@8363
   220
extension AccountSettingsViewModel: VerifiableAccountDelegate {
dz@8363
   221
    public func didEndVerification(result: Result<Void, Error>) {
dz@8364
   222
        switch result {
dz@8364
   223
        case .success(()):
dz@8364
   224
            do {
dz@8364
   225
                try verifiableAccount?.save()
dz@8364
   226
            } catch {
dz@8364
   227
                Logger.frontendLogger.log(error: error)
dz@8364
   228
                Logger.frontendLogger.errorAndCrash("Unexpected error on saving the account")
dz@8364
   229
            }
dz@8364
   230
        case .failure(let error):
dz@8364
   231
            if let imapError = error as? ImapSyncError {
dz@8364
   232
                delegate?.didVerify(
dz@8364
   233
                    result: .imapError(imapError), accountInput: verifiableAccount)
dz@8364
   234
            } else if let smtpError = error as? SmtpSendError {
dz@8364
   235
                delegate?.didVerify(
dz@8364
   236
                    result: .smtpError(smtpError), accountInput: verifiableAccount)
dz@8364
   237
            } else {
dz@8364
   238
                Logger.frontendLogger.log(error: error)
dz@8364
   239
                Logger.frontendLogger.errorAndCrash("Unexpected error")
dz@8364
   240
            }
dz@8364
   241
        }
dz@8363
   242
    }
dz@8363
   243
}