DownAndOutController.m

//
//  DownAndOutController.m
//  DownAndOutª
//
//  Created by Fritz Anderson on Mon Aug 18 2003.
//  Copyright (c) 2003 Trustees of the University of Chicago. All rights reserved.
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation; either version 2 of the License, or
//  (at your option) any later version.
//  
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//  
//  You should have received a copy of the GNU General Public License
//  along with this program; if not, write to the Free Software
//  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
//

#define USE_DAPREFERENCES   1

#import "DownAndOutController.h"
#import "DownAndOutView.h"
#import "DAApplication.h"

#if USE_DAPREFERENCES
#include "DAPreferences.h"

static DAPreferences *          sPreferences = nil;
#endif

#define DEFAULT_LOGOUT      300

#define DEFAULT_SHOULD_MOVE YES

#define DEFAULT_MESSAGE     @"%n will log out automatically in %t"

#define DEFAULT_RADIUS      30

@implementation DownAndOutController

static DownAndOutController *   sMainController = nil;
static NSArray *                sTags = nil;

+ (void) initialize
{
    if (self == [DownAndOutController class]) {
        NSDictionary *          stdDefaults = [NSDictionary dictionaryWithObjectsAndKeys:
            [NSNumber numberWithInt: DEFAULT_LOGOUT], @"secondsBeforeLogout",
            [NSNumber numberWithBool: DEFAULT_SHOULD_MOVE], @"shouldMove",
            [NSNumber numberWithBool: NO], @"usesImage",
            [NSNumber numberWithInt: DEFAULT_RADIUS], @"clockRadius",
            DEFAULT_MESSAGE, @"messageText",
            nil];
#if USE_DAPREFERENCES
        sPreferences = [[DAPreferences alloc] initWithDefaults: stdDefaults];
#else
        ScreenSaverDefaults *   defaults = [ScreenSaverDefaults defaultsForModuleWithName: MODULE_NAME];
        [defaults registerDefaults: stdDefaults];
#endif

        sTags = [[NSArray alloc] initWithObjects:
            @"%s", @"%t", @"%n", @"%u", nil
            ];      
    }
}

- (id) initAsPreview: (BOOL) inIsPreview forView: (DownAndOutView *) aView
{
#if USE_DAPREFERENCES
    DAPreferences *         defaults = sPreferences;
#else
    ScreenSaverDefaults *   defaults = [ScreenSaverDefaults defaultsForModuleWithName: MODULE_NAME];
#endif
    
    view = aView;
    
    if (!inIsPreview && !sMainController)
        sMainController = self;

    secondsBeforeLogout = [defaults integerForKey: @"secondsBeforeLogout"];
    shouldMove = [defaults boolForKey: @"shouldMove"];
    
    if ([defaults boolForKey: @"usesImage"]) {
        NSData *    imageData = [defaults dataForKey: @"addedImage"];
        
        addedImage = (NSImage *) [NSUnarchiver unarchiveObjectWithData: imageData];
        [addedImage retain];
    }
    else
        addedImage = nil;
    
    messageText = [[defaults stringForKey: @"messageText"] retain];
    
    [view setClockRadius: [defaults integerForKey: @"clockRadius"]];  //  This is important to initializing the view.
        
    return self;
}

- (void) dispose
{
    [messageText release];
    [addedImage release];
}

- (NSPanel *) panel
{
    return panel;
}

- (void) colorMessageField
{
    NSTextStorage *     storage = [messageField textStorage];
    NSString *          content = [storage string];
    NSRange             contentRange = NSMakeRange(0, [content length]);
    
    [storage removeAttribute: NSForegroundColorAttributeName
                       range: contentRange];
    
    NSEnumerator *      iter = [sTags objectEnumerator];
    NSString *          curr;
    
    while (curr = [iter nextObject]) {
        NSRange         found = { 0, 0 };
        do {
            found = [content rangeOfString: curr
                                   options: NSLiteralSearch 
                                     range: NSMakeRange(NSMaxRange(found),
                                        contentRange.length - NSMaxRange(found))];
            if (found.location != NSNotFound) {
                [storage addAttribute: NSForegroundColorAttributeName
                                value: [NSColor blueColor] 
                                range: found];
            }
        } while (NSLocationInRange(found.location, contentRange));
    }
}

- (void) awakeFromNib
{
#if USE_DAPREFERENCES
    DAPreferences *         defaults = sPreferences;
#else
    ScreenSaverDefaults *   defaults = [ScreenSaverDefaults defaultsForModuleWithName: MODULE_NAME];
#endif

    [self setSecondsBeforeLogout: secondsBeforeLogout];
    if (shouldMove)
        [movementMatrix selectCellAtRow: 0 column: 0];
    else
        [movementMatrix selectCellAtRow: 1 column: 0];
    [imageWell setImage: addedImage];
    [imageClear setEnabled: (addedImage != nil)];
        
    [[messageField textStorage]
        replaceCharactersInRange: NSMakeRange(0, [[messageField textStorage] length])
        withString: messageText];
    [self colorMessageField];

    [sizeSlider setFloatValue: [defaults integerForKey: @"clockRadius"]];
}

- (void) textDidChange: (NSNotification *) notice
{
    if ([notice object] == messageField)
        [self colorMessageField];
}

- (IBAction) moveSelected: (id) sender
{
    
}

- (BOOL) shouldMove { return shouldMove; }

- (IBAction) endButton: (id) sender
{
    if (sender == okButton) {
        secondsBeforeLogout = [timeStepper intValue];
#if USE_DAPREFERENCES
        DAPreferences *         defaults = sPreferences;
#else
        ScreenSaverDefaults *   defaults = [ScreenSaverDefaults defaultsForModuleWithName: MODULE_NAME];
#endif
        [defaults setInteger: secondsBeforeLogout forKey: @"secondsBeforeLogout"];
        
        int                     row = [movementMatrix selectedRow];
        shouldMove = (row == 0);
        [defaults setBool: shouldMove forKey: @"shouldMove"];
        
        [self setAddedImage: [imageWell image]];
        if (addedImage) {
            [defaults setBool: YES forKey: @"usesImage"];
            NSData *    imageData = [NSArchiver archivedDataWithRootObject: addedImage];
            [defaults setObject: imageData forKey: @"addedImage"];              
        }
        else {
            [defaults setBool: NO forKey: @"usesImage"];
            [defaults removeObjectForKey: @"addedImage"];
        }
        
        messageText = [[[messageField textStorage] string] copy];
        [defaults setObject: messageText forKey: @"messageText"];

        float       value = [sizeSlider floatValue];
        [view setClockRadius: value];
        [defaults setInteger: (int) value forKey: @"clockRadius"];

#if USE_DAPREFERENCES
        [defaults flushToFile];
#endif
    }
    
    [[NSApplication sharedApplication] endSheet: panel];
}

- (IBAction) sliderAction: (id) sender
{
}

- (NSString *) messageText { return messageText; }

- (NSString *) messageTextForSeconds: (int) seconds
{
    NSMutableString *   retval = [NSMutableString stringWithString: messageText];
    
    NSString *          userID = (NSString *) CSCopyUserName(YES);
    [retval replaceOccurrencesOfString: @"%u" withString: userID
            options: NSLiteralSearch range: NSMakeRange(0, [retval length])];
    
    NSString *          userName = (NSString *) CSCopyUserName(NO);
    if (userName && [userName length]) {
        [retval replaceOccurrencesOfString: @"%n" withString: userName
            options: NSLiteralSearch range: NSMakeRange(0, [retval length])];
    }
    else {
        [retval replaceOccurrencesOfString: @"%n" withString: userID
            options: NSLiteralSearch range: NSMakeRange(0, [retval length])];
    }
    [userName release];
    [userID release];
    
    NSString *          secondsInt = [NSString stringWithFormat: @"%d", seconds];
    [retval replaceOccurrencesOfString: @"%s" withString: secondsInt
            options: NSLiteralSearch range: NSMakeRange(0, [retval length])];

    NSString *          secondsMss = [NSString stringWithFormat: @"%d:%02d", seconds/60, seconds%60];
    [retval replaceOccurrencesOfString: @"%t" withString: secondsMss
            options: NSLiteralSearch range: NSMakeRange(0, [retval length])];
    
    return retval;
}

- (void) setMessageText: (NSString *) newText
{
    if (newText != messageText) {
        [messageText release];
        messageText = [newText retain];

        [[messageField textStorage]
                replaceCharactersInRange: NSMakeRange(0, [[messageField textStorage] length])
                              withString: newText];
        [self colorMessageField];
    }
}

- (void) setTimeTextMinutes: (int) mins seconds: (int) secs
{
    NSString *      str = [NSString stringWithFormat: @"%d:%02d", mins, secs];
    [timeField setStringValue: str];
}

- (void) setSecondsBeforeLogout: (int) seconds
{
    if (seconds < 10)
        seconds = 10;
    
    [self setTimeTextMinutes: (seconds / 60) seconds: (seconds % 60)];
    [timeStepper setIntValue: seconds];
}

- (int) secondsBeforeLogout
{
    return secondsBeforeLogout;
}

- (IBAction) stepper: (id) sender
{
    int     value = [sender intValue];
    [self setSecondsBeforeLogout: value];
}

- (IBAction) timeText: (id) sender
{
    NSString *      str = [sender stringValue];
    int             minutes = 0;
    int             seconds = 0;
    
    int             i, iLimit = [str length];
    int             state = 0;
    
    for (i = 0; i < iLimit && state < 9; i++) {
        unichar         ch = [str characterAtIndex: i];
        switch (state) {
            case 0:
                if (ch >= '0' && ch <= '9') {
                    seconds *= 10;
                    seconds += (ch - '0');
                }
                else if (ch == ':' || ch == ' ' || ch == '.') {
                    minutes = 60 * seconds;
                    seconds = 0;
                    state = 1;
                }
                else {
                    state = 9;
                }
                break;
            case 1:
                if (ch >= '0' && ch <= '9') {
                    seconds *= 10;
                    seconds += (ch - '0');
                }
                else {
                    state = 9;
                    return;
                }
                break;
        }
    }
    
    [self setSecondsBeforeLogout: seconds + minutes];
}

- (NSMutableArray *) closableUIApplications
{
    NSArray *           exempt = [NSArray arrayWithObjects: @"Finder", nil];
    NSMutableArray *    retval = [DAApplication launchedApplications];
    int                 i;
    
    //  Remove exempt applications from the list of running UI apps
    for (i = [retval count] - 1; i >= 0; i--) {
        DAApplication *     curr = [retval objectAtIndex: i];
        if ([exempt containsObject: [curr name]]) {
            [retval removeObject: curr];
        }
    }
    
    return retval;
}

- (void) quitApplications
{
    if (self != sMainController)
        return;
    
    NSMutableArray *    apps = [self closableUIApplications];
    [apps makeObjectsPerformSelector: @selector(quitWithoutSaving)];
}

- (void) killApplications
{
    if (self != sMainController)
        return;
    
    NSMutableArray *    apps = [self closableUIApplications];
    [apps makeObjectsPerformSelector: @selector(killObj:) withObject: [NSNumber numberWithInt: 9]];
}

- (NSString *) nameOfThisApplication
{
    return [[[NSWorkspace sharedWorkspace] activeApplication] objectForKey: @"NSApplicationName"];
}

- (void) logOut
{
    if (self != sMainController)
        return;
    
    NSAppleScript *     script = [[NSAppleScript alloc] initWithSource:
        @"ignoring application responses\r"
        @"tell application \"loginwindow\" to Çevent aevtrlgoÈ\r"
        @"end ignoring"];
    //  logo is the event that asks permission to log out;
    //  rlgo doesn't ask.
    NSAppleEventDescriptor *    result;
    NSDictionary *              errorDict;
    
    result = [script executeAndReturnError: &errorDict];
    if (! result)
        NSLog(@"ErrorDict = %@", errorDict);    
}

- (void) makeTouchFile
{
    NSString *          userID = (NSString *) CSCopyUserName(YES);
    NSString *          touchPath = [NSString stringWithFormat: @"/tmp/%@.DownAndOut", userID];
    NSMutableString *   content = [NSMutableString string];
    NSEnumerator *      iter = [[DAApplication quitApplicationNames] objectEnumerator];
    NSString *          curr;
    
    while (curr = [iter nextObject]) {
        if (![[DAApplication killedApplicationNames] containsObject: curr])
            [content appendFormat: @"Quit:\t%@\n", curr];
    }
    
    iter = [[DAApplication killedApplicationNames] objectEnumerator];
    while (curr = [iter nextObject])
        [content appendFormat: @"Killed:\t%@\n", curr];
    
    [content writeToFile: touchPath atomically: NO];
}

- (IBAction) restoreDefaults: (id) sender
{
    [self setSecondsBeforeLogout: DEFAULT_LOGOUT];
    if (DEFAULT_SHOULD_MOVE)
        [movementMatrix selectCellAtRow: 0 column: 0];
    else
        [movementMatrix selectCellAtRow: 1 column: 0];
    [imageWell setImage: nil];
    [imageClear setEnabled: NO];
    
    [[messageField textStorage]
        replaceCharactersInRange: NSMakeRange(0, [[messageField textStorage] length])
                      withString: DEFAULT_MESSAGE];
    [self colorMessageField];
}

- (NSImage *) addedImage { return addedImage; }

- (void) setAddedImage: (NSImage *) anImage
{
    if (anImage != addedImage) {
        [addedImage release];
        addedImage = [anImage retain];
        
        [imageWell setImage: anImage];
        [imageClear setEnabled: (addedImage != nil)];
    }
}

- (IBAction) imageAction: (id) sender
{
    if (sender == imageClear) {
        [imageWell setImage: nil];
    }
    else if (sender == imageSet) {
        NSOpenPanel *   openPanel = [NSOpenPanel openPanel];
        
        [openPanel setCanChooseDirectories: NO];
        [openPanel setAllowsMultipleSelection: NO];
        
        int             answer = [openPanel runModalForTypes: [NSImage imageFileTypes]];
        
        if (answer == NSOKButton) {
            NSImage *   newImage = [[NSImage alloc] initWithContentsOfFile: [openPanel filename]];
            if (newImage)
                [imageWell setImage: newImage];
            [newImage release];
        }
    }
    else if (sender == imageWell) {
        //  [self setAddedImage: [imageWell image]];
    }
    
    [imageClear setEnabled: ([imageWell image] != nil)];
}

@end

---