mirror of
https://github.com/ksherlock/ample.git
synced 2024-10-31 15:04:56 +00:00
841 lines
22 KiB
Objective-C
841 lines
22 KiB
Objective-C
//
|
|
// DownloadWindowController.m
|
|
// Ample
|
|
//
|
|
// Created by Kelvin Sherlock on 9/2/2020.
|
|
// Copyright © 2020 Kelvin Sherlock. All rights reserved.
|
|
//
|
|
|
|
#import "Ample.h"
|
|
#import "DownloadWindowController.h"
|
|
#import "Menu.h"
|
|
|
|
|
|
static NSString *Extensions[] = { @"zip", @"7z" };
|
|
|
|
|
|
enum {
|
|
kTagZip = 1,
|
|
kTag7z = 2,
|
|
};
|
|
|
|
@interface DownloadExtensionTransformer: NSValueTransformer
|
|
@end
|
|
|
|
@implementation DownloadExtensionTransformer
|
|
|
|
+(void)load {
|
|
[NSValueTransformer setValueTransformer: [DownloadExtensionTransformer new] forName: @"FormatTransformer"];
|
|
|
|
}
|
|
|
|
+ (Class)transformedValueClass {
|
|
return [NSString class];
|
|
}
|
|
|
|
+ (BOOL)allowsReverseTransformation {
|
|
return YES;
|
|
}
|
|
|
|
-(id)transformedValue:(id)value {
|
|
// string to number.
|
|
if ([@"zip" isEqualToString: value])
|
|
return @(kTagZip);
|
|
if ([@"7z" isEqualToString: value])
|
|
return @(kTag7z);
|
|
return @0;
|
|
}
|
|
|
|
-(id)reverseTransformedValue:(id)value {
|
|
// number back to string.
|
|
switch ([value intValue]) {
|
|
case kTagZip: return @"zip";
|
|
case kTag7z: return @"7z";
|
|
default: return @"";
|
|
}
|
|
}
|
|
|
|
+(unsigned)stringToNumber: (NSString *)string {
|
|
if ([@"zip" isEqualToString: string])
|
|
return kTagZip;
|
|
if ([@"7z" isEqualToString: string])
|
|
return kTag7z;
|
|
return 0;
|
|
}
|
|
|
|
+(NSString *)numberToString: (unsigned)number {
|
|
switch (number) {
|
|
case kTagZip: return @"zip";
|
|
case kTag7z: return @"7z";
|
|
default: return @"";
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
enum {
|
|
ItemMissing = 0,
|
|
ItemFound,
|
|
ItemDownloading,
|
|
ItemDownloaded,
|
|
ItemCanceled,
|
|
ItemError
|
|
};
|
|
|
|
@interface DownloadItem : NSObject
|
|
|
|
@property NSString *value;
|
|
@property NSString *name;
|
|
@property NSError *error;
|
|
@property NSString *pathName;
|
|
@property NSURLSessionDownloadTask *task;
|
|
@property NSURL *localURL;
|
|
|
|
@property NSUInteger status;
|
|
@property NSUInteger index;
|
|
@property NSUInteger fileSize;
|
|
|
|
@property (readonly) NSColor *titleColor;
|
|
@property (readonly) NSColor *descriptionColor;
|
|
|
|
|
|
-(void)cancelDownload;
|
|
-(void)beginDownloadWithTask:(NSURLSessionDownloadTask *)task;
|
|
-(void)completeWithError: (NSError *)error;
|
|
-(NSString *)statusDescription;
|
|
|
|
-(void)refresh: (NSURL *)localURL;
|
|
@end
|
|
|
|
|
|
@interface DownloadItemArrayController : NSArrayController
|
|
|
|
@property(readonly, copy) NSArray<NSString *> *automaticRearrangementKeyPaths;
|
|
|
|
@end
|
|
|
|
@implementation DownloadItemArrayController
|
|
|
|
-(NSArray<NSString *> *)automaticRearrangementKeyPaths {
|
|
return @[@"localURL"]; // , @"error", @"task", @"statusDescription"];
|
|
}
|
|
|
|
|
|
@end
|
|
|
|
|
|
@interface DownloadWindowController ()
|
|
@property (weak) IBOutlet NSTableView *tableView;
|
|
@property (weak) IBOutlet NSPopUpButton *formatButton;
|
|
@property (weak) IBOutlet NSTextField *downloadField;
|
|
@property NSString *downloadExtension;
|
|
|
|
/* filter buttons */
|
|
@property (weak) IBOutlet NSButton *allFilterButton;
|
|
@property (weak) IBOutlet NSButton *missingFilterButton;
|
|
|
|
@property (strong) IBOutlet NSArrayController *arrayController;
|
|
|
|
|
|
|
|
@end
|
|
|
|
@implementation DownloadWindowController {
|
|
|
|
NSArray *_items;
|
|
NSURL *_romFolder;
|
|
NSURL *_defaultDownloadURL;
|
|
NSURL *_downloadURL;
|
|
|
|
NSURLSession *_session;
|
|
NSMutableDictionary *_taskIndex;
|
|
NSUserDefaults *_defaults;
|
|
|
|
NSArray<NSButton *> *_filterButtons;
|
|
}
|
|
|
|
+(instancetype)sharedInstance {
|
|
static DownloadWindowController *me = nil;
|
|
if (!me) {
|
|
me = [self new];
|
|
}
|
|
return me;
|
|
}
|
|
|
|
+ (void)restoreWindowWithIdentifier:(nonnull NSUserInterfaceItemIdentifier)identifier state:(nonnull NSCoder *)state completionHandler:(nonnull void (^)(NSWindow * _Nullable, NSError * _Nullable))completionHandler {
|
|
NSLog(@"restore rom manager window");
|
|
|
|
NSWindowController *controller = [DownloadWindowController sharedInstance];
|
|
NSWindow *w = [controller window];
|
|
[w restoreStateWithCoder: state];
|
|
completionHandler(w, nil);
|
|
}
|
|
|
|
|
|
#if 0
|
|
- (void)encodeWithCoder:(nonnull NSCoder *)coder {
|
|
|
|
}
|
|
#endif
|
|
|
|
-(NSString *)windowNibName {
|
|
return @"DownloadWindow";
|
|
}
|
|
|
|
-(void)windowWillLoad {
|
|
_defaults = [NSUserDefaults standardUserDefaults];
|
|
|
|
// set here so binding works.
|
|
NSString *s = [_defaults stringForKey: kDownloadExtension];
|
|
if (![s length]) s = [_defaults stringForKey: kDefaultDownloadExtension];
|
|
|
|
_downloadExtension = s;
|
|
}
|
|
|
|
- (void)windowDidLoad {
|
|
[super windowDidLoad];
|
|
#if 0
|
|
NSWindow *window = [self window];
|
|
// disabled for now ... restoration happens before defaults are loaded.
|
|
[window setRestorable: YES];
|
|
[window setRestorationClass: [self class]];
|
|
#endif
|
|
|
|
_filterButtons = @[
|
|
_allFilterButton,
|
|
_missingFilterButton
|
|
];
|
|
|
|
|
|
// Implement this method to handle any initialization after your window controller's window has been loaded from its nib file.
|
|
|
|
NSError *error = nil;
|
|
NSBundle *bundle = [NSBundle mainBundle];
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
|
|
#ifdef AMPLE_LITE
|
|
NSURL *url = [bundle URLForResource: @"roms~extra" withExtension: @"plist"];
|
|
#else
|
|
NSURL *url = [bundle URLForResource: @"roms" withExtension: @"plist"];
|
|
#endif
|
|
NSArray *roms = [NSArray arrayWithContentsOfURL: url];
|
|
|
|
NSURL *sd = SupportDirectory();
|
|
|
|
_romFolder = [sd URLByAppendingPathComponent: @"roms"];
|
|
|
|
[fm createDirectoryAtURL: _romFolder withIntermediateDirectories: YES attributes: nil error: &error];
|
|
|
|
// so blank URL isn't overwritten.
|
|
NSString *s = [_defaults stringForKey: kDefaultDownloadURL];
|
|
_defaultDownloadURL = [NSURL URLWithString: s];
|
|
[_downloadField setPlaceholderString: s];
|
|
|
|
s = [_defaults stringForKey: kDownloadURL];
|
|
if ([s length]) {
|
|
[_downloadField setStringValue: s];
|
|
_downloadURL = [NSURL URLWithString: s];
|
|
} else {
|
|
_downloadURL = _defaultDownloadURL;
|
|
}
|
|
|
|
[self initializeExtensionMenu];
|
|
|
|
|
|
[self setCurrentROM: @""];
|
|
[self setCurrentCount: 0];
|
|
[self setTotalCount: [roms count]];
|
|
[self setErrorCount: 0];
|
|
|
|
|
|
NSMutableArray *tmp = [NSMutableArray arrayWithCapacity: [roms count]];
|
|
unsigned ix = 0;
|
|
for (NSDictionary *d in roms) {
|
|
|
|
DownloadItem *item = [DownloadItem new];
|
|
[item setValue: [d objectForKey: @"value"]];
|
|
[item setName: [d objectForKey: @"description"]];
|
|
[item setIndex: ix++];
|
|
|
|
[tmp addObject: item];
|
|
}
|
|
_items = tmp;
|
|
[self refreshROMs: nil];
|
|
[_arrayController setContent: _items];
|
|
|
|
//[_tableView reloadData];
|
|
NSURLSessionConfiguration *config = [NSURLSessionConfiguration defaultSessionConfiguration];
|
|
_session = [NSURLSession sessionWithConfiguration: config delegate: self delegateQueue: nil];
|
|
_taskIndex = [NSMutableDictionary dictionaryWithCapacity: [_items count]];
|
|
|
|
//[self download];
|
|
}
|
|
|
|
|
|
|
|
#if 0
|
|
-(void)validateURL: (NSString *)url {
|
|
NSURL *v;
|
|
|
|
if (![url length]) {
|
|
_effectiveURL = [NSURL URLWithString: _downloadURL];
|
|
[_downloadField setTextColor: nil];
|
|
return;
|
|
}
|
|
|
|
v = [NSURL URLWithString: url];
|
|
if (v) {
|
|
_effectiveURL = v;
|
|
[_downloadField setTextColor: nil];
|
|
} else {
|
|
_effectiveURL = [NSURL URLWithString: _downloadURL];
|
|
[_downloadField setTextColor: [NSColor systemRedColor]];
|
|
}
|
|
}
|
|
#endif
|
|
|
|
-(void)downloadItem: (DownloadItem *)item {
|
|
|
|
if (!_session) {
|
|
NSURLSessionConfiguration *config = [NSURLSessionConfiguration defaultSessionConfiguration];
|
|
_session = [NSURLSession sessionWithConfiguration: config delegate: self delegateQueue: nil];
|
|
}
|
|
|
|
NSURLSessionDownloadTask *task;
|
|
NSString *s = [item value];
|
|
NSString *path = [s stringByAppendingPathExtension: _downloadExtension];
|
|
NSURL *url = [_downloadURL URLByAppendingPathComponent: path];
|
|
|
|
task = [_session downloadTaskWithURL: url];
|
|
|
|
[item beginDownloadWithTask: task];
|
|
[_taskIndex setObject: item forKey: task];
|
|
|
|
[task resume];
|
|
|
|
}
|
|
|
|
-(void)download {
|
|
|
|
// run in thread?
|
|
//unsigned count = 0;
|
|
|
|
for (DownloadItem *item in _items) {
|
|
|
|
NSURLSessionDownloadTask *task;
|
|
NSString *s = [item value];
|
|
NSString *path = [s stringByAppendingPathExtension: _downloadExtension];
|
|
NSURL *url = [_downloadURL URLByAppendingPathComponent: path];
|
|
|
|
task = [_session downloadTaskWithURL: url];
|
|
[_taskIndex setObject: item forKey: task];
|
|
|
|
[item setTask: task];
|
|
|
|
[task resume];
|
|
|
|
//++count;
|
|
//if (count >= 2) break;
|
|
}
|
|
[self setActive: YES];
|
|
|
|
}
|
|
|
|
-(DownloadItem *)clickedItem {
|
|
NSInteger row = [_tableView clickedRow];
|
|
if (row < 0 || row >= [_items count]) return nil;
|
|
return [[_arrayController arrangedObjects] objectAtIndex: row];
|
|
//return [_items objectAtIndex: row];
|
|
}
|
|
#if 0
|
|
-(void)redrawRow: (NSUInteger)row {
|
|
|
|
//NSRect r = [_tableView rectOfRow: row];
|
|
//[_tableView setNeedsDisplayInRect: r];
|
|
|
|
NSIndexSet *rIx = [NSIndexSet indexSetWithIndex: row];
|
|
NSIndexSet *cIx = [NSIndexSet indexSetWithIndex: 0];
|
|
|
|
[_tableView reloadDataForRowIndexes: rIx columnIndexes: cIx];
|
|
}
|
|
#endif
|
|
|
|
-(void)initializeExtensionMenu {
|
|
|
|
unsigned tag;
|
|
// mark default download extension.
|
|
NSString *defaultExt = [_defaults stringForKey: kDefaultDownloadExtension];
|
|
tag = [DownloadExtensionTransformer stringToNumber: defaultExt];
|
|
|
|
NSMenuItem *item = [[_formatButton menu] itemWithTag: tag];
|
|
if (item) {
|
|
[item setAttributedTitle: ItalicMenuString([item title])];
|
|
}
|
|
|
|
#if 0
|
|
// handled via binding.
|
|
NSString *ext = [_defaults stringForKey: kDownloadExtension];
|
|
if ([ext length]) {
|
|
ix = [DownloadExtensionTransformer stringToNumber: ext];
|
|
}
|
|
|
|
[_formatButton selectItemWithTag: tag];
|
|
#endif
|
|
}
|
|
|
|
-(void) defocus {
|
|
[[self window] makeFirstResponder: nil];
|
|
}
|
|
|
|
#pragma mark - IBActions
|
|
|
|
-(IBAction)cancelAll:(id)sender {
|
|
|
|
for (DownloadItem *item in _items) {
|
|
[item cancelDownload];
|
|
}
|
|
|
|
[_session invalidateAndCancel];
|
|
_session = nil;
|
|
[_taskIndex removeAllObjects];
|
|
[self setCurrentCount: 0];
|
|
[self setActive: NO];
|
|
}
|
|
|
|
- (IBAction)downloadMissing:(id)sender {
|
|
|
|
[self defocus];
|
|
BOOL delta = NO;
|
|
for (DownloadItem *item in _items) {
|
|
NSURL *url = [item localURL];
|
|
id task = [item task];
|
|
if (!url && !task) {
|
|
[self downloadItem: item];
|
|
delta = YES;
|
|
}
|
|
}
|
|
|
|
if (delta) {
|
|
[self setActive: YES];
|
|
}
|
|
}
|
|
- (IBAction)showRomFolder:(id)sender {
|
|
NSWorkspace *ws = [NSWorkspace sharedWorkspace];
|
|
|
|
[ws openURL: _romFolder];
|
|
}
|
|
|
|
-(IBAction)refreshROMs: (id)sender {
|
|
|
|
NSString *romdir = [SupportDirectoryPath() stringByAppendingPathComponent: @"roms"];
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
|
|
for (DownloadItem *item in _items) {
|
|
NSDictionary *attr = nil;
|
|
NSString *value = [item value];
|
|
NSString *s = [romdir stringByAppendingPathComponent: value];
|
|
NSString *path;
|
|
|
|
for (unsigned i = 0; i < 2; ++i) {
|
|
path = [s stringByAppendingPathExtension: Extensions[i]];
|
|
attr = [fm attributesOfItemAtPath: path error: nil];
|
|
if (attr) break;
|
|
}
|
|
|
|
if (attr) {
|
|
[item setStatus: ItemFound];
|
|
[item setLocalURL: [NSURL fileURLWithPath: path]];
|
|
[item setFileSize: [attr fileSize]];
|
|
} else {
|
|
[item setStatus: ItemMissing];
|
|
[item setLocalURL: nil];
|
|
[item setFileSize: 0];
|
|
}
|
|
}
|
|
}
|
|
|
|
- (IBAction)showInFinder:(id)sender {
|
|
DownloadItem *item = [self clickedItem];
|
|
if (!item) return;
|
|
NSURL *url = [item localURL];
|
|
if (!url) return;
|
|
|
|
NSWorkspace *ws = [NSWorkspace sharedWorkspace];
|
|
[ws activateFileViewerSelectingURLs: @[url]];
|
|
}
|
|
|
|
- (IBAction)moveToTrash:(id)sender {
|
|
NSError *error = nil;
|
|
|
|
DownloadItem *item = [self clickedItem];
|
|
if (!item) return;
|
|
NSURL *url = [item localURL];
|
|
if (!url) return;
|
|
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
if ([fm trashItemAtURL: url resultingItemURL: NULL error: &error]) {
|
|
[item refresh: nil];
|
|
} else {
|
|
[self presentError: error];
|
|
}
|
|
|
|
}
|
|
|
|
- (IBAction)download:(id)sender {
|
|
DownloadItem *item = [self clickedItem];
|
|
if (!item) return;
|
|
|
|
[self defocus];
|
|
|
|
[self downloadItem: item];
|
|
[self setActive: YES];
|
|
}
|
|
- (IBAction)cancel:(id)sender {
|
|
DownloadItem *item = [self clickedItem];
|
|
if (!item) return;
|
|
|
|
[item cancelDownload];
|
|
}
|
|
|
|
// binding screws up with placeholder.
|
|
-(IBAction)downloadURLChanged: (NSTextField *)sender {
|
|
NSString *value;
|
|
value = [sender stringValue];
|
|
if (![value length]) {
|
|
[_defaults removeObjectForKey: kDownloadURL];
|
|
_downloadURL = _defaultDownloadURL;
|
|
return;
|
|
}
|
|
// [self validateURL: value];
|
|
[_defaults setValue: value forKey: kDownloadURL];
|
|
_downloadURL = [NSURL URLWithString: value];
|
|
}
|
|
- (IBAction)downloadExtensionChanged:(id)sender {
|
|
[_defaults setValue: _downloadExtension forKey: kDownloadExtension];
|
|
}
|
|
|
|
|
|
- (IBAction)filterButton:(id)sender {
|
|
|
|
NSPredicate *p = nil;
|
|
NSUInteger tag = [sender tag];
|
|
[sender setState: NSControlStateValueOn];
|
|
|
|
|
|
for (NSButton *b in _filterButtons) {
|
|
if (b != sender) [b setState: NSControlStateValueOff];
|
|
}
|
|
switch (tag) {
|
|
case 1: // all
|
|
default:
|
|
[_arrayController setFilterPredicate: nil];
|
|
break;
|
|
case 2: // missing.
|
|
p = [NSPredicate predicateWithBlock: ^BOOL(DownloadItem *item, NSDictionary *bindings){
|
|
NSURL *localURL = [item localURL];
|
|
return localURL == nil;
|
|
}];
|
|
|
|
[_arrayController setFilterPredicate: p];
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
#pragma mark - NSURLSessionDelegate
|
|
|
|
static NSInteger TaskStatusCode(NSURLSessionTask *task) {
|
|
NSURLResponse *response = [task response];
|
|
if ([response isKindOfClass: [NSHTTPURLResponse class]]) {
|
|
return [(NSHTTPURLResponse *)response statusCode];
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
|
|
-(void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error {
|
|
|
|
|
|
if (error) NSLog(@"Download error: %@", error);
|
|
|
|
NSInteger statusCode = TaskStatusCode(task);
|
|
if (!error && statusCode != 200) {
|
|
// treat as an error.
|
|
NSDictionary *info = @{
|
|
NSURLErrorKey: [[task originalRequest] URL],
|
|
NSLocalizedDescriptionKey: [NSHTTPURLResponse localizedStringForStatusCode: statusCode],
|
|
};
|
|
error = [NSError errorWithDomain: NSURLErrorDomain code: NSURLErrorFileDoesNotExist userInfo: info];
|
|
}
|
|
|
|
|
|
// not sure if strictly necessary but this happens in a background thread
|
|
// and these are used in KVO binding. Also, main thread only
|
|
// means no race conditions.
|
|
dispatch_async(dispatch_get_main_queue(), ^(void){
|
|
if (error) {
|
|
[self setErrorCount: self->_errorCount + 1];
|
|
} else {
|
|
[self setCurrentCount: self->_currentCount + 1];
|
|
}
|
|
NSMutableDictionary *taskIndex = self->_taskIndex;
|
|
DownloadItem *item = [taskIndex objectForKey: task];
|
|
[taskIndex removeObjectForKey: task];
|
|
|
|
if ([taskIndex count] == 0) {
|
|
[self setActive: NO];
|
|
}
|
|
|
|
if (item) {
|
|
[item completeWithError: error];
|
|
}
|
|
});
|
|
|
|
}
|
|
|
|
- (void)URLSession:(NSURLSession *)session downloadTask:(nonnull NSURLSessionDownloadTask *)task didFinishDownloadingToURL:(nonnull NSURL *)location {
|
|
|
|
|
|
// NSLog(@"%@", task);
|
|
// NSLog(@"%@", [task response]);
|
|
|
|
if (TaskStatusCode(task) != 200) return;
|
|
|
|
|
|
// need to move to the destination directory...
|
|
// file deleted after this function returns, so can't move asynchronously.
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
NSURL *src = [[task originalRequest] URL];
|
|
NSURL *dest = [_romFolder URLByAppendingPathComponent: [src lastPathComponent]];
|
|
NSError *error = nil;
|
|
|
|
[fm moveItemAtURL: location toURL: dest error: &error];
|
|
|
|
dispatch_async(dispatch_get_main_queue(), ^(void){
|
|
NSMutableDictionary *taskIndex = self->_taskIndex;
|
|
DownloadItem *item = [taskIndex objectForKey: task];
|
|
|
|
[item refresh: dest];
|
|
});
|
|
|
|
NSLog(@"%@", src);
|
|
}
|
|
|
|
|
|
|
|
-(NSURLCredential *)credentialForChallenge: (NSURLAuthenticationChallenge *)challenge {
|
|
|
|
|
|
//if ([challenge previousFailureCount]) return nil;
|
|
|
|
NSURLCredential *credential = nil;
|
|
NSURLProtectionSpace *space = [challenge protectionSpace];
|
|
|
|
|
|
OSStatus status;
|
|
NSDictionary *query;
|
|
CFTypeRef item = nil;
|
|
|
|
|
|
query = @{
|
|
(id)kSecClass: (id)kSecClassInternetPassword,
|
|
(id)kSecReturnData: (id)kCFBooleanTrue,
|
|
(id)kSecReturnAttributes: (id)kCFBooleanTrue,
|
|
(id)kSecAttrServer: [space host],
|
|
//(id)kSecAttrProtocol: [space protocol],
|
|
(id)kSecMatchLimit: (id)kSecMatchLimitOne,
|
|
};
|
|
|
|
status = SecItemCopyMatching((CFDictionaryRef)query, &item);
|
|
//NSLog(@"%@", query);
|
|
if (status != 0) return nil;
|
|
|
|
NSDictionary *d = (__bridge NSDictionary *)item;
|
|
//NSLog(@"%@", d);
|
|
|
|
NSString *account = [d objectForKey: (id)kSecAttrAccount];
|
|
NSData *passwordData = [d objectForKey: (id)kSecValueData];
|
|
NSString *password = [[NSString alloc] initWithData: passwordData encoding: NSUTF8StringEncoding];
|
|
|
|
|
|
credential = [NSURLCredential credentialWithUser: account password: password persistence: NSURLCredentialPersistenceForSession];
|
|
|
|
return credential;
|
|
|
|
}
|
|
|
|
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didReceiveChallenge:(NSURLAuthenticationChallenge *)challenge completionHandler:(void (^)(NSURLSessionAuthChallengeDisposition disposition, NSURLCredential *credential))completionHandler {
|
|
|
|
NSLog(@"challenge: %@", challenge);
|
|
|
|
if ([challenge previousFailureCount]) {
|
|
completionHandler(NSURLSessionAuthChallengeCancelAuthenticationChallenge, nil);
|
|
return;
|
|
}
|
|
|
|
NSURLProtectionSpace *space = [challenge protectionSpace];
|
|
NSString *method = [space authenticationMethod];
|
|
|
|
if ([NSURLAuthenticationMethodHTTPBasic isEqualToString: method] ||
|
|
[NSURLAuthenticationMethodNTLM isEqualToString: method] ||
|
|
[NSURLAuthenticationMethodHTTPDigest isEqualToString: method]) {
|
|
|
|
NSURLCredential *credential = [self credentialForChallenge: challenge];
|
|
|
|
if (credential) {
|
|
completionHandler(NSURLSessionAuthChallengeUseCredential, credential);
|
|
return;
|
|
}
|
|
}
|
|
|
|
completionHandler(NSURLSessionAuthChallengePerformDefaultHandling, nil);
|
|
}
|
|
|
|
@end
|
|
|
|
|
|
|
|
|
|
@implementation DownloadItem
|
|
|
|
-(void)beginDownloadWithTask:(NSURLSessionDownloadTask *)task {
|
|
[self setTask: task];
|
|
[self setError: nil];
|
|
if (task) [self setStatus: ItemDownloading];
|
|
}
|
|
|
|
-(void)cancelDownload {
|
|
if (!_task) return;
|
|
|
|
[_task cancel];
|
|
[self setTask: nil];
|
|
[self setStatus: ItemCanceled];
|
|
}
|
|
|
|
-(void)completeWithError: (NSError *)error {
|
|
[self setTask: nil];
|
|
if (error) {
|
|
[self setError: error];
|
|
[self setStatus: ItemError];
|
|
} else {
|
|
// what if there was an error moving it?
|
|
[self setError: nil];
|
|
[self setStatus: ItemDownloaded];
|
|
}
|
|
}
|
|
|
|
|
|
-(void)refresh: (NSURL *)localURL {
|
|
NSDictionary *attr = nil;
|
|
NSError * error = nil;
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
|
|
if (localURL) {
|
|
attr = [fm attributesOfItemAtPath: [localURL path] error: &error];
|
|
} else {
|
|
NSString *romdir = [SupportDirectoryPath() stringByAppendingPathComponent: @"roms"];
|
|
NSString *s = [romdir stringByAppendingPathComponent: _value];
|
|
|
|
NSString *path = nil;
|
|
for (unsigned i = 0; i < 2; ++i) {
|
|
path = [s stringByAppendingPathExtension: Extensions[i]];
|
|
attr = [fm attributesOfItemAtPath: path error: nil];
|
|
if (attr) {
|
|
localURL = [NSURL fileURLWithPath: path];
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (attr && localURL) {
|
|
[self setLocalURL: localURL];
|
|
[self setFileSize: [attr fileSize]];
|
|
[self setStatus: ItemFound];
|
|
} else {
|
|
[self setLocalURL: nil];
|
|
[self setFileSize: 0];
|
|
if (_status == ItemFound || _status == ItemDownloaded)
|
|
[self setStatus: ItemMissing];
|
|
}
|
|
|
|
}
|
|
|
|
+(NSSet *)keyPathsForValuesAffectingStatusDescription {
|
|
return [NSSet setWithObjects: @"error", @"status", nil];
|
|
}
|
|
-(NSString *)statusDescription {
|
|
|
|
static NSString *Names[] = {
|
|
@"ROM missing",
|
|
@"ROM found",
|
|
@"Downloading…",
|
|
@"Downloaded",
|
|
@"Canceled",
|
|
@"Error"
|
|
};
|
|
if (_error) return [_error localizedDescription];
|
|
|
|
if (_status > sizeof(Names)/sizeof(Names[0])) return @"Unknown";
|
|
return Names[_status];
|
|
}
|
|
|
|
+(NSSet *)keyPathsForValuesAffectingTitleColor {
|
|
return [NSSet setWithObject: @"localURL"];
|
|
}
|
|
-(NSColor *)titleColor {
|
|
return _localURL ? nil : [NSColor systemRedColor];
|
|
}
|
|
+(NSSet *)keyPathsForValuesAffectingDescriptionColor {
|
|
return [NSSet setWithObject: @"error"];
|
|
}
|
|
-(NSColor *)descriptionColor {
|
|
return _error ? [NSColor systemRedColor] : nil;
|
|
}
|
|
|
|
@end
|
|
|
|
|
|
|
|
@implementation DownloadWindowController (Menu)
|
|
|
|
enum {
|
|
kOpenInFinder = 1,
|
|
kDownload,
|
|
kCancel,
|
|
};
|
|
|
|
- (BOOL)validateMenuItem:(NSMenuItem *)menuItem {
|
|
|
|
SEL action = [menuItem action];
|
|
if (action == @selector(downloadExtensionChanged:)) return YES;
|
|
|
|
DownloadItem *item = [self clickedItem];
|
|
if (!item) return NO;
|
|
#if 0
|
|
NSInteger row = [_tableView clickedRow];
|
|
if (row < 0) return NO;
|
|
DownloadItem *item = [[_arrayController arrangedObjects] objectAtIndex: row]; //[_items objectAtIndex: row];
|
|
#endif
|
|
|
|
NSUInteger status = [item status];
|
|
|
|
if (action == @selector(moveToTrash:)) {
|
|
return [item localURL] != nil;
|
|
}
|
|
if (action == @selector(showInFinder:)) {
|
|
return [item localURL] != nil;
|
|
}
|
|
if (action == @selector(download:)) {
|
|
return YES;
|
|
}
|
|
if (action == @selector(cancel:)) {
|
|
return status == ItemDownloading;
|
|
}
|
|
|
|
|
|
return NO;
|
|
}
|
|
|
|
@end
|