Added new methods to return numbers. Cleaned up code.
[ITFoundation.git] / ITAppleEventCenter.m
index ceaf358..78eb8f8 100755 (executable)
@@ -21,7 +21,7 @@ static ITAppleEventCenter *_sharedAECenter = nil;
 
 + (id)sharedCenter
 {
-    if( _sharedAECenter != nil ) {
+    if( _sharedAECenter ) {
         return _sharedAECenter;
     } else {   
         _sharedAECenter = [[ITAppleEventCenter alloc] init];
@@ -29,49 +29,33 @@ static ITAppleEventCenter *_sharedAECenter = nil;
     }
 }
 
-- (NSString*)runAEWithRequestedKey:(NSString*)key eventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
+- (NSString*)sendAEWithRequestedKey:(NSString*)key eventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
 {
+    //Add error checking...
     AEEventClass eClass = *((unsigned long*)[eventClass UTF8String]);
-    AEEventID    eID    = *((unsigned long*)[eventID UTF8String]);
+    AEEventID   eID    = *((unsigned long*)[eventID UTF8String]);
     
     const char *sendString = [[NSString stringWithFormat:@"'----':obj { form:'prop', want:type('prop'), seld:type('%s'), from:'null'() }", [key UTF8String]] UTF8String];
     NSString  *_finalString;
     
-    // Variables for building and sending the event    
     AppleEvent sendEvent, replyEvent;
-    //AEDesc replyDesc;
     AEIdleUPP upp = NewAEIdleUPP(&MyAEIdleCallback);
     
-    // Variables for getting the result
-    char *charResult;
     DescType resultType;
     Size resultSize, charResultSize;
     int pid;
     
-    // Variables for error checking
     AEBuildError buildError;
     OSStatus err;
     OSErr err2, err3;
-    //Ptr buffer;
-    //Size length;
-    
-    // Start Code
-    // ^ Most pointless comment EVAR!
-    // ^^ Nope, that one is
     
-    if (GetProcessPID(&psn, &pid) == noErr) {
-        if (pid ==0) {
-            NSLog(@"Process doesn't exist! Exiting.");
-            return;
-        }
-    } else {
-        NSLog(@"Error getting PID of application to send to! Exiting.");
-        return;
+    if ((GetProcessPID(&psn, &pid) == noErr) && (pid == 0)) {
+        NSLog(@"Error getting PID of application! Exiting.");
+        return nil;
     }
     
     NSLog(@"_sendString: %s", sendString);
     
-    // The problem is, if I use eClass and eID in place of 'core' and 'getd' respectively, It won't build a valid AppleEvent :'(
     err = AEBuildAppleEvent(eClass, eID, typeProcessSerialNumber,(ProcessSerialNumber*)&psn, sizeof(ProcessSerialNumber), kAutoGenerateReturnID, 0, &sendEvent, &buildError, sendString);
     
     [self printCarbonDesc:&sendEvent];
@@ -87,51 +71,310 @@ static ITAppleEventCenter *_sharedAECenter = nil;
     if (err) {
         NSLog(@"Send Error: %i",err);
     } else {
-
-        unichar* result=0;
+        unichar *result = 0;
         
         err2 = AESizeOfParam(&replyEvent, keyDirectObject, &resultType, &resultSize);
-        result=malloc(resultSize);
-        //err2 = AEGetParamDesc(&replyEvent, keyDirectObject, typeWildCard, &replyDesc);
-        
-        if (err2) { NSLog(@"Error After AESizeOfParam: %i", err2); } else {
-        //if (err2) { NSLog(@"Error After AEGetParamDesc: %i", err2); } else {
+        result = malloc(resultSize);
         
-        err3 = AEGetParamPtr(&replyEvent, keyDirectObject, resultType, NULL, result, resultSize, &charResultSize);
-        //length = AEGetDescDataSize(&replyDesc);
-        //buffer = malloc(length);
+        if (err2) {
+            NSLog(@"Error After AESizeOfParam: %i", err2);
+        } else {
+            err3 = AEGetParamPtr(&replyEvent, keyDirectObject, resultType, NULL, result, resultSize, &charResultSize);
+            
+            if (err3) {
+                NSLog(@"Error After AEGetParamPtr: %i", err3);
+            } else {
+                _finalString = [[NSString stringWithCharacters:result length:charResultSize/sizeof(unichar)] copy];
+            }
+        }
+        free(result);
+    }
     
-        //err3 = AEGetDescData(&replyDesc, buffer, length);
-        
-        if (err3) { NSLog(@"Error After AEGetParamPtr: %i", err3); } else {
-        
-        // if (err3) { NSLog(@"Error After AEGetDescData: %i", err3); } else {
+    AEDisposeDesc(&sendEvent);
+    AEDisposeDesc(&replyEvent);
+    
+    return _finalString;
+}
+
+- (long)sendAEWithRequestedKeyForNumber:(NSString*)key eventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
+{
+    //Add error checking...
+    AEEventClass eClass = *((unsigned long*)[eventClass UTF8String]);
+    AEEventID    eID    = *((unsigned long*)[eventID UTF8String]);
+    
+    const char *sendString = [[NSString stringWithFormat:@"'----':obj { form:'prop', want:type('prop'), seld:type('%s'), from:'null'() }", [key UTF8String]] UTF8String];
+    long result = 0;
+    
+    AppleEvent sendEvent, replyEvent;
+    AEIdleUPP upp = NewAEIdleUPP(&MyAEIdleCallback);
+    
+    DescType resultType;
+    Size resultSize, charResultSize;
+    int pid;
+    
+    AEBuildError buildError;
+    OSStatus err;
+    OSErr err2, err3;
+    
+    if ((GetProcessPID(&psn, &pid) == noErr) && (pid == 0)) {
+        NSLog(@"Error getting PID of application! Exiting.");
+        return nil;
+    }
+    
+    NSLog(@"_sendString: %s", sendString);
+    
+    err = AEBuildAppleEvent(eClass, eID, typeProcessSerialNumber,(ProcessSerialNumber*)&psn, sizeof(ProcessSerialNumber), kAutoGenerateReturnID, 0, &sendEvent, &buildError, sendString);
+    
+    [self printCarbonDesc:&sendEvent];
+    
+    if (err) {
+        NSLog(@"%d:%d at \"%@\"",buildError.fError,buildError.fErrorPos,[sendString substringToIndex:buildError.fErrorPos]);
+    }
+    
+    err = AESend(&sendEvent, &replyEvent, kAEWaitReply, kAENormalPriority, kNoTimeOut, upp, NULL);
+    
+    [self printCarbonDesc:&replyEvent];
+    
+    if (err) {
+        NSLog(@"Send Error: %i",err);
+    } else {
+        err2 = AESizeOfParam(&replyEvent, keyDirectObject, &resultType, &resultSize);
         
-        _finalString = [[NSString stringWithCharacters:result length:charResultSize/sizeof(unichar)] copy];
+        if (err2) {
+            NSLog(@"Error After AESizeOfParam: %i", err2);
+        } else {
+            err3 = AEGetParamPtr(&replyEvent, keyDirectObject, resultType, NULL, &result, resultSize, &charResultSize);
+            
+            if (err3) {
+                NSLog(@"Error After AEGetParamPtr: %i", err3);
+            } else {
+                NSLog(@"%i", result);
+            }
+        }
+    }
+    
+    AEDisposeDesc(&sendEvent);
+    AEDisposeDesc(&replyEvent);
+    
+    return result;
+}
+
+- (NSString*)sendTwoTierAEWithRequestedKey:(NSString*)key fromObjectByKey:(NSString*)object eventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
+{
+    //Add error checking...
+    AEEventClass eClass = *((unsigned long*)[eventClass UTF8String]);
+    AEEventID    eID    = *((unsigned long*)[eventID UTF8String]);
+    
+    const char *sendString = [[NSString stringWithFormat:@"'----':obj { form:'prop', want:type('prop'), seld:type('%s'), from:obj { form:'prop', want:type('prop'), seld:type('%s'), from:'null'() } }", [key UTF8String], [object UTF8String]] UTF8String];
+    NSString  *_finalString;
+    
+    AppleEvent sendEvent, replyEvent;
+    AEIdleUPP upp = NewAEIdleUPP(&MyAEIdleCallback);
+    
+    DescType resultType;
+    Size resultSize, charResultSize;
+    int pid;
+    
+    AEBuildError buildError;
+    OSStatus err;
+    OSErr err2, err3;
+    
+    if ((GetProcessPID(&psn, &pid) == noErr) && (pid == 0)) {
+        NSLog(@"Error getting PID of application! Exiting.");
+        return nil;
+    }
+    
+    NSLog(@"_sendString: %s", sendString);
+    
+    err = AEBuildAppleEvent(eClass, eID, typeProcessSerialNumber,(ProcessSerialNumber*)&psn, sizeof(ProcessSerialNumber), kAutoGenerateReturnID, 0, &sendEvent, &buildError, sendString);
+    
+    [self printCarbonDesc:&sendEvent];
+    
+    if (err) {
+        NSLog(@"%d:%d at \"%@\"",buildError.fError,buildError.fErrorPos,[sendString substringToIndex:buildError.fErrorPos]);
+    }
+    
+    err = AESend(&sendEvent, &replyEvent, kAEWaitReply, kAENormalPriority, kNoTimeOut, upp, NULL);
+    
+    [self printCarbonDesc:&replyEvent];
+    
+    if (err) {
+        NSLog(@"Send Error: %i",err);
+    } else {
+        unichar *result = 0;
         
-        /* _finalString = [NSString stringWithCString:buffer length:length];
+        err2 = AESizeOfParam(&replyEvent, keyDirectObject, &resultType, &resultSize);
+        result = malloc(resultSize);
         
-        if ( (! [_finalString isEqualToString:@""])      &&
-            ([_finalString characterAtIndex:0] == '\"') &&
-            ([_finalString characterAtIndex:[_finalString length] - 1] == '\"') ) {
-            _finalString = [_finalString substringWithRange:NSMakeRange(1, [_finalString length] - 2)];
+        if (err2) {
+            NSLog(@"Error After AESizeOfParam: %i", err2);
+        } else {
+            err3 = AEGetParamPtr(&replyEvent, keyDirectObject, resultType, NULL, result, resultSize, &charResultSize);
+            
+            if (err3) {
+                NSLog(@"Error After AEGetParamPtr: %i", err3);
+            } else {
+                _finalString = [[NSString stringWithCharacters:result length:charResultSize/sizeof(unichar)] copy];
+            }
         }
-        free(buffer);
-        buffer = nil; */
+        free(result);
+    }
+    
+    AEDisposeDesc(&sendEvent);
+    AEDisposeDesc(&replyEvent);
+    
+    return _finalString;
+}
 
-        }
-        }
+- (long)sendTwoTierAEWithRequestedKeyForNumber:(NSString*)key fromObjectByKey:(NSString*)object eventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
+{
+    //Add error checking...
+    AEEventClass eClass = *((unsigned long*)[eventClass UTF8String]);
+    AEEventID    eID    = *((unsigned long*)[eventID UTF8String]);
+    
+    const char *sendString = [[NSString stringWithFormat:@"'----':obj { form:'prop', want:type('prop'), seld:type('%s'), from:obj { form:'prop', want:type('prop'), seld:type('%s'), from:'null'() } }", [key UTF8String], [object UTF8String]] UTF8String];
+    long result = 0;
+    
+    AppleEvent sendEvent, replyEvent;
+    AEIdleUPP upp = NewAEIdleUPP(&MyAEIdleCallback);
+    
+    DescType resultType;
+    Size resultSize, charResultSize;
+    int pid;
+    
+    AEBuildError buildError;
+    OSStatus err;
+    OSErr err2, err3;
+    
+    if ((GetProcessPID(&psn, &pid) == noErr) && (pid == 0)) {
+        NSLog(@"Error getting PID of application! Exiting.");
+        return nil;
+    }
+    
+    NSLog(@"_sendString: %s", sendString);
+    
+    err = AEBuildAppleEvent(eClass, eID, typeProcessSerialNumber,(ProcessSerialNumber*)&psn, sizeof(ProcessSerialNumber), kAutoGenerateReturnID, 0, &sendEvent, &buildError, sendString);
+    
+    [self printCarbonDesc:&sendEvent];
+    
+    if (err) {
+        NSLog(@"%d:%d at \"%@\"",buildError.fError,buildError.fErrorPos,[sendString substringToIndex:buildError.fErrorPos]);
     }
     
-    //NSLog(@"Result Size: %i", charResultSize);
-    //NSLog(@"Result: %c", charResult);
+    err = AESend(&sendEvent, &replyEvent, kAEWaitReply, kAENormalPriority, kNoTimeOut, upp, NULL);
+    
+    [self printCarbonDesc:&replyEvent];
+    
+    if (err) {
+        NSLog(@"Send Error: %i",err);
+    } else {
+        err2 = AESizeOfParam(&replyEvent, keyDirectObject, &resultType, &resultSize);
+        
+        if (err2) {
+            NSLog(@"Error After AESizeOfParam: %i", err2);
+        } else {
+            err3 = AEGetParamPtr(&replyEvent, keyDirectObject, resultType, NULL, &result, resultSize, &charResultSize);
+            
+            if (err3) {
+                NSLog(@"Error After AEGetParamPtr: %i", err3);
+            } else {
+                NSLog(@"%i", result);
+            }
+        }
+    }
     
     AEDisposeDesc(&sendEvent);
     AEDisposeDesc(&replyEvent);
     
+    return result;
+}
+
+- (NSString *)sendAEWithRequestedArray:(NSArray *)array eventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
+{
+    NSString *buildString = [NSString stringWithFormat:@"{ form:'prop', want:type('prop'), seld:type('%s'), from:'null'() }", [[array objectAtIndex:0] UTF8String]];
+    const char *sendString;
+    int i;
+    NSString  *_finalString;
+    
+    AEEventClass eClass = *((unsigned long*)[eventClass UTF8String]);
+    AEEventID    eID    = *((unsigned long*)[eventID UTF8String]);
+    
+    AppleEvent sendEvent, replyEvent;
+    AEIdleUPP upp = NewAEIdleUPP(&MyAEIdleCallback);
+    
+    DescType resultType;
+    Size resultSize, charResultSize;
+    int pid;
+    
+    AEBuildError buildError;
+    OSStatus err;
+    OSErr err2, err3;
+    
+    for (i = 1; i < [array count]; i++) {
+        NSString *from = [NSString stringWithFormat:@"{ form:'prop', want:type('prop'), seld:type('%s'), from:obj %@ }",
+                    [[array objectAtIndex:i] UTF8String], buildString];
+        buildString = from;
+    }
+    buildString = [@"'----':obj " stringByAppendingString:buildString];
+    sendString = [buildString UTF8String];
+    
+    if ((GetProcessPID(&psn, &pid) == noErr) && (pid == 0)) {
+        NSLog(@"Error getting PID of application! Exiting.");
+        return nil;
+    }
+    
+    NSLog(@"_sendString: %s", sendString);
+    
+    err = AEBuildAppleEvent(eClass, eID, typeProcessSerialNumber,(ProcessSerialNumber*)&psn, sizeof(ProcessSerialNumber), kAutoGenerateReturnID, 0, &sendEvent, &buildError, sendString);
+    
+    [self printCarbonDesc:&sendEvent];
+    
+    if (err) {
+        NSLog(@"%d:%d at \"%@\"",buildError.fError,buildError.fErrorPos,[sendString substringToIndex:buildError.fErrorPos]);
+    }
+    
+    err = AESend(&sendEvent, &replyEvent, kAEWaitReply, kAENormalPriority, kNoTimeOut, upp, NULL);
+    
+    [self printCarbonDesc:&replyEvent];
+    
+    if (err) {
+        NSLog(@"Send Error: %i",err);
+    } else {
+        unichar *result = 0;
+        
+        err2 = AESizeOfParam(&replyEvent, keyDirectObject, &resultType, &resultSize);
+        result=malloc(resultSize);
+        
+        if (err2) {
+            NSLog(@"Error After AESizeOfParam: %i", err2);
+        } else {
+            err3 = AEGetParamPtr(&replyEvent, keyDirectObject, resultType, NULL, result, resultSize, &charResultSize);
+            if (err3) {
+                NSLog(@"Error After AEGetParamPtr: %i", err3);
+            } else {
+                _finalString = [[NSString stringWithCharacters:result length:charResultSize/sizeof(unichar)] copy];
+            }
+        }
+        free(result);
+    }
     return _finalString;
 }
 
+- (void)sendAEWithEventClass:(NSString*)eventClass eventID:(NSString*)eventID appPSN:(ProcessSerialNumber)psn
+{
+    //Add error checking...
+    AEEventClass eClass = *((unsigned long*)[eventClass UTF8String]);
+    AEEventID    eID    = *((unsigned long*)[eventID UTF8String]);
+    AppleEvent event, reply;
+
+    AEBuildAppleEvent(eClass, eID, typeProcessSerialNumber,(ProcessSerialNumber*)&psn, sizeof(ProcessSerialNumber), kAutoGenerateReturnID, kAnyTransactionID, &event, nil, "");
+
+    AESend(&event, &reply, kAENoReply, kAENormalPriority, kAEDefaultTimeout, nil, nil);
+    
+    AEDisposeDesc(&event);
+    AEDisposeDesc(&reply);
+}
+
 - (void)printCarbonDesc:(AEDesc*)desc {
     Handle xx;
     AEPrintDescToHandle(desc,&xx);