pEpForiOS/UI/EmailDisplay/EmailListViewController.swift
author Xavier Algarra <xavier@pep-project.org>
Tue, 25 Oct 2016 12:35:39 +0200
branchIOS-230
changeset 870 dd67c28b7449
parent 867 89ccccb16970
child 935 16f7ad421c63
permissions -rw-r--r--
save draft message to the correct folder
dirk@30
     1
//
dirk@30
     2
//  EmailListViewController.swift
dirk@30
     3
//  pEpForiOS
dirk@30
     4
//
dirk@30
     5
//  Created by Dirk Zimmermann on 16/04/16.
dirk@30
     6
//  Copyright © 2016 p≡p Security S.A. All rights reserved.
dirk@30
     7
//
dirk@30
     8
dirk@30
     9
import Foundation
dirk@30
    10
import UIKit
dirk@31
    11
import CoreData
dirk@808
    12
dirk@810
    13
import MessageModel
dirk@810
    14
dirk@784
    15
fileprivate func < <T : Comparable>(lhs: T?, rhs: T?) -> Bool {
dirk@784
    16
  switch (lhs, rhs) {
dirk@784
    17
  case let (l?, r?):
dirk@784
    18
    return l < r
dirk@784
    19
  case (nil, _?):
dirk@784
    20
    return true
dirk@784
    21
  default:
dirk@784
    22
    return false
dirk@784
    23
  }
dirk@784
    24
}
dirk@784
    25
dirk@784
    26
fileprivate func > <T : Comparable>(lhs: T?, rhs: T?) -> Bool {
dirk@784
    27
  switch (lhs, rhs) {
dirk@784
    28
  case let (l?, r?):
dirk@784
    29
    return l > r
dirk@784
    30
  default:
dirk@784
    31
    return rhs < lhs
dirk@784
    32
  }
dirk@784
    33
}
dirk@784
    34
dirk@30
    35
dirk@854
    36
