1 files changed, 79 insertions, 0 deletions
diff --git a/Foundation/GTMNSString+FindFolder.m b/Foundation/GTMNSString+FindFolder.m
new file mode 100644
index 0000000..7d2f3b5
--- /dev/null
+++ b/Foundation/GTMNSString+FindFolder.m
@@ -0,0 +1,79 @@
+//
+//  GTMNSString+FindFolder.m
+//
+//  Copyright 2006-2008 Google Inc.
+//
+//  Licensed under the Apache License, Version 2.0 (the "License"); you may not
+//  use this file except in compliance with the License.  You may obtain a copy
+//  of the License at
+// 
+//  http://www.apache.org/licenses/LICENSE-2.0
+// 
+//  Unless required by applicable law or agreed to in writing, software
+//  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+//  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+//  License for the specific language governing permissions and limitations under
+//  the License.
+//
+
+#import "GTMNSString+FindFolder.h"
+#import "GTMGarbageCollection.h"
+
+@implementation NSString (GTMStringFindFolderAdditions)
+
++ (NSString *)gtm_stringWithPathForFolder:(OSType)theFolderType 
+                                 inDomain:(short)theDomain
+                                 doCreate:(BOOL)doCreate {
+  
+  NSString* folderPath = nil;
+  FSRef folderRef;
+  
+  OSErr err = FSFindFolder(theDomain, theFolderType, doCreate, &folderRef);
+  if (err == noErr) {
+    
+    CFURLRef folderURL = CFURLCreateFromFSRef(kCFAllocatorSystemDefault, &folderRef);
+    if (folderURL) {
+      
+      folderPath = GTMNSMakeCollectable(CFURLCopyFileSystemPath(folderURL, kCFURLPOSIXPathStyle));
+      [folderPath autorelease];
+      
+      CFRelease(folderURL);
+    }
+  }
+  return folderPath;
+}
+
++ (NSString *)gtm_stringWithPathForFolder:(OSType)theFolderType 
+                            subfolderName:(NSString *)subfolderName
+                                 inDomain:(short)theDomain
+                                 doCreate:(BOOL)doCreate {
+  NSString *resultPath = nil;
+  NSString *subdirPath = nil;
+  NSString *parentFolderPath = [self gtm_stringWithPathForFolder:theFolderType
+                                                        inDomain:theDomain
+                                                        doCreate:doCreate];
+  if (parentFolderPath) {
+    
+    // find the path to the subdirectory
+    subdirPath = [parentFolderPath stringByAppendingPathComponent:subfolderName];
+
+    NSFileManager* fileMgr = [NSFileManager defaultManager];
+    BOOL isDir = NO;
+    if ([fileMgr fileExistsAtPath:subdirPath isDirectory:&isDir] && isDir) {
+      // it already exists
+      resultPath = subdirPath;
+    } else if (doCreate) {
+      
+      // create the subdirectory with the parent folder's attributes
+      NSDictionary* attrs = [fileMgr fileAttributesAtPath:parentFolderPath
+                                             traverseLink:YES];
+      if ([fileMgr createDirectoryAtPath:subdirPath
+                              attributes:attrs]) {
+        resultPath = subdirPath;
+      }
+    }
+  }
+  return resultPath;
+}
+
+@end