/*

 * Copyright (c) 2007-2009 Dave Dribin

 * 

 * 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.

 */

/*

 *  This class is based on CInvocationGrabber:

 *

 *  Copyright (c) 2007, Toxic Software

 *  All rights reserved.

 *  Redistribution and use in source and binary forms, with or without

 *  modification, are permitted provided that the following conditions are

 *  met:

 *  

 *  * Redistributions of source code must retain the above copyright notice,

 *  this list of conditions and the following disclaimer.

 *  

 *  * 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.

 *  

 *  * Neither the name of the Toxic Software nor the names of its

 *  contributors may be used to endorse or promote products derived from

 *  this software without specific prior written permission.

 *  

 *  THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``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 REGENTS 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 "DDInvocationGrabber.h"

@implementation DDInvocationGrabber

+ (id)invocationGrabber

{

    return([[[self alloc] init] autorelease]);

}

- (id)init

{

    _target = nil;

    _invocation = nil;

    _forwardInvokesOnMainThread = NO;

    _invocationThread = nil;

    _waitUntilDone = NO;

    return self;

}

- (void)dealloc

{

    [self setTarget:NULL];

    [self setInvocation:NULL];

    //

    [super dealloc];

}

#pragma mark -

- (id)target

{

    return _target;

}

- (void)setTarget:(id)inTarget

{

    if (_target != inTarget)

	{

        [_target autorelease];

        _target = [inTarget retain];

	}

}

- (NSInvocation *)invocation

{

    return _invocation;

}

- (void)setInvocation:(NSInvocation *)inInvocation

{

    if (_invocation != inInvocation)

	{

        [_invocation autorelease];

        _invocation = [inInvocation retain];

	}

}

- (BOOL)forwardInvokesOnMainThread;

{

    return _forwardInvokesOnMainThread;

}

- (void)setForwardInvokesOnMainThread:(BOOL)forwardInvokesOnMainThread;

{

    _forwardInvokesOnMainThread = forwardInvokesOnMainThread;

}

#if (MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_5)

- (void)setInvokesOnThread:(NSThread *)thread;

{

    _invocationThread = thread;

}

#endif

- (BOOL)waitUntilDone;

{

    return _waitUntilDone;

}

- (void)setWaitUntilDone:(BOOL)waitUntilDone;

{

    _waitUntilDone = waitUntilDone;

}

#pragma mark -

- (NSMethodSignature *)methodSignatureForSelector:(SEL)selector

{

    return [[self target] methodSignatureForSelector:selector];

}

- (void)forwardInvocation:(NSInvocation *)ioInvocation

{

    [ioInvocation setTarget:[self target]];

    [self setInvocation:ioInvocation];

    BOOL invokeOnOtherThread = _forwardInvokesOnMainThread || (_invocationThread != nil);

    if (invokeOnOtherThread && !_waitUntilDone)

    {

        [_invocation retainArguments];

    }

    if (_forwardInvokesOnMainThread)

    {

        [_invocation performSelectorOnMainThread:@selector(invoke)

                                      withObject:nil

                                   waitUntilDone:_waitUntilDone];

    }

#if (MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_5)

    if (_invocationThread != nil)

    {

        [_invocation performSelector:@selector(invoke)

                            onThread:_invocationThread

                          withObject:nil

                       waitUntilDone:_waitUntilDone];

    }

#endif

}

@end

#pragma mark -

@implementation DDInvocationGrabber (DDnvocationGrabber_Conveniences)

- (id)prepareWithInvocationTarget:(id)inTarget

{

    [self setTarget:inTarget];

    return(self);

}

@end