// ServerConnectionProvider.cpp

#include "ServerConnectionProvider.h"

#include <AutoLocker.h>

#include "ExtendedServerInfo.h"
#include "ServerConnection.h"

// constructor
ServerConnectionProvider::ServerConnectionProvider(VolumeManager* volumeManager,
	ExtendedServerInfo* serverInfo,
	vnode_id connectionBrokenTarget)
	:
	BReferenceable(),
	fLock("server connection provider"),
	fVolumeManager(volumeManager),
	fServerInfo(serverInfo),
	fServerConnection(NULL),
	fConnectionBrokenTarget(connectionBrokenTarget)
{
	if (fServerInfo)
		fServerInfo->AcquireReference();
}

// destructor
ServerConnectionProvider::~ServerConnectionProvider()
{
	AutoLocker<Locker> _(fLock);
	if (fServerConnection) {
		fServerConnection->Close();
		fServerConnection->ReleaseReference();
	}

	if (fServerInfo)
		fServerInfo->ReleaseReference();
}

// Init
status_t
ServerConnectionProvider::Init()
{
	return B_OK;
}

// GetServerConnection
status_t
ServerConnectionProvider::GetServerConnection(
	ServerConnection** serverConnection)
{
	AutoLocker<Locker> _(fLock);

	// if there is no server connection yet, create one
	if (!fServerConnection) {
		fServerConnection = new(std::nothrow) ServerConnection(fVolumeManager,
			fServerInfo);
		if (!fServerConnection)
			return B_NO_MEMORY;
		status_t error = fServerConnection->Init(fConnectionBrokenTarget);
		if (error != B_OK)
			return error;
	}

	if (!fServerConnection->IsConnected())
		return B_ERROR;

	fServerConnection->AcquireReference();
	*serverConnection = fServerConnection;
	return B_OK;
}

// GetExistingServerConnection
ServerConnection*
ServerConnectionProvider::GetExistingServerConnection()
{
	AutoLocker<Locker> _(fLock);

	// if there is no server connection yet, create one
	if (!fServerConnection || !fServerConnection->IsConnected())
		return NULL;

	fServerConnection->AcquireReference();
	return fServerConnection;
}

// CloseServerConnection
void
ServerConnectionProvider::CloseServerConnection()
{
	AutoLocker<Locker> _(fLock);
	if (fServerConnection)
		fServerConnection->Close();
}