/*
 * Copyright 1999, Be Incorporated.
 * Copyright (c) 1999-2000, Eric Moon.
 * 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.
 *
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF TITLE, NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR 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.
 */


////////////////////////////////////////////////////////////
// MultiInvoker.cpp
// ----------------
// Implements the MultiInvoker class.
//

#include <Messenger.h>
#include "MultiInvoker.h"

MultiInvoker::MultiInvoker()
{
	m_message = 0;
	m_timeout = B_INFINITE_TIMEOUT;
	m_replyHandler = 0;
}

MultiInvoker::MultiInvoker(const MultiInvoker& src)
{
	Clone(src);
}

MultiInvoker::~MultiInvoker()
{
	Clear();
}

MultiInvoker& MultiInvoker::operator=(const MultiInvoker& src)
{
	if (this != &src) {
		Clear();
		Clone(src);
	}
	return *this;
}

void MultiInvoker::Clear()
{
	delete m_message;
	int32 i=CountTargets();
	while (--i >=0) {
		RemoveTarget(i);
	}
}

void MultiInvoker::Clone(const MultiInvoker& src)
{
	m_message = new BMessage(*src.Message());
	int32 len=src.CountTargets();
	for (int32 i=0; i<len; i++) {
		AddTarget(src.TargetAt(i));
	}
	m_timeout = src.Timeout();
	m_replyHandler = src.HandlerForReply();
}

void MultiInvoker::SetMessage(BMessage* message)
{
	delete m_message;
	m_message = message;
}

BMessage* MultiInvoker::Message() const
{
	return m_message;
}

uint32 MultiInvoker::Command() const
{
	return (m_message) ? m_message->what : 0;
}

status_t MultiInvoker::AddTarget(const BHandler* h, const BLooper* loop)
{
	status_t err;
	BMessenger* msgr = new BMessenger(h, loop, &err);
	if (err == B_OK)
		m_messengers.AddItem(msgr);
	else
		delete msgr;
	return err;
}

status_t MultiInvoker::AddTarget(BMessenger* msgr)
{
	if (msgr) {
		m_messengers.AddItem(msgr);
		return B_OK;
	} else {
		return B_BAD_VALUE;
	}
}

void MultiInvoker::RemoveTarget(const BHandler* h)
{
	int32 i = IndexOfTarget(h);
	if (i >= 0)
		RemoveTarget(i);
}

void MultiInvoker::RemoveTarget(int32 index)
{
	BMessenger* msgr = static_cast<BMessenger*>
		(m_messengers.RemoveItem(index));
	delete msgr;
}

int32 MultiInvoker::IndexOfTarget(const BHandler* h) const
{
	int32 len = CountTargets();
	for (int32 i=0; i<len; i++) {
		BMessenger* msgr = MessengerAt(i);
		if (msgr && msgr->Target(0) == h) {
			return i;
		}
	}
	return -1;
}

int32 MultiInvoker::CountTargets() const
{
	return m_messengers.CountItems();
}

BHandler* MultiInvoker::TargetAt(int32 index, BLooper** looper) const
{
	BMessenger* msgr = MessengerAt(index);
	if (msgr) {
		return msgr->Target(looper);
	} else {
		if (looper) *looper = 0;
		return 0;
	}
}

BMessenger* MultiInvoker::MessengerAt(int32 index) const
{
	return static_cast<BMessenger*>
		(m_messengers.ItemAt(index));
}

bool MultiInvoker::IsTargetLocal(int32 index) const
{
	BMessenger* msgr = MessengerAt(index);
	return (msgr) ? msgr->IsTargetLocal() : false;
}

void MultiInvoker::SetTimeout(bigtime_t timeout)
{
	m_timeout = timeout;
}

bigtime_t MultiInvoker::Timeout() const
{
	return m_timeout;
}

void MultiInvoker::SetHandlerForReply(BHandler* h)
{
	m_replyHandler = h;
}

BHandler* MultiInvoker::HandlerForReply() const
{
	return m_replyHandler;
}

status_t MultiInvoker::Invoke(BMessage* msg)
{
	BMessage* sendMsg = (msg) ? msg : m_message;
	if (! sendMsg)
		return B_BAD_VALUE;

	status_t err, finalResult=B_OK;
	BMessage replyMsg;
	int32 len = CountTargets();
	for (int32 i=0; i<len; i++) {
		BMessenger* msgr = MessengerAt(i);
		if (msgr) {
			err = msgr->SendMessage(sendMsg,
				HandlerForReply(), m_timeout);
			if (err != B_OK) finalResult = err;
		}
	}
	return finalResult;
}