DomainListWindowController.m

//
//  DomainListWindowController.m
//  SelfControl
//
//  Created by Charlie Stigler on 2/7/09.
//  Copyright 2009 Eyebeam. 

// This file is part of SelfControl.
// 
// SelfControl is free software:  you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of 
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#import "DomainListWindowController.h"

@implementation DomainListWindowController

- (DomainListWindowController*)init {
  if(self = [super initWithWindowNibName:@"DomainList"]) {
      
    defaults_ = [NSUserDefaults standardUserDefaults];
    
    NSArray* curArray = [defaults_ arrayForKey: @"HostBlacklist"];
    if(curArray == nil)
      domainList_ = [NSMutableArray arrayWithCapacity: 10];
    else
      domainList_ = [curArray mutableCopy];
      
    [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  }
  
  return self;
}

- (void)awakeFromNib {
  [[self window] makeKeyAndOrderFront: self];
}

- (IBAction)addDomain:(id)sender
{
  [domainList_ addObject:@""];
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [domainListTableView_ reloadData];
  [domainListTableView_ selectRow:([domainList_ count] - 1)
             byExtendingSelection:NO];
  [domainListTableView_ editColumn: 0 row:([domainList_ count] - 1)
                         withEvent:nil
                            select:YES];
}

- (IBAction)removeDomain:(id)sender
{
  NSIndexSet* selected = [domainListTableView_ selectedRowIndexes];
  
  // This isn't the most efficient way to do this, but the code is much cleaner
  // than other methods and the domain blacklist will probably never be large
  // enough for it to be an issue.
  unsigned int index = [selected firstIndex];
  int shift = 0;
  while (index != NSNotFound) {
    if (index < 0 || (index - shift) >= [domainList_ count])
      break;
    [domainList_ removeObjectAtIndex: index - shift];
    shift++;
    index = [selected indexGreaterThanIndex: index];
  }
  
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [domainListTableView_ reloadData];
  
  [[NSNotificationCenter defaultCenter] postNotificationName: @"SCConfigurationChangedNotification"
                                                      object: self];
}

- (int)numberOfRowsInTableView:(NSTableView *)aTableView {
  return [domainList_ count];
}

- (id)tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *)aTableColumn row:(int)rowIndex {
  return [domainList_ objectAtIndex:rowIndex];
}

