source: trunk/macosx/MessageWindowController.m @ 3035

/******************************************************************************
 * $Id: MessageWindowController.m 3035 2007-09-10 19:28:15Z livings124 $
 *
 * Copyright (c) 2006-2007 Transmission authors and contributors
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 *****************************************************************************/

#import "MessageWindowController.h"
#import <transmission.h>

#define LEVEL_ERROR 0
#define LEVEL_INFO  1
#define LEVEL_DEBUG 2

#define UPDATE_SECONDS  0.6
#define MAX_MESSAGES    2500

@interface MessageWindowController (Private)

- (NSString *) stringForMessage: (NSDictionary *) message;
- (void) setDebugWarningHidden: (BOOL) hide;

@end

@implementation MessageWindowController

- (id) initWithWindowNibName: (NSString *) name
{
    if ((self = [super initWithWindowNibName: name]))
    {
        fMessages = [[NSMutableArray alloc] init];
        
        fTimer = [NSTimer scheduledTimerWithTimeInterval: UPDATE_SECONDS target: self
                    selector: @selector(updateLog:) userInfo: nil repeats: YES];
        
        int level = [[NSUserDefaults standardUserDefaults] integerForKey: @"MessageLevel"];
        if (level == TR_MSG_ERR)
            [fLevelButton selectItemAtIndex: LEVEL_ERROR];
        else if (level == TR_MSG_INF)
            [fLevelButton selectItemAtIndex: LEVEL_INFO];
        else if (level == TR_MSG_DBG)
            [fLevelButton selectItemAtIndex: LEVEL_DEBUG];
        else
        {
            level = TR_MSG_ERR;
            [fLevelButton selectItemAtIndex: LEVEL_ERROR];
            [[NSUserDefaults standardUserDefaults] setInteger: level forKey: @"MessageLevel"];
        }
        
        tr_setMessageLevel(level);
        tr_setMessageQueuing(1);
    }
    return self;
}

- (void) dealloc
{
    [fTimer invalidate];
    [fMessages release];
    
    [super dealloc];
}

- (void) awakeFromNib
{
    [[self window] center];
    
    int level = tr_getMessageLevel();
    if (level == TR_MSG_ERR)
        [fLevelButton selectItemAtIndex: LEVEL_ERROR];
    else if (level == TR_MSG_INF)
        [fLevelButton selectItemAtIndex: LEVEL_INFO];
    else if (level == TR_MSG_DBG)
        [fLevelButton selectItemAtIndex: LEVEL_DEBUG];
    else
    {
        level = TR_MSG_ERR;
        [fLevelButton selectItemAtIndex: LEVEL_ERROR];
        [[NSUserDefaults standardUserDefaults] setInteger: level forKey: @"MessageLevel"];
    }
    
    [self setDebugWarningHidden: level != TR_MSG_DBG];
}

- (void) updateLog: (NSTimer *) timer
{
    tr_msg_list_t * messages, * currentMessage;
    if ((messages = tr_getQueuedMessages()) == NULL)
        return;
    
    NSString * levelString;
    for (currentMessage = messages; currentMessage != NULL; currentMessage = currentMessage->next)
    {
        int level = currentMessage->level;
        if (level == TR_MSG_ERR)
            levelString = @"Error";
        else if (level == TR_MSG_INF)
            levelString = @"Info";
        else if (level == TR_MSG_DBG)
            levelString = @"Debug";
        else
            levelString = @"???";
        
        //remove the first line if at max number of lines
        /*if (fLines == MAX_LINES)
        {
            unsigned int loc = [[fTextView string] rangeOfString: @"\n"].location;
            if (loc != NSNotFound)
                [[fTextView textStorage] deleteCharactersInRange: NSMakeRange(0, loc + 1)];
        }*/
        
        [fMessages addObject: [NSDictionary dictionaryWithObjectsAndKeys:
                                [NSString stringWithUTF8String: currentMessage->message], @"Message",
                                [NSDate dateWithTimeIntervalSince1970: currentMessage->when], @"Date",
                                levelString, @"Level", nil]];
    }
    
    #warning still needed?
    int total = [fMessages count];
    if (total > MAX_MESSAGES)
        [fMessages removeObjectsInRange: NSMakeRange(0, total-MAX_MESSAGES)];
    
    [fMessageView reloadData];
    
    tr_freeMessageList(messages);
}

- (int) numberOfRowsInTableView: (NSTableView *) tableView
{
    return [fMessages count];
}

- (id) tableView: (NSTableView *) tableView objectValueForTableColumn: (NSTableColumn *) column row: (int) row
{
    NSString * ident = [column identifier];
    NSDictionary * message = [fMessages objectAtIndex: row];

    if ([ident isEqualToString: @"Date"])
        return [message objectForKey: @"Date"];
    else if ([ident isEqualToString: @"Level"])
        return [message objectForKey: @"Level"];
    else
        return [message objectForKey: @"Message"];
}

- (NSString *) tableView: (NSTableView *) tableView toolTipForCell: (NSCell *) cell rect: (NSRectPointer) rect
                tableColumn: (NSTableColumn *) column row: (int) row mouseLocation: (NSPoint) mouseLocation
{
    return [self stringForMessage: [fMessages objectAtIndex: row]];
}

- (void) changeLevel: (id) sender
{
    [self updateLog: nil];
    
    int selection = [fLevelButton indexOfSelectedItem], level;
    if (selection == LEVEL_INFO)
        level = TR_MSG_INF;
    else if (selection == LEVEL_DEBUG)
        level = TR_MSG_DBG;
    else
        level = TR_MSG_ERR;
    
    [self setDebugWarningHidden: level != TR_MSG_DBG];
    
    tr_setMessageLevel(level);
    [[NSUserDefaults standardUserDefaults] setInteger: level forKey: @"MessageLevel"];
}

- (void) clearLog: (id) sender
{
    [fMessages removeAllObjects];
    [fMessageView reloadData];
}

- (void) writeToFile: (id) sender
{
    //create the text to output
    NSMutableArray * messageStrings = [NSMutableArray arrayWithCapacity: [fMessages count]];
    
    NSEnumerator * enumerator = [fMessages objectEnumerator];
    NSDictionary * message;
    while ((message = [enumerator nextObject]))
        [messageStrings addObject: [self stringForMessage: message]];
    
    NSString * fileString = [[messageStrings componentsJoinedByString: @"\n"] retain];
    
    NSSavePanel * panel = [NSSavePanel savePanel];
    [panel setRequiredFileType: @"txt"];
    [panel setCanSelectHiddenExtension: YES];
    
    [panel beginSheetForDirectory: nil file: NSLocalizedString(@"untitled", "Save log panel -> default file name")
            modalForWindow: [self window] modalDelegate: self
            didEndSelector: @selector(writeToFileSheetClosed:returnCode:contextInfo:) contextInfo: fileString];
}

- (void) writeToFileSheetClosed: (NSSavePanel *) panel returnCode: (int) code contextInfo: (NSString *) string
{
    if (code == NSOKButton)
    {
        if (![string writeToFile: [panel filename] atomically: YES encoding: NSUTF8StringEncoding error: nil])
        {
            NSAlert * alert = [[NSAlert alloc] init];
            [alert addButtonWithTitle: NSLocalizedString(@"OK", "Save log alert panel -> button")];
            [alert setMessageText: [NSString stringWithFormat: NSLocalizedString(@"Log Could Not Be Saved",
                                    "Save log alert panel -> title")]];
            [alert setInformativeText: [NSString stringWithFormat: 
                    NSLocalizedString(@"There was a problem creating the file \"%@\".",
                                        "Save log alert panel -> message"), [[panel filename] lastPathComponent]]];
            [alert setAlertStyle: NSWarningAlertStyle];
            
            [alert runModal];
            [alert release];
        }
    }
    
    [string release];
}

@end

@implementation MessageWindowController (Private)

- (NSString *) stringForMessage: (NSDictionary *) message
{
    return [NSString stringWithFormat: @"%@ %@ %@", [message objectForKey: @"Date"],
                [message objectForKey: @"Level"], [message objectForKey: @"Message"]];
}

- (void) setDebugWarningHidden: (BOOL) hide
{
    [fDebugWarningField setHidden: hide];
    [fDebugWarningIcon setHidden: hide];
}

@end