source: trunk/Sparkle/SUHost.m @ 1046

//
//  SUHost.m
//  Sparkle
//
//  Copyright 2008 Andy Matuschak. All rights reserved.
//

#import "SUHost.h"

#import "SUSystemProfiler.h"
#import <sys/mount.h> // For statfs for isRunningOnReadOnlyVolume

@implementation SUHost

- (id)initWithBundle:(NSBundle *)aBundle
{
    if (aBundle == nil) aBundle = [NSBundle mainBundle];
    if ((self = [super init]))
    {
        bundle = [aBundle retain];
        if (![bundle bundleIdentifier])
            NSLog(@"Sparkle Error: the bundle being updated at %@ has no CFBundleIdentifier! This will cause preference read/write to not work properly.", [bundle bundlePath]);
    }
    return self;
}

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

- (NSString *)description { return [NSString stringWithFormat:@"%@ <%@>", [self class], [self bundlePath]]; }

- (NSBundle *)bundle
{
    return bundle;
}

- (NSString *)bundlePath
{
    return [bundle bundlePath];
}

- (NSString *)name
{
    NSString *name = [bundle objectForInfoDictionaryKey:@"CFBundleDisplayName"];
    if (name) return name;
    
    name = [self objectForInfoDictionaryKey:@"CFBundleName"];
    if (name) return name;
    
    return [[[NSFileManager defaultManager] displayNameAtPath:[bundle bundlePath]] stringByDeletingPathExtension];
}

- (NSString *)version
{
    return [bundle objectForInfoDictionaryKey:@"CFBundleVersion"];
}

- (NSString *)displayVersion
{
    NSString *shortVersionString = [bundle objectForInfoDictionaryKey:@"CFBundleShortVersionString"];
    if (shortVersionString)
        return shortVersionString;
    else
        return [self version]; // Fall back on the normal version string.
}

- (NSImage *)icon
{
    // Cache the application icon.
    NSString *iconPath = [bundle pathForResource:[bundle objectForInfoDictionaryKey:@"CFBundleIconFile"] ofType:@"icns"];
    // According to the OS X docs, "CFBundleIconFile - This key identifies the file containing
    // the icon for the bundle. The filename you specify does not need to include the .icns
    // extension, although it may."
    //
    // However, if it *does* include the '.icns' the above method fails (tested on OS X 10.3.9) so we'll also try:
    if (!iconPath)
        iconPath = [bundle pathForResource:[bundle objectForInfoDictionaryKey:@"CFBundleIconFile"] ofType: nil];
    NSImage *icon = [[[NSImage alloc] initWithContentsOfFile:iconPath] autorelease];
    // Use a default icon if none is defined.
    if (!icon) { icon = [[NSWorkspace sharedWorkspace] iconForFileType:NSFileTypeForHFSTypeCode(kGenericApplicationIcon)]; }
    return icon;
}

- (BOOL)isRunningOnReadOnlyVolume
{   
    struct statfs statfs_info;
    statfs([[bundle bundlePath] fileSystemRepresentation], &statfs_info);
    return (statfs_info.f_flags & MNT_RDONLY);
}

- (BOOL)isBackgroundApplication
{
    ProcessSerialNumber PSN;
    GetCurrentProcess(&PSN);
    NSDictionary * processInfo = (NSDictionary *)ProcessInformationCopyDictionary(&PSN, kProcessDictionaryIncludeAllInformationMask);
    BOOL isElement = [[processInfo objectForKey:@"LSUIElement"] boolValue];
    if (processInfo)
        CFRelease(processInfo);
    return isElement;
}

- (NSString *)publicDSAKey
{
    // Maybe the key is just a string in the Info.plist.
    NSString *key = [bundle objectForInfoDictionaryKey:SUPublicDSAKeyKey];
    if (key) { return key; }
    
    // More likely, we've got a reference to a Resources file by filename:
    NSString *keyFilename = [self objectForInfoDictionaryKey:SUPublicDSAKeyFileKey];
    if (!keyFilename) { return nil; }
    NSError *ignoreErr;
    return [NSString stringWithContentsOfFile:[bundle pathForResource:keyFilename ofType:nil] encoding:NSASCIIStringEncoding error: &ignoreErr];
}

- (NSArray *)systemProfile
{
    return [[SUSystemProfiler sharedSystemProfiler] systemProfileArrayForHost:self];
}

- (id)objectForInfoDictionaryKey:(NSString *)key
{
    return [bundle objectForInfoDictionaryKey:key];
}

- (BOOL)boolForInfoDictionaryKey:(NSString *)key
{
    return [[self objectForInfoDictionaryKey:key] boolValue];
}

- (id)objectForUserDefaultsKey:(NSString *)defaultName
{
    // Under Tiger, CFPreferencesCopyAppValue doesn't get values from NSRegistratioDomain, so anything
    // passed into -[NSUserDefaults registerDefaults:] is ignored.  The following line falls
    // back to using NSUserDefaults, but only if the host bundle is the main bundle.
    if (bundle == [NSBundle mainBundle])
        return [[NSUserDefaults standardUserDefaults] objectForKey:defaultName];
    
    CFPropertyListRef obj = CFPreferencesCopyAppValue((CFStringRef)defaultName, (CFStringRef)[bundle bundleIdentifier]);
#if MAC_OS_X_VERSION_MAX_ALLOWED > MAC_OS_X_VERSION_10_4
    return [NSMakeCollectable(obj) autorelease];
#else
    return [(id)obj autorelease];
#endif  
}

- (void)setObject:(id)value forUserDefaultsKey:(NSString *)defaultName;
{
    // If we're using a .app, we'll use the standard user defaults mechanism; otherwise, we have to get CF-y.
    if (bundle == [NSBundle mainBundle])
    {
        [[NSUserDefaults standardUserDefaults] setObject:value forKey:defaultName];
    }
    else
    {
        CFPreferencesSetValue((CFStringRef)defaultName, value, (CFStringRef)[bundle bundleIdentifier],  kCFPreferencesCurrentUser,  kCFPreferencesAnyHost);
        CFPreferencesSynchronize((CFStringRef)[bundle bundleIdentifier], kCFPreferencesCurrentUser, kCFPreferencesAnyHost);
    }
}

- (BOOL)boolForUserDefaultsKey:(NSString *)defaultName
{
    if (bundle == [NSBundle mainBundle])
        return [[NSUserDefaults standardUserDefaults] boolForKey:defaultName];
    
    BOOL value;
    CFPropertyListRef plr = CFPreferencesCopyAppValue((CFStringRef)defaultName, (CFStringRef)[bundle bundleIdentifier]);
    if (plr == NULL)
        value = NO;
    else
    {
        value = (BOOL)CFBooleanGetValue((CFBooleanRef)plr);
        CFRelease(plr);
    }
    return value;
}

- (void)setBool:(BOOL)value forUserDefaultsKey:(NSString *)defaultName
{
    // If we're using a .app, we'll use the standard user defaults mechanism; otherwise, we have to get CF-y.
    if (bundle == [NSBundle mainBundle])
    {
        [[NSUserDefaults standardUserDefaults] setBool:value forKey:defaultName];
    }
    else
    {
        CFPreferencesSetValue((CFStringRef)defaultName, (CFBooleanRef)[NSNumber numberWithBool:value], (CFStringRef)[bundle bundleIdentifier],  kCFPreferencesCurrentUser,  kCFPreferencesAnyHost);
        CFPreferencesSynchronize((CFStringRef)[bundle bundleIdentifier], kCFPreferencesCurrentUser, kCFPreferencesAnyHost);
    }
}

- (id)objectForKey:(NSString *)key {
    return [self objectForUserDefaultsKey:key] ?: [self objectForInfoDictionaryKey:key];
}

- (BOOL)boolForKey:(NSString *)key {
    return [self objectForUserDefaultsKey:key] ? [self boolForUserDefaultsKey:key] : [self boolForInfoDictionaryKey:key];
}

+ (NSString *)systemVersionString
{
    // This returns a version string of the form X.Y.Z
    // There may be a better way to deal with the problem that gestaltSystemVersionMajor
    //  et al. are not defined in 10.3, but this is probably good enough.
    NSString* verStr = nil;
#if MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_4
    SInt32 major, minor, bugfix;
    OSErr err1 = Gestalt(gestaltSystemVersionMajor, &major);
    OSErr err2 = Gestalt(gestaltSystemVersionMinor, &minor);
    OSErr err3 = Gestalt(gestaltSystemVersionBugFix, &bugfix);
    if (!err1 && !err2 && !err3)
    {
        verStr = [NSString stringWithFormat:@"%d.%d.%d", major, minor, bugfix];
    }
    else
#endif
    {
        NSString *versionPlistPath = @"/System/Library/CoreServices/SystemVersion.plist";
        verStr = [[[NSDictionary dictionaryWithContentsOfFile:versionPlistPath] objectForKey:@"ProductVersion"] retain];
    }
    return verStr;
}

@end