1
0
mirror of https://github.com/vcmi/vcmi.git synced 2024-12-20 20:23:03 +02:00
vcmi/osx-vcmibuilder/vcmibuilder/AppDelegate.m
2012-12-15 07:24:25 +00:00

311 lines
11 KiB
Objective-C

#import "AppDelegate.h"
@implementation AppDelegate
- (void)applicationDidFinishLaunching:(NSNotification*)aNotification
{
installationCompleted = NO;
outputDir = [[[NSBundle mainBundle] bundlePath] stringByAppendingString:@"/../../Data"];
tempDir = NSTemporaryDirectory();
}
- (BOOL)applicationShouldTerminateAfterLastWindowClosed:(NSApplication*)sender
{
return YES;
}
- (void)download:(NSURLDownload*)download didReceiveResponse:(NSURLResponse*)response
{
self->bytesRecieved = 0;
self->bytesExpected = [response expectedContentLength];
}
- (void)download:(NSURLDownload*)download didReceiveDataOfLength:(NSUInteger)length
{
self->bytesRecieved += length;
[self showProgressText:[NSString stringWithFormat:@"Downloading WoG archive: %3.1f Mb / %3.1f Mb",
self->bytesRecieved / 1024.0f / 1024.0f, self->bytesExpected / 1024.0f / 1024.0f]];
}
- (void)download:(NSURLDownload*)download decideDestinationWithSuggestedFilename:(NSString*)filename
{
[download setDestination:[tempDir stringByAppendingString:@"/wog.zip"] allowOverwrite:YES];
}
- (void)downloadDidFinish:(NSURLDownload*)download
{
[self showProgressText:@"Downloading WoG archive: completed"];
[self nextAction];
}
- (void)download:(NSURLDownload*)download didFailWithError:(NSError*)error
{
[self showProgressText:@"Downloading WoG archive: failed"];
[self showErrorText:[error localizedDescription]];
}
- (void)nextAction
{
dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
SEL sel = NSSelectorFromString(actions[0]);
[actions removeObjectAtIndex:0];
@try {
[self performSelector:sel];
}
@catch (NSException* e) {
[self showErrorText:[e name]];
}
});
}
- (int)runTask:(NSString*)executable withArgs:(NSArray*)args withWorkingDir:(NSString*)workingDir withPipe:(NSPipe*)pipe
{
if (![executable hasPrefix:@"/usr/"]) {
executable = [[[NSBundle mainBundle] resourcePath] stringByAppendingString:executable];
}
NSTask* task = [[NSTask alloc] init];
[task setLaunchPath:executable];
if (workingDir != nil) {
[task setCurrentDirectoryPath:workingDir];
}
if (pipe != nil) {
[task setStandardOutput:pipe];
}
[task setArguments:args];
[task launch];
[task waitUntilExit];
return [task terminationStatus];
}
- (void)validateAction
{
// Before starting anything run validations
if (![[NSFileManager defaultManager] fileExistsAtPath:[self.cd1TextField stringValue]]) {
return [self showErrorText:@"Please select existing file"];
}
// Show progress controls
[self.progressIndicator setHidden:NO];
[self.progressIndicator startAnimation:self];
[self showProgressText:@"Installing VCMI..."];
[self nextAction];
}
- (void)downloadWogArchive
{
// First of all we need to download WoG archive
// Downloading should be done on main thread because of callbacks
dispatch_async(dispatch_get_main_queue(), ^{
NSURL* url = [NSURL URLWithString:@"http://download.vcmi.eu/WoG/wog.zip"];
self.download = [[NSURLDownload alloc] initWithRequest:[NSURLRequest requestWithURL:url] delegate:self];
});
}
- (void)unzipWogArchive
{
// Then we unzip downloaded WoG archive
[self showProgressText:@"Unzipping WoG archive"];
if ([self runTask:@"/usr/bin/unzip" withArgs:@[@"-qo", [tempDir stringByAppendingString:@"/wog.zip"], @"-d", outputDir] withWorkingDir:nil withPipe:nil] != 0) {
return [self showErrorText:@"Failed to unzip WoG archive"];
}
[self nextAction];
}
- (void)extractGameData
{
// Then we extract game data from provided iso files using unshield or from innosetup exe
if ([[self.cd1TextField stringValue] hasSuffix:@".exe"]) {
[self innoexctract];
} else {
[self unshield];
}
[self nextAction];
}
- (void)innoexctract
{
// Extraction via innoextact is pretty straightforward
[self showProgressText:@"Extracting game data using innoextract..."];
if ([self runTask:@"/innoextract" withArgs:@[[self.cd1TextField stringValue]] withWorkingDir:tempDir withPipe:nil] != 0) {
[self showErrorText:@"Failed to exctract game data using innoextract"];
}
dataDir = [tempDir stringByAppendingString:@"/app"];
}
- (NSString*)attachDiskImage:(NSString*)path
{
[self showProgressText:[NSString stringWithFormat:@"Mounting image \"%@\"", path]];
// Run hdiutil to mount specified disk image
NSPipe* pipe = [NSPipe pipe];
if ([self runTask:@"/usr/bin/hdiutil" withArgs:@[@"attach", path] withWorkingDir:nil withPipe:pipe] != 0) {
[NSException raise:[NSString stringWithFormat:@"Failed to mount \"%@\"", path] format:nil];
}
// Capture hdiutil output to get mounted disk image filesystem path
NSFileHandle* file = [pipe fileHandleForReading];
NSString* output = [[NSString alloc] initWithData:[file readDataToEndOfFile] encoding:NSUTF8StringEncoding];
NSRegularExpression* regex = [NSRegularExpression regularExpressionWithPattern:@"(/Volumes/.*)$" options:0 error:nil];
NSTextCheckingResult* match = [regex firstMatchInString:output options:0 range:NSMakeRange(0, [output length])];
return [output substringWithRange:[match range]];
}
- (void)detachDiskImage:(NSString*)mountedPath
{
if ([self runTask:@"/usr/bin/hdiutil" withArgs:@[@"detach", mountedPath] withWorkingDir:nil withPipe:nil] != 0) {
[NSException raise:[NSString stringWithFormat:@"Failed to unmount \"%@\"", mountedPath] format:nil];
}
}
- (void)unshield
{
// In case of iso files we should mount them first
// If CD2 is not specified use the same path as for CD1
NSString* cd1 = [self attachDiskImage:[self.cd1TextField stringValue]];
NSString* cd2 = [[self.cd2TextField stringValue] isEqualToString:@""] ? cd1 : [self attachDiskImage:[self.cd2TextField stringValue]];
// Extract
[self showProgressText:@"Extracting game data using unshield..."];
if ([self runTask:@"/unshield" withArgs:@[@"-d", tempDir, @"x", [cd1 stringByAppendingString:@"/_setup/data1.cab"]] withWorkingDir:tempDir withPipe:nil] != 0) {
return [self showErrorText:@"Failed to extract game data using unshield"];
}
dataDir = [tempDir stringByAppendingString:@"/Heroes3"];
if (![[NSFileManager defaultManager] fileExistsAtPath:dataDir]) {
// Some releases have "Program_Files" folder instead of "Heroes3"
dataDir = [tempDir stringByAppendingString:@"/Program_Files"];
}
// Unmount CD1. Unmount CD2 if needed
[self detachDiskImage:cd1];
if (![cd1 isEqualToString:cd2]) {
[self detachDiskImage:cd2];
}
}
- (void)extractionCompleted
{
// After game data is extracted we should move it to destination place
[self showProgressText:@"Moving items into place"];
NSFileManager* fileManager = [NSFileManager defaultManager];
[fileManager moveItemAtPath:[dataDir stringByAppendingString:@"/Data"] toPath:[outputDir stringByAppendingString:@"/Data"] error:nil];
[fileManager moveItemAtPath:[dataDir stringByAppendingString:@"/Maps"] toPath:[outputDir stringByAppendingString:@"/Maps"] error:nil];
if ([fileManager fileExistsAtPath:[dataDir stringByAppendingString:@"/MP3"] isDirectory:nil]) {
[fileManager moveItemAtPath:[dataDir stringByAppendingString:@"/MP3"] toPath:[outputDir stringByAppendingString:@"/Mp3"] error:nil];
} else {
[fileManager moveItemAtPath:[dataDir stringByAppendingString:@"/Mp3"] toPath:[outputDir stringByAppendingString:@"/Mp3"] error:nil];
}
// After everythin is complete we create marker file. VCMI will look for this file to exists on startup and
// will run this setup otherwise
system([[NSString stringWithFormat:@"touch %@/game_data_prepared", outputDir] UTF8String]);
[self showProgressText:@"Installation complete"];
[self.installButton setTitle:@"Run VCMI"];
[self.progressIndicator stopAnimation:self];
[NSApp requestUserAttention:NSCriticalRequest];
// Hide all progress related controls
[self.progressIndicator setHidden:YES];
[self.progressIndicator stopAnimation:self];
[self.progressLabel setHidden:YES];
[self.installButton setEnabled:YES];
installationCompleted = YES;
}
- (void)selectFile:(NSArray*)fileTypes withTextField:(NSTextField*)textField
{
NSOpenPanel* openPanel = [NSOpenPanel openPanel];
[openPanel setCanChooseFiles:YES];
[openPanel setAllowedFileTypes:fileTypes];
[openPanel setAllowsMultipleSelection:NO];
if ([openPanel runModal] == NSOKButton) {
NSString* path = [[openPanel URL] path];
[textField setStringValue:path];
}
}
- (IBAction)selectCD1:(id)sender
{
[self selectFile:@[@"iso", @"exe"] withTextField:self.cd1TextField];
}
- (IBAction)selectCD2:(id)sender
{
[self selectFile:@[@"iso"] withTextField:self.cd2TextField];
}
- (IBAction)install:(id)sender
{
if (installationCompleted) {
// Run vcmi
system([[NSString stringWithFormat:@"open %@/../../..", [[NSBundle mainBundle] bundlePath]] UTF8String]);
[NSApp terminate: nil];
} else {
// Run installation
[self.cd1Button setEnabled:NO];
[self.cd2Button setEnabled:NO];
[self.installButton setEnabled:NO];
actions = [NSMutableArray arrayWithObjects:
@"validateAction",
@"downloadWogArchive",
@"unzipWogArchive",
@"extractGameData",
@"extractionCompleted",
nil
];
[self nextAction];
}
}
- (void)showProgressText:(NSString*)text
{
// All GUI updates should be done on main thread
dispatch_async(dispatch_get_main_queue(), ^{
[self.progressLabel setHidden:NO];
[self.progressLabel setStringValue:text];
});
}
- (void)showErrorText:(NSString*)text
{
// All GUI updates should be done on main thread
dispatch_async(dispatch_get_main_queue(), ^{
// Show error alert
NSAlert *alert = [[NSAlert alloc] init];
[alert setMessageText:@"Error"];
[alert setInformativeText:text];
[alert beginSheetModalForWindow:self.window modalDelegate:nil didEndSelector:nil contextInfo:nil];
// Enable select file buttons again
[self.cd1Button setEnabled:YES];
[self.cd2Button setEnabled:YES];
[self.installButton setEnabled:YES];
// Hide all progress related controls
[self.progressIndicator setHidden:YES];
[self.progressIndicator stopAnimation:self];
[self.progressLabel setHidden:YES];
});
}
@end