ThunderbirdPreferenceParser.m

//
//  ThunderbirdPreferenceParser.m
//  SelfControl
//
//  Created by Charlie Stigler on 2/17/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 "ThunderbirdPreferenceParser.h"

NSString* const kThunderbirdSupportFolderPath = @"~/Library/Thunderbird";

@implementation ThunderbirdPreferenceParser

+ (NSString*)pathToSupportFolder {
  return [kThunderbirdSupportFolderPath stringByExpandingTildeInPath];
}

+ (BOOL)thunderbirdIsInstalled {
  NSString* profilesIniPath = [[self pathToSupportFolder] stringByAppendingPathComponent: @"profiles.ini"];
  return [[NSFileManager defaultManager] isReadableFileAtPath: profilesIniPath];
}

+ (NSString*)pathToDefaultProfile {
  NSString* profilesIniPath = [[self pathToSupportFolder] stringByAppendingPathComponent: @"profiles.ini"];
  if(![[NSFileManager defaultManager] isReadableFileAtPath: profilesIniPath])
    return nil;
  
  NSString* profilesIniContents = [NSString stringWithContentsOfFile: profilesIniPath];
  NSScanner* profilesIniScanner = [NSScanner scannerWithString: profilesIniContents];
  NSMutableArray* profiles = [NSMutableArray arrayWithCapacity: 1];   
  
  [profilesIniScanner scanUpToString: @"[Profile" intoString: NULL];
  [profilesIniScanner scanUpToCharactersFromSet: [NSCharacterSet newlineCharacterSet]
                                     intoString: NULL];
  
  if([profilesIniScanner isAtEnd])
    return nil;
  
  do { // for each profile...
    NSString* subString;
    NSMutableDictionary* profile = [NSMutableDictionary dictionaryWithCapacity: 4];
    [profilesIniScanner scanUpToString: @"\n\n"
                            intoString: &subString];
    
    NSScanner* subStringScanner = [NSScanner scannerWithString: subString];
    
    while(![subStringScanner isAtEnd]) { // scan for more attributes
      NSString* key;
      
      // Read in the key and value
      [subStringScanner scanUpToCharactersFromSet: [NSCharacterSet characterSetWithCharactersInString: @"="]
                        
                                       intoString: &key];
      
      [subStringScanner scanCharactersFromSet: [NSCharacterSet characterSetWithCharactersInString: @"="]
                                   intoString: NULL];
      
      // deal with each possible attribute
      if([key isEqual: @"Name"]) {
        NSString* profileName;
        [subStringScanner scanUpToCharactersFromSet: [NSCharacterSet newlineCharacterSet]
                                         intoString: &profileName];
        [profile setObject: profileName forKey: @"Name"];
      }
      else if([key isEqual: @"IsRelative"]) {
        int isRelative = 1;
        [subStringScanner scanInt: &isRelative];
        [profile setObject: [NSNumber numberWithInt: isRelative] forKey: @"IsRelative"];
      }
      else if([key isEqual: @"Default"]) {
        int isDefault = 0;
        [subStringScanner scanInt: &isDefault];
        [profile setObject: [NSNumber numberWithInt: isDefault] forKey: @"Default"];
      }
      else if([key isEqual: @"Path"]) {
        NSString* path;
        [subStringScanner scanUpToCharactersFromSet: [NSCharacterSet newlineCharacterSet]
                                         intoString: &path];
        [profile setObject: path forKey: @"Path"];
      }
    }
    
    [profiles addObject: profile]; // add the profile into the array of profiles
  } while(![profilesIniScanner isAtEnd]);
  
  NSDictionary* defaultProfile = nil;
  
  for(int i = 0; i < [profiles count]; i++) {
    NSDictionary* p = [profiles objectAtIndex: i];
    if([[p objectForKey: @"Default"] isEqual: [NSNumber numberWithInt: 1]]) {
      defaultProfile = p;
      break;
    }
    if([[p objectForKey: @"Name"] isEqual: @"default"])
      defaultProfile = p;
  }
  
  if(defaultProfile == nil) {
    defaultProfile = [profiles objectAtIndex: 0];
  }
  if(defaultProfile == nil)
    return nil;
  
  NSString* pathToProfile = [defaultProfile objectForKey: @"Path"];
  if(pathToProfile == nil)
    return nil;
  
  if([[defaultProfile objectForKey: @"IsRelative"] isEqual: [NSNumber numberWithInt: 0]])
    return [pathToProfile stringByStandardizingPath];
  else
    return [[[self pathToSupportFolder] stringByAppendingPathComponent: pathToProfile] stringByStandardizingPath];  
}

+ (NSString*)pathToPrefsJsFile {
  NSString* pathToDefaultProfile = [self pathToDefaultProfile];
  if(pathToDefaultProfile == nil)
    return nil;
  return [pathToDefaultProfile stringByAppendingPathComponent: @"prefs.js"];
}

+ (NSArray*)incomingHostnames {
  NSString* pathToPrefsJsFile = [self pathToPrefsJsFile];
  if(pathToPrefsJsFile == nil)
    return [NSArray array];
  if(![[NSFileManager defaultManager] isReadableFileAtPath: pathToPrefsJsFile])
    return [NSArray array];
  
  // NSArray* prefsJsLines = [[NSString stringWithContentsOfFile: pathToPrefsJsFile] componentsSeparatedByCharactersInSet: [NSCharacterSet newlineCharacterSet]];
  // The old implementation of this was better.  The new one misses alternative
  // line breaks like \r.  10.4 doesn't have a componentSeparatedByCharactersInSet
  // method though, so this'll have to do.
  NSArray* prefsJsLines = [[NSString stringWithContentsOfFile: pathToPrefsJsFile] componentsSeparatedByString: @"\n"];
  NSMutableArray* hostnames = [NSMutableArray arrayWithCapacity: 10];
  
  for(int i = 0; i < [prefsJsLines count]; i++) {
    NSString* line = [prefsJsLines objectAtIndex: i];
    // All of the asterisks are necessary for globbing so that any amount of
    // whitespace will work.
    if([line isLike: @"*user_pref(*\"mail.server.server*.hostname\"*,*\"*\"*)*;*"]) {
      NSArray* parts = [line  componentsSeparatedByString: @"\""];
      // If the hostname is "Local Folders", it's a special Thunderbird thing,
      // and obviously not something we can block.
      if([parts count] >= 4 && ![[parts objectAtIndex: 3] isEqual: @"Local Folders"]) {
        [hostnames addObject: [[parts objectAtIndex: 3] stringByAppendingString: @":110"]];
      }
    }
    else if([line isLike: @"*user_pref(*\"mail.server.server*.port\"*,*)*;*"]) {
      NSArray* parts = [line  componentsSeparatedByString: @","];
      parts = [[parts objectAtIndex: 1]  componentsSeparatedByString: @")"];
      int portNumber = [[[parts objectAtIndex: 0] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]] intValue];
      NSString* alteredHost = [[[hostnames objectAtIndex: ([hostnames count] - 1)]  componentsSeparatedByString: @":"] objectAtIndex: 0];
      alteredHost = [alteredHost stringByAppendingFormat: @":%d", portNumber];
      [hostnames replaceObjectAtIndex: ([hostnames count] - 1) withObject: alteredHost];
    }
  }
  
  return hostnames;
}

+ (NSArray*)outgoingHostnames {
  NSString* pathToPrefsJsFile = [self pathToPrefsJsFile];
  if(pathToPrefsJsFile == nil)
    return [NSArray array];
  if(![[NSFileManager defaultManager] isReadableFileAtPath: pathToPrefsJsFile])
    return [NSArray array];
  
  NSArray* prefsJsLines = [[NSString stringWithContentsOfFile: pathToPrefsJsFile]  componentsSeparatedByString: @"\n"];
  NSMutableArray* hostnames = [NSMutableArray arrayWithCapacity: 10];
  
  for(int i = 0; i < [prefsJsLines count]; i++) {
    NSString* line = [prefsJsLines objectAtIndex: i];
    if([line isLike: @"*user_pref(*\"mail.smtpserver.smtp*.hostname\"*,*\"*\"*)*;*"]) {
      NSArray* parts = [line componentsSeparatedByString: @"\""];
      if([parts count] >= 4 && ![[parts objectAtIndex: 3] isEqual: @"Local Folders"]) {
        [hostnames addObject: [[parts objectAtIndex: 3] stringByAppendingString: @":25"]];
      }
    }
    // If there's a port number, add it to the last added hostname.  Yes, it could
    // technically be associated with a different hostname, but only if the user
    // was manually editing the preferences file and changed it stupidly.
    else if([line isLike: @"*user_pref(*\"mail.smtpserver.smtp*.port\"*,*)*;*"]) {
      NSArray* parts = [line  componentsSeparatedByString: @","];
      parts = [[parts objectAtIndex: 1] componentsSeparatedByString: @")"];
      int portNumber = [[[parts objectAtIndex: 0] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]] intValue];
      NSString* alteredHost = [[[hostnames objectAtIndex: ([hostnames count] - 1)] componentsSeparatedByString: @":"] objectAtIndex: 0];
      alteredHost = [alteredHost stringByAppendingFormat: @":%d", portNumber];
      [hostnames replaceObjectAtIndex: ([hostnames count] - 1) withObject: alteredHost];
    }
  }
  
  return hostnames;
}

@end