mirror of
https://github.com/swisspol/GCDWebServer.git
synced 2026-04-24 00:00:04 +08:00
Added GCDWebUploader
This commit is contained in:
Pierre-Olivier Latour
@@ -0,0 +1,255 @@
|
||||
/*
|
||||
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* _footer;
|
||||
}
|
||||
@end
|
||||
|
||||
static NSDictionary* _GetInfoForFile(NSString* path) {
|
||||
NSDictionary* attributes = [[NSFileManager defaultManager] attributesOfItemAtPath:path error:NULL];
|
||||
if ([[attributes objectForKey:NSFileType] isEqualToString:NSFileTypeRegular]) {
|
||||
NSString* name = [path lastPathComponent];
|
||||
NSString* file = GCDWebServerEscapeURLString(name);
|
||||
return @{
|
||||
@"url": [@"/download?file=" stringByAppendingString:file],
|
||||
@"name": name,
|
||||
@"size": [attributes objectForKey:NSFileSize],
|
||||
@"type": GCDWebServerGetMimeTypeForExtension([name pathExtension]),
|
||||
@"deleteType": @"DELETE",
|
||||
@"deleteUrl": [@"/delete?file=" stringByAppendingString:file]
|
||||
};
|
||||
}
|
||||
return nil;
|
||||
}
|
||||
|
||||
@implementation GCDWebUploader
|
||||
|
||||
@synthesize uploadDirectory=_uploadDirectory, delegate=_delegate, allowedFileExtensions=_allowedExtensions, showHiddenFiles=_showHidden, title=_title, header=_header, footer=_footer;
|
||||
|
||||
- (BOOL)_checkFileExtension:(NSString*)fileName {
|
||||
if (_allowedExtensions && ![_allowedExtensions containsObject:[[fileName pathExtension] lowercaseString]]) {
|
||||
return NO;
|
||||
}
|
||||
return YES;
|
||||
}
|
||||
|
||||
- (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];
|
||||
GCDWebUploader* __unsafe_unretained uploader = self; // Avoid retain-cycles with self
|
||||
|
||||
// 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) {
|
||||
|
||||
NSString* title = uploader.title;
|
||||
if (title == nil) {
|
||||
title = [NSString stringWithFormat:[siteBundle localizedStringForKey:@"TITLE" value:@"" table:nil],
|
||||
[[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleDisplayName"]];
|
||||
}
|
||||
NSString* header = uploader.header;
|
||||
if (header == nil) {
|
||||
header = [siteBundle localizedStringForKey:@"HEADER" value:@"" table:nil];
|
||||
}
|
||||
NSString* footer = uploader.footer;
|
||||
if (footer == nil) {
|
||||
#if TARGET_OS_IPHONE
|
||||
NSString* name = [[UIDevice currentDevice] name];
|
||||
#else
|
||||
CFStringRef name = SCDynamicStoreCopyComputerName(NULL, NULL);
|
||||
#endif
|
||||
footer = [NSString stringWithFormat:[siteBundle localizedStringForKey:@"FOOTER" value:@"" table:nil],
|
||||
[[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleDisplayName"],
|
||||
[[NSBundle mainBundle] objectForInfoDictionaryKey:@"CFBundleShortVersionString"],
|
||||
name];
|
||||
}
|
||||
return [GCDWebServerDataResponse responseWithHTMLTemplate:[siteBundle pathForResource:@"index" ofType:@"html"]
|
||||
variables:@{
|
||||
@"title": title,
|
||||
@"header": header,
|
||||
@"footer": footer
|
||||
}];
|
||||
|
||||
}];
|
||||
|
||||
// File listing
|
||||
[self addHandlerForMethod:@"GET" path:@"/list" requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
|
||||
|
||||
NSString* basePath = uploader.uploadDirectory;
|
||||
BOOL showHidden = uploader.showHiddenFiles;
|
||||
NSArray* contents = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:basePath error:NULL];
|
||||
if (contents) {
|
||||
NSMutableArray* files = [NSMutableArray array];
|
||||
for (NSString* path in [contents sortedArrayUsingSelector:@selector(localizedStandardCompare:)]) {
|
||||
if (showHidden || ![path hasPrefix:@"."]) {
|
||||
if ([uploader _checkFileExtension:path]) {
|
||||
NSDictionary* info = _GetInfoForFile([basePath stringByAppendingPathComponent:path]);
|
||||
if (info) {
|
||||
[files addObject:info];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return [GCDWebServerDataResponse responseWithJSONObject:@{@"files": files}];
|
||||
} else {
|
||||
return [GCDWebServerResponse responseWithStatusCode:500];
|
||||
}
|
||||
|
||||
}];
|
||||
|
||||
// File download
|
||||
[self addHandlerForMethod:@"GET" path:@"/download" requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
|
||||
|
||||
NSString* file = [[request query] objectForKey:@"file"];
|
||||
NSString* path = [uploader.uploadDirectory stringByAppendingPathComponent:file];
|
||||
BOOL isDirectory;
|
||||
if ([[NSFileManager defaultManager] fileExistsAtPath:path isDirectory:&isDirectory] && !isDirectory) {
|
||||
return [GCDWebServerFileResponse responseWithFile:path 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[]"];
|
||||
NSString* fileName = file.fileName;
|
||||
if ([uploader _checkFileExtension:fileName]) {
|
||||
NSString* path = nil;
|
||||
int retries = 0;
|
||||
while (1) {
|
||||
if (retries > 0) {
|
||||
path = [uploader.uploadDirectory stringByAppendingPathComponent:[[[fileName stringByDeletingPathExtension] stringByAppendingFormat:@" (%i)", retries] stringByAppendingPathExtension:[fileName pathExtension]]];
|
||||
} else {
|
||||
path = [uploader.uploadDirectory stringByAppendingPathComponent:fileName];
|
||||
}
|
||||
if (![[NSFileManager defaultManager] fileExistsAtPath:path]) {
|
||||
break;
|
||||
}
|
||||
++retries;
|
||||
}
|
||||
NSError* error = nil;
|
||||
if ([[NSFileManager defaultManager] moveItemAtPath:file.temporaryPath toPath:path error:&error]) {
|
||||
dispatch_async(dispatch_get_main_queue(), ^{
|
||||
[uploader.delegate webUploader:uploader didUploadFile:[path lastPathComponent]];
|
||||
});
|
||||
NSDictionary* info = _GetInfoForFile(path);
|
||||
return [GCDWebServerDataResponse responseWithJSONObject:@{@"files": @[info]} contentType:contentType];
|
||||
} else {
|
||||
return [GCDWebServerDataResponse responseWithJSONObject:@{
|
||||
@"files": @[@{
|
||||
@"name": fileName,
|
||||
@"size": @0,
|
||||
@"error": [error localizedDescription]
|
||||
}]
|
||||
} contentType:contentType];
|
||||
}
|
||||
} else {
|
||||
return [GCDWebServerDataResponse responseWithJSONObject:@{
|
||||
@"files": @[@{
|
||||
@"name": fileName,
|
||||
@"size": @0,
|
||||
@"error": [siteBundle localizedStringForKey:@"UNSUPPORTED_FILE_EXTENSION" value:@"" table:nil]
|
||||
}]
|
||||
} contentType:contentType];
|
||||
}
|
||||
|
||||
}];
|
||||
|
||||
// File deletion
|
||||
[self addHandlerForMethod:@"DELETE" path:@"/delete" requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
|
||||
|
||||
NSString* file = [[request query] objectForKey:@"file"];
|
||||
NSString* path = [uploader.uploadDirectory stringByAppendingPathComponent:file];
|
||||
BOOL isDirectory;
|
||||
if ([[NSFileManager defaultManager] fileExistsAtPath:path isDirectory:&isDirectory] && !isDirectory) {
|
||||
BOOL success = [[NSFileManager defaultManager] removeItemAtPath:path error:NULL];
|
||||
if (success) {
|
||||
dispatch_async(dispatch_get_main_queue(), ^{
|
||||
[uploader.delegate webUploader:uploader didDeleteFile:file];
|
||||
});
|
||||
}
|
||||
return [GCDWebServerResponse responseWithStatusCode:(success ? 204 : 500)];
|
||||
// TODO: Contrary to the documentation at https://github.com/blueimp/jQuery-File-Upload/wiki/Setup, jquery.fileupload-ui.js ignores the returned JSON
|
||||
// return [GCDWebServerDataResponse responseWithJSONObject:@{@"files": @[@{file: [NSNumber numberWithBool:success]}]}];
|
||||
} else {
|
||||
return [GCDWebServerResponse responseWithStatusCode:404];
|
||||
}
|
||||
|
||||
}];
|
||||
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
#if !__has_feature(objc_arc)
|
||||
|
||||
- (void)dealloc {
|
||||
[_uploadDirectory release];
|
||||
[_allowedExtensions release];
|
||||
[_title release];
|
||||
[_header release];
|
||||
[_footer release];
|
||||
|
||||
[super dealloc];
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
@end
|
||||
Reference in New Issue
Block a user