class EmailListViewController: UITableViewController {
dirk@591
    37
    struct EmailListConfig {
dirk@591
    38
        let appConfig: AppConfig
dirk@583
    39
dirk@855
    40
        let account: Account?
dirk@855
    41
dirk@854
    42
        /** The folder to display, if it exists */
dirk@854
    43
        let folder: Folder?
dirk@854
    44
    }
dirk@583
    45
dirk@854
    46
    var comp = "EmailListViewController"
dirk@58
    47
dirk@583
    48
    struct UIState {
dirk@583
    49
        var isSynching: Bool = false
dirk@583
    50
    }
dirk@583
    51
    
dirk@273
    52
    let segueShowEmail = "segueShowEmail"
dirk@397
    53
    let segueCompose = "segueCompose"
dirk@397
    54
    let segueUserSettings = "segueUserSettings"
dirk@30
    55
dirk@583
    56
    var config: EmailListConfig!
dirk@583
    57
dirk@275
    58
    var state = UIState()
dirk@342
    59
    let dateFormatter = UIHelper.dateFormatterEmailList()
dirk@275
    60
dirk@502
    61
    /**
dirk@502
    62
     The default background color for an email cell, as determined the first time a cell is
dirk@502
    63
     created.
dirk@502
    64
     */
dirk@502
    65
    var defaultCellBackgroundColor: UIColor?
dirk@502
    66
dirk@502
    67
    /**
dirk@502
    68
     Indicates whether `defaultCellBackgroundColor` has been determined or not.
dirk@502
    69
     */
dirk@502
    70
    var determinedCellBackgroundColor: Bool = false
dirk@502
    71
dirk@594
    72
    var refreshController: UIRefreshControl!
dirk@594
    73
dirk@705
    74
    /**
dirk@705
    75
     The message that should be saved as a draft when compose gets aborted.
dirk@705
    76
     */
dirk@841
    77
    var draftMessageToStore: Message?
dirk@705
    78
dirk@719
    79
    /**
dirk@719
    80
     When the user taps on a draft email, this is the message that was selected
dirk@719
    81
     and should be given to the compose view.
dirk@719
    82
     */
dirk@841
    83
    var draftMessageToCompose: Message?
dirk@719
    84
dirk@745
    85
    required init?(coder aDecoder: NSCoder) {
dirk@745
    86
        super.init(coder: aDecoder)
dirk@745
    87
        self.comp = "EmailListViewController"
dirk@745
    88
    }
dirk@745
    89
dirk@855
    90
    func isRead(message: Message)-> Bool {
dirk@855
    91
        return message.imapFlags.seen
ana@614
    92
    }
ana@614
    93
dirk@855
    94
    func isImportant(message: Message)-> Bool {
dirk@855
    95
        return message.imapFlags.flagged
ana@614
    96
    }
ana@614
    97
dirk@275
    98
    override func viewDidLoad() {
dirk@356
    99
        UIHelper.variableCellHeightsTableView(self.tableView)
dirk@275
   100
    }
dirk@31
   101
dirk@784
   102
    override func viewWillAppear(_ animated: Bool) {
dirk@854
   103
        updateModel()
ana@207
   104
        super.viewWillAppear(animated)
ana@207
   105
    }
ana@152
   106
dirk@784
   107
    @IBAction func mailSentSegue(_ segue: UIStoryboardSegue) {
dirk@452
   108
    }
dirk@452
   109
dirk@784
   110
    @IBAction func backFromComposeWithoutSavingDraftSegue(_ segue: UIStoryboardSegue) {
dirk@705
   111
    }
dirk@705
   112
dirk@784
   113
    @IBAction func backFromComposeSaveDraftSegue(_ segue: UIStoryboardSegue) {
dirk@867
   114
        guard let message = draftMessageToStore else {
dirk@705
   115
            return
dirk@705
   116
        }
dirk@707
   117
dirk@707
   118
        state.isSynching = true
dirk@707
   119
        updateUI()
dirk@707
   120
dirk@867
   121
        message.imapFlags.draft = true
dirk@867
   122
dirk@854
   123
        // TODO: IOS 222: Save as draft
dirk@867
   124
        if let folder = draftMessageToStore?.parent as? Folder {
dirk@867
   125
            if folder.folderType == .drafts {
dirk@867
   126
                message.save()
dirk@867
   127
                return
dirk@867
   128
            }
dirk@867
   129
        }
dirk@867
   130
dirk@867
   131
        guard let account = config.account else {
dirk@867
   132
            return
dirk@867
   133
        }
xavier@870
   134
        
xavier@870
   135
        if let folder = account.folder(ofType: FolderType.drafts) {
xavier@870
   136
            folder.save(message: message)
xavier@870
   137
            return
xavier@870
   138
        }
dirk@705
   139
    }
dirk@705
   140
dirk@854
   141
    func updateModel() {
dirk@31
   142
    }
dirk@31
   143
dirk@58
   144
    // MARK: - UI State
dirk@58
   145
dirk@58
   146
    func updateUI() {
dirk@784
   147
        UIApplication.shared.isNetworkActivityIndicatorVisible = state.isSynching
dirk@748
   148
        if !state.isSynching {
dirk@594
   149
            self.refreshControl?.endRefreshing()
dirk@58
   150
        }
dirk@58
   151
    }
dirk@58
   152
dirk@31
   153
    // MARK: - UITableViewDataSource
dirk@31
   154
dirk@784
   155
    override func numberOfSections(in tableView: UITableView) -> Int {
dirk@854
   156
        if let _ = config.folder {
dirk@854
   157
            return 1
dirk@209
   158
        }
dirk@854
   159
        return 0
dirk@31
   160
    }
dirk@31
   161
dirk@784
   162
    override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
dirk@854
   163
        if let fol = config.folder {
dirk@854
   164
            return fol.messageCount()
dirk@31
   165
        }
dirk@31
   166
        return 0
dirk@31
   167
    }
dirk@31
   168
dirk@784
   169
    override func tableView(_ tableView: UITableView,
dirk@784
   170
                            cellForRowAt indexPath: IndexPath) -> UITableViewCell {
dirk@784
   171
        let cell = tableView.dequeueReusableCell(
dirk@784
   172
            withIdentifier: "EmailListViewCell", for: indexPath) as! EmailListViewCell
dirk@502
   173
        if !determinedCellBackgroundColor {
dirk@502
   174
            defaultCellBackgroundColor = cell.backgroundColor
dirk@502
   175
            determinedCellBackgroundColor = true
dirk@502
   176
        }
dirk@31
   177
        configureCell(cell, indexPath: indexPath)
dirk@31
   178
        return cell
dirk@31
   179
    }
