SharedWebCredentialViewService/SWCViewController.m

   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
  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 colorWithWhite:0.1 alpha:0.005];
  92
  93         self.textLabel.textColor = [UIColor blackColor];
  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 : @"--";
 100
 101         self.detailTextLabel.textColor = [UIColor darkGrayColor];
 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 : @"--";
 108
 109         self.backgroundView = [[UIView alloc] init];
 110         self.backgroundView.backgroundColor = self.backgroundColor;
 111
 112         self.imageView.image = [self _checkmarkImage: NO];
 113         self.imageView.hidden = YES;
 114
 115     }
 116
 117     return self;
 118 }
 119
 120 - (void)setTicked: (BOOL) selected
 121 {
 122     _isTicked = selected;
 123 }
 124
 125 - (void)layoutSubviews
 126 {
 127
 128     if (_bottomLine) {
 129         CGFloat scale = [[UIScreen mainScreen] scale];
 130         [_bottomLine setFrame:CGRectMake(0, self.frame.size.height - (1 / scale), self.frame.size.width, 1 / scale)];
 131     }
 132
 133     if (_bottomLineSelected) {
 134         CGFloat scale = [[UIScreen mainScreen] scale];
 135         [_bottomLineSelected setFrame:CGRectMake(0, self.frame.size.height - (1 / scale), self.frame.size.width, 1 / scale)];
 136     }
 137
 138     if (_topLine) {
 139         CGFloat scale = [[UIScreen mainScreen] scale];
 140         [_topLine setFrame:CGRectMake(0, 0, self.frame.size.width, 1 / scale)];
 141     }
 142
 143     if (_topLineSelected) {
 144         CGFloat scale = [[UIScreen mainScreen] scale];
 145         [_topLineSelected setFrame:CGRectMake(0, 0, self.frame.size.width, 1 / scale)];
 146     }
 147
 148     if (_isTicked)
 149     {
 150         self.imageView.hidden = NO;
 151     } else {
 152         self.imageView.hidden = YES;
 153     }
 154
 155     [super layoutSubviews];
 156
 157 }
 158
 159 - (void)setShowSeparator:(BOOL)showSeparator {
 160     if (_showSeparator != showSeparator) {
 161         _showSeparator = showSeparator;
 162
 163         if (_showSeparator) {
 164             if (!_bottomLine) {
 165                 CGRect rectZero = CGRectMake(0, 0, 0, 0);
 166                 _bottomLine = [[UIView alloc] initWithFrame:rectZero];
 167                 _bottomLine.backgroundColor = [UIColor colorWithWhite:.5 alpha:.5];
 168                 [self.backgroundView addSubview:_bottomLine];
 169             }
 170             if (!_bottomLineSelected) {
 171                 CGRect rectZero = CGRectMake(0, 0, 0, 0);
 172                 _bottomLineSelected = [[UIView alloc] initWithFrame:rectZero];
 173                 _bottomLineSelected.backgroundColor = [UIColor colorWithWhite:.5 alpha:.5];
 174                 [self.selectedBackgroundView addSubview: _bottomLineSelected];
 175             }
 176
 177         } else {
 178             if (_bottomLine) {
 179                 [_bottomLine removeFromSuperview];
 180                 _bottomLine = nil;
 181             }
 182             if (_bottomLineSelected) {
 183                 [_bottomLineSelected removeFromSuperview];
 184                 _bottomLineSelected = nil;
 185             }
 186         }
 187     }
 188 }
 189
 190 - (void)setShowTopSeparator:(BOOL)showTopSeparator {
 191     if (_showTopSeparator != showTopSeparator) {
 192         _showTopSeparator = showTopSeparator;
 193
 194         if (_showTopSeparator) {
 195             if (!_topLine) {
 196                 CGRect rectZero = CGRectMake(0, 0, 0, 0);
 197                 _topLine = [[UIView alloc] initWithFrame:rectZero];
 198                 _topLine.backgroundColor = [UIColor colorWithWhite:.5 alpha:.5];
 199                 [self.backgroundView addSubview:_topLine];
 200             }
 201             if (!_topLineSelected) {
 202                 CGRect rectZero = CGRectMake(0, 0, 0, 0);
 203                 _topLineSelected = [[UIView alloc] initWithFrame:rectZero];
 204                 _topLineSelected.backgroundColor = [UIColor colorWithWhite:.5 alpha:.5];
 205                 [self.selectedBackgroundView addSubview: _topLineSelected];
 206             }
 207
 208         } else {
 209             if (_topLine) {
 210                 [_topLine removeFromSuperview];
 211                 _topLine = nil;
 212             }
 213             if (_topLineSelected) {
 214                 [_topLineSelected removeFromSuperview];
 215                 _topLineSelected = nil;
 216             }
 217         }
 218     }
 219 }
 220
 221 @end
 222
 223
 224 //
 225 // SWCViewController
 226 //
 227
 228 @interface SWCViewController ()
 229 {
 230     NSMutableArray       *_credentials; // array of NSDictionary
 231     UILabel              *_topLabel;
 232     UILabel              *_middleLabel;
 233     UITableView          *_table;
 234     NSDictionary         *_selectedDict;
 235     NSIndexPath          *_selectedCell;
 236 }
 237
 238 @end
 239
 240 @implementation SWCViewController
 241
 242 - (NSDictionary *)selectedItem
 243 {
 244     return _selectedDict;
 245 }
 246
 247 - (void)setCredentials:(NSArray *)inArray
 248 {
 249     NSMutableArray *credentials = [[NSMutableArray alloc] initWithArray:inArray];
 250     [credentials sortUsingSelector:@selector(compareCredentialDictionaryAscending:)];
 251     _credentials = credentials;
 252     if (_table)
 253         [_table reloadData];
 254 }
 255
 256 - (void)_enableTable
 257 {
 258     [_table setUserInteractionEnabled:YES];
 259 }
 260
 261 - (UITableView *)tableView
 262 {
 263     if (_table == nil) {
 264         _table = [[UITableView alloc] init];
 265         [_table setTranslatesAutoresizingMaskIntoConstraints:NO];
 266         [_table setAutoresizingMask:UIViewAutoresizingNone];
 267         [_table setBackgroundColor:[UIColor clearColor]];
 268         [_table setSeparatorStyle:UITableViewCellSeparatorStyleNone];
 269     }
 270     [_table sizeToFit];
 271
 272     return (UITableView *)_table;
 273 }
 274
 275 -(void)loadView
 276 {
 277
 278     UIView* view = [[UIView alloc] init];
 279
 280     UITableView* table = [self tableView];
 281     [table setDelegate: self];
 282     [table setDataSource: self];
 283
 284     [view addSubview: table];
 285
 286     CFErrorRef error = NULL;
 287     audit_token_t auditToken = {};
 288     memset(&auditToken, 0, sizeof(auditToken));
 289     CFArrayRef credentialList = swca_copy_pairs(swca_copy_pairs_request_id, &auditToken, &error);
 290     if (error) {
 291         NSLog(@"Unable to get accounts: %@", [(__bridge NSError*)error localizedDescription]);
 292     }
 293
 294     [self setCredentials:(__bridge NSArray*)credentialList];
 295     if (credentialList) {
 296         CFRelease(credentialList);
 297     }
 298
 299
 300     NSDictionary* views = NSDictionaryOfVariableBindings(table);
 301
 302     if ([_credentials count] > 2)
 303     {
 304
 305         NSDictionary *metrics = @{@"height":@120.0};
 306         [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|-[table]-|"  options:0 metrics:metrics views:views]];
 307         [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[table(height)]-|" options:0 metrics:metrics views:views]];
 308
 309         CGFloat scale = [[UIScreen mainScreen] scale];
 310         table.layer.borderWidth = 1.0 / scale;
 311         table.layer.borderColor = [UIColor colorWithWhite:.5 alpha:.5].CGColor;
 312
 313         [self setPreferredContentSize:CGSizeMake(0,140)];
 314
 315     } else if ([_credentials count] == 2) {
 316
 317         NSDictionary *metrics = @{@"height":@90.0};
 318         [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|[table]|"  options:0 metrics:metrics views:views]];
 319         [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[table(height)]-|" options:0 metrics:metrics views:views]];
 320
 321         [self setPreferredContentSize:CGSizeMake(0,90)];
 322         [table setScrollEnabled: NO];
 323
 324     } else {  // [_credentials count] == 1
 325
 326         NSDictionary *metrics = @{@"height":@45.0};
 327         [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|[table]|"  options:0 metrics:metrics views:views]];
 328         [view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|[table(height)]|" options:0 metrics:metrics views:views]];
 329
 330         [self setPreferredContentSize:CGSizeMake(0,45)];
 331         [table setScrollEnabled: NO];
 332
 333     }
 334
 335     [self setView:view];
 336 }
 337
 338 -(void)viewWillAppear:(BOOL)animated
 339 {
 340
 341     // Select the first cell by default
 342
 343     NSDictionary *dict = [_credentials objectAtIndex: 0];
 344     _selectedDict = dict;
 345     _selectedCell = [NSIndexPath indexPathForItem:0 inSection: 0];
 346     SWCItemCell *cell = (SWCItemCell *)[_table cellForRowAtIndexPath: _selectedCell];
 347     [cell setTicked: YES];
 348     [_table selectRowAtIndexPath: _selectedCell animated: NO scrollPosition: UITableViewScrollPositionTop];
 349
 350     CFErrorRef error = NULL;
 351     audit_token_t auditToken = {};
 352     memset(&auditToken, 0, sizeof(auditToken));
 353     bool result = swca_set_selection(swca_set_selection_request_id,
 354                                      &auditToken, (__bridge CFDictionaryRef)dict, &error);
 355     if (!result) {
 356         NSLog(@"Unable to select item: %@", [(__bridge NSError*)error localizedDescription]);
 357     }
 358
 359     [super viewWillAppear:animated];
 360 }
 361
 362
 363 //
 364 // UITableView delegate methods
 365 //
 366
 367 - (NSInteger)tableView:(UITableView *)table numberOfRowsInSection:(NSInteger)section
 368 {
 369     return [_credentials count];
 370 }
 371
 372 - (void) tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
 373 {
 374     NSUInteger row = indexPath.row;
 375
 376     NSDictionary *dict = [_credentials objectAtIndex:row];
 377     _selectedDict = dict;
 378
 379     CFErrorRef error = NULL;
 380     audit_token_t auditToken = {};
 381     memset(&auditToken, 0, sizeof(auditToken));
 382     bool result = swca_set_selection(swca_set_selection_request_id,
 383                                      &auditToken, (__bridge CFDictionaryRef)dict, &error);
 384     if (!result) {
 385         NSLog(@"Unable to select item: %@", [(__bridge NSError*)error localizedDescription]);
 386     }
 387
 388     _selectedCell = indexPath;
 389     SWCItemCell *cell = (SWCItemCell *)[tableView cellForRowAtIndexPath: indexPath];
 390     [cell setTicked: YES];
 391     [cell layoutSubviews];
 392 }
 393
 394 - (void) tableView:(UITableView *)tableView didDeselectRowAtIndexPath:(NSIndexPath *)indexPath
 395 {
 396     SWCItemCell *cell = (SWCItemCell *)[tableView cellForRowAtIndexPath: indexPath];
 397     [cell setTicked: NO];
 398     [cell layoutSubviews];
 399
 400 }
 401
 402 - (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
 403 {
 404
 405     NSDictionary *dict = [_credentials objectAtIndex:[indexPath row]];
 406     SWCItemCell *cell = [[SWCItemCell alloc] initWithDictionary:dict];
 407
 408     // show separator on top cell if there's only or or two items
 409     if ([_credentials count] <= 2) {
 410         cell.showTopSeparator = YES;
 411     } else {
 412         cell.showSeparator = YES;
 413
 414         if (indexPath.row == 0)
 415         {
 416             cell.showTopSeparator = YES;
 417         }
 418
 419     }
 420
 421     if (_selectedCell == indexPath)
 422     {
 423         [cell setTicked: YES];
 424     } else {
 425         [cell setTicked: NO];
 426     }
 427
 428     return cell;
 429 }
 430
 431
 432 @end