- (void)tableView:(NSTableView *)aTableView
   setObjectValue:(id)theObject
   forTableColumn:(NSTableColumn *)aTableColumn
              row:(int)rowIndex {
  // All of this is just code to standardize and clean up the input value.
  // This'll remove whitespace and lowercase the string.
  NSString* str = [[theObject stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]] lowercaseString];
  
  if([str rangeOfCharacterFromSet: [NSCharacterSet newlineCharacterSet]].location != NSNotFound) {
    // only hits LF linebreaks, but componentsSeparatedByCharacterSet won't work on 10.4
    NSArray* listComponents = [str componentsSeparatedByString: @"\n"];
    
    for(int i = 0; i < [listComponents count]; i++) {
      if(i == 0) {
        [self tableView: aTableView setObjectValue: [listComponents objectAtIndex: i] forTableColumn: aTableColumn row: rowIndex];
      }
      else {
        [domainList_ addObject:@""];
        [self tableView: aTableView setObjectValue: [listComponents objectAtIndex: i] forTableColumn: aTableColumn row: [domainList_ count] - 1];
      }
    }
    
    [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
    [domainListTableView_ reloadData];
    
    return;
  }
  
  // Remove "http://" if a user tried to put that in
  NSArray* splitString = [str componentsSeparatedByString: @"http://"];
  for(int i = 0; i < [splitString count]; i++) {
    if(![[splitString objectAtIndex: i] isEqual: @""]) {
      str = [splitString objectAtIndex: i];
      break;
    }
  }
  
  // Remove "https://" if a user tried to put that in
  splitString = [str componentsSeparatedByString: @"https://"];
  for(int i = 0; i < [splitString count]; i++) {
    if(![[splitString objectAtIndex: i] isEqual: @""]) {
      str = [splitString objectAtIndex: i];
      break;
    }
  }  
  
  // Remove URL login names/passwords (username:password@host) if a user tried to put that in
  splitString = [str componentsSeparatedByString: @"@"];
  str = [splitString lastObject];
  
  // Delete anything after a "/" in case a user tried to copy-paste a web address.
  // str = [[str componentsSeparatedByString: @"/"] objectAtIndex: 0];
  
  int maskLength = -1;
  int portNum = -1;
  
  splitString = [str componentsSeparatedByString: @"/"];

  str = [splitString objectAtIndex: 0];
  
  NSString* stringToSearchForPort = str;
  
  if([splitString count] >= 2) {
    maskLength = [[splitString objectAtIndex: 1] intValue];
    // If the int value is 0, we couldn't find a valid integer representation
    // in the split off string
    if(maskLength == 0)
      maskLength = -1;
    
    stringToSearchForPort = [splitString objectAtIndex: 1];
  }
  
  splitString = [stringToSearchForPort componentsSeparatedByString: @":"];
  
  if(stringToSearchForPort == str) {
    str = [splitString objectAtIndex: 0];
  }
  
  if([splitString count] >= 2) {
    portNum = [[splitString objectAtIndex: 1] intValue];
    // If the int value is 0, we couldn't find a valid integer representation
    // in the split off string
    if(portNum == 0)
      portNum = -1;
  }
    
  if([str isEqual: @""] && portNum == -1)
    [domainList_ removeObjectAtIndex: rowIndex];
  else {
    NSString* maskString;
    NSString* portString;
    if(maskLength == -1)
      maskString = @"";
    else
      maskString = [NSString stringWithFormat: @"/%d", maskLength];
    if(portNum == -1)
      portString = @"";
    else
      portString = [NSString stringWithFormat: @":%d", portNum];
    str = [NSString stringWithFormat: @"%@%@%@", str, maskString, portString];
    [domainList_ replaceObjectAtIndex:rowIndex withObject:str];
  }
  
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [aTableView reloadData];
  [[NSNotificationCenter defaultCenter] postNotificationName: @"SCConfigurationChangedNotification"
                                                      object: self];  
}

