1  //
  2  //  SWCViewController.m
  3  //  SharedWebCredentialViewService
  4  //
  5  //  Copyright (c) 2014 Apple Inc. All Rights Reserved.
  6  //
  7  
  8  #import <Foundation/NSXPCConnection.h>
  9  #import "SWCViewController.h"
 10  #import <UIKit/UIViewController_Private.h>
 11  #import <UIKit/UIFont_Private.h>
 12  #import <UIKit/UIImage_SymbolImagePrivate.h>
 13  #import <UIKit/UIAlertController_Private.h>
 14  #import <UIKit/UITableViewCell_Private.h>
 15  
 16  #include <bsm/libbsm.h>
 17  #include <ipc/securityd_client.h>
 18  
 19  #include "SharedWebCredential/swcagent_client.h"
 20  
 21  #import "SWCViewController.h"
 22  
 23  const NSString* SWC_PASSWORD_KEY = @"spwd";
 24  const NSString* SWC_ACCOUNT_KEY  = @"acct";
 25  const NSString* SWC_SERVER_KEY   = @"srvr";
 26  
 27  //
 28  // SWCDictionaryAdditions
 29  //
 30  
 31  @interface NSDictionary (SWCDictionaryAdditions)
 32  - (NSComparisonResult) compareCredentialDictionaryAscending:(NSDictionary *)other;
 33  @end
 34  
 35  @implementation NSDictionary (SWCDictionaryAdditions)
 36  - (NSComparisonResult)compareCredentialDictionaryAscending:(NSDictionary *)other
 37  {
 38      NSComparisonResult result;
 39      NSString *str1 = [self objectForKey:SWC_ACCOUNT_KEY], *str2 = [other objectForKey:SWC_ACCOUNT_KEY];
 40      if (!str1) str1 = @"";
 41      if (!str2) str2 = @"";
 42      
 43      // primary sort by account name
 44      result = [str1 localizedCaseInsensitiveCompare:str2];
 45      if (result == NSOrderedSame) {
 46          // secondary sort by domain name
 47          NSString *str3 = [self objectForKey:SWC_SERVER_KEY], *str4 = [other objectForKey:SWC_SERVER_KEY];
 48          if (!str3) str3 = @"";
 49          if (!str4) str4 = @"";
 50          
 51          result = [str3 localizedCaseInsensitiveCompare:str4];
 52      }
 53      
 54      return result;
 55  }
 56  @end
 57  
 58  
 59  //
 60  // SWCItemCell
 61  //
 62  @interface SWCItemCell : UITableViewCell
 63  {
 64      NSDictionary *_dict;
 65      BOOL _isTicked;
 66      UIView *_bottomLine;
 67      UIView *_bottomLineSelected;
 68      UIView *_topLine;
 69      UIView *_topLineSelected;
 70      BOOL _showSeparator;
 71      BOOL _showTopSeparator;
 72  }
 73  
 74  - (id)initWithDictionary:(NSDictionary *)dict;
 75  @property (nonatomic, readonly) id userInfo;
 76  @property (nonatomic, assign) BOOL showSeparator;
 77  @end
 78  
 79  @implementation SWCItemCell
 80  
 81  @synthesize showSeparator = _showSeparator;
 82  
 83  - (id)initWithDictionary:(NSDictionary *)dict
 84  {
 85      if ((self = [super initWithStyle:UITableViewCellStyleSubtitle reuseIdentifier:nil]))
 86      {
 87          _dict = dict;
 88          
 89          self.selectionStyle = UITableViewCellSelectionStyleNone;
 90          
 91          self.backgroundColor = [UIColor systemBackgroundColor];
 92          
 93          self.textLabel.textColor = [UIColor labelColor];
 94          self.textLabel.textAlignment = NSTextAlignmentLeft;
 95          self.textLabel.adjustsFontSizeToFitWidth = YES;
 96          self.textLabel.baselineAdjustment = UIBaselineAdjustmentAlignCenters;
 97          
 98          NSString *title = [dict objectForKey:SWC_ACCOUNT_KEY];
 99          self.textLabel.text = title ? title : NSLocalizedString(@"--", nil);
100          
101          self.detailTextLabel.textColor = [UIColor secondaryLabelColor];
102          self.detailTextLabel.textAlignment = NSTextAlignmentLeft;
103          self.detailTextLabel.adjustsFontSizeToFitWidth = YES;
104          self.detailTextLabel.baselineAdjustment = UIBaselineAdjustmentAlignCenters;
105          
106          NSString *subtitle = [dict objectForKey:SWC_SERVER_KEY];
107          self.detailTextLabel.text = subtitle ? subtitle : NSLocalizedString(@"--", nil);
108          
109          self.backgroundView = [[UIView alloc] init];
110          self.backgroundView.backgroundColor = [UIColor systemBackgroundColor];
111  
112          self.imageView.image = [UIImage systemImageNamed:@"checkmark"];
113          self.imageView.hidden = YES;
114      }
115      
116      return self;
117  }
118  
119  - (void)setTicked: (BOOL) selected
120  {
121      _isTicked = selected;
122  }
123  
124  - (void)layoutSubviews
125  {
126      
127      if (_bottomLine) {
128          CGFloat scale = [[UIScreen mainScreen] scale];
129          [_bottomLine setFrame:CGRectMake(0, self.frame.size.height - (1 / scale), self.frame.size.width, 1 / scale)];
130      }
131      
132      if (_bottomLineSelected) {
133          CGFloat scale = [[UIScreen mainScreen] scale];
134          [_bottomLineSelected setFrame:CGRectMake(0, self.frame.size.height - (1 / scale), self.frame.size.width, 1 / scale)];
135      }
136      
137      if (_topLine) {
138          CGFloat scale = [[UIScreen mainScreen] scale];
139          [_topLine setFrame:CGRectMake(0, 0, self.frame.size.width, 1 / scale)];
140      }
141      
142      if (_topLineSelected) {
143          CGFloat scale = [[UIScreen mainScreen] scale];
144          [_topLineSelected setFrame:CGRectMake(0, 0, self.frame.size.width, 1 / scale)];
145      }
146      
147      if (_isTicked)
148      {
149          self.imageView.hidden = NO;
150      } else {
151          self.imageView.hidden = YES;
152      }
153      
154      [super layoutSubviews];
155      
156  }
157  
158  - (void)setShowSeparator:(BOOL)showSeparator {
159      if (_showSeparator != showSeparator) {
160          _showSeparator = showSeparator;
161          
162          if (_showSeparator) {
163              if (!_bottomLine) {
164                  CGRect rectZero = CGRectMake(0, 0, 0, 0);
165                  _bottomLine = [[UIView alloc] initWithFrame:rectZero];
166                  _bottomLine.backgroundColor = [UIColor separatorColor];
167                  [self.backgroundView addSubview:_bottomLine];
168              }
169              if (!_bottomLineSelected) {
170                  CGRect rectZero = CGRectMake(0, 0, 0, 0);
171                  _bottomLineSelected = [[UIView alloc] initWithFrame:rectZero];
172                  _bottomLineSelected.backgroundColor = [UIColor separatorColor];
173                  [self.selectedBackgroundView addSubview: _bottomLineSelected];
174              }
175              
176          } else {
177              if (_bottomLine) {
178                  [_bottomLine removeFromSuperview];
179                  _bottomLine = nil;
180              }
181              if (_bottomLineSelected) {
182                  [_bottomLineSelected removeFromSuperview];
183                  _bottomLineSelected = nil;
184              }
185          }
186      }
187  }
188  
189  - (void)setShowTopSeparator:(BOOL)showTopSeparator {
190      if (_showTopSeparator != showTopSeparator) {
191          _showTopSeparator = showTopSeparator;
192          
193          if (_showTopSeparator) {
194              if (!_topLine) {
195                  CGRect rectZero = CGRectMake(0, 0, 0, 0);
196                  _topLine = [[UIView alloc] initWithFrame:rectZero];
197                  _topLine.backgroundColor = [UIColor separatorColor];
198                  [self.backgroundView addSubview:_topLine];
199              }
200              if (!_topLineSelected) {
201                  CGRect rectZero = CGRectMake(0, 0, 0, 0);
202                  _topLineSelected = [[UIView alloc] initWithFrame:rectZero];
203                  _topLineSelected.backgroundColor = [UIColor separatorColor];
204                  [self.selectedBackgroundView addSubview: _topLineSelected];
205              }
206              
207          } else {
208              if (_topLine) {
209                  [_topLine removeFromSuperview];
210                  _topLine = nil;
211              }
212              if (_topLineSelected) {
213                  [_topLineSelected removeFromSuperview];
214                  _topLineSelected = nil;
215              }
216          }
217      }
218  }
219  
220  @end
221  
222  
223  //
224  // SWCViewController
225  //
226  
227  @interface SWCViewController ()
228  {
229      NSMutableArray       *_credentials; // array of NSDictionary
230      UILabel              *_topLabel;
231      UILabel              *_middleLabel;
232      UITableView          *_table;
233      NSDictionary         *_selectedDict;
234      NSIndexPath          *_selectedCell;
235  }
236  
237  @end
238  
239  @implementation SWCViewController
240  
241  - (NSDictionary *)selectedItem
242  {
243      return _selectedDict;
244  }
245  
246  - (void)setCredentials:(NSArray *)inArray
247  {
248      NSMutableArray *credentials = [[NSMutableArray alloc] initWithArray:inArray];
249      [credentials sortUsingSelector:@selector(compareCredentialDictionaryAscending:)];
250      _credentials = credentials;
251      if (_table)
252          [_table reloadData];
253  }
254  
255  - (void)_enableTable
256  {
257      [_table setUserInteractionEnabled:YES];
258  }
259  
260  - (UITableView *)tableView
261  {
262      if (_table == nil) {
263          _table = [[UITableView alloc] init];
264          [_table setTranslatesAutoresizingMaskIntoConstraints:NO];
265          [_table setAutoresizingMask:UIViewAutoresizingNone];
266          [_table setBackgroundColor:[UIColor systemBackgroundColor]];
267          [_table setSeparatorStyle:UITableViewCellSeparatorStyleNone];
268      }
269      [_table sizeToFit];
270      
271      return (UITableView *)_table;
272  }
273  
274  -(void)loadView
275  {
276      
277      UIView* view = [[UIView alloc] init];
278      
279      UITableView* table = [self tableView];
280      [table setDelegate: self];
281      [table setDataSource: self];
282      
283      [view addSubview: table];
284      
285      CFErrorRef error = NULL;
286      audit_token_t auditToken = {};
287      memset(&auditToken, 0, sizeof(auditToken));
288      CFArrayRef credentialList = swca_copy_pairs(swca_copy_pairs_request_id, &auditToken, &error);
289      if (error) {
290          NSLog(@"Unable to get accounts: %@", [(__bridge NSError*)error localizedDescription]);
291      }
292      
293      [self setCredentials:(__bridge NSArray*)credentialList];
294      if (credentialList) {
295          CFRelease(credentialList);
296      }
297      
298      table.frame = CGRectMake(0.0, 0.0, 300.0, 45.0);
299      [table layoutIfNeeded]; // so autolayout can do its thing and then we can measure a cell
300      UITableViewCell* cell = table.visibleCells.firstObject;
301      CGFloat heightForOneRow = cell ? CGRectGetHeight(cell.frame) : 45.0;
302      
303      NSDictionary* views = NSDictionaryOfVariableBindings(table);
304      
305      if ([_credentials count] > 2)
306      {
307          
308          CGFloat manyCredentialsHeight = CGFloatMax((heightForOneRow * 2) + 30.0, 120.0);
309          NSDictionary *metrics = @{@"height":@(manyCredentialsHeight)};
310          [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|-[table]-|"  options:0 metrics:metrics views:views]];
311          [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[table(height)]-|" options:0 metrics:metrics views:views]];
312          
313          CGFloat scale = [[UIScreen mainScreen] scale];
314          table.layer.borderWidth = 1.0 / scale;
315          table.layer.borderColor = [UIColor opaqueSeparatorColor].CGColor;
316  
317          [self setPreferredContentSize:CGSizeMake(0, manyCredentialsHeight + 20.0)];
318          
319      } else if ([_credentials count] == 2) {
320          
321          NSDictionary *metrics = @{@"height":@(heightForOneRow * 2)};
322          [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|[table]|"  options:0 metrics:metrics views:views]];
323          [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[table(height)]-|" options:0 metrics:metrics views:views]];
324          
325          [self setPreferredContentSize:CGSizeMake(0, heightForOneRow * 2)];
326          [table setScrollEnabled: NO];
327          
328      } else {  // [_credentials count] == 1
329          
330          NSDictionary *metrics = @{@"height":@(heightForOneRow)};
331          [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|[table]|"  options:0 metrics:metrics views:views]];
332          [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[table(height)]|" options:0 metrics:metrics views:views]];
333          
334          [self setPreferredContentSize:CGSizeMake(0, heightForOneRow)];
335          [table setScrollEnabled: NO];
336          
337      }
338      
339      [self setView:view];
340  }
341  
342  -(void)viewWillAppear:(BOOL)animated
343  {
344      
345      // Select the first cell by default
346      
347      NSDictionary *dict = [_credentials objectAtIndex: 0];
348      _selectedDict = dict;
349      _selectedCell = [NSIndexPath indexPathForItem:0 inSection: 0];
350      SWCItemCell *cell = (SWCItemCell *)[_table cellForRowAtIndexPath: _selectedCell];
351      [cell setTicked: YES];
352      [cell layoutSubviews];
353      [_table selectRowAtIndexPath: _selectedCell animated: NO scrollPosition: UITableViewScrollPositionTop];
354      
355      CFErrorRef error = NULL;
356      audit_token_t auditToken = {};
357      memset(&auditToken, 0, sizeof(auditToken));
358      bool result = swca_set_selection(swca_set_selection_request_id,
359                                       &auditToken, (__bridge CFDictionaryRef)dict, &error);
360      if (!result) {
361          NSLog(@"Unable to select item: %@", [(__bridge NSError*)error localizedDescription]);
362      }
363      
364      [super viewWillAppear:animated];
365  }
366  
367  
368  //
369  // UITableView delegate methods
370  //
371  
372  - (NSInteger)tableView:(UITableView *)table numberOfRowsInSection:(NSInteger)section
373  {
374      return [_credentials count];
375  }
376  
377  - (void) tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
378  {
379      NSUInteger row = indexPath.row;
380      
381      NSDictionary *dict = [_credentials objectAtIndex:row];
382      _selectedDict = dict;
383      
384      CFErrorRef error = NULL;
385      audit_token_t auditToken = {};
386      memset(&auditToken, 0, sizeof(auditToken));
387      bool result = swca_set_selection(swca_set_selection_request_id,
388                                       &auditToken, (__bridge CFDictionaryRef)dict, &error);
389      if (!result) {
390          NSLog(@"Unable to select item: %@", [(__bridge NSError*)error localizedDescription]);
391      }
392      
393      _selectedCell = indexPath;
394      SWCItemCell *cell = (SWCItemCell *)[tableView cellForRowAtIndexPath: indexPath];
395      [cell setTicked: YES];
396      [cell layoutSubviews];
397  }
398  
399  - (void) tableView:(UITableView *)tableView didDeselectRowAtIndexPath:(NSIndexPath *)indexPath
400  {
401      SWCItemCell *cell = (SWCItemCell *)[tableView cellForRowAtIndexPath: indexPath];
402      [cell setTicked: NO];
403      [cell layoutSubviews];
404      
405  }
406  
407  - (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
408  {
409      
410      NSDictionary *dict = [_credentials objectAtIndex:[indexPath row]];
411      SWCItemCell *cell = [[SWCItemCell alloc] initWithDictionary:dict];
412      
413      // show separator on top cell if there's only or or two items
414      if ([_credentials count] <= 2) {
415          cell.showTopSeparator = YES;
416      } else {
417          cell.showSeparator = YES;
418          
419          if (indexPath.row == 0)
420          {
421              cell.showTopSeparator = YES;
422          }
423          
424      }
425      
426      if (_selectedCell == indexPath)
427      {
428          [cell setTicked: YES];
429      } else {
430          [cell setTicked: NO];
431      }
432      
433      return cell;
434  }
435  
436  
437  @end