dirk@31
   180
dirk@861
   181
    /**
dirk@861
   182
     The message at the given position.
dirk@861
   183
     */
dirk@861
   184
    func messageAt(indexPath: IndexPath) -> Message? {
dirk@861
   185
        if let fol = config.folder {
dirk@861
   186
            return fol.messageByIndex(indexPath.row)
dirk@861
   187
        }
dirk@861
   188
        return nil
dirk@861
   189
    }
dirk@861
   190
dirk@719
   191
    // MARK: - UITableViewDelegate
dirk@719
   192
dirk@784
   193
    override func tableView(_ tableView: UITableView,
dirk@784
   194
                            didSelectRowAt indexPath: IndexPath) {
dirk@719
   195
        draftMessageToCompose = nil
dirk@719
   196
dirk@784
   197
        let cell = tableView.cellForRow(at: indexPath)
dirk@719
   198
dirk@855
   199
        if let fol = config.folder {
dirk@855
   200
            if fol.folderType == .drafts {
dirk@855
   201
                draftMessageToCompose = messageAt(indexPath: indexPath)
dirk@784
   202
                performSegue(withIdentifier: segueCompose, sender: cell)
dirk@719
   203
                return
dirk@719
   204
            }
dirk@719
   205
        }
dirk@719
   206
dirk@784
   207
        performSegue(withIdentifier: segueShowEmail, sender: cell)
dirk@719
   208
    }
dirk@719
   209
dirk@784
   210
    override func tableView(_ tableView: UITableView, editActionsForRowAt
dirk@784
   211
        indexPath: IndexPath)-> [UITableViewRowAction]? {
dirk@784
   212
        let cell = tableView.cellForRow(at: indexPath) as! EmailListViewCell
dirk@855
   213
        if let email = messageAt(indexPath: indexPath) {
dirk@855
   214
            let isFlagAction = createIsFlagAction(message: email, cell: cell)
dirk@855
   215
            let deleteAction = createDeleteAction(cell)
dirk@855
   216
            let isReadAction = createIsReadAction(message: email, cell: cell)
dirk@855
   217
            return [deleteAction,isFlagAction,isReadAction]
dirk@855
   218
        }
dirk@855
   219
        return nil
dirk@744
   220
    }
dirk@744
   221
dirk@719
   222
    // MARK: - Misc
dirk@719
   223