- (void)tableView:(NSTableView *)tableView
  willDisplayCell:(id)cell
   forTableColumn:(NSTableColumn *)tableColumn
              row:(int)row {
  // this method is really inefficient. rewrite/optimize later.
  [defaults_ synchronize];
  
  // Initialize the cell's text color to black
  [cell setTextColor: [NSColor blackColor]];
  NSString* str = [[cell title] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
  if([str isEqual: @""]) return;
  if([defaults_ boolForKey: @"HighlightInvalidHosts"]) {
    // Validate the value as either an IP or a hostname.  In case of failure,
    // we'll make its text color red.
        
    int maskLength = -1;
    int portNum = -1;
    
    NSArray* splitString = [str componentsSeparatedByString: @"/"];
    
    str = [splitString objectAtIndex: 0];
    
    NSString* stringToSearchForPort = str;
    
    if([splitString count] >= 2) {
      maskLength = [[splitString objectAtIndex: 1] intValue];
      // If the int value is 0, we couldn't find a valid integer representation
      // in the split off string
      if(maskLength == 0)
        maskLength = -1;
      
      stringToSearchForPort = [splitString objectAtIndex: 1];
    }
    
    splitString = [stringToSearchForPort componentsSeparatedByString: @":"];
    
    if(stringToSearchForPort == str) {
      str = [splitString objectAtIndex: 0];
    }
    
    if([splitString count] >= 2) {
      portNum = [[splitString objectAtIndex: 1] intValue];
      // If the int value is 0, we couldn't find a valid integer representation
      // in the split off string
      if(portNum == 0)
        portNum = -1;
    }
        
    BOOL isIP;
    
    NSString* ipValidationRegex = @"^([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\\.([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$";
    NSPredicate *ipRegexTester = [NSPredicate
                                  predicateWithFormat:@"SELF MATCHES %@",
                                  ipValidationRegex];      
    isIP = [ipRegexTester evaluateWithObject: str];
    
    if(!isIP) {
      NSString* hostnameValidationRegex = @"^([a-zA-Z0-9]([a-zA-Z0-9\\-]{0,61}[a-zA-Z0-9])?\\.)+[a-zA-Z]{2,6}$";
      NSPredicate *hostnameRegexTester = [NSPredicate
                                          predicateWithFormat:@"SELF MATCHES %@",
                                          hostnameValidationRegex
                                          ];
      if(![hostnameRegexTester evaluateWithObject: str] && ![str isEqualToString: @"*"] && ![str isEqualToString: @""]) {
        [cell setTextColor: [NSColor redColor]];
        return;
      }
    }
    
    // We shouldn't have a mask length if it's not an IP, fail
    if(!isIP && maskLength != -1) {
      [cell setTextColor: [NSColor redColor]];
      return;
    }
    
    if(([str isEqualToString: @"*"] || [str isEqualToString: @""]) && portNum == -1) {
      [cell setTextColor: [NSColor redColor]];
      return;
    }
    
    [cell setTextColor: [NSColor blackColor]];
  }
}

- (IBAction)importIncomingMailServersFromThunderbird:(id)sender {
  NSArray* arr = [HostImporter incomingMailHostnamesFromThunderbird];
  for(int i = 0; i < [arr count]; i++) {
    // Check for dupes
    if(![domainList_ containsObject: [arr objectAtIndex: i]])
      [domainList_ addObject: [arr objectAtIndex: i]];
  }
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [domainListTableView_ reloadData];
  [[NSNotificationCenter defaultCenter] postNotificationName: @"SCConfigurationChangedNotification"
                                                      object: self];
}

- (IBAction)importOutgoingMailServersFromThunderbird:(id)sender {
  NSArray* arr = [HostImporter outgoingMailHostnamesFromThunderbird];
  for(int i = 0; i < [arr count]; i++) {
    // Check for dupes
    if(![domainList_ containsObject: [arr objectAtIndex: i]])
      [domainList_ addObject: [arr objectAtIndex: i]];
  }
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [domainListTableView_ reloadData];
  [[NSNotificationCenter defaultCenter] postNotificationName: @"SCConfigurationChangedNotification"
                                                      object: self];
}


- (IBAction)importIncomingMailServersFromMail:(id)sender {
  NSArray* arr = [HostImporter incomingMailHostnamesFromMail];
  for(int i = 0; i < [arr count]; i++) {
    // Check for dupes
    if(![domainList_ containsObject: [arr objectAtIndex: i]])
      [domainList_ addObject: [arr objectAtIndex: i]];
  }
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [domainListTableView_ reloadData];
  [[NSNotificationCenter defaultCenter] postNotificationName: @"SCConfigurationChangedNotification"
                                                      object: self];
}


- (IBAction)importOutgoingMailServersFromMail:(id)sender {
  NSArray* arr = [HostImporter outgoingMailHostnamesFromMail];
  for(int i = 0; i < [arr count]; i++) {
    // Check for dupes
    if(![domainList_ containsObject: [arr objectAtIndex: i]])
      [domainList_ addObject: [arr objectAtIndex: i]];
  }
  [defaults_ setObject: domainList_ forKey: @"HostBlacklist"];
  [domainListTableView_ reloadData];
  [[NSNotificationCenter defaultCenter] postNotificationName: @"SCConfigurationChangedNotification"
                                                      object: self];
}

- (void)dealloc {
  [super dealloc];
}

@end