mirror of
https://github.com/swisspol/GCDWebServer.git
synced 2024-10-08 17:14:17 +08:00
256 lines
12 KiB
Objective-C
256 lines
12 KiB
Objective-C
/*
|
|
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
|