lupin/LupinBrowser/LPDocument.m

/*
 * Copyright (c) 2004-2005 The Lupin Project. All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without modification, are permitted 
 * provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the above copyright notice, this list of conditions 
 *    and the following disclaimer.
 * 
 * 2. 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.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE LUPIN PROJECT ``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 THE LUPIN PROJECT OR CONTRIBUTORS 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 "LPDocument.h"

@implementation LPDocument

- (id)init
{
        self = [super init];
        if (self) {
        // Add your subclass-specific initialization here.
        // If an error occurs here, send a [self release] message and return nil.
                _archiveAllUncompressMode = NO;
    }
    return self;
}


- (void)dealloc
{
        if (_plugins) [_plugins release];
        if (_plugin) [_plugin release];
        if (_infoDict) [_infoDict release];
        if (_droppedTemporaryPath) [_droppedTemporaryPath release];
        if (_selectedArray) [_selectedArray release];
        [super dealloc];
}

// make and add WindowController
- (void)makeWindowControllers
{
        NSLog(@"makeWindowControllers");
        NSLog([self fileType]);
        LPMainWindowController* windowController;
        if ([[self fileType] isEqualToString:@"LupinCompressType"]) {
                windowController = [[LPMainWindowController alloc] initWithWindowNibName:@"LPProgressDocument"];
        } else if ([[self fileType] isEqualToString:@"LupinUncompressType"]) {
                windowController = [[LPMainWindowController alloc] initWithWindowNibName:@"LPProgressDocument"];
        } else {
                windowController = [[LPMainWindowController alloc] initWithWindowNibName:@"LPBrowseDocument"];
        }
        [self addWindowController:windowController];
        [windowController release];
}

/*
 - (NSString *)windowNibName
 {
         // If you need to use a subclass of NSWindowController or if your document supports multiple NSWindowControllers, you should remove this method and override -makeWindowControllers instead.
         return @"LPBrowseDocument";
 }
*/ 

/*
- (void)windowControllerDidLoadNib:(NSWindowController *) aController
{
        NSLog(@"windowControllerDidLoadNib");
        [super windowControllerDidLoadNib:aController];
}
*/

- (NSData *)dataRepresentationOfType:(NSString *)aType
{
    // Insert code here to write your document from the given data.  You can also choose to override -fileWrapperRepresentationOfType: or -writeToFile:ofType: instead.
    return nil;
}

/*
- (BOOL)loadDataRepresentation:(NSData *)data ofType:(NSString *)aType
{
    // Insert code here to read your document from the given data.  You can also choose to override -loadFileWrapperRepresentation:ofType: or -readFromFile:ofType: instead.
    return YES;
}
*/

//���������������������������������������������������������������������������������
- (BOOL)readFromFile:(NSString *)fileName ofType:(NSString *)docType
{
        NSLog(@"fileName=%@ type=%@",fileName,docType);
        NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
        LPFile* infoDict;
        BOOL hierarchy = [defaults boolForKey:@"inspectIsHierarchy"];
        int result;
        if ([docType isEqualToString:@"LupinCompressType"]) {
                return YES;
        }
        
        _plugins = [[[NSDocumentController sharedDocumentController] pluginsController] acceptablePlugins:fileName];
        [_plugins retain];
        
        if ([docType isEqualToString:@"LupinUncompressType"]) {
                if ([_plugins count] != 0) {
                        _plugin = [_plugins lastObject];
                        [_plugin retain];
                        [_plugins removeLastObject];
                }
                if (!_plugin) {
                        return NO;
                }
                return YES;
        } else if ([docType isEqualToString:@"LupinInspectType"]) {
                NSEnumerator* enumerator = [_plugins objectEnumerator];
                while (_plugin = [enumerator nextObject]) {
                        infoDict = [self buildArchiveInfoDictFromPlugin:_plugin hierarchy:hierarchy];
                        if (infoDict) break;
                }
                if (!infoDict) {
                        return NO;
                } else {
                        _infoDict = infoDict;
                        [_infoDict retain];
                        [_plugin retain];
                        return YES;
                }
        } else if ([_plugins count] == 0) {
                // ���������������������������������������������������������������������������������������������������NO������������
                switch ([defaults integerForKey:@"generalFileDropped"]) {
                        case 1:
                                break;
                        case 2:
                                result = NSRunAlertPanel(NSLocalizedString(@"DialogCompressChooseArchiveTitle",nil),
                                                                                 [NSString stringWithFormat:NSLocalizedString(@"DialogAskCreateArchiveMessage",nil),fileName],
                                                                                 NSLocalizedString(@"DialogYes",nil),
                                                                                 NSLocalizedString(@"DialogNo",nil),
                                                                                 nil);
                                if (result == NSCancelButton)
                                        break;
                                case 0:
                                        [[NSDocumentController sharedDocumentController] compressFiles:[NSArray arrayWithObject:fileName]];
                }
                return NO;
        }
        switch ([defaults integerForKey:@"generalArchiveDropped"]) {
                case 0:
                        break;
                case 2:
                        result = NSRunAlertPanel(NSLocalizedString(@"DialogAskUncompressOrInspectTitle",nil),
                                                                         [NSString stringWithFormat:NSLocalizedString(@"DialogAskUncompressOrInspectMessage",nil),fileName],
                                                                         NSLocalizedString(@"DialogInspect",nil),
                                                                         NSLocalizedString(@"DialogUncompress",nil),
                                                                         nil);
                        if (result == NSOKButton) break;
                        case 1:
                                [[NSDocumentController sharedDocumentController] uncompressFile:fileName];
                                return NO;
        }
        //������������������������������������
        NSEnumerator* enumerator = [_plugins objectEnumerator];
        while (_plugin = [enumerator nextObject]) {
                infoDict = [self buildArchiveInfoDictFromPlugin:_plugin hierarchy:hierarchy];
                if (infoDict) break;
        }
        if (!infoDict) {
                return NO;
        } else {
                _infoDict = infoDict;
                [_infoDict retain];
                [_plugin retain];
                return YES;
        }
}