dirk@855
   224
    func configureCell(_ theCell: UITableViewCell, indexPath: IndexPath) {
dirk@745
   225
        guard let cell = theCell as? EmailListViewCell else {
dirk@745
   226
            return
dirk@745
   227
        }
dirk@855
   228
        if let email = messageAt(indexPath: indexPath) {
dirk@855
   229
            if let pEpRating = email.pEpRating {
dirk@855
   230
                let privacyColor = PEPUtil.pEpColorFromRating(pEpRating)
dirk@502
   231
                if let uiColor = UIHelper.textBackgroundUIColorFromPrivacyColor(privacyColor) {
dirk@502
   232
                    cell.backgroundColor = uiColor
dirk@502
   233
                } else {
dirk@502
   234
                    if determinedCellBackgroundColor {
dirk@502
   235
                        cell.backgroundColor = defaultCellBackgroundColor
dirk@502
   236
                    }
dirk@502
   237
                }
dirk@502
   238
            }
dirk@855
   239
            UIHelper.putString(email.from?.displayString, toLabel: cell.senderLabel)
dirk@855
   240
            UIHelper.putString(email.shortMessage, toLabel: cell.subjectLabel)
dirk@572
   241
dirk@572
   242
            // Snippet
dirk@565
   243
            if let text = email.longMessage {
dirk@572
   244
                let theText = text.replaceNewLinesWith(" ").trimmedWhiteSpace()
dirk@565
   245
                UIHelper.putString(theText, toLabel: cell.summaryLabel)
dirk@572
   246
            } else if let html = email.longMessageFormatted {
dirk@568
   247
                var text = html.extractTextFromHTML()
dirk@572
   248
                text = text.replaceNewLinesWith(" ").trimmedWhiteSpace()
dirk@568
   249
                UIHelper.putString(text, toLabel: cell.summaryLabel)
dirk@568
   250
            } else {
dirk@568
   251
                UIHelper.putString(nil, toLabel: cell.summaryLabel)
dirk@565
   252
            }
dirk@32
   253
dirk@855
   254
            if let receivedDate = email.received {
dirk@788
   255
                UIHelper.putString(dateFormatter.string(from: receivedDate as Date),
dirk@342
   256
                                   toLabel: cell.dateLabel)
dirk@32
   257
            } else {
dirk@342
   258
                UIHelper.putString(nil, toLabel: cell.dateLabel)
dirk@32
   259
            }
ana@620
   260
dirk@855
   261
            if (isImportant(message: email) && isRead(message: email)) {
dirk@784
   262
                cell.isImportantImage.isHidden = false
dirk@784
   263
                cell.isImportantImage.backgroundColor = UIColor.orange
ana@666
   264
            }
dirk@855
   265
            else if (isImportant(message: email) && !isRead(message: email)) {
dirk@784
   266
                cell.isImportantImage.isHidden = false
dirk@784
   267
                cell.isImportantImage.backgroundColor = UIColor.blue
ana@656
   268
                cell.isImportantImage.layer.borderWidth = 2
dirk@784
   269
                cell.isImportantImage.layer.borderColor = UIColor.orange.cgColor
dirk@855
   270
            } else if (!isImportant(message: email) && isRead(message: email)) {
dirk@784
   271
                    cell.isImportantImage.isHidden = true
dirk@855
   272
            } else if (!isImportant(message: email) && !isRead(message: email)) {
dirk@784
   273
                cell.isImportantImage.isHidden = false
dirk@784
   274
                cell.isImportantImage.backgroundColor = UIColor.blue
ana@620
   275
            }
dirk@31
   276
        }
dirk@30
   277
    }
dirk@30
   278
dirk@784
   279
    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
dirk@606
   280
        // Make sure the current account is set, if defined
dirk@606
   281
        config.appConfig.currentAccount = config.account
dirk@606
   282
dirk@361
   283
        if segue.identifier == segueCompose {
dirk@784
   284
            let destination = segue.destination
dirk@434
   285
                as! ComposeViewController
dirk@583
   286
            destination.appConfig = config.appConfig
dirk@719
   287
            if let draft = draftMessageToCompose {
dirk@855
   288
                draft.imapFlags.seen = true
dirk@723
   289
                config.appConfig.model.save()
dirk@723
   290
dirk@719
   291
                destination.originalMessage = draft
dirk@784
   292
                destination.composeMode = .composeDraft
dirk@719
   293
            }
dirk@273
   294
        } else if segue.identifier == segueShowEmail {
dirk@273
   295
            guard
dirk@784
   296
                let vc = segue.destination as? EmailViewController,
dirk@273
   297
                let cell = sender as? UITableViewCell,
dirk@784
   298
                let indexPath = self.tableView.indexPath(for: cell),
dirk@855
   299
                let email = messageAt(indexPath: indexPath) else {
dirk@273
   300
                    return
dirk@273
   301
            }
dirk@583
   302
            vc.appConfig = config.appConfig
dirk@273
   303
            vc.message = email
ana@246
   304
        }
ana@246
   305
    }
dirk@744
   306
dirk@855
   307
    func syncFlagsToServer(message: Message) {
dirk@855
   308
        // TODO: IOS 222: Sync flags back to server
dirk@744
   309
    }
dirk@744
   310
