UnrealEngineUWP/Engine/Source/Developer/FriendsAndChat/Private/Models/ChatViewModel.cpp

// Copyright 1998-2014 Epic Games, Inc. All Rights Reserved.

#include "FriendsAndChatPrivatePCH.h"
#include "ChatItemViewModel.h"
#include "ChatViewModel.h"

class FChatViewModelImpl
	: public FChatViewModel
{
public:

	virtual TArray< TSharedRef<class FChatItemViewModel > >& GetFilteredChatList() override
	{
		return FilteredChatLists;
	}

	virtual FReply  HandleSelectionChanged(TSharedRef<FChatItemViewModel> ItemSelected) override
	{
		SetChatChannel(ItemSelected->GetMessageType(), ItemSelected->GetFriendID().ToString());
		if(SelectedChatChannel == EChatMessageType::Whisper)
		{
			RecentPlayerList.AddUnique(ItemSelected->GetFriendID().ToString());
		}

		return FReply::Handled();
	}

	virtual FText GetViewGroupText() const override
	{
		return EChatMessageType::ToText(SelectedViewChannel);
	}

	virtual FText GetChatGroupText() const override
	{
		return SelectedChatChannel == EChatMessageType::Whisper && SelectedFriend.IsValid() ? FText::FromString(SelectedFriend->GetName()) : EChatMessageType::ToText(SelectedChatChannel);
	}

	virtual void EnumerateChatChannelOptionsList(TArray<EChatMessageType::Type>& OUTChannelType) override
	{
		OUTChannelType.Add(EChatMessageType::Global);
		OUTChannelType.Add(EChatMessageType::Party);
		OUTChannelType.Add(EChatMessageType::Network);
	}

	virtual void SetChatChannel(const EChatMessageType::Type NewOption, FString InSelectedFriend) override
	{
		// To Do - set the chat channel. Disabled for now
//		SelectedFriend = FText::FromString(InSelectedFriend);
		SelectedChatChannel = NewOption;
	}

	virtual void SetViewChannel(const EChatMessageType::Type NewOption) override
	{
		SelectedViewChannel = NewOption;
		SelectedChatChannel = NewOption;
		FilterChatList();
	}

	virtual void SendMessage(const FText NewMessage) override
	{
		switch(SelectedChatChannel)
		{
			case EChatMessageType::Whisper:
			{
				if (SelectedFriend.IsValid())
				{
					MessageManager.Pin()->SendPrivateMessage(SelectedFriend->GetUniqueID().Get(), NewMessage.ToString());
					TSharedPtr< FFriendChatMessage > ChatItem = MakeShareable(new FFriendChatMessage());
					if (SelectedFriend.IsValid())
					{
						ChatItem->FromName = FText::FromString(SelectedFriend->GetName());
					}
					else
					{
						ChatItem->FromName = FText::FromString("Unknown");
					}

					ChatItem->Message = NewMessage;
					ChatItem->MessageType = SelectedChatChannel;
					ChatItem->MessageTimeText = FText::AsTime(FDateTime::Now());
					ChatItem->bIsFromSelf = true;
					ChatLists.Add(FChatItemViewModelFactory::Create(ChatItem.ToSharedRef(), SharedThis(this)));
					FilterChatList();
				}
			}
			break;
			case EChatMessageType::Global:
			{
				//@todo samz - send message to specific room (empty room name will send to all rooms)
				MessageManager.Pin()->SendRoomMessage(FString(), NewMessage.ToString());
			}
			break;
			case EChatMessageType::Network:
			{
				MessageManager.Pin()->SendNetworkMessage(NewMessage.ToString());
			}
			break;
		}
	}

	virtual void SetTimeDisplayTransparency(const float TimeTransparency)
	{
		TimeDisplayTransaprency = TimeTransparency;
	}

	virtual const float GetTimeTransparency() const
	{
		return TimeDisplayTransaprency;
	}

	virtual EChatMessageType::Type GetChatChannelType() const
	{
		return SelectedChatChannel;
	}
	
	virtual const TArray<FString>& GetRecentOptions() const override
	{
		return RecentPlayerList;
	}

	virtual void SetChatFriend(TSharedPtr<IFriendItem> ChatFriend) override
	{
		SelectedFriend = ChatFriend;
		SelectedChatChannel = EChatMessageType::Whisper;
	}

	DECLARE_DERIVED_EVENT(FChatViewModelImpl , FChatViewModel::FChatListUpdated, FChatListUpdated);
	virtual FChatListUpdated& OnChatListUpdated() override
	{
		return ChatListUpdatedEvent;
	}

private:
	void Initialize()
	{
		MessageManager.Pin()->OnChatMessageRecieved().AddSP(this, &FChatViewModelImpl::HandleMessageReceived);
		FilterChatList();
	}

	void FilterChatList()
	{
		FilteredChatLists = ChatLists;
		ChatListUpdatedEvent.Broadcast();
	}


	void HandleMessageReceived( const TSharedRef<FFriendChatMessage> NewMessage)
	{
		ChatLists.Add(FChatItemViewModelFactory::Create(NewMessage, SharedThis(this)));
		FilterChatList();
	}

	FChatViewModelImpl(const TSharedRef<FFriendsMessageManager>& MessageManager)
		: SelectedViewChannel(EChatMessageType::Global)
		, SelectedChatChannel(EChatMessageType::Global)
		, MessageManager(MessageManager)
		, TimeDisplayTransaprency(0.f)
	{
	}

private:
	TArray<TSharedRef<FChatItemViewModel> > ChatLists;
	TArray<TSharedRef<FChatItemViewModel> > FilteredChatLists;
	FChatListUpdated ChatListUpdatedEvent;
	TArray<FString> RecentPlayerList;
	TSharedPtr<IFriendItem> SelectedFriend;

	EChatMessageType::Type SelectedViewChannel;
	EChatMessageType::Type SelectedChatChannel;
	TWeakPtr<FFriendsMessageManager> MessageManager;
	float TimeDisplayTransaprency;

private:
	friend FChatViewModelFactory;
};

TSharedRef< FChatViewModel > FChatViewModelFactory::Create(
	const TSharedRef<FFriendsMessageManager>& MessageManager
	)
{
	TSharedRef< FChatViewModelImpl > ViewModel(new FChatViewModelImpl(MessageManager));
	ViewModel->Initialize();
	return ViewModel;
}