- (LPFile*)buildArchiveInfoDictFromPlugin:(NSObject<LupinPluginProtocol>*)plugin hierarchy:(BOOL)hierarchy
{
        LPFile* infoDict;
        infoDict = [plugin buildArchiveInfoDict:hierarchy password:NULL];
        if (infoDict) {
                return infoDict;
        } else {
                return NULL;
        }
}

/*
//������������������������������������������������������������������������������
- (void)archiveInfoDictDidBuild:(LPFile*)infoDict error:(LPError)error
{
        NSLog(@"archiveInfoDictDidBuild");
        if (error == LPErrorNotImplementMethod || error  == LPErrorNotSupportFormat) {
                [_plugin release];
                if ([_plugins count] > 0) {
                        _plugin = [_plugins lastObject];
                        [_plugin retain];
                        [_plugins removeLastObject];
                        [_plugin buildArchiveInfoDict:[self fileName] hierarchy:[[NSUserDefaults standardUserDefaults] boolForKey:@"inspectIsHierarchy"] password:nil delegate:self selector:@selector(archiveInfoDictDidBuild:error:)];
                        return;
                }
        } else if (error == LPErrorRequestPassword || error == LPErrorBadPassword) {
                NSLog(@"asking password");
                NSString* password = [[self mainWindowController] askPassword];
                if (password) {
                        [_plugin buildArchiveInfoDict:[self fileName] hierarchy:[[NSUserDefaults standardUserDefaults] boolForKey:@"inspectIsHierarchy"] password:password delegate:self selector:@selector(archiveInfoDictDidBuild:error:)];
                        return;
                }
        }
        if (error != LPErrorNoError) {
                [[self mainWindowController] showErrorAlertWithErrorID:error];
        }
        if (!infoDict) {
                NSLog(@"Can't load archive!");
                [self close];
                return;
        }
        _infoDict = infoDict;
        [_infoDict retain];
        [[self mainWindowController] archiveInfoDictDidBuild];
}
*/

- (LPFile*)infoDict
{
        return _infoDict;
}

- (LPMainWindowController*)mainWindowController
{
        return [[self windowControllers] objectAtIndex:0];
}

- (void)setPlugin:(NSObject<LupinPluginProtocol>*)plugin
{
        _plugin = plugin;
        [_plugin retain];
}

- (NSObject<LupinPluginProtocol>*)plugin
{
        return _plugin;
}

#pragma mark -
#pragma mark Drag&Drop

- (void)setSelectedArray:(NSArray*)selectedArray
{
        if (_selectedArray) [_selectedArray release];
        _selectedArray = [NSArray arrayWithArray:selectedArray];
        [_selectedArray retain];
}

#pragma mark -
#pragma mark Uncompress

- (void)archiveUncompress:(NSString*)dropPath
{
        NSFileManager *manager = [NSFileManager defaultManager];
        NSDictionary *dic = [NSDictionary dictionaryWithObjectsAndKeys:
                [NSDate date],NSFileModificationDate,
                @"owner",@"NSFileOwnerAccountName",
                @"group",@"NSFileGroupOwnerAccountName",
                nil,@"NSFilePosixPermissions",
                [NSNumber numberWithBool:YES],@"NSFileExtensionHidden",
                nil];
        
        [manager changeCurrentDirectoryPath:dropPath];
        NSString* temporaryDirectoryName = @"Lupin";
        int random;
        srand(time(NULL));
        while (1) {
                random = rand();
                if ([manager fileExistsAtPath:temporaryDirectoryName])
                        temporaryDirectoryName = [NSString stringWithFormat:@"Lupin%d",random];
                else
                        break;
        }
        _droppedTemporaryPath = [dropPath stringByAppendingPathComponent:temporaryDirectoryName];
        if (![manager createDirectoryAtPath:temporaryDirectoryName attributes:dic]) {
                NSLog(@"Can't create directory at %@",temporaryDirectoryName);
                [[self mainWindowController] showErrorAlert:NSLocalizedString(@"DialogCantCreateDirectoryTitle",nil)
                                                                                           info:[NSString stringWithFormat:NSLocalizedString(@"DialogCantCreateDirectoryInfo",nil),_droppedTemporaryPath]];
                [[self mainWindowController] archiveDidUncompress]; //! ���������������
                return;
        }
        //���������������������������������������������������������������������������������������������
        [[NSDocumentController sharedDocumentController] addDirectory:_droppedTemporaryPath];
        
        [_droppedTemporaryPath retain];
        [[self mainWindowController] archiveWillUncompress];
        [_plugin archiveUncompress:_droppedTemporaryPath files:_selectedArray password:nil delegate:self selector:@selector(archiveDidUncompress:)];
        [self showProcessWindow];
}

- (void)archiveAllUncompress:(NSString*)dropPath
{
        NSFileManager *manager = [NSFileManager defaultManager];
        NSDictionary *dic = [NSDictionary dictionaryWithObjectsAndKeys:
                [NSDate date],NSFileModificationDate,
                @"owner",@"NSFileOwnerAccountName",
                @"group",@"NSFileGroupOwnerAccountName",
                nil,@"NSFilePosixPermissions",
                [NSNumber numberWithBool:YES],@"NSFileExtensionHidden",
                nil];
        
        [manager changeCurrentDirectoryPath:dropPath];
        NSString* temporaryDirectoryName = @"Lupin";
        int random;
        srand(time(NULL));
        while (1) {
                random = rand();
                if ([manager fileExistsAtPath:temporaryDirectoryName])
                        temporaryDirectoryName = [NSString stringWithFormat:@"Lupin%d",random];
                else
                        break;
        }
        _droppedTemporaryPath = [dropPath stringByAppendingPathComponent:temporaryDirectoryName];
        if (![manager createDirectoryAtPath:temporaryDirectoryName attributes:dic]) {
                NSLog(@"Can't create directory at %@",temporaryDirectoryName);
                [[self mainWindowController] showErrorAlert:NSLocalizedString(@"DialogCantCreateDirectoryTitle",nil)
                                                                                           info:[NSString stringWithFormat:NSLocalizedString(@"DialogCantCreateDirectoryInfo",nil),_droppedTemporaryPath]];
                [[self mainWindowController] archiveDidUncompress]; //! ���������������
                return;
        }
        //���������������������������������������������������������������������������������������������
        [[NSDocumentController sharedDocumentController] addDirectory:_droppedTemporaryPath];
        
        [_droppedTemporaryPath retain];
        [[self mainWindowController] archiveWillUncompress];
        _archiveAllUncompressMode = YES;
        [_plugin archiveAllUncompress:_droppedTemporaryPath exclude:nil password:nil delegate:self selector:@selector(archiveDidUncompress:)];
        [self showProcessWindow];
}

- (void)archiveDidUncompress:(LPError)error
{
        NSLog(@"archiveDidUncompress");
        NSLog(@"Uncompressing Error #%d",error);
        if (error == LPErrorNotImplementMethod || error  == LPErrorNotSupportFormat) {
                [_plugin release];
                if ([_plugins count] > 0) {
                        _plugin = [_plugins lastObject];
                        [_plugin retain];
                        [_plugins removeLastObject];
                        [_plugin archiveAllUncompress:_droppedTemporaryPath exclude:nil password:nil delegate:self selector:@selector(archiveDidUncompress:)];
                        return;
                }
        } else if (error == LPErrorRequestPassword || error == LPErrorBadPassword) {
                NSLog(@"asking password");
                NSString* password = [[self mainWindowController] askPassword];
                if (password) {
                        if (_archiveAllUncompressMode)
                                [_plugin archiveAllUncompress:_droppedTemporaryPath exclude:nil password:password delegate:self selector:@selector(archiveDidUncompress:)];
                        else
                                [_plugin archiveUncompress:_droppedTemporaryPath files:_selectedArray password:password delegate:self selector:@selector(archiveDidUncompress:)];
                        return;
                }
        }
        if (error != LPErrorNoError) {
                [[self mainWindowController] showErrorAlertWithErrorID:error];
        }
        [_progressTimer invalidate];
        [_progressTimer release];
        NSFileManager* manager = [NSFileManager defaultManager];
        NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
        NSEnumerator* enumerator;
        NSMutableArray* array;
        NSString* filePath;
        
        array = [NSMutableArray array];
        if (_archiveAllUncompressMode) {
                _archiveAllUncompressMode = NO;
                //_droppedTemporaryPath������������������������������_droppedTemporaryPath������������������������������������������
                enumerator = [[manager directoryContentsAtPath:_droppedTemporaryPath] objectEnumerator];
                while ((filePath = [enumerator nextObject])) {
                        [array addObject:[_droppedTemporaryPath stringByAppendingPathComponent:filePath]];
                }
        } else {
                enumerator = [_selectedArray objectEnumerator];
                LPFile* file;
                while ((file = [enumerator nextObject])) {
                        [array addObject:[_droppedTemporaryPath stringByAppendingPathComponent:[file path]]];
                }
        }
        enumerator = [array objectEnumerator];
        NSString* toPath;
        BOOL allOverwrite = ([defaults integerForKey:@"uncompressExistMatrix"] == 1);
        while (((filePath = [enumerator nextObject]) && [manager fileExistsAtPath:filePath])) {
                toPath = [[_droppedTemporaryPath stringByDeletingLastPathComponent] stringByAppendingPathComponent:[filePath lastPathComponent]];
                if ([self moveFileFrom:filePath to:toPath onlyFile:([defaults integerForKey:@"uncompressOverwriteMatrix"] == 0) allOverwrite:&allOverwrite] == LPDialogCancel)
                        break;
        }
        // ���������������������������������������
        // ������������������[NSUserDefaults standardUserDefaults]������������������������������������������������
        // ������������������������������������������������������������
        if ([manager removeFileAtPath:_droppedTemporaryPath handler:nil])
                [[NSDocumentController sharedDocumentController] removeDirectory:_droppedTemporaryPath];
        else
                [[self mainWindowController] showErrorAlert:NSLocalizedString(@"DialogCantRemoveTemporaryFolderTitle",nil)
                                                                                           info:[NSString stringWithFormat:NSLocalizedString(@"DialogCantRemoveTemporaryFolderInfo",nil),_droppedTemporaryPath]];
        [[self mainWindowController] archiveDidUncompress];
}

// onlyFile���NO���������������Mac OS X������YES������������Windows������������������������������������������������
// NO���������������������������������������������������������������������������������������������������������������������
// YES������������������������������������������������������������������������������������������������������������������������������������������������
- (LPDialogResult)moveFileFrom:(NSString*)fromPath to:(NSString*)toPath onlyFile:(BOOL)onlyFile allOverwrite:(BOOL*)allOverwrite
{
        NSFileManager* manager = [NSFileManager defaultManager];
        BOOL result,isDir;
        LPDialogResult dialogResult = LPDialogYES;
        if (![manager fileExistsAtPath:fromPath isDirectory:&isDir]) //! ������������������������������������������������������������������������������������������������������
                return LPDialogNO;
        if (![manager fileExistsAtPath:toPath]) { //! ���������������������������������
                result = [manager movePath:fromPath toPath:toPath handler:nil];
                if (!result) {
                        NSLog(@"Failed(1) to move from %@ to %@",fromPath,toPath);
                        return LPDialogNO;
                }
        } else if (!onlyFile || !isDir) { //! ���������������������������������������������������������������������������������������������������������
                if (!*allOverwrite) //! ���������������������������������������������������������������������������������������
                        dialogResult = [[self mainWindowController] showChoiceAlert:allOverwrite
                                                                                                                                  title:NSLocalizedString(@"DialogAlreadyExistsTitle",nil) 
                                                                                                                  defaultButton:NSLocalizedString(@"DialogYes",nil) 
                                                                                                                alternateButton:NSLocalizedString(@"DialogNo",nil) 
                                                                                                                        otherButton:NSLocalizedString(@"DialogCancel",nil) 
                                                                                                                                message:[NSString stringWithFormat:NSLocalizedString(@"DialogAlreadyExistsInfo",nil),toPath]];
                if (allOverwrite || dialogResult == LPDialogYES)
                        [manager removeFileAtPath:toPath handler:nil];
                else if (dialogResult == LPDialogCancel) //! ������������������������������������������������������������������������������
                        return LPDialogCancel;
                result = [manager movePath:fromPath toPath:toPath handler:nil];
                if (!result) {
                        NSLog(@"Failed(2) to move from %@ to %@",fromPath,toPath);
                        return LPDialogNO;
                }
        } else { //! ������������������������������������������������������
                NSEnumerator* enumerator = [[manager subpathsAtPath:fromPath] objectEnumerator];
                NSString* filePath;
                while (filePath = [enumerator nextObject]) {
                        if ([self moveFileFrom:[fromPath stringByAppendingPathComponent:filePath] to:[toPath stringByAppendingPathComponent:filePath] onlyFile:onlyFile allOverwrite:allOverwrite] == LPDialogCancel)
                                return LPDialogCancel;
                }
        }
        return LPDialogYES;
}