dirk@855
   311
    func createIsFlagAction(message: Message, cell: EmailListViewCell) -> UITableViewRowAction {
dirk@744
   312
        // preparing the title action to show when user swipe
dirk@744
   313
        var localizedIsFlagTitle = " "
dirk@855
   314
        if (isImportant(message: message)) {
dirk@855
   315
            localizedIsFlagTitle = NSLocalizedString(
dirk@855
   316
                "Unflag",
dirk@855
   317
                comment: "Unflag button title in swipe action on EmailListViewController")
dirk@744
   318
        } else {
dirk@855
   319
            localizedIsFlagTitle = NSLocalizedString(
dirk@855
   320
                "Flag",
dirk@855
   321
                comment: "Flag button title in swipe action on EmailListViewController")
dirk@744
   322
        }
dirk@744
   323
dirk@744
   324
        // preparing action to trigger when user swipe
dirk@784
   325
        let isFlagCompletionHandler: (UITableViewRowAction, IndexPath) -> Void =
dirk@744
   326
            { (action, indexPath) in
dirk@855
   327
                if (self.isImportant(message: message)) {
dirk@855
   328
                    message.imapFlags.flagged = false
dirk@744
   329
dirk@744
   330
                } else {
dirk@855
   331
                    message.imapFlags.flagged = true
dirk@744
   332
                }
dirk@855
   333
                self.syncFlagsToServer(message: message)
dirk@784
   334
                self.tableView.reloadRows(at: [indexPath], with: .none)
dirk@744
   335
        }
dirk@744
   336
        // creating the action
dirk@784
   337
        let isFlagAction = UITableViewRowAction(style: .default, title: localizedIsFlagTitle,
dirk@744
   338
                                                handler: isFlagCompletionHandler)
dirk@744
   339
        // changing default action color
dirk@784
   340
        isFlagAction.backgroundColor = UIColor.orange
dirk@744
   341
dirk@744
   342
        return isFlagAction
dirk@744
   343
    }
dirk@744
   344
dirk@784
   345
    func createDeleteAction (_ cell: EmailListViewCell) -> UITableViewRowAction {
dirk@744
   346
dirk@744
   347
        // preparing the title action to show when user swipe
dirk@855
   348
        let localizedDeleteTitle = NSLocalizedString(
dirk@855
   349
            "Erase",
dirk@855
   350
            comment: "Erase button title in swipe action on EmailListViewController")
dirk@744
   351
dirk@784
   352
        let deleteCompletionHandler: (UITableViewRowAction, IndexPath) -> Void =
dirk@744
   353
            { (action, indexPath) in
dirk@855
   354
                let message = self.messageAt(indexPath: indexPath)
dirk@855
   355
                message?.imapFlags.deleted = true
dirk@855
   356
                self.syncFlagsToServer(message: message!)
dirk@744
   357
        }
dirk@744
   358
dirk@744
   359
        // creating the action
dirk@784
   360
        let deleteAction = UITableViewRowAction(style: .default, title: localizedDeleteTitle,
dirk@744
   361
                                                handler: deleteCompletionHandler)
dirk@744
   362
        return deleteAction
dirk@744
   363
    }
dirk@744
   364
dirk@855
   365
    func createIsReadAction(message: Message, cell: EmailListViewCell) -> UITableViewRowAction {
dirk@744
   366
        // preparing the title action to show when user swipe
dirk@744
   367
        var localizedisReadTitle = " "
dirk@855
   368
        if (isRead(message: message)) {
dirk@855
   369
            localizedisReadTitle = NSLocalizedString(
dirk@855
   370
                "Unread",
dirk@855
   371
                comment: "Unread button title in swipe action on EmailListViewController")
dirk@744
   372
        } else {
dirk@855
   373
            localizedisReadTitle = NSLocalizedString(
dirk@855
   374
                "Read",
dirk@855
   375
                comment: "Read button title in swipe action on EmailListViewController")
dirk@744
   376
        }
dirk@744
   377
dirk@744
   378
        // creating the action
dirk@784
   379
        let isReadCompletionHandler: (UITableViewRowAction, IndexPath) -> Void =
dirk@744
   380
            { (action, indexPath) in
dirk@855
   381
                if (self.isRead(message: message)) {
dirk@855
   382
                    message.imapFlags.seen = false
dirk@744
   383
                } else {
dirk@855
   384
                    message.imapFlags.seen = true
dirk@744
   385
                }
dirk@855
   386
                self.syncFlagsToServer(message: message)
dirk@784
   387
                self.tableView.reloadRows(at: [indexPath], with: .none)
dirk@744
   388
        }
dirk@784
   389
        let isReadAction = UITableViewRowAction(style: .default, title: localizedisReadTitle,
dirk@744
   390
                                                handler: isReadCompletionHandler)
dirk@784
   391
        isReadAction.backgroundColor = UIColor.blue
dirk@744
   392
dirk@744
   393
        return isReadAction
dirk@744
   394
    }
dirk@784
   395
}