1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
|
//
// MasterViewController.m
// iOS UI Test
//
// Created by Jonathan Willing on 4/8/13.
// Copyright (c) 2013 AppJon. All rights reserved.
//
#import "MasterViewController.h"
#import <MailCore/MailCore.h>
#import "FXKeychain.h"
#import "MCTMsgViewController.h"
#import "GTMOAuth2ViewControllerTouch.h"
#import "MCTTableViewCell.h"
#define CLIENT_ID @"the-client-id"
#define CLIENT_SECRET @"the-client-secret"
#define KEYCHAIN_ITEM_NAME @"MailCore OAuth 2.0 Token"
@interface MasterViewController ()
@property (nonatomic, strong) NSArray *messages;
@property (nonatomic, strong) MCOIMAPOperation *imapCheckOp;
@property (nonatomic, strong) MCOIMAPSession *imapSession;
@property (nonatomic, strong) MCOIMAPFetchMessagesOperation *imapMessagesFetchOp;
@end
@implementation MasterViewController
- (void)viewDidLoad {
[super viewDidLoad];
[[NSUserDefaults standardUserDefaults] registerDefaults:@{ HostnameKey: @"imap.gmail.com" }];
if ([[NSUserDefaults standardUserDefaults] boolForKey:@"OAuth2Enabled"]) {
[self startOAuth2];
}
else {
[self startLogin];
}
}
- (void) startLogin
{
NSString *username = [[NSUserDefaults standardUserDefaults] objectForKey:UsernameKey];
NSString *password = [[FXKeychain defaultKeychain] objectForKey:PasswordKey];
NSString *hostname = [[NSUserDefaults standardUserDefaults] objectForKey:HostnameKey];
if (!username.length || !password.length) {
[self performSelector:@selector(showSettingsViewController:) withObject:nil afterDelay:0.5];
return;
}
[self loadAccountWithUsername:username password:password hostname:hostname oauth2Token:nil];
}
- (void) startOAuth2
{
GTMOAuth2Authentication * auth = [GTMOAuth2ViewControllerTouch authForGoogleFromKeychainForName:KEYCHAIN_ITEM_NAME
clientID:CLIENT_ID
clientSecret:CLIENT_SECRET];
if ([auth refreshToken] == nil) {
MasterViewController * __weak weakSelf = self;
GTMOAuth2ViewControllerTouch *viewController = [GTMOAuth2ViewControllerTouch controllerWithScope:@"https://mail.google.com/"
clientID:CLIENT_ID
clientSecret:CLIENT_SECRET
keychainItemName:KEYCHAIN_ITEM_NAME
completionHandler:^(GTMOAuth2ViewControllerTouch *viewController, GTMOAuth2Authentication *retrievedAuth, NSError *error) {
[weakSelf loadWithAuth:retrievedAuth];
}];
[self.navigationController pushViewController:viewController
animated:YES];
}
else {
[auth beginTokenFetchWithDelegate:self
didFinishSelector:@selector(auth:finishedRefreshWithFetcher:error:)];
}
}
- (void)auth:(GTMOAuth2Authentication *)auth
finishedRefreshWithFetcher:(GTMHTTPFetcher *)fetcher
error:(NSError *)error {
[self loadWithAuth:auth];
}
- (void)loadWithAuth:(GTMOAuth2Authentication *)auth
{
NSString *hostname = [[NSUserDefaults standardUserDefaults] objectForKey:HostnameKey];
[self loadAccountWithUsername:[auth userEmail] password:nil hostname:hostname oauth2Token:[auth accessToken]];
}
- (void)loadAccountWithUsername:(NSString *)username
password:(NSString *)password
hostname:(NSString *)hostname
oauth2Token:(NSString *)oauth2Token
{
self.imapSession = [[MCOIMAPSession alloc] init];
self.imapSession.hostname = hostname;
self.imapSession.port = 993;
self.imapSession.username = username;
self.imapSession.password = password;
if (oauth2Token != nil) {
self.imapSession.OAuth2Token = oauth2Token;
self.imapSession.authType = MCOAuthTypeXOAuth2;
}
self.imapSession.connectionType = MCOConnectionTypeTLS;
MasterViewController * __weak weakSelf = self;
self.imapSession.connectionLogger = ^(void * connectionID, MCOConnectionLogType type, NSData * data) {
@synchronized(weakSelf) {
if (type != MCOConnectionLogTypeSentPrivate) {
NSLog(@"event logged:%p %i withData: %@", connectionID, type, [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding]);
}
}
};
NSLog(@"checking account");
self.imapCheckOp = [self.imapSession checkAccountOperation];
[self.imapCheckOp start:^(NSError *error) {
MasterViewController *strongSelf = weakSelf;
NSLog(@"finished checking account.");
if (error == nil) {
[strongSelf loadEmails];
} else {
NSLog(@"error loading account: %@", error);
}
strongSelf.imapCheckOp = nil;
}];
}
- (void)loadEmails {
MCOIMAPMessagesRequestKind requestKind = (MCOIMAPMessagesRequestKind)
(MCOIMAPMessagesRequestKindHeaders | MCOIMAPMessagesRequestKindStructure |
MCOIMAPMessagesRequestKindInternalDate | MCOIMAPMessagesRequestKindHeaderSubject |
MCOIMAPMessagesRequestKindFlags);
self.imapMessagesFetchOp = [self.imapSession fetchMessagesByUIDOperationWithFolder:@"INBOX"
requestKind:requestKind
uids:[MCOIndexSet indexSetWithRange:MCORangeMake(1, UINT64_MAX)]];
[self.imapMessagesFetchOp setProgress:^(unsigned int progress) {
//NSLog(@"progress: %u", progress);
}];
__weak MasterViewController *weakSelf = self;
[self.imapMessagesFetchOp start:^(NSError *error, NSArray *messages, MCOIndexSet *vanishedMessages) {
MasterViewController *strongSelf = weakSelf;
NSLog(@"fetched all messages.");
NSSortDescriptor *sort = [NSSortDescriptor sortDescriptorWithKey:@"header.date" ascending:NO];
strongSelf.messages = [messages sortedArrayUsingDescriptors:@[sort]];
[strongSelf.tableView reloadData];
}];
}
- (void)didReceiveMemoryWarning {
[super didReceiveMemoryWarning];
NSLog(@"%s",__PRETTY_FUNCTION__);
}
#pragma mark - Table View
- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
return 1;
}
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
return self.messages.count;
}
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
MCTTableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:@"Cell" forIndexPath:indexPath];
MCOIMAPMessage *message = self.messages[indexPath.row];
cell.textLabel.text = message.header.subject;
[cell.messageRenderingOperation cancel];
cell.messageRenderingOperation = [self.imapSession plainTextBodyRenderingOperationWithMessage:message
folder:@"INBOX"];
[cell.messageRenderingOperation start:^(NSString * plainTextBodyString, NSError * error) {
cell.detailTextLabel.text = plainTextBodyString;
cell.messageRenderingOperation = nil;
}];
return cell;
}
- (void)showSettingsViewController:(id)sender {
[self.imapMessagesFetchOp cancel];
SettingsViewController *settingsViewController = [[SettingsViewController alloc] initWithNibName:nil bundle:nil];
settingsViewController.delegate = self;
[self presentViewController:settingsViewController animated:YES completion:nil];
}
- (void)settingsViewControllerFinished:(SettingsViewController *)viewController {
[self dismissViewControllerAnimated:YES completion:nil];
NSString *username = [[NSUserDefaults standardUserDefaults] stringForKey:UsernameKey];
NSString *password = [[FXKeychain defaultKeychain] objectForKey:PasswordKey];
NSString *hostname = [[NSUserDefaults standardUserDefaults] objectForKey:HostnameKey];
if (![username isEqualToString:self.imapSession.username] ||
![password isEqualToString:self.imapSession.password] ||
![hostname isEqualToString:self.imapSession.hostname]) {
self.imapSession = nil;
[self loadAccountWithUsername:username password:password hostname:hostname oauth2Token:nil];
}
}
- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {
MCOIMAPMessage *msg = self.messages[indexPath.row];
MCTMsgViewController *vc = [[MCTMsgViewController alloc] init];
vc.folder = @"INBOX";
vc.message = msg;
vc.session = self.imapSession;
[self.navigationController pushViewController:vc animated:YES];
}
@end
|