#pragma mark -
#pragma mark Compress

- (void)archiveCompress:(NSString*)archiveName files:(NSArray*)files exclude:(NSArray*)exclude currentPath:(NSString*)currentPath compress:(LPCompress*)compress option:(NSDictionary*)option
{
        NSLog([files description]);
        [[self mainWindowController] archiveWillCompress];
        [_plugin archiveCompress:archiveName files:files exclude:exclude currentPath:currentPath compress:compress option:option delegate:self selector:@selector(archiveDidCompress:)];
        [self showProcessWindow];
}

- (void)archiveDidCompress:(LPError)error
{
        NSLog(@"Compressing Error #%d",error);
        if (error != LPErrorNoError)
                [[self mainWindowController] showErrorAlertWithErrorID:error];
        [_progressTimer invalidate];
        [_progressTimer release];
        [[self mainWindowController] archiveDidCompress];
}

#pragma mark -
#pragma mark Showing Information

- (void)showProcessWindow
{
        NSLog(@"showProcessWindow");
         _progressTimer = [[NSTimer scheduledTimerWithTimeInterval:0.05                                                 // ������������
                                                                                                           target:[self mainWindowController]   // ���������������������������������
                                                                                                         selector:@selector(progressUpdate)             // ������������������������������
                                                                                                         userInfo:nil                                                   // ���������������������
                                                                                                          repeats:YES] retain];                                 // ���������������������
        
        [_progressTimer fire];
}

#pragma mark -
#pragma mark IBOutlet
- (void)uncompress:(id)sender
{
        NSLog(@"(LPDocument) Uncompress selected.");
        
        [[self mainWindowController] archiveWillUncompress];
        
        NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
        NSString* destination = NULL;
        switch ([defaults integerForKey:@"uncompressCreateAtMatrix"]) {
                case 0:
                        destination = [[self fileName] stringByDeletingLastPathComponent];
                        break;
                case 1:
                        destination = [NSHomeDirectory() stringByAppendingPathComponent:@"Desktop"];
                        break;
                case 2:
                        destination = NSHomeDirectory();
                        break;
                case 3:
                        break;
                case 4:
                        destination = [defaults objectForKey:@"uncompressCreateAt"];
                        break;
        }
        if (destination) {
                [self archiveAllUncompress:destination];
                return;
        }
        NSOpenPanel* openPanel = [NSOpenPanel openPanel];
        [openPanel setTitle:NSLocalizedString(@"DialogUncompressChooseDestinationTitle",nil)];
        [openPanel setMessage:NSLocalizedString(@"DialogUncompressChooseDestinationMessage",nil)];
        [openPanel setPrompt:NSLocalizedString(@"DialogChooseFolderButtonTitle",nil)];
        [openPanel setCanChooseDirectories:YES];
        [openPanel setCanChooseFiles:NO];
        [openPanel beginSheetForDirectory:NSHomeDirectory() file:nil types:[NSArray arrayWithObjects:@"txt",nil] modalForWindow:[[self mainWindowController] window] modalDelegate:self didEndSelector:@selector(openPanelDidEnd:returnCode:contextInfo:) contextInfo:NULL];
}

- (void)openPanelDidEnd:(NSOpenPanel *)sheet returnCode:(int)returnCode contextInfo:(void *)contextInfo
{
        NSLog(@"panelDidEnd");
        NSLog([sheet filename]);
        [sheet close]; //! ���������������������archiveAllUncompress������������������������������������������������������������������������������������������������������
        if (returnCode == NSOKButton) {
                [self archiveAllUncompress:[sheet filename]];
        } else if ([[self fileType] isEqualToString:@"LupinUncompressType"]) {
                //���������������������������������������������������������������������������
                [self close];
        }
}
@end