-
-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathExtensions.cs
494 lines (433 loc) · 14.5 KB
/
Extensions.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
using Bloodcraft.Services;
using Bloodcraft.Utilities;
using Il2CppInterop.Runtime;
using ProjectM;
using ProjectM.Gameplay.Systems;
using ProjectM.Network;
using ProjectM.Scripting;
using ProjectM.Shared;
using Stunlock.Core;
using System.Runtime.InteropServices;
using Unity.Entities;
using Unity.Mathematics;
using Unity.Transforms;
using static Bloodcraft.Services.LocalizationService;
using static Bloodcraft.Services.PlayerService;
namespace Bloodcraft;
internal static class Extensions // probably need to organize this soon
{
static EntityManager EntityManager => Core.EntityManager;
static ServerGameManager ServerGameManager => Core.ServerGameManager;
static SystemService SystemService => Core.SystemService;
static PrefabCollectionSystem PrefabCollectionSystem => SystemService.PrefabCollectionSystem;
const string EMPTY_KEY = "LocalizationKey.Empty";
public delegate void WithRefHandler<T>(ref T item);
public static void With<T>(this Entity entity, WithRefHandler<T> action) where T : struct
{
T item = entity.ReadRW<T>();
action(ref item);
EntityManager.SetComponentData(entity, item);
}
public static void AddWith<T>(this Entity entity, WithRefHandler<T> action) where T : struct
{
if (!entity.Has<T>())
{
entity.Add<T>();
}
entity.With(action);
}
public unsafe static void Write<T>(this Entity entity, T componentData) where T : struct
{
ComponentType componentType = new(Il2CppType.Of<T>());
TypeIndex typeIndex = componentType.TypeIndex;
byte[] byteArray = StructureToByteArray(componentData);
int size = Marshal.SizeOf<T>();
fixed (byte* byteData = byteArray)
{
EntityManager.SetComponentDataRaw(entity, typeIndex, byteData, size);
}
}
static byte[] StructureToByteArray<T>(T structure) where T : struct
{
int size = Marshal.SizeOf(structure);
byte[] byteArray = new byte[size];
IntPtr ptr = Marshal.AllocHGlobal(size);
Marshal.StructureToPtr(structure, ptr, true);
Marshal.Copy(ptr, byteArray, 0, size);
Marshal.FreeHGlobal(ptr);
return byteArray;
}
unsafe static T ReadRW<T>(this Entity entity) where T : struct
{
ComponentType componentType = new(Il2CppType.Of<T>());
TypeIndex typeIndex = componentType.TypeIndex;
void* componentData = EntityManager.GetComponentDataRawRW(entity, typeIndex);
return Marshal.PtrToStructure<T>(new IntPtr(componentData));
}
public unsafe static T ReadRO<T>(this Entity entity) where T : struct
{
ComponentType componentType = new(Il2CppType.Of<T>());
TypeIndex typeIndex = componentType.TypeIndex;
void* componentData = EntityManager.GetComponentDataRawRO(entity, typeIndex);
return Marshal.PtrToStructure<T>(new IntPtr(componentData));
}
public static DynamicBuffer<T> ReadBuffer<T>(this Entity entity) where T : struct
{
return EntityManager.GetBuffer<T>(entity);
}
public static DynamicBuffer<T> AddBuffer<T>(this Entity entity) where T : struct
{
return EntityManager.AddBuffer<T>(entity);
}
public unsafe static void* GetComponentData(this Entity entity, TypeIndex typeIndex)
{
return EntityManager.GetComponentDataRawRO(entity, typeIndex);
}
public static bool TryGetComponent<T>(this Entity entity, out T componentData) where T : struct
{
componentData = default;
if (entity.Has<T>())
{
componentData = entity.ReadRO<T>();
return true;
}
return false;
}
public static bool TryRemoveComponent<T>(this Entity entity) where T : struct
{
if (entity.Has<T>())
{
entity.Remove<T>();
return true;
}
return false;
}
public static bool Has<T>(this Entity entity)
{
return EntityManager.HasComponent(entity, new(Il2CppType.Of<T>()));
}
public static bool Has(this Entity entity, ComponentType componentType)
{
return EntityManager.HasComponent(entity, componentType);
}
public static string GetPrefabName(this PrefabGUID prefabGUID)
{
return PrefabCollectionSystem.PrefabGuidToNameDictionary.TryGetValue(prefabGUID, out string prefabName) ? $"{prefabName} {prefabGUID}" : "String.Empty";
}
public static string GetLocalizedName(this PrefabGUID prefabGUID)
{
string localizedName = GetNameFromGuidString(GetGuidString(prefabGUID));
if (!string.IsNullOrEmpty(localizedName))
{
return localizedName;
}
return EMPTY_KEY;
}
public static void Add<T>(this Entity entity)
{
EntityManager.AddComponent(entity, new(Il2CppType.Of<T>()));
}
public static void Add(this Entity entity, ComponentType componentType)
{
EntityManager.AddComponent(entity, componentType);
}
public static void Remove<T>(this Entity entity)
{
EntityManager.RemoveComponent(entity, new(Il2CppType.Of<T>()));
}
public static bool TryGetFollowedPlayer(this Entity entity, out Entity player)
{
player = Entity.Null;
if (entity.TryGetComponent(out Follower follower))
{
if (follower.Followed._Value.TryGetPlayer(out player))
{
return true;
}
}
return false;
}
public static bool TryGetPlayer(this Entity entity, out Entity player)
{
player = Entity.Null;
if (entity.Has<PlayerCharacter>())
{
player = entity;
return true;
}
return false;
}
public static bool IsPlayer(this Entity entity)
{
if (entity.Has<VampireTag>())
{
return true;
}
return false;
}
public static bool IsDifferentPlayer(this Entity entity, Entity target)
{
if (entity.IsPlayer() && target.IsPlayer() && !entity.Equals(target))
{
return true;
}
return false;
}
public static bool IsFollowingPlayer(this Entity entity)
{
if (entity.TryGetComponent(out Follower follower))
{
if (follower.Followed._Value.IsPlayer())
{
return true;
}
}
return false;
}
public static Entity GetBuffTarget(this Entity entity)
{
return CreateGameplayEventServerUtility.GetBuffTarget(EntityManager, entity);
}
public static Entity GetPrefabEntity(this Entity entity)
{
return ServerGameManager.GetPrefabEntity(entity.ReadRO<PrefabGUID>());
}
public static Entity GetSpellTarget(this Entity entity)
{
return CreateGameplayEventServerUtility.GetSpellTarget(EntityManager, entity);
}
public static bool TryGetTeamEntity(this Entity entity, out Entity teamEntity)
{
teamEntity = Entity.Null;
if (entity.TryGetComponent(out TeamReference teamReference))
{
Entity teamReferenceEntity = teamReference.Value._Value;
if (teamReferenceEntity.Exists())
{
teamEntity = teamReferenceEntity;
return true;
}
}
return false;
}
public static bool Exists(this Entity entity)
{
return EntityManager.Exists(entity);
}
public static bool IsDisabled(this Entity entity)
{
return entity.Has<Disabled>();
}
public static bool IsVBlood(this Entity entity)
{
return entity.Has<VBloodConsumeSource>();
}
public static bool IsGateBoss(this Entity entity)
{
return entity.Has<VBloodUnit>() && !entity.Has<VBloodConsumeSource>();
}
public static bool IsVBloodOrGateBoss(this Entity entity)
{
return entity.Has<VBloodUnit>();
}
public static ulong GetSteamId(this Entity entity)
{
if (entity.TryGetComponent(out PlayerCharacter playerCharacter))
{
return playerCharacter.UserEntity.ReadRO<User>().PlatformId;
}
else if (entity.TryGetComponent(out User user))
{
return user.PlatformId;
}
return 0;
}
public static NetworkId GetNetworkId(this Entity entity)
{
if (entity.TryGetComponent(out NetworkId networkId))
{
return networkId;
}
return NetworkId.Empty;
}
public static void ForEach<T>(this IEnumerable<T> source, Action<T> action)
{
foreach (var item in source)
{
action(item);
}
}
public static bool TryGetPlayerInfo(this ulong steamId, out PlayerInfo playerInfo)
{
if (PlayerCache.TryGetValue(steamId, out playerInfo)) return true;
else if (OnlineCache.TryGetValue(steamId, out playerInfo)) return true;
return false;
}
public static PrefabGUID GetPrefabGuid(this Entity entity)
{
if (entity.TryGetComponent(out PrefabGUID prefabGUID)) return prefabGUID;
return PrefabGUID.Empty;
}
public static Entity GetUserEntity(this Entity character)
{
if (character.TryGetComponent(out PlayerCharacter playerCharacter)) return playerCharacter.UserEntity;
return Entity.Null;
}
public static User GetUser(this Entity entity)
{
if (entity.TryGetComponent(out PlayerCharacter playerCharacter) && playerCharacter.UserEntity.TryGetComponent(out User user)) return user;
else if (entity.TryGetComponent(out user)) return user;
return User.Empty;
}
public static bool HasBuff(this Entity entity, PrefabGUID buffPrefabGUID)
{
return ServerGameManager.HasBuff(entity, buffPrefabGUID.ToIdentifier());
}
public static bool TryGetBuff(this Entity entity, PrefabGUID buffPrefabGUID, out Entity buffEntity)
{
if (ServerGameManager.TryGetBuff(entity, buffPrefabGUID.ToIdentifier(), out buffEntity))
{
return true;
}
return false;
}
public static bool TryApplyBuff(this Entity entity, PrefabGUID buffPrefabGUID)
{
if (Buffs.TryApplyBuff(entity, buffPrefabGUID))
{
return true;
}
return false;
}
public static bool TryRemoveBuff(this Entity entity, PrefabGUID buffPrefabGuid)
{
if (entity.TryGetBuff(buffPrefabGuid, out Entity buffEntity))
{
buffEntity.Destroy();
return true;
}
return false;
}
public static bool TryApplyAndGetBuff(this Entity entity, PrefabGUID buffPrefabGUID, out Entity buffEntity)
{
buffEntity = Entity.Null;
if (Buffs.TryApplyBuff(entity, buffPrefabGUID) && entity.TryGetBuff(buffPrefabGUID, out buffEntity))
{
return true;
}
return false;
}
public static bool TryApplyBuffWithOwner(Entity target, Entity owner, PrefabGUID buffPrefabGuid)
{
if (target.TryApplyAndGetBuff(buffPrefabGuid, out Entity buffEntity) && buffEntity.Has<EntityOwner>())
{
buffEntity.With((ref EntityOwner entityOwner) =>
{
entityOwner.Owner = owner;
});
return true;
}
return false;
}
public static unsafe bool TryGetBuffer<T>(this Entity entity, out DynamicBuffer<T> dynamicBuffer) where T : struct
{
if (ServerGameManager.TryGetBuffer(entity, out dynamicBuffer))
{
return true;
}
return false;
}
public static float3 GetAimPosition(this Entity entity)
{
if (entity.TryGetComponent(out EntityInput entityInput))
{
return entityInput.AimPosition;
}
return float3.zero;
}
public static bool TryGetPosition(this Entity entity, out float3 position)
{
position = float3.zero;
if (entity.TryGetComponent(out Translation translation))
{
position = translation.Value;
return true;
}
return false;
}
public static float3 GetPosition(this Entity entity)
{
if (entity.TryGetComponent(out Translation translation))
{
return translation.Value;
}
return float3.zero;
}
public static bool TryGetMatch(this HashSet<(ulong, ulong)> hashSet, ulong value, out (ulong, ulong) matchingPair)
{
matchingPair = default;
foreach (var pair in hashSet)
{
if (pair.Item1 == value || pair.Item2 == value)
{
matchingPair = pair;
return true;
}
}
return false;
}
public static bool TryGetMatchPairInfo(this (ulong, ulong) matchPair, out (PlayerInfo, PlayerInfo) matchPairInfo)
{
matchPairInfo = default;
ulong playerOne = matchPair.Item1;
ulong playerTwo = matchPair.Item2;
if (playerOne.TryGetPlayerInfo(out PlayerInfo playerOneInfo) && playerTwo.TryGetPlayerInfo(out PlayerInfo playerTwoInfo))
{
matchPairInfo = (playerOneInfo, playerTwoInfo);
return true;
}
return false;
}
public static bool IsCustomSpawned(this Entity entity)
{
if (entity.TryGetComponent(out IsMinion isMinion) && isMinion.Value)
{
return true;
}
return false;
}
public static void Destroy(this Entity entity)
{
if (entity.Exists()) DestroyUtility.Destroy(EntityManager, entity);
}
public static void SetTeam(this Entity entity, Entity teamSource)
{
if (entity.Has<Team>() && entity.Has<TeamReference>() && teamSource.TryGetComponent(out Team sourceTeam) && teamSource.TryGetComponent(out TeamReference sourceTeamReference))
{
Entity teamRefEntity = sourceTeamReference.Value._Value;
int teamId = sourceTeam.Value;
entity.With((ref TeamReference teamReference) =>
{
teamReference.Value._Value = teamRefEntity;
});
entity.With((ref Team team) =>
{
team.Value = teamId;
});
}
}
public static void SetFaction(this Entity entity, PrefabGUID factionPrefabGUID)
{
if (entity.Has<FactionReference>())
{
entity.With((ref FactionReference factionReference) =>
{
factionReference.FactionGuid._Value = factionPrefabGUID;
});
}
}
public static bool IsAllies(this Entity entity, Entity player)
{
return ServerGameManager.IsAllies(entity, player);
}
}