| | 1 | | using System.Collections.Generic; |
| | 2 | | using DCL.Interface; |
| | 3 | | using UnityEngine; |
| | 4 | |
|
| | 5 | | namespace DCL.Chat.HUD |
| | 6 | | { |
| | 7 | | [CreateAssetMenu(fileName = "PoolChatEntryFactory", menuName = "DCL/Social/PoolChatEntryFactory")] |
| | 8 | | public class PoolChatEntryFactory : ScriptableObject, IChatEntryFactory |
| | 9 | | { |
| | 10 | | private const string POOL_NAME_PREFIX = "ChatMessages_"; |
| | 11 | | private const int PRE_INSTANTIATED_ENTRIES = 30; |
| | 12 | |
|
| | 13 | | [SerializeField] private DefaultChatEntry defaultMessagePrefab; |
| | 14 | | [SerializeField] private DefaultChatEntry systemMessagePrefab; |
| | 15 | | [SerializeField] private DefaultChatEntry privateReceivedMessagePrefab; |
| | 16 | | [SerializeField] private DefaultChatEntry privateSentMessagePrefab; |
| | 17 | | [SerializeField] private DefaultChatEntry publicReceivedMessagePrefab; |
| | 18 | | [SerializeField] private DefaultChatEntry publicSentMessagePrefab; |
| | 19 | |
|
| 3 | 20 | | private readonly Dictionary<ChatEntry, PoolableObject> pooledObjects = |
| | 21 | | new Dictionary<ChatEntry, PoolableObject>(); |
| | 22 | |
|
| | 23 | | public ChatEntry Create(ChatEntryModel model) |
| | 24 | | { |
| 0 | 25 | | if (model.messageType == ChatMessage.Type.SYSTEM) |
| 0 | 26 | | return GetEntryFromPool("SYSTEM", systemMessagePrefab); |
| | 27 | |
|
| 0 | 28 | | if (model.messageType == ChatMessage.Type.PUBLIC) |
| | 29 | | { |
| 0 | 30 | | if (model.subType == ChatEntryModel.SubType.RECEIVED) |
| 0 | 31 | | return GetEntryFromPool("PUBLIC_RECEIVED", publicReceivedMessagePrefab); |
| 0 | 32 | | if (model.subType == ChatEntryModel.SubType.SENT) |
| 0 | 33 | | return GetEntryFromPool("PUBLIC_SENT", publicSentMessagePrefab); |
| | 34 | | } |
| 0 | 35 | | else if (model.messageType == ChatMessage.Type.PRIVATE) |
| | 36 | | { |
| 0 | 37 | | if (model.subType == ChatEntryModel.SubType.RECEIVED) |
| 0 | 38 | | return GetEntryFromPool("PRIVATE_RECEIVED", privateReceivedMessagePrefab); |
| 0 | 39 | | if (model.subType == ChatEntryModel.SubType.SENT) |
| 0 | 40 | | return GetEntryFromPool("PRIVATE_SENT", privateSentMessagePrefab); |
| | 41 | | } |
| 0 | 42 | | return GetEntryFromPool("DEFAULT", defaultMessagePrefab); |
| | 43 | | } |
| | 44 | |
|
| | 45 | | public void Destroy(ChatEntry entry) |
| | 46 | | { |
| 0 | 47 | | if (!pooledObjects.TryGetValue(entry, out var pooledObj)) return; |
| 0 | 48 | | pooledObj.Release(); |
| 0 | 49 | | pooledObjects.Remove(entry); |
| 0 | 50 | | } |
| | 51 | |
|
| | 52 | | private DefaultChatEntry GetEntryFromPool(string poolName, DefaultChatEntry prefab) |
| | 53 | | { |
| 0 | 54 | | var pool = GetPool(poolName, prefab); |
| 0 | 55 | | var pooledObj = pool.Get(); |
| 0 | 56 | | var entry = pooledObj.gameObject.GetComponent<DefaultChatEntry>(); |
| 0 | 57 | | pooledObjects[entry] = pooledObj; |
| 0 | 58 | | return entry; |
| | 59 | | } |
| | 60 | |
|
| | 61 | | private Pool GetPool(string poolName, DefaultChatEntry prefab) |
| | 62 | | { |
| 0 | 63 | | var poolId = POOL_NAME_PREFIX + poolName + GetInstanceID(); |
| 0 | 64 | | var entryPool = PoolManager.i.GetPool(poolId); |
| 0 | 65 | | if (entryPool != null) return entryPool; |
| | 66 | |
|
| 0 | 67 | | entryPool = PoolManager.i.AddPool( |
| | 68 | | poolId, |
| | 69 | | Instantiate(prefab).gameObject, |
| | 70 | | maxPrewarmCount: PRE_INSTANTIATED_ENTRIES, |
| | 71 | | isPersistent: true); |
| 0 | 72 | | entryPool.ForcePrewarm(); |
| | 73 | |
|
| 0 | 74 | | return entryPool; |
| | 75 | | } |
| | 76 | | } |
| | 77 | | } |