1 files changed, 10 insertions, 2 deletions
diff --git a/On Task.m b/On Task.m
index a4e67e5..5976fa0 100644
--- a/On Task.m
+++ b/On Task.m
@@ -6,7 +6,7 @@ int main (int argc, const char * argv[]) {
     NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init];
 	
 	// Get Application Support dir path
-	NSString *supportPath = [[NSFileManager defaultManager] applicationSupportDirectory];
+	NSString *applicationSupportPath = [[NSFileManager defaultManager] applicationSupportDirectory];
 	
 	// Get a date string
 	[NSDateFormatter setDefaultFormatterBehavior:NSDateFormatterBehavior10_4];
@@ -15,9 +15,17 @@ int main (int argc, const char * argv[]) {
 	NSString *date = [df stringFromDate:[NSDate date]];
 	
     // # Take screenshot
+	// Make screenshots directory if needed
+	NSString *screenshotsPath = [applicationSupportPath stringByAppendingString:@"/Screenshots"];
+	if (![[NSFileManager defaultManager] fileExistsAtPath:screenshotsPath]) {
+		if (![[NSFileManager defaultManager] createDirectoryAtPath:screenshotsPath attributes:nil]) {
+			NSLog(@"Error: Could not create folder %@", screenshotsPath);
+		}
+	}
+	
 	NSTask *takeScreenshot = [[NSTask alloc] init];
 	[takeScreenshot setLaunchPath:@"/usr/sbin/screencapture"];
-	[takeScreenshot setArguments:[NSArray arrayWithObjects:@"-SxC", [supportPath stringByAppendingString:[NSString stringWithFormat:@"/%@.png", date]], nil]];
+	[takeScreenshot setArguments:[NSArray arrayWithObjects:@"-SxC", [applicationSupportPath stringByAppendingString:[NSString stringWithFormat:@"/Screenshots/%@.png", date]], nil]];
 	[takeScreenshot launch];
 	[takeScreenshot waitUntilExit];
 	[takeScreenshot release];