quantum-space-buddies/QSB/Utility/Extensions.cs

using Cysharp.Threading.Tasks;
using Mirror;
using OWML.Common;
using QSB.Player;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Runtime.Serialization.Formatters.Binary;
using System.Text;
using UnityEngine;
using Object = UnityEngine.Object;

namespace QSB.Utility;

public static class Extensions
{
	#region UNITY

	public static Quaternion TransformRotation(this Transform transform, Quaternion localRotation)
		=> transform.rotation * localRotation;

	public static GameObject InstantiateInactive(this GameObject original)
	{
		if (!original.activeSelf)
		{
			return Object.Instantiate(original);
		}

		original.SetActive(false);
		var copy = Object.Instantiate(original);
		original.SetActive(true);
		return copy;
	}

	public static GameObject FindChild(this GameObject g, string childPath) =>
			g.transform.Find(childPath)?.gameObject;

	#endregion

	#region MIRROR

	public static uint GetPlayerId(this NetworkConnectionToClient conn)
	{
		if (conn.identity == null)
		{
			// wtf
			DebugLog.ToConsole($"Error - GetPlayerId on {conn} has no identity.", MessageType.Error);
			return uint.MaxValue;
		}

		return conn.identity.netId;
	}

	public static NetworkConnectionToClient GetNetworkConnection(this uint playerId)
	{
		var conn = NetworkServer.connections.Values.FirstOrDefault(x => playerId == x.GetPlayerId());
		if (conn == default)
		{
			DebugLog.ToConsole($"Error - GetNetworkConnection on {playerId} found no connection\n{Environment.StackTrace}", MessageType.Error);
		}

		return conn;
	}

	public static void SpawnWithServerOwnership(this GameObject go) =>
		NetworkServer.Spawn(go, NetworkServer.localConnection);

	#endregion

	#region C#

	public static float Map(this float value, float inputFrom, float inputTo, float outputFrom, float outputTo, bool clamp)
	{
		var mappedValue = (value - inputFrom) / (inputTo - inputFrom) * (outputTo - outputFrom) + outputFrom;

		return clamp
			? Mathf.Clamp(mappedValue, outputTo, outputFrom)
			: mappedValue;
	}

	public static void ForEach<T>(this IEnumerable<T> enumerable, Action<T> action)
	{
		foreach (var item in enumerable)
		{
			action(item);
		}
	}

	public static TSource MinBy<TSource, TKey>(this IEnumerable<TSource> source, Func<TSource, TKey> keySelector)
	{
		var comparer = Comparer<TKey>.Default;
		var yk = default(TKey);
		var y = default(TSource);
		var hasValue = false;
		foreach (var x in source)
		{
			var xk = keySelector(x);
			if (!hasValue)
			{
				hasValue = true;
				yk = xk;
				y = x;
			}
			else if (comparer.Compare(xk, yk) < 0)
			{
				yk = xk;
				y = x;
			}
		}

		if (!hasValue)
		{
			throw new InvalidOperationException("Sequence contains no elements");
		}

		return y;
	}

	public static TSource MaxBy<TSource, TKey>(this IEnumerable<TSource> source, Func<TSource, TKey> keySelector)
	{
		var comparer = Comparer<TKey>.Default;
		var yk = default(TKey);
		var y = default(TSource);
		var hasValue = false;
		foreach (var x in source)
		{
			var xk = keySelector(x);
			if (!hasValue)
			{
				hasValue = true;
				yk = xk;
				y = x;
			}
			else if (comparer.Compare(xk, yk) > 0)
			{
				yk = xk;
				y = x;
			}
		}

		if (!hasValue)
		{
			throw new InvalidOperationException("Sequence contains no elements");
		}

		return y;
	}

	public static bool IsInRange<T>(this IList<T> list, int index) => index >= 0 && index < list.Count;

	public static IEnumerable<Type> GetDerivedTypes(this Type type)
	{
		var assemblies = QSBCore.Addons.Values
			.Select(x => x.GetType().Assembly)
			.Append(type.Assembly);

		if (QSBCore.QSBNHAssembly != null)
		{
			assemblies = assemblies.Append(QSBCore.QSBNHAssembly);
		}

		return assemblies.SelectMany(x => x.GetTypes())
			.Where(x => !x.IsInterface && !x.IsAbstract && type.IsAssignableFrom(x))
			.OrderBy(x => x.FullName);
	}

	public static Guid ToGuid(this int value)
	{
		var bytes = new byte[16];
		BitConverter.GetBytes(value).CopyTo(bytes, 0);
		return new Guid(bytes);
	}

	public static void Try(this object self, string doingWhat, Action action)
	{
		try
		{
			action();
		}
		catch (Exception e)
		{
			DebugLog.ToConsole($"{self} - error {doingWhat} : {e}", MessageType.Error);
		}
	}

	public static async UniTask Try(this object self, string doingWhat, Func<UniTask> func)
	{
		try
		{
			await func().SuppressCancellationThrow();
		}
		catch (Exception e)
		{
			DebugLog.ToConsole($"{self} - error {doingWhat} : {e}", MessageType.Error);
		}
	}

	// https://stackoverflow.com/a/24031467
	public static string GetMD5Hash(this IEnumerable<string> list)
	{
		using var md5 = System.Security.Cryptography.MD5.Create();

		var longString = string.Concat(list);
		var bytes = Encoding.ASCII.GetBytes(longString);
		var hashBytes = md5.ComputeHash(bytes);

		var sb = new StringBuilder();
		for (var i = 0; i < hashBytes.Length; i++)
		{
			sb.Append(hashBytes[i].ToString("X2"));
		}

		return sb.ToString();
	}

	public static string GetMD5Hash(this string input)
	{
		using var md5 = System.Security.Cryptography.MD5.Create();

		var bytes = Encoding.ASCII.GetBytes(input);
		var hashBytes = md5.ComputeHash(bytes);

		var sb = new StringBuilder();
		for (var i = 0; i < hashBytes.Length; i++)
		{
			sb.Append(hashBytes[i].ToString("X2"));
		}

		return sb.ToString();
	}

	/// <summary>
	/// only works for c# serializable objects
	/// </summary>
	public static byte[] ToBytes(this object obj)
	{
		using var ms = new MemoryStream();
		var bf = new BinaryFormatter();
		bf.Serialize(ms, obj);
		var bytes = ms.ToArray();
		return bytes;
	}

	/// <summary>
	/// only works for c# serializable objects
	/// </summary>
	public static object ToObject(this byte[] bytes)
	{
		using var ms = new MemoryStream(bytes);
		var bf = new BinaryFormatter();
		var obj = bf.Deserialize(ms);
		return obj;
	}

	#endregion
}