| /* |
| Copyright (c) 2012-2014, Pierre-Olivier Latour |
| All rights reserved. |
| |
| Redistribution and use in source and binary forms, with or without |
| modification, are permitted provided that the following conditions are met: |
| * Redistributions of source code must retain the above copyright |
| notice, this list of conditions and the following disclaimer. |
| * Redistributions in binary form must reproduce the above copyright |
| notice, this list of conditions and the following disclaimer in the |
| documentation and/or other materials provided with the distribution. |
| * The name of Pierre-Olivier Latour may not be used to endorse |
| or promote products derived from this software without specific |
| prior written permission. |
| |
| THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
| ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| DISCLAIMED. IN NO EVENT SHALL PIERRE-OLIVIER LATOUR BE LIABLE FOR ANY |
| DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
| (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
| LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
| ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
| SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #import <TargetConditionals.h> |
| #if TARGET_OS_IPHONE |
| #import <UIKit/UIKit.h> |
| #else |
| #import <SystemConfiguration/SystemConfiguration.h> |
| #endif |
| |
| #import "GCDWebUploader.h" |
| |
| @interface GCDWebUploader () { |
| @private |
| NSString* _uploadDirectory; |
| id<GCDWebUploaderDelegate> delegate; |
| NSArray* _allowedExtensions; |
| BOOL _showHidden; |
| NSString* _title; |
| NSString* _header; |
| NSString* _prologue; |
| NSString* _epilogue; |
| NSString* _footer; |
| } |
| @end |
| |
| @implementation GCDWebUploader |
| |
| @synthesize uploadDirectory=_uploadDirectory, delegate=_delegate, allowedFileExtensions=_allowedExtensions, showHiddenFiles=_showHidden, |
| title=_title, header=_header, prologue=_prologue, epilogue=_epilogue, footer=_footer; |
| |
| - (BOOL)_checkFileExtension:(NSString*)fileName { |
| if (_allowedExtensions && ![_allowedExtensions containsObject:[[fileName pathExtension] lowercaseString]]) { |
| return NO; |
| } |
| return YES; |
| } |
| |
| - (NSString*) _uniquePathForPath:(NSString*)path { |
| if ([[NSFileManager defaultManager] fileExistsAtPath:path]) { |
| NSString* directory = [path stringByDeletingLastPathComponent]; |
| NSString* file = [path lastPathComponent]; |
| NSString* base = [file stringByDeletingPathExtension]; |
| NSString* extension = [file pathExtension]; |
| int retries = 0; |
| do { |
| if (extension.length) { |
| path = [directory stringByAppendingPathComponent:[[base stringByAppendingFormat:@" (%i)", ++retries] stringByAppendingPathExtension:extension]]; |
| } else { |
| path = [directory stringByAppendingPathComponent:[base stringByAppendingFormat:@" (%i)", ++retries]]; |
| } |
| } while ([[NSFileManager defaultManager] fileExistsAtPath:path]); |
| } |
| return path; |
| } |
| |
| - (id)initWithUploadDirectory:(NSString*)path { |
| if ((self = [super init])) { |
| NSBundle* siteBundle = [NSBundle bundleWithPath:[[NSBundle mainBundle] pathForResource:@"GCDWebUploader" ofType:@"bundle"]]; |
| if (siteBundle == nil) { |
| #if !__has_feature(objc_arc) |
| [self release]; |
| #endif |
| return nil; |
| } |
| _uploadDirectory = [[path stringByStandardizingPath] copy]; |
| #if __has_feature(objc_arc) |
| __unsafe_unretained GCDWebUploader* uploader = self; |
| #else |
| __block GCDWebUploader* uploader = self; |
| #endif |
| |
| // Resource files |
| [self addGETHandlerForBasePath:@"/" directoryPath:[siteBundle resourcePath] indexFilename:nil cacheAge:3600 allowRangeRequests:NO]; |
| |
| // Web page |
| [self addHandlerForMethod:@"GET" path:@"/" requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| #if TARGET_OS_IPHONE |
| NSString* device = [[UIDevice currentDevice] name]; |
| #else |
| #if __has_feature(objc_arc) |
| NSString* device = CFBridgingRelease(SCDynamicStoreCopyComputerName(NULL, NULL)); |
| #else |
| NSString* device = [(id)SCDynamicStoreCopyComputerName(NULL, NULL) autorelease]; |
| #endif |
| #endif |
| NSString* title = uploader.title; |
| if (title == nil) { |
| title = [[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleDisplayName"]; |
| #if !TARGET_OS_IPHONE |
| if (title == nil) { |
| title = [[NSProcessInfo processInfo] processName]; |
| } |
| #endif |
| } |
| NSString* header = uploader.header; |
| if (header == nil) { |
| header = title; |
| } |
| NSString* prologue = uploader.prologue; |
| if (prologue == nil) { |
| prologue = [siteBundle localizedStringForKey:@"PROLOGUE" value:@"" table:nil]; |
| } |
| NSString* epilogue = uploader.epilogue; |
| if (epilogue == nil) { |
| epilogue = [siteBundle localizedStringForKey:@"EPILOGUE" value:@"" table:nil]; |
| } |
| NSString* footer = uploader.footer; |
| if (footer == nil) { |
| NSString* name = [[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleDisplayName"]; |
| NSString* version = [[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleShortVersionString"]; |
| #if !TARGET_OS_IPHONE |
| if (!name && !version) { |
| name = @"OS X"; |
| version = [[NSProcessInfo processInfo] operatingSystemVersionString]; |
| } |
| #endif |
| footer = [NSString stringWithFormat:[siteBundle localizedStringForKey:@"FOOTER_FORMAT" value:@"" table:nil], name, version]; |
| } |
| return [GCDWebServerDataResponse responseWithHTMLTemplate:[siteBundle pathForResource:@"index" ofType:@"html"] |
| variables:@{ |
| @"device": device, |
| @"title": title, |
| @"header": header, |
| @"prologue": prologue, |
| @"epilogue": epilogue, |
| @"footer": footer |
| }]; |
| |
| }]; |
| |
| // File listing |
| [self addHandlerForMethod:@"GET" path:@"/list" requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| NSString* relativePath = [[request query] objectForKey:@"path"]; |
| NSString* absolutePath = [uploader.uploadDirectory stringByAppendingPathComponent:relativePath]; |
| BOOL isDirectory; |
| if ([[NSFileManager defaultManager] fileExistsAtPath:absolutePath isDirectory:&isDirectory]) { |
| if (isDirectory) { |
| BOOL showHidden = uploader.showHiddenFiles; |
| NSArray* contents = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:absolutePath error:NULL]; |
| if (contents) { |
| NSMutableArray* array = [NSMutableArray array]; |
| for (NSString* item in [contents sortedArrayUsingSelector:@selector(localizedStandardCompare:)]) { |
| if (showHidden || ![item hasPrefix:@"."]) { |
| NSDictionary* attributes = [[NSFileManager defaultManager] attributesOfItemAtPath:[absolutePath stringByAppendingPathComponent:item] error:NULL]; |
| NSString* type = [attributes objectForKey:NSFileType]; |
| if ([type isEqualToString:NSFileTypeRegular] && [uploader _checkFileExtension:item]) { |
| [array addObject:@{ |
| @"path": [relativePath stringByAppendingPathComponent:item], |
| @"name": item, |
| @"size": [attributes objectForKey:NSFileSize] |
| }]; |
| } else if ([type isEqualToString:NSFileTypeDirectory]) { |
| [array addObject:@{ |
| @"path": [[relativePath stringByAppendingPathComponent:item] stringByAppendingString:@"/"], |
| @"name": item |
| }]; |
| } |
| } |
| } |
| return [GCDWebServerDataResponse responseWithJSONObject:array]; |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:500]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:400]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:404]; |
| } |
| |
| }]; |
| |
| // File download |
| [self addHandlerForMethod:@"GET" path:@"/download" requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| NSString* relativePath = [[request query] objectForKey:@"path"]; |
| NSString* absolutePath = [uploader.uploadDirectory stringByAppendingPathComponent:relativePath]; |
| BOOL isDirectory; |
| if ([[NSFileManager defaultManager] fileExistsAtPath:absolutePath isDirectory:&isDirectory]) { |
| if (isDirectory) { |
| return [GCDWebServerResponse responseWithStatusCode:400]; |
| } else { |
| if ([uploader.delegate respondsToSelector:@selector(webUploader:didDownloadFileAtPath: )]) { |
| dispatch_async(dispatch_get_main_queue(), ^{ |
| [uploader.delegate webUploader:uploader didDownloadFileAtPath:absolutePath]; |
| }); |
| } |
| return [GCDWebServerFileResponse responseWithFile:absolutePath isAttachment:YES]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:404]; |
| } |
| |
| }]; |
| |
| // File upload |
| [self addHandlerForMethod:@"POST" path:@"/upload" requestClass:[GCDWebServerMultiPartFormRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| // Required when using iFrame transport (see https://github.com/blueimp/jQuery-File-Upload/wiki/Setup) |
| NSRange range = [[request.headers objectForKey:@"Accept"] rangeOfString:@"application/json" options:NSCaseInsensitiveSearch]; |
| NSString* contentType = (range.location != NSNotFound ? @"application/json" : @"text/plain; charset=utf-8"); |
| |
| GCDWebServerMultiPartFile* file = [[(GCDWebServerMultiPartFormRequest*)request files] objectForKey:@"files[]"]; |
| if ((![file.fileName hasPrefix:@"."] || uploader.showHiddenFiles) && [uploader _checkFileExtension:file.fileName]) { |
| NSString* relativePath = [(GCDWebServerMultiPartArgument*)[[(GCDWebServerURLEncodedFormRequest*)request arguments] objectForKey:@"path"] string]; |
| NSString* absolutePath = [uploader _uniquePathForPath:[[uploader.uploadDirectory stringByAppendingPathComponent:relativePath] stringByAppendingPathComponent:file.fileName]]; |
| if ([uploader shouldUploadFileAtPath:absolutePath withTemporaryFile:file.temporaryPath]) { |
| NSError* error = nil; |
| if ([[NSFileManager defaultManager] moveItemAtPath:file.temporaryPath toPath:absolutePath error:&error]) { |
| if ([uploader.delegate respondsToSelector:@selector(webUploader:didUploadFileAtPath:)]) { |
| dispatch_async(dispatch_get_main_queue(), ^{ |
| [uploader.delegate webUploader:uploader didUploadFileAtPath:absolutePath]; |
| }); |
| } |
| return [GCDWebServerDataResponse responseWithJSONObject:@{} contentType:contentType]; |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:500]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:403]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:400]; |
| } |
| |
| }]; |
| |
| // File and folder moving |
| [self addHandlerForMethod:@"POST" path:@"/move" requestClass:[GCDWebServerURLEncodedFormRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| NSString* oldRelativePath = [[(GCDWebServerURLEncodedFormRequest*)request arguments] objectForKey:@"oldPath"]; |
| NSString* oldAbsolutePath = [uploader.uploadDirectory stringByAppendingPathComponent:oldRelativePath]; |
| BOOL isDirectory; |
| if ([[NSFileManager defaultManager] fileExistsAtPath:oldAbsolutePath isDirectory:&isDirectory]) { |
| NSString* newRelativePath = [[(GCDWebServerURLEncodedFormRequest*)request arguments] objectForKey:@"newPath"]; |
| if (!uploader.showHiddenFiles) { |
| for (NSString* component in [newRelativePath pathComponents]) { |
| if ([component hasPrefix:@"."]) { |
| return [GCDWebServerResponse responseWithStatusCode:400]; |
| } |
| } |
| } |
| if (!isDirectory && ![uploader _checkFileExtension:newRelativePath]) { |
| return [GCDWebServerResponse responseWithStatusCode:400]; |
| } |
| NSString* newAbsolutePath = [uploader _uniquePathForPath:[uploader.uploadDirectory stringByAppendingPathComponent:newRelativePath]]; |
| if ([uploader shouldMoveItemFromPath:oldAbsolutePath toPath:newAbsolutePath]) { |
| if ([[NSFileManager defaultManager] moveItemAtPath:oldAbsolutePath toPath:newAbsolutePath error:NULL]) { |
| if ([uploader.delegate respondsToSelector:@selector(webUploader:didMoveItemFromPath:toPath:)]) { |
| dispatch_async(dispatch_get_main_queue(), ^{ |
| [uploader.delegate webUploader:uploader didMoveItemFromPath:oldAbsolutePath toPath:newAbsolutePath]; |
| }); |
| } |
| return [GCDWebServerDataResponse responseWithJSONObject:@{}]; |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:500]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:403]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:404]; |
| } |
| |
| }]; |
| |
| // File deletion |
| [self addHandlerForMethod:@"POST" path:@"/delete" requestClass:[GCDWebServerURLEncodedFormRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| NSString* relativePath = [[(GCDWebServerURLEncodedFormRequest*)request arguments] objectForKey:@"path"]; |
| NSString* absolutePath = [uploader.uploadDirectory stringByAppendingPathComponent:relativePath]; |
| if ([[NSFileManager defaultManager] fileExistsAtPath:absolutePath]) { |
| if ([[NSFileManager defaultManager] removeItemAtPath:absolutePath error:NULL]) { |
| if ([uploader.delegate respondsToSelector:@selector(webUploader:didDeleteItemAtPath:)]) { |
| dispatch_async(dispatch_get_main_queue(), ^{ |
| [uploader.delegate webUploader:uploader didDeleteItemAtPath:absolutePath]; |
| }); |
| } |
| return [GCDWebServerDataResponse responseWithJSONObject:@{}]; |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:500]; |
| } |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:404]; |
| } |
| |
| }]; |
| |
| // Directory creation |
| [self addHandlerForMethod:@"POST" path:@"/create" requestClass:[GCDWebServerURLEncodedFormRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) { |
| |
| NSString* relativePath = [[(GCDWebServerURLEncodedFormRequest*)request arguments] objectForKey:@"path"]; |
| if (!uploader.showHiddenFiles) { |
| for (NSString* component in [relativePath pathComponents]) { |
| if ([component hasPrefix:@"."]) { |
| return [GCDWebServerResponse responseWithStatusCode:400]; |
| } |
| } |
| } |
| NSString* absolutePath = [uploader _uniquePathForPath:[uploader.uploadDirectory stringByAppendingPathComponent:relativePath]]; |
| if ([[NSFileManager defaultManager] createDirectoryAtPath:absolutePath withIntermediateDirectories:YES attributes:nil error:NULL]) { |
| if ([uploader.delegate respondsToSelector:@selector(webUploader:didCreateDirectoryAtPath:)]) { |
| dispatch_async(dispatch_get_main_queue(), ^{ |
| [uploader.delegate webUploader:uploader didCreateDirectoryAtPath:absolutePath]; |
| }); |
| } |
| return [GCDWebServerDataResponse responseWithJSONObject:@{}]; |
| } else { |
| return [GCDWebServerResponse responseWithStatusCode:500]; |
| } |
| |
| }]; |
| |
| } |
| return self; |
| } |
| |
| #if !__has_feature(objc_arc) |
| |
| - (void)dealloc { |
| [_uploadDirectory release]; |
| [_allowedExtensions release]; |
| [_title release]; |
| [_header release]; |
| [_prologue release]; |
| [_epilogue release]; |
| [_footer release]; |
| |
| [super dealloc]; |
| } |
| |
| #endif |
| |
| @end |
| |
| @implementation GCDWebUploader (Subclassing) |
| |
| - (BOOL)shouldUploadFileAtPath:(NSString*)path withTemporaryFile:(NSString*)tempPath { |
| return YES; |
| } |
| |
| - (BOOL)shouldMoveItemFromPath:(NSString*)fromPath toPath:(NSString*)toPath { |
| return YES; |
| } |
| |
| @end |
