Added Ellie Common/Abstractions
This commit is contained in:
parent
e58268e339
commit
97e81ac0f4
31 changed files with 1458 additions and 0 deletions
19
src/EllieBot/_common/Abstractions/AsyncLazy.cs
Normal file
19
src/EllieBot/_common/Abstractions/AsyncLazy.cs
Normal file
|
@ -0,0 +1,19 @@
|
||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public class AsyncLazy<T> : Lazy<Task<T>>
|
||||||
|
{
|
||||||
|
public AsyncLazy(Func<T> valueFactory)
|
||||||
|
: base(() => Task.Run(valueFactory))
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public AsyncLazy(Func<Task<T>> taskFactory)
|
||||||
|
: base(() => Task.Run(taskFactory))
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public TaskAwaiter<T> GetAwaiter()
|
||||||
|
=> Value.GetAwaiter();
|
||||||
|
}
|
|
@ -0,0 +1,46 @@
|
||||||
|
using OneOf;
|
||||||
|
using OneOf.Types;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class BotCacheExtensions
|
||||||
|
{
|
||||||
|
public static async ValueTask<T?> GetOrDefaultAsync<T>(this IBotCache cache, TypedKey<T> key)
|
||||||
|
{
|
||||||
|
var result = await cache.GetAsync(key);
|
||||||
|
if (result.TryGetValue(out var val))
|
||||||
|
return val;
|
||||||
|
|
||||||
|
return default;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static TypedKey<byte[]> GetImgKey(Uri uri)
|
||||||
|
=> new($"image:{uri}");
|
||||||
|
|
||||||
|
public static ValueTask SetImageDataAsync(this IBotCache c, string key, byte[] data)
|
||||||
|
=> c.SetImageDataAsync(new Uri(key), data);
|
||||||
|
public static async ValueTask SetImageDataAsync(this IBotCache c, Uri key, byte[] data)
|
||||||
|
=> await c.AddAsync(GetImgKey(key), data, expiry: TimeSpan.FromHours(48));
|
||||||
|
|
||||||
|
public static async ValueTask<OneOf<byte[], None>> GetImageDataAsync(this IBotCache c, Uri key)
|
||||||
|
=> await c.GetAsync(GetImgKey(key));
|
||||||
|
|
||||||
|
public static async Task<TimeSpan?> GetRatelimitAsync(
|
||||||
|
this IBotCache c,
|
||||||
|
TypedKey<long> key,
|
||||||
|
TimeSpan length)
|
||||||
|
{
|
||||||
|
var now = DateTime.UtcNow;
|
||||||
|
var nowB = now.ToBinary();
|
||||||
|
|
||||||
|
var cachedValue = await c.GetOrAddAsync(key,
|
||||||
|
() => Task.FromResult(now.ToBinary()),
|
||||||
|
expiry: length);
|
||||||
|
|
||||||
|
if (cachedValue == nowB)
|
||||||
|
return null;
|
||||||
|
|
||||||
|
var diff = now - DateTime.FromBinary(cachedValue);
|
||||||
|
return length - diff;
|
||||||
|
}
|
||||||
|
}
|
47
src/EllieBot/_common/Abstractions/Cache/IBotCache.cs
Normal file
47
src/EllieBot/_common/Abstractions/Cache/IBotCache.cs
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
using OneOf;
|
||||||
|
using OneOf.Types;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public interface IBotCache
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Adds an item to the cache
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="key">Key to add</param>
|
||||||
|
/// <param name="value">Value to add to the cache</param>
|
||||||
|
/// <param name="expiry">Optional expiry</param>
|
||||||
|
/// <param name="overwrite">Whether old value should be overwritten</param>
|
||||||
|
/// <typeparam name="T">Type of the value</typeparam>
|
||||||
|
/// <returns>Returns whether add was sucessful. Always true unless ovewrite = false</returns>
|
||||||
|
ValueTask<bool> AddAsync<T>(TypedKey<T> key, T value, TimeSpan? expiry = null, bool overwrite = true);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Get an element from the cache
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="key">Key</param>
|
||||||
|
/// <typeparam name="T">Type of the value</typeparam>
|
||||||
|
/// <returns>Either a value or <see cref="None"/></returns>
|
||||||
|
ValueTask<OneOf<T, None>> GetAsync<T>(TypedKey<T> key);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Remove a key from the cache
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="key">Key to remove</param>
|
||||||
|
/// <typeparam name="T">Type of the value</typeparam>
|
||||||
|
/// <returns>Whether there was item</returns>
|
||||||
|
ValueTask<bool> RemoveAsync<T>(TypedKey<T> key);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Get the key if it exists or add a new one
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="key">Key to get and potentially add</param>
|
||||||
|
/// <param name="createFactory">Value creation factory</param>
|
||||||
|
/// <param name="expiry">Optional expiry</param>
|
||||||
|
/// <typeparam name="T">Type of the value</typeparam>
|
||||||
|
/// <returns>The retrieved or newly added value</returns>
|
||||||
|
ValueTask<T?> GetOrAddAsync<T>(
|
||||||
|
TypedKey<T> key,
|
||||||
|
Func<Task<T?>> createFactory,
|
||||||
|
TimeSpan? expiry = null);
|
||||||
|
}
|
71
src/EllieBot/_common/Abstractions/Cache/MemoryBotCache.cs
Normal file
71
src/EllieBot/_common/Abstractions/Cache/MemoryBotCache.cs
Normal file
|
@ -0,0 +1,71 @@
|
||||||
|
using Microsoft.Extensions.Caching.Memory;
|
||||||
|
using OneOf;
|
||||||
|
using OneOf.Types;
|
||||||
|
|
||||||
|
// ReSharper disable InconsistentlySynchronizedField
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public sealed class MemoryBotCache : IBotCache
|
||||||
|
{
|
||||||
|
// needed for overwrites and Delete return value
|
||||||
|
private readonly object _cacheLock = new object();
|
||||||
|
private readonly MemoryCache _cache;
|
||||||
|
|
||||||
|
public MemoryBotCache()
|
||||||
|
{
|
||||||
|
_cache = new MemoryCache(new MemoryCacheOptions());
|
||||||
|
}
|
||||||
|
|
||||||
|
public ValueTask<bool> AddAsync<T>(TypedKey<T> key, T value, TimeSpan? expiry = null, bool overwrite = true)
|
||||||
|
{
|
||||||
|
if (overwrite)
|
||||||
|
{
|
||||||
|
using var item = _cache.CreateEntry(key.Key);
|
||||||
|
item.Value = value;
|
||||||
|
item.AbsoluteExpirationRelativeToNow = expiry;
|
||||||
|
return new(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
lock (_cacheLock)
|
||||||
|
{
|
||||||
|
if (_cache.TryGetValue(key.Key, out var old) && old is not null)
|
||||||
|
return new(false);
|
||||||
|
|
||||||
|
using var item = _cache.CreateEntry(key.Key);
|
||||||
|
item.Value = value;
|
||||||
|
item.AbsoluteExpirationRelativeToNow = expiry;
|
||||||
|
return new(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public async ValueTask<T?> GetOrAddAsync<T>(
|
||||||
|
TypedKey<T> key,
|
||||||
|
Func<Task<T?>> createFactory,
|
||||||
|
TimeSpan? expiry = null)
|
||||||
|
=> await _cache.GetOrCreateAsync(key.Key,
|
||||||
|
async ce =>
|
||||||
|
{
|
||||||
|
ce.AbsoluteExpirationRelativeToNow = expiry;
|
||||||
|
var val = await createFactory();
|
||||||
|
return val;
|
||||||
|
});
|
||||||
|
|
||||||
|
public ValueTask<OneOf<T, None>> GetAsync<T>(TypedKey<T> key)
|
||||||
|
{
|
||||||
|
if (!_cache.TryGetValue(key.Key, out var val) || val is null)
|
||||||
|
return new(new None());
|
||||||
|
|
||||||
|
return new((T)val);
|
||||||
|
}
|
||||||
|
|
||||||
|
public ValueTask<bool> RemoveAsync<T>(TypedKey<T> key)
|
||||||
|
{
|
||||||
|
lock (_cacheLock)
|
||||||
|
{
|
||||||
|
var toReturn = _cache.TryGetValue(key.Key, out var old) && old is not null;
|
||||||
|
_cache.Remove(key.Key);
|
||||||
|
return new(toReturn);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,88 @@
|
||||||
|
using System.Diagnostics;
|
||||||
|
|
||||||
|
namespace System.Collections.Generic;
|
||||||
|
|
||||||
|
[DebuggerDisplay("{_backingStore.Count}")]
|
||||||
|
public sealed class ConcurrentHashSet<T> : IReadOnlyCollection<T>, ICollection<T> where T : notnull
|
||||||
|
{
|
||||||
|
private readonly ConcurrentDictionary<T, bool> _backingStore;
|
||||||
|
|
||||||
|
public ConcurrentHashSet()
|
||||||
|
=> _backingStore = new();
|
||||||
|
|
||||||
|
public ConcurrentHashSet(IEnumerable<T> values, IEqualityComparer<T>? comparer = null)
|
||||||
|
=> _backingStore = new(values.Select(x => new KeyValuePair<T, bool>(x, true)), comparer);
|
||||||
|
|
||||||
|
public IEnumerator<T> GetEnumerator()
|
||||||
|
=> _backingStore.Keys.GetEnumerator();
|
||||||
|
|
||||||
|
IEnumerator IEnumerable.GetEnumerator()
|
||||||
|
=> GetEnumerator();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Adds the specified item to the <see cref="ConcurrentHashSet{T}" />.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="item">The item to add.</param>
|
||||||
|
/// <returns>
|
||||||
|
/// true if the items was added to the <see cref="ConcurrentHashSet{T}" />
|
||||||
|
/// successfully; false if it already exists.
|
||||||
|
/// </returns>
|
||||||
|
/// <exception cref="T:System.OverflowException">
|
||||||
|
/// The <see cref="ConcurrentHashSet{T}" />
|
||||||
|
/// contains too many items.
|
||||||
|
/// </exception>
|
||||||
|
public bool Add(T item)
|
||||||
|
=> _backingStore.TryAdd(item, true);
|
||||||
|
|
||||||
|
void ICollection<T>.Add(T item)
|
||||||
|
=> Add(item);
|
||||||
|
|
||||||
|
public void Clear()
|
||||||
|
=> _backingStore.Clear();
|
||||||
|
|
||||||
|
public bool Contains(T item)
|
||||||
|
=> _backingStore.ContainsKey(item);
|
||||||
|
|
||||||
|
public void CopyTo(T[] array, int arrayIndex)
|
||||||
|
{
|
||||||
|
ArgumentNullException.ThrowIfNull(array);
|
||||||
|
|
||||||
|
if (arrayIndex < 0)
|
||||||
|
throw new ArgumentOutOfRangeException(nameof(arrayIndex));
|
||||||
|
|
||||||
|
if (arrayIndex >= array.Length)
|
||||||
|
throw new ArgumentOutOfRangeException(nameof(arrayIndex));
|
||||||
|
|
||||||
|
CopyToInternal(array, arrayIndex);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CopyToInternal(T[] array, int arrayIndex)
|
||||||
|
{
|
||||||
|
var len = array.Length;
|
||||||
|
foreach (var (k, _) in _backingStore)
|
||||||
|
{
|
||||||
|
if (arrayIndex >= len)
|
||||||
|
throw new IndexOutOfRangeException(nameof(arrayIndex));
|
||||||
|
|
||||||
|
array[arrayIndex++] = k;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ICollection<T>.Remove(T item)
|
||||||
|
=> TryRemove(item);
|
||||||
|
|
||||||
|
public bool TryRemove(T item)
|
||||||
|
=> _backingStore.TryRemove(item, out _);
|
||||||
|
|
||||||
|
public void RemoveWhere(Func<T, bool> predicate)
|
||||||
|
{
|
||||||
|
foreach (var elem in this.Where(predicate))
|
||||||
|
TryRemove(elem);
|
||||||
|
}
|
||||||
|
|
||||||
|
public int Count
|
||||||
|
=> _backingStore.Count;
|
||||||
|
|
||||||
|
public bool IsReadOnly
|
||||||
|
=> false;
|
||||||
|
}
|
|
@ -0,0 +1,148 @@
|
||||||
|
using System.Collections;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public interface IIndexed
|
||||||
|
{
|
||||||
|
int Index { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public class IndexedCollection<T> : IList<T>
|
||||||
|
where T : class, IIndexed
|
||||||
|
{
|
||||||
|
public List<T> Source { get; }
|
||||||
|
|
||||||
|
public int Count
|
||||||
|
=> Source.Count;
|
||||||
|
|
||||||
|
public bool IsReadOnly
|
||||||
|
=> false;
|
||||||
|
|
||||||
|
public virtual T this[int index]
|
||||||
|
{
|
||||||
|
get => Source[index];
|
||||||
|
set
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
value.Index = index;
|
||||||
|
Source[index] = value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private readonly object _locker = new();
|
||||||
|
|
||||||
|
public IndexedCollection()
|
||||||
|
=> Source = new();
|
||||||
|
|
||||||
|
public IndexedCollection(IEnumerable<T> source)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
Source = source.OrderBy(x => x.Index).ToList();
|
||||||
|
UpdateIndexes();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public int IndexOf(T item)
|
||||||
|
=> item?.Index ?? -1;
|
||||||
|
|
||||||
|
public IEnumerator<T> GetEnumerator()
|
||||||
|
=> Source.GetEnumerator();
|
||||||
|
|
||||||
|
IEnumerator IEnumerable.GetEnumerator()
|
||||||
|
=> Source.GetEnumerator();
|
||||||
|
|
||||||
|
public void Add(T item)
|
||||||
|
{
|
||||||
|
ArgumentNullException.ThrowIfNull(item);
|
||||||
|
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
item.Index = Source.Count;
|
||||||
|
Source.Add(item);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public virtual void Clear()
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
Source.Clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public bool Contains(T item)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
return Source.Contains(item);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void CopyTo(T[] array, int arrayIndex)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
Source.CopyTo(array, arrayIndex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public virtual bool Remove(T item)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
if (Source.Remove(item))
|
||||||
|
{
|
||||||
|
for (var i = 0; i < Source.Count; i++)
|
||||||
|
{
|
||||||
|
if (Source[i].Index != i)
|
||||||
|
Source[i].Index = i;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public virtual void Insert(int index, T item)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
Source.Insert(index, item);
|
||||||
|
for (var i = index; i < Source.Count; i++)
|
||||||
|
Source[i].Index = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public virtual void RemoveAt(int index)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
Source.RemoveAt(index);
|
||||||
|
for (var i = index; i < Source.Count; i++)
|
||||||
|
Source[i].Index = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void UpdateIndexes()
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
for (var i = 0; i < Source.Count; i++)
|
||||||
|
{
|
||||||
|
if (Source[i].Index != i)
|
||||||
|
Source[i].Index = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static implicit operator List<T>(IndexedCollection<T> x)
|
||||||
|
=> x.Source;
|
||||||
|
|
||||||
|
public List<T> ToList()
|
||||||
|
=> Source.ToList();
|
||||||
|
}
|
66
src/EllieBot/_common/Abstractions/EllieRandom.cs
Normal file
66
src/EllieBot/_common/Abstractions/EllieRandom.cs
Normal file
|
@ -0,0 +1,66 @@
|
||||||
|
#nullable disable
|
||||||
|
using System.Security.Cryptography;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public sealed class EllieRandom : Random
|
||||||
|
{
|
||||||
|
private readonly RandomNumberGenerator _rng;
|
||||||
|
|
||||||
|
public EllieRandom()
|
||||||
|
=> _rng = RandomNumberGenerator.Create();
|
||||||
|
|
||||||
|
public override int Next()
|
||||||
|
{
|
||||||
|
var bytes = new byte[sizeof(int)];
|
||||||
|
_rng.GetBytes(bytes);
|
||||||
|
return Math.Abs(BitConverter.ToInt32(bytes, 0));
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Generates a random integer between 0 (inclusive) and
|
||||||
|
/// a specified exclusive upper bound using a cryptographically strong random number generator.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="maxValue">Exclusive max value</param>
|
||||||
|
/// <returns>A random number</returns>
|
||||||
|
public override int Next(int maxValue)
|
||||||
|
=> RandomNumberGenerator.GetInt32(maxValue);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Generates a random integer between a specified inclusive lower bound and a
|
||||||
|
/// specified exclusive upper bound using a cryptographically strong random number generator.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="minValue">Inclusive min value</param>
|
||||||
|
/// <param name="maxValue">Exclusive max value</param>
|
||||||
|
/// <returns>A random number</returns>
|
||||||
|
public override int Next(int minValue, int maxValue)
|
||||||
|
=> RandomNumberGenerator.GetInt32(minValue, maxValue);
|
||||||
|
|
||||||
|
public long NextLong(long minValue, long maxValue)
|
||||||
|
{
|
||||||
|
ArgumentOutOfRangeException.ThrowIfGreaterThan(minValue, maxValue);
|
||||||
|
if (minValue == maxValue)
|
||||||
|
return minValue;
|
||||||
|
var bytes = new byte[sizeof(long)];
|
||||||
|
_rng.GetBytes(bytes);
|
||||||
|
var sign = Math.Sign(BitConverter.ToInt64(bytes, 0));
|
||||||
|
return (sign * BitConverter.ToInt64(bytes, 0) % (maxValue - minValue)) + minValue;
|
||||||
|
}
|
||||||
|
|
||||||
|
public override void NextBytes(byte[] buffer)
|
||||||
|
=> _rng.GetBytes(buffer);
|
||||||
|
|
||||||
|
protected override double Sample()
|
||||||
|
{
|
||||||
|
var bytes = new byte[sizeof(double)];
|
||||||
|
_rng.GetBytes(bytes);
|
||||||
|
return Math.Abs((BitConverter.ToDouble(bytes, 0) / double.MaxValue) + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
public override double NextDouble()
|
||||||
|
{
|
||||||
|
var bytes = new byte[sizeof(double)];
|
||||||
|
_rng.GetBytes(bytes);
|
||||||
|
return BitConverter.ToDouble(bytes, 0);
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,62 @@
|
||||||
|
using System.Security.Cryptography;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
// made for expressions because they almost never get added
|
||||||
|
// and they get looped through constantly
|
||||||
|
public static class ArrayExtensions
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Create a new array from the old array + new element at the end
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="input">Input array</param>
|
||||||
|
/// <param name="added">Item to add to the end of the output array</param>
|
||||||
|
/// <typeparam name="T">Type of the array</typeparam>
|
||||||
|
/// <returns>A new array with the new element at the end</returns>
|
||||||
|
public static T[] With<T>(this T[] input, T added)
|
||||||
|
{
|
||||||
|
var newExprs = new T[input.Length + 1];
|
||||||
|
Array.Copy(input, 0, newExprs, 0, input.Length);
|
||||||
|
newExprs[input.Length] = added;
|
||||||
|
return newExprs;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Creates a new array by applying the specified function to every element in the input array
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="arr">Array to modify</param>
|
||||||
|
/// <param name="f">Function to apply</param>
|
||||||
|
/// <typeparam name="TIn">Orignal type of the elements in the array</typeparam>
|
||||||
|
/// <typeparam name="TOut">Output type of the elements of the array</typeparam>
|
||||||
|
/// <returns>New array with updated elements</returns>
|
||||||
|
public static TOut[] Map<TIn, TOut>(this TIn[] arr, Func<TIn, TOut> f)
|
||||||
|
=> Array.ConvertAll(arr, x => f(x));
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Creates a new array by applying the specified function to every element in the input array
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="col">Array to modify</param>
|
||||||
|
/// <param name="f">Function to apply</param>
|
||||||
|
/// <typeparam name="TIn">Orignal type of the elements in the array</typeparam>
|
||||||
|
/// <typeparam name="TOut">Output type of the elements of the array</typeparam>
|
||||||
|
/// <returns>New array with updated elements</returns>
|
||||||
|
public static TOut[] Map<TIn, TOut>(this IReadOnlyCollection<TIn> col, Func<TIn, TOut> f)
|
||||||
|
{
|
||||||
|
var toReturn = new TOut[col.Count];
|
||||||
|
|
||||||
|
var i = 0;
|
||||||
|
foreach (var item in col)
|
||||||
|
toReturn[i++] = f(item);
|
||||||
|
|
||||||
|
return toReturn;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static T? RandomOrDefault<T>(this T[] data)
|
||||||
|
{
|
||||||
|
if (data.Length == 0)
|
||||||
|
return default;
|
||||||
|
|
||||||
|
var index = RandomNumberGenerator.GetInt32(0, data.Length);
|
||||||
|
return data[index];
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,97 @@
|
||||||
|
using System.Security.Cryptography;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class EnumerableExtensions
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Concatenates the members of a collection, using the specified separator between each member.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="data">Collection to join</param>
|
||||||
|
/// <param name="separator">
|
||||||
|
/// The character to use as a separator. separator is included in the returned string only if
|
||||||
|
/// values has more than one element.
|
||||||
|
/// </param>
|
||||||
|
/// <param name="func">Optional transformation to apply to each element before concatenation.</param>
|
||||||
|
/// <typeparam name="T">The type of the members of values.</typeparam>
|
||||||
|
/// <returns>
|
||||||
|
/// A string that consists of the members of values delimited by the separator character. -or- Empty if values has
|
||||||
|
/// no elements.
|
||||||
|
/// </returns>
|
||||||
|
public static string Join<T>(this IEnumerable<T> data, char separator, Func<T, string>? func = null)
|
||||||
|
=> string.Join(separator, data.Select(func ?? (x => x?.ToString() ?? string.Empty)));
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Concatenates the members of a collection, using the specified separator between each member.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="data">Collection to join</param>
|
||||||
|
/// <param name="separator">
|
||||||
|
/// The string to use as a separator.separator is included in the returned string only if values
|
||||||
|
/// has more than one element.
|
||||||
|
/// </param>
|
||||||
|
/// <param name="func">Optional transformation to apply to each element before concatenation.</param>
|
||||||
|
/// <typeparam name="T">The type of the members of values.</typeparam>
|
||||||
|
/// <returns>
|
||||||
|
/// A string that consists of the members of values delimited by the separator character. -or- Empty if values has
|
||||||
|
/// no elements.
|
||||||
|
/// </returns>
|
||||||
|
public static string Join<T>(this IEnumerable<T> data, string separator, Func<T, string>? func = null)
|
||||||
|
=> string.Join(separator, data.Select(func ?? (x => x?.ToString() ?? string.Empty)));
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Randomize element order by performing the Fisher-Yates shuffle
|
||||||
|
/// </summary>
|
||||||
|
/// <typeparam name="T">Item type</typeparam>
|
||||||
|
/// <param name="items">Items to shuffle</param>
|
||||||
|
public static IReadOnlyList<T> Shuffle<T>(this IEnumerable<T> items)
|
||||||
|
{
|
||||||
|
var list = items.ToArray();
|
||||||
|
var n = list.Length;
|
||||||
|
while (n-- > 1)
|
||||||
|
{
|
||||||
|
var k = RandomNumberGenerator.GetInt32(n);
|
||||||
|
(list[k], list[n]) = (list[n], list[k]);
|
||||||
|
}
|
||||||
|
|
||||||
|
return list;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Initializes a new instance of the <see cref="ConcurrentDictionary{TKey,TValue}" /> class
|
||||||
|
/// that contains elements copied from the specified <see cref="IEnumerable{T}" />
|
||||||
|
/// has the default concurrency level, has the default initial capacity,
|
||||||
|
/// and uses the default comparer for the key type.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="dict">
|
||||||
|
/// The <see cref="IEnumerable{T}" /> whose elements are copied to the new
|
||||||
|
/// <see cref="ConcurrentDictionary{TKey,TValue}" />.
|
||||||
|
/// </param>
|
||||||
|
/// <returns>A new instance of the <see cref="ConcurrentDictionary{TKey,TValue}" /> class</returns>
|
||||||
|
public static ConcurrentDictionary<TKey, TValue> ToConcurrent<TKey, TValue>(
|
||||||
|
this IEnumerable<KeyValuePair<TKey, TValue>> dict)
|
||||||
|
where TKey : notnull
|
||||||
|
=> new(dict);
|
||||||
|
|
||||||
|
public static IndexedCollection<T> ToIndexed<T>(this IEnumerable<T> enumerable)
|
||||||
|
where T : class, IIndexed
|
||||||
|
=> new(enumerable);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Creates a task that will complete when all of the <see cref="Task{TResult}" /> objects in an enumerable
|
||||||
|
/// collection have completed
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="tasks">The tasks to wait on for completion.</param>
|
||||||
|
/// <typeparam name="TResult">The type of the completed task.</typeparam>
|
||||||
|
/// <returns>A task that represents the completion of all of the supplied tasks.</returns>
|
||||||
|
public static Task<TResult[]> WhenAll<TResult>(this IEnumerable<Task<TResult>> tasks)
|
||||||
|
=> Task.WhenAll(tasks);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Creates a task that will complete when all of the <see cref="Task" /> objects in an enumerable
|
||||||
|
/// collection have completed
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="tasks">The tasks to wait on for completion.</param>
|
||||||
|
/// <returns>A task that represents the completion of all of the supplied tasks.</returns>
|
||||||
|
public static Task WhenAll(this IEnumerable<Task> tasks)
|
||||||
|
=> Task.WhenAll(tasks);
|
||||||
|
}
|
|
@ -0,0 +1,7 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class Extensions
|
||||||
|
{
|
||||||
|
public static long ToTimestamp(this in DateTime value)
|
||||||
|
=> (value.Ticks - 621355968000000000) / 10000000;
|
||||||
|
}
|
|
@ -0,0 +1,35 @@
|
||||||
|
using System.Net.Http.Headers;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class HttpClientExtensions
|
||||||
|
{
|
||||||
|
public static HttpClient AddFakeHeaders(this HttpClient http)
|
||||||
|
{
|
||||||
|
AddFakeHeaders(http.DefaultRequestHeaders);
|
||||||
|
return http;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void AddFakeHeaders(this HttpHeaders dict)
|
||||||
|
{
|
||||||
|
dict.Clear();
|
||||||
|
dict.Add("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8");
|
||||||
|
dict.Add("User-Agent",
|
||||||
|
"Mozilla/5.0 (Windows NT 6.1) AppleWebKit/535.1 (KHTML, like Gecko) Chrome/14.0.835.202 Safari/535.1");
|
||||||
|
}
|
||||||
|
|
||||||
|
public static bool IsImage(this HttpResponseMessage msg)
|
||||||
|
=> IsImage(msg, out _);
|
||||||
|
|
||||||
|
public static bool IsImage(this HttpResponseMessage msg, out string? mimeType)
|
||||||
|
{
|
||||||
|
mimeType = msg.Content.Headers.ContentType?.MediaType;
|
||||||
|
if (mimeType is "image/png" or "image/jpeg" or "image/gif")
|
||||||
|
return true;
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static long GetContentLength(this HttpResponseMessage msg)
|
||||||
|
=> msg.Content.Headers.ContentLength ?? long.MaxValue;
|
||||||
|
}
|
|
@ -0,0 +1,10 @@
|
||||||
|
using OneOf.Types;
|
||||||
|
using OneOf;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class OneOfExtensions
|
||||||
|
{
|
||||||
|
public static bool TryGetValue<T>(this OneOf<T, None> oneOf, out T value)
|
||||||
|
=> oneOf.TryPickT0(out value, out _);
|
||||||
|
}
|
|
@ -0,0 +1,22 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public delegate TOut PipeFunc<TIn, out TOut>(in TIn a);
|
||||||
|
public delegate TOut PipeFunc<TIn1, TIn2, out TOut>(in TIn1 a, in TIn2 b);
|
||||||
|
|
||||||
|
public static class PipeExtensions
|
||||||
|
{
|
||||||
|
public static TOut Pipe<TIn, TOut>(this TIn a, Func<TIn, TOut> fn)
|
||||||
|
=> fn(a);
|
||||||
|
|
||||||
|
public static TOut Pipe<TIn, TOut>(this TIn a, PipeFunc<TIn, TOut> fn)
|
||||||
|
=> fn(a);
|
||||||
|
|
||||||
|
public static TOut Pipe<TIn1, TIn2, TOut>(this (TIn1, TIn2) a, PipeFunc<TIn1, TIn2, TOut> fn)
|
||||||
|
=> fn(a.Item1, a.Item2);
|
||||||
|
|
||||||
|
public static (TIn, TExtra) With<TIn, TExtra>(this TIn a, TExtra b)
|
||||||
|
=> (a, b);
|
||||||
|
|
||||||
|
public static async Task<TOut> Pipe<TIn, TOut>(this Task<TIn> a, Func<TIn, TOut> fn)
|
||||||
|
=> fn(await a);
|
||||||
|
}
|
150
src/EllieBot/_common/Abstractions/Extensions/StringExtensions.cs
Normal file
150
src/EllieBot/_common/Abstractions/Extensions/StringExtensions.cs
Normal file
|
@ -0,0 +1,150 @@
|
||||||
|
using EllieBot.Common.Yml;
|
||||||
|
using System.Text;
|
||||||
|
using System.Text.RegularExpressions;
|
||||||
|
|
||||||
|
namespace EllieBot.Extensions;
|
||||||
|
|
||||||
|
public static class StringExtensions
|
||||||
|
{
|
||||||
|
private static readonly HashSet<char> _lettersAndDigits =
|
||||||
|
[
|
||||||
|
..Enumerable.Range(48, 10)
|
||||||
|
.Concat(Enumerable.Range(65, 26))
|
||||||
|
.Concat(Enumerable.Range(97, 26))
|
||||||
|
.Select(x => (char)x)
|
||||||
|
];
|
||||||
|
|
||||||
|
private static readonly Regex _filterRegex = new(@"discord(?:\.gg|\.io|\.me|\.li|(?:app)?\.com\/invite)\/(\w+)",
|
||||||
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
||||||
|
|
||||||
|
private static readonly Regex _codePointRegex =
|
||||||
|
new(@"(\\U(?<code>[a-zA-Z0-9]{8})|\\u(?<code>[a-zA-Z0-9]{4})|\\x(?<code>[a-zA-Z0-9]{2}))",
|
||||||
|
RegexOptions.Compiled);
|
||||||
|
|
||||||
|
public static string PadBoth(this string str, int length)
|
||||||
|
{
|
||||||
|
var spaces = length - str.Length;
|
||||||
|
var padLeft = (spaces / 2) + str.Length;
|
||||||
|
return str.PadLeft(padLeft, ' ').PadRight(length, ' ');
|
||||||
|
}
|
||||||
|
|
||||||
|
public static string StripHtml(this string input)
|
||||||
|
=> Regex.Replace(input, "<.*?>", string.Empty);
|
||||||
|
|
||||||
|
public static string? TrimTo(this string? str, int maxLength, bool hideDots = false)
|
||||||
|
{
|
||||||
|
if (hideDots)
|
||||||
|
{
|
||||||
|
return str?.Substring(0, Math.Min(str?.Length ?? 0, maxLength));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (str is null || str.Length <= maxLength)
|
||||||
|
return str;
|
||||||
|
|
||||||
|
return string.Concat(str.AsSpan(0, maxLength - 1), "…");
|
||||||
|
}
|
||||||
|
|
||||||
|
public static string ToTitleCase(this string str)
|
||||||
|
{
|
||||||
|
var tokens = str.Split([" "], StringSplitOptions.RemoveEmptyEntries);
|
||||||
|
for (var i = 0; i < tokens.Length; i++)
|
||||||
|
{
|
||||||
|
var token = tokens[i];
|
||||||
|
tokens[i] = token[..1].ToUpperInvariant() + token[1..];
|
||||||
|
}
|
||||||
|
|
||||||
|
return tokens.Join(" ").Replace(" Of ", " of ").Replace(" The ", " the ");
|
||||||
|
}
|
||||||
|
|
||||||
|
//http://www.dotnetperls.com/levenshtein
|
||||||
|
public static int LevenshteinDistance(this string s, string t)
|
||||||
|
{
|
||||||
|
var n = s.Length;
|
||||||
|
var m = t.Length;
|
||||||
|
var d = new int[n + 1, m + 1];
|
||||||
|
|
||||||
|
// Step 1
|
||||||
|
if (n == 0)
|
||||||
|
return m;
|
||||||
|
|
||||||
|
if (m == 0)
|
||||||
|
return n;
|
||||||
|
|
||||||
|
// Step 2
|
||||||
|
for (var i = 0; i <= n; d[i, 0] = i++)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
for (var j = 0; j <= m; d[0, j] = j++)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
// Step 3
|
||||||
|
for (var i = 1; i <= n; i++)
|
||||||
|
//Step 4
|
||||||
|
for (var j = 1; j <= m; j++)
|
||||||
|
{
|
||||||
|
// Step 5
|
||||||
|
var cost = t[j - 1] == s[i - 1] ? 0 : 1;
|
||||||
|
|
||||||
|
// Step 6
|
||||||
|
d[i, j] = Math.Min(Math.Min(d[i - 1, j] + 1, d[i, j - 1] + 1), d[i - 1, j - 1] + cost);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Step 7
|
||||||
|
return d[n, m];
|
||||||
|
}
|
||||||
|
|
||||||
|
public static async Task<Stream> ToStream(this string str)
|
||||||
|
{
|
||||||
|
var ms = new MemoryStream();
|
||||||
|
var sw = new StreamWriter(ms);
|
||||||
|
await sw.WriteAsync(str);
|
||||||
|
await sw.FlushAsync();
|
||||||
|
ms.Position = 0;
|
||||||
|
return ms;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static bool IsDiscordInvite(this string str)
|
||||||
|
=> _filterRegex.IsMatch(str);
|
||||||
|
|
||||||
|
public static string Unmention(this string str)
|
||||||
|
=> str.Replace("@", "ම", StringComparison.InvariantCulture);
|
||||||
|
|
||||||
|
public static string SanitizeMentions(this string str, bool sanitizeRoleMentions = false)
|
||||||
|
{
|
||||||
|
str = str.Replace("@everyone", "@everyοne", StringComparison.InvariantCultureIgnoreCase)
|
||||||
|
.Replace("@here", "@һere", StringComparison.InvariantCultureIgnoreCase);
|
||||||
|
if (sanitizeRoleMentions)
|
||||||
|
str = str.SanitizeRoleMentions();
|
||||||
|
|
||||||
|
return str;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static string SanitizeRoleMentions(this string str)
|
||||||
|
=> str.Replace("<@&", "<ම&", StringComparison.InvariantCultureIgnoreCase);
|
||||||
|
|
||||||
|
public static string SanitizeAllMentions(this string str)
|
||||||
|
=> str.SanitizeMentions().SanitizeRoleMentions();
|
||||||
|
|
||||||
|
public static string ToBase64(this string plainText)
|
||||||
|
{
|
||||||
|
var plainTextBytes = Encoding.UTF8.GetBytes(plainText);
|
||||||
|
return Convert.ToBase64String(plainTextBytes);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static string GetInitials(this string txt, string glue = "")
|
||||||
|
=> txt.Split(' ').Select(x => x.FirstOrDefault()).Join(glue);
|
||||||
|
|
||||||
|
public static bool IsAlphaNumeric(this string txt)
|
||||||
|
=> txt.All(c => _lettersAndDigits.Contains(c));
|
||||||
|
|
||||||
|
public static string UnescapeUnicodeCodePoints(this string input)
|
||||||
|
=> _codePointRegex.Replace(input,
|
||||||
|
me =>
|
||||||
|
{
|
||||||
|
var str = me.Groups["code"].Value;
|
||||||
|
var newString = str.UnescapeUnicodeCodePoint();
|
||||||
|
return newString;
|
||||||
|
});
|
||||||
|
}
|
35
src/EllieBot/_common/Abstractions/Helpers/LogSetup.cs
Normal file
35
src/EllieBot/_common/Abstractions/Helpers/LogSetup.cs
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
using Serilog.Events;
|
||||||
|
using Serilog.Sinks.SystemConsole.Themes;
|
||||||
|
using System.Text;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class LogSetup
|
||||||
|
{
|
||||||
|
public static void SetupLogger(object source)
|
||||||
|
{
|
||||||
|
Log.Logger = new LoggerConfiguration().MinimumLevel.Override("Microsoft", LogEventLevel.Information)
|
||||||
|
.MinimumLevel.Override("System", LogEventLevel.Information)
|
||||||
|
.MinimumLevel.Override("Microsoft.AspNetCore", LogEventLevel.Warning)
|
||||||
|
.Enrich.FromLogContext()
|
||||||
|
.WriteTo.Console(LogEventLevel.Information,
|
||||||
|
theme: GetTheme(),
|
||||||
|
outputTemplate:
|
||||||
|
"[{Timestamp:HH:mm:ss} {Level:u3}] | #{LogSource} | {Message:lj}{NewLine}{Exception}")
|
||||||
|
.Enrich.WithProperty("LogSource", source)
|
||||||
|
.CreateLogger();
|
||||||
|
|
||||||
|
Console.OutputEncoding = Encoding.UTF8;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static ConsoleTheme GetTheme()
|
||||||
|
{
|
||||||
|
if (Environment.OSVersion.Platform == PlatformID.Unix)
|
||||||
|
return AnsiConsoleTheme.Code;
|
||||||
|
#if DEBUG
|
||||||
|
return AnsiConsoleTheme.Code;
|
||||||
|
#else
|
||||||
|
return ConsoleTheme.None;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,7 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public static class StandardConversions
|
||||||
|
{
|
||||||
|
public static double CelsiusToFahrenheit(double cel)
|
||||||
|
=> (cel * 1.8f) + 32;
|
||||||
|
}
|
100
src/EllieBot/_common/Abstractions/Kwum.cs
Normal file
100
src/EllieBot/_common/Abstractions/Kwum.cs
Normal file
|
@ -0,0 +1,100 @@
|
||||||
|
using System.Runtime.CompilerServices;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
// needs proper invalid input check (character array input out of range)
|
||||||
|
// needs negative number support
|
||||||
|
// ReSharper disable once InconsistentNaming
|
||||||
|
#pragma warning disable CS8981
|
||||||
|
public readonly struct kwum : IEquatable<kwum>
|
||||||
|
#pragma warning restore CS8981
|
||||||
|
{
|
||||||
|
private const string VALID_CHARACTERS = "23456789abcdefghijkmnpqrstuvwxyz";
|
||||||
|
private readonly int _value;
|
||||||
|
|
||||||
|
public kwum(int num)
|
||||||
|
=> _value = num;
|
||||||
|
|
||||||
|
public kwum(in char c)
|
||||||
|
{
|
||||||
|
if (!IsValidChar(c))
|
||||||
|
throw new ArgumentException("Character needs to be a valid kwum character.", nameof(c));
|
||||||
|
|
||||||
|
_value = InternalCharToValue(c);
|
||||||
|
}
|
||||||
|
|
||||||
|
public kwum(in ReadOnlySpan<char> input)
|
||||||
|
{
|
||||||
|
_value = 0;
|
||||||
|
for (var index = 0; index < input.Length; index++)
|
||||||
|
{
|
||||||
|
var c = input[index];
|
||||||
|
if (!IsValidChar(c))
|
||||||
|
throw new ArgumentException("All characters need to be a valid kwum characters.", nameof(input));
|
||||||
|
|
||||||
|
_value += VALID_CHARACTERS.IndexOf(c) * (int)Math.Pow(VALID_CHARACTERS.Length, input.Length - index - 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||||
|
private static int InternalCharToValue(in char c)
|
||||||
|
=> VALID_CHARACTERS.IndexOf(c);
|
||||||
|
|
||||||
|
public static bool TryParse(in ReadOnlySpan<char> input, out kwum value)
|
||||||
|
{
|
||||||
|
value = default;
|
||||||
|
foreach (var c in input)
|
||||||
|
{
|
||||||
|
if (!IsValidChar(c))
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
value = new(input);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static kwum operator +(kwum left, kwum right)
|
||||||
|
=> new(left._value + right._value);
|
||||||
|
|
||||||
|
public static bool operator ==(kwum left, kwum right)
|
||||||
|
=> left._value == right._value;
|
||||||
|
|
||||||
|
public static bool operator !=(kwum left, kwum right)
|
||||||
|
=> !(left == right);
|
||||||
|
|
||||||
|
public static implicit operator long(kwum kwum)
|
||||||
|
=> kwum._value;
|
||||||
|
|
||||||
|
public static implicit operator int(kwum kwum)
|
||||||
|
=> kwum._value;
|
||||||
|
|
||||||
|
public static implicit operator kwum(int num)
|
||||||
|
=> new(num);
|
||||||
|
|
||||||
|
public static bool IsValidChar(char c)
|
||||||
|
=> VALID_CHARACTERS.Contains(c);
|
||||||
|
|
||||||
|
public override string ToString()
|
||||||
|
{
|
||||||
|
var count = VALID_CHARACTERS.Length;
|
||||||
|
var localValue = _value;
|
||||||
|
var arrSize = (int)Math.Log(localValue, count) + 1;
|
||||||
|
Span<char> chars = new char[arrSize];
|
||||||
|
while (localValue > 0)
|
||||||
|
{
|
||||||
|
localValue = Math.DivRem(localValue, count, out var rem);
|
||||||
|
chars[--arrSize] = VALID_CHARACTERS[rem];
|
||||||
|
}
|
||||||
|
|
||||||
|
return new(chars);
|
||||||
|
}
|
||||||
|
|
||||||
|
public override bool Equals(object? obj)
|
||||||
|
=> obj is kwum kw && kw == this;
|
||||||
|
|
||||||
|
public bool Equals(kwum other)
|
||||||
|
=> other == this;
|
||||||
|
|
||||||
|
public override int GetHashCode()
|
||||||
|
=> _value.GetHashCode();
|
||||||
|
}
|
80
src/EllieBot/_common/Abstractions/PubSub/EventPubSub.cs
Normal file
80
src/EllieBot/_common/Abstractions/PubSub/EventPubSub.cs
Normal file
|
@ -0,0 +1,80 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public class EventPubSub : IPubSub
|
||||||
|
{
|
||||||
|
private readonly Dictionary<string, Dictionary<Delegate, List<Func<object, ValueTask>>>> _actions = new();
|
||||||
|
private readonly object _locker = new();
|
||||||
|
|
||||||
|
public Task Sub<TData>(in TypedKey<TData> key, Func<TData, ValueTask> action)
|
||||||
|
where TData : notnull
|
||||||
|
{
|
||||||
|
Func<object, ValueTask> localAction = obj => action((TData)obj);
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
if (!_actions.TryGetValue(key.Key, out var keyActions))
|
||||||
|
{
|
||||||
|
keyActions = new();
|
||||||
|
_actions[key.Key] = keyActions;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!keyActions.TryGetValue(action, out var sameActions))
|
||||||
|
{
|
||||||
|
sameActions = new();
|
||||||
|
keyActions[action] = sameActions;
|
||||||
|
}
|
||||||
|
|
||||||
|
sameActions.Add(localAction);
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Task Pub<TData>(in TypedKey<TData> key, TData data)
|
||||||
|
where TData : notnull
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
if (_actions.TryGetValue(key.Key, out var actions))
|
||||||
|
// if this class ever gets used, this needs to be properly implemented
|
||||||
|
// 1. ignore all valuetasks which are completed
|
||||||
|
// 2. run all other tasks in parallel
|
||||||
|
return actions.SelectMany(kvp => kvp.Value).Select(action => action(data).AsTask()).WhenAll();
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Task Unsub<TData>(in TypedKey<TData> key, Func<TData, ValueTask> action)
|
||||||
|
{
|
||||||
|
lock (_locker)
|
||||||
|
{
|
||||||
|
// get subscriptions for this action
|
||||||
|
if (_actions.TryGetValue(key.Key, out var actions))
|
||||||
|
// get subscriptions which have the same action hash code
|
||||||
|
// note: having this as a list allows for multiple subscriptions of
|
||||||
|
// the same insance's/static method
|
||||||
|
{
|
||||||
|
if (actions.TryGetValue(action, out var sameActions))
|
||||||
|
{
|
||||||
|
// remove last subscription
|
||||||
|
sameActions.RemoveAt(sameActions.Count - 1);
|
||||||
|
|
||||||
|
// if the last subscription was the only subscription
|
||||||
|
// we can safely remove this action's dictionary entry
|
||||||
|
if (sameActions.Count == 0)
|
||||||
|
{
|
||||||
|
actions.Remove(action);
|
||||||
|
|
||||||
|
// if our dictionary has no more elements after
|
||||||
|
// removing the entry
|
||||||
|
// it's safe to remove it from the key's subscriptions
|
||||||
|
if (actions.Count == 0)
|
||||||
|
_actions.Remove(key.Key);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
10
src/EllieBot/_common/Abstractions/PubSub/IPubSub.cs
Normal file
10
src/EllieBot/_common/Abstractions/PubSub/IPubSub.cs
Normal file
|
@ -0,0 +1,10 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public interface IPubSub
|
||||||
|
{
|
||||||
|
public Task Pub<TData>(in TypedKey<TData> key, TData data)
|
||||||
|
where TData : notnull;
|
||||||
|
|
||||||
|
public Task Sub<TData>(in TypedKey<TData> key, Func<TData, ValueTask> action)
|
||||||
|
where TData : notnull;
|
||||||
|
}
|
7
src/EllieBot/_common/Abstractions/PubSub/ISeria.cs
Normal file
7
src/EllieBot/_common/Abstractions/PubSub/ISeria.cs
Normal file
|
@ -0,0 +1,7 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public interface ISeria
|
||||||
|
{
|
||||||
|
byte[] Serialize<T>(T data);
|
||||||
|
T? Deserialize<T>(byte[]? data);
|
||||||
|
}
|
61
src/EllieBot/_common/Abstractions/QueueRunner.cs
Normal file
61
src/EllieBot/_common/Abstractions/QueueRunner.cs
Normal file
|
@ -0,0 +1,61 @@
|
||||||
|
using System.Threading.Channels;
|
||||||
|
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public sealed class QueueRunner
|
||||||
|
{
|
||||||
|
private readonly Channel<Func<Task>> _channel;
|
||||||
|
private readonly int _delayMs;
|
||||||
|
|
||||||
|
public QueueRunner(int delayMs = 0, int maxCapacity = -1)
|
||||||
|
{
|
||||||
|
ArgumentOutOfRangeException.ThrowIfNegative(delayMs);
|
||||||
|
|
||||||
|
_delayMs = delayMs;
|
||||||
|
_channel = maxCapacity switch
|
||||||
|
{
|
||||||
|
0 or < -1 => throw new ArgumentOutOfRangeException(nameof(maxCapacity)),
|
||||||
|
-1 => Channel.CreateUnbounded<Func<Task>>(new UnboundedChannelOptions()
|
||||||
|
{
|
||||||
|
SingleReader = true,
|
||||||
|
SingleWriter = false,
|
||||||
|
AllowSynchronousContinuations = true,
|
||||||
|
}),
|
||||||
|
_ => Channel.CreateBounded<Func<Task>>(new BoundedChannelOptions(maxCapacity)
|
||||||
|
{
|
||||||
|
Capacity = maxCapacity,
|
||||||
|
FullMode = BoundedChannelFullMode.DropOldest,
|
||||||
|
SingleReader = true,
|
||||||
|
SingleWriter = false,
|
||||||
|
AllowSynchronousContinuations = true
|
||||||
|
})
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task RunAsync(CancellationToken cancel = default)
|
||||||
|
{
|
||||||
|
while (true)
|
||||||
|
{
|
||||||
|
var func = await _channel.Reader.ReadAsync(cancel);
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
await func();
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
Log.Warning(ex, "Exception executing a staggered func: {ErrorMessage}", ex.Message);
|
||||||
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
if (_delayMs != 0)
|
||||||
|
{
|
||||||
|
await Task.Delay(_delayMs, cancel);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public ValueTask EnqueueAsync(Func<Task> action)
|
||||||
|
=> _channel.Writer.WriteAsync(action);
|
||||||
|
}
|
30
src/EllieBot/_common/Abstractions/TypedKey.cs
Normal file
30
src/EllieBot/_common/Abstractions/TypedKey.cs
Normal file
|
@ -0,0 +1,30 @@
|
||||||
|
namespace Ellie.Common;
|
||||||
|
|
||||||
|
public readonly struct TypedKey<TData>
|
||||||
|
{
|
||||||
|
public string Key { get; }
|
||||||
|
|
||||||
|
public TypedKey(in string key)
|
||||||
|
=> Key = key;
|
||||||
|
|
||||||
|
public static implicit operator TypedKey<TData>(in string input)
|
||||||
|
=> new(input);
|
||||||
|
|
||||||
|
public static implicit operator string(in TypedKey<TData> input)
|
||||||
|
=> input.Key;
|
||||||
|
|
||||||
|
public static bool operator ==(in TypedKey<TData> left, in TypedKey<TData> right)
|
||||||
|
=> left.Key == right.Key;
|
||||||
|
|
||||||
|
public static bool operator !=(in TypedKey<TData> left, in TypedKey<TData> right)
|
||||||
|
=> !(left == right);
|
||||||
|
|
||||||
|
public override bool Equals(object? obj)
|
||||||
|
=> obj is TypedKey<TData> o && o == this;
|
||||||
|
|
||||||
|
public override int GetHashCode()
|
||||||
|
=> Key?.GetHashCode() ?? 0;
|
||||||
|
|
||||||
|
public override string ToString()
|
||||||
|
=> Key;
|
||||||
|
}
|
48
src/EllieBot/_common/Abstractions/YamlHelper.cs
Normal file
48
src/EllieBot/_common/Abstractions/YamlHelper.cs
Normal file
|
@ -0,0 +1,48 @@
|
||||||
|
#nullable disable
|
||||||
|
namespace EllieBot.Common.Yml;
|
||||||
|
|
||||||
|
public static class YamlHelper
|
||||||
|
{
|
||||||
|
// https://github.com/aaubry/YamlDotNet/blob/0f4cc205e8b2dd8ef6589d96de32bf608a687c6f/YamlDotNet/Core/Scanner.cs#L1687
|
||||||
|
/// <summary>
|
||||||
|
/// This is modified code from yamldotnet's repo which handles parsing unicode code points
|
||||||
|
/// it is needed as yamldotnet doesn't support unescaped unicode characters
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="point">Unicode code point</param>
|
||||||
|
/// <returns>Actual character</returns>
|
||||||
|
public static string UnescapeUnicodeCodePoint(this string point)
|
||||||
|
{
|
||||||
|
var character = 0;
|
||||||
|
|
||||||
|
// Scan the character value.
|
||||||
|
|
||||||
|
foreach (var c in point)
|
||||||
|
{
|
||||||
|
if (!IsHex(c))
|
||||||
|
return point;
|
||||||
|
|
||||||
|
character = (character << 4) + AsHex(c);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check the value and write the character.
|
||||||
|
|
||||||
|
if (character is (>= 0xD800 and <= 0xDFFF) or > 0x10FFFF)
|
||||||
|
return point;
|
||||||
|
|
||||||
|
return char.ConvertFromUtf32(character);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static bool IsHex(char c)
|
||||||
|
=> c is (>= '0' and <= '9') or (>= 'A' and <= 'F') or (>= 'a' and <= 'f');
|
||||||
|
|
||||||
|
public static int AsHex(char c)
|
||||||
|
{
|
||||||
|
if (c <= '9')
|
||||||
|
return c - '0';
|
||||||
|
|
||||||
|
if (c <= 'F')
|
||||||
|
return c - 'A' + 10;
|
||||||
|
|
||||||
|
return c - 'a' + 10;
|
||||||
|
}
|
||||||
|
}
|
78
src/EllieBot/_common/Abstractions/creds/IBotCredentials.cs
Normal file
78
src/EllieBot/_common/Abstractions/creds/IBotCredentials.cs
Normal file
|
@ -0,0 +1,78 @@
|
||||||
|
#nullable disable
|
||||||
|
namespace EllieBot;
|
||||||
|
|
||||||
|
public interface IBotCredentials
|
||||||
|
{
|
||||||
|
string Token { get; }
|
||||||
|
string GoogleApiKey { get; }
|
||||||
|
ICollection<ulong> OwnerIds { get; set; }
|
||||||
|
bool UsePrivilegedIntents { get; }
|
||||||
|
string RapidApiKey { get; }
|
||||||
|
|
||||||
|
Creds.DbOptions Db { get; }
|
||||||
|
string OsuApiKey { get; }
|
||||||
|
int TotalShards { get; }
|
||||||
|
Creds.PatreonSettings Patreon { get; }
|
||||||
|
string CleverbotApiKey { get; }
|
||||||
|
string Gpt3ApiKey { get; }
|
||||||
|
RestartConfig RestartCommand { get; }
|
||||||
|
Creds.VotesSettings Votes { get; }
|
||||||
|
string BotListToken { get; }
|
||||||
|
string RedisOptions { get; }
|
||||||
|
string LocationIqApiKey { get; }
|
||||||
|
string TimezoneDbApiKey { get; }
|
||||||
|
string CoinmarketcapApiKey { get; }
|
||||||
|
string TrovoClientId { get; }
|
||||||
|
string CoordinatorUrl { get; set; }
|
||||||
|
string TwitchClientId { get; set; }
|
||||||
|
string TwitchClientSecret { get; set; }
|
||||||
|
GoogleApiConfig Google { get; set; }
|
||||||
|
BotCacheImplemenation BotCache { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface IVotesSettings
|
||||||
|
{
|
||||||
|
string TopggServiceUrl { get; set; }
|
||||||
|
string TopggKey { get; set; }
|
||||||
|
string DiscordsServiceUrl { get; set; }
|
||||||
|
string DiscordsKey { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface IPatreonSettings
|
||||||
|
{
|
||||||
|
public string ClientId { get; set; }
|
||||||
|
public string AccessToken { get; set; }
|
||||||
|
public string RefreshToken { get; set; }
|
||||||
|
public string ClientSecret { get; set; }
|
||||||
|
public string CampaignId { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface IRestartConfig
|
||||||
|
{
|
||||||
|
string Cmd { get; set; }
|
||||||
|
string Args { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public class RestartConfig : IRestartConfig
|
||||||
|
{
|
||||||
|
public string Cmd { get; set; }
|
||||||
|
public string Args { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public enum BotCacheImplemenation
|
||||||
|
{
|
||||||
|
Memory,
|
||||||
|
Redis
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface IDbOptions
|
||||||
|
{
|
||||||
|
string Type { get; set; }
|
||||||
|
string ConnectionString { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public interface IGoogleApiConfig
|
||||||
|
{
|
||||||
|
string SearchId { get; init; }
|
||||||
|
string ImageSearchId { get; init; }
|
||||||
|
}
|
|
@ -0,0 +1,8 @@
|
||||||
|
namespace EllieBot;
|
||||||
|
|
||||||
|
public interface IBotCredsProvider
|
||||||
|
{
|
||||||
|
public void Reload();
|
||||||
|
public IBotCredentials GetCreds();
|
||||||
|
public void ModifyCredsFile(Action<IBotCredentials> func);
|
||||||
|
}
|
35
src/EllieBot/_common/Abstractions/strings/CommandStrings.cs
Normal file
35
src/EllieBot/_common/Abstractions/strings/CommandStrings.cs
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
#nullable disable
|
||||||
|
using YamlDotNet.Core;
|
||||||
|
using YamlDotNet.Serialization;
|
||||||
|
|
||||||
|
namespace EllieBot.Services;
|
||||||
|
|
||||||
|
// public sealed record class CommandStrings
|
||||||
|
// {
|
||||||
|
// [YamlMember(Alias = "desc")]
|
||||||
|
// public string Desc { get; set; }
|
||||||
|
//
|
||||||
|
// [YamlMember(Alias = "args")]
|
||||||
|
// public string[] Args { get; set; }
|
||||||
|
// }
|
||||||
|
|
||||||
|
public sealed record class CommandStrings
|
||||||
|
{
|
||||||
|
[YamlMember(Alias = "desc")]
|
||||||
|
public string Desc { get; set; }
|
||||||
|
|
||||||
|
[YamlMember(Alias = "ex")]
|
||||||
|
public string[] Examples { get; set; }
|
||||||
|
|
||||||
|
[YamlMember(Alias = "params")]
|
||||||
|
public Dictionary<string, CommandStringParam>[] Params { get; set; }
|
||||||
|
}
|
||||||
|
|
||||||
|
public sealed record class CommandStringParam
|
||||||
|
{
|
||||||
|
// [YamlMember(Alias = "type", ScalarStyle = ScalarStyle.DoubleQuoted)]
|
||||||
|
// public string Type { get; set; }
|
||||||
|
|
||||||
|
[YamlMember(Alias = "desc", ScalarStyle = ScalarStyle.DoubleQuoted)]
|
||||||
|
public string Desc { get; set; }
|
||||||
|
}
|
16
src/EllieBot/_common/Abstractions/strings/IBotStrings.cs
Normal file
16
src/EllieBot/_common/Abstractions/strings/IBotStrings.cs
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
#nullable disable
|
||||||
|
using System.Globalization;
|
||||||
|
|
||||||
|
namespace EllieBot.Common;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Defines methods to retrieve and reload bot strings
|
||||||
|
/// </summary>
|
||||||
|
public interface IBotStrings
|
||||||
|
{
|
||||||
|
string GetText(string key, ulong? guildId = null, params object[] data);
|
||||||
|
string GetText(string key, CultureInfo locale, params object[] data);
|
||||||
|
void Reload();
|
||||||
|
CommandStrings GetCommandStrings(string commandName, ulong? guildId = null);
|
||||||
|
CommandStrings GetCommandStrings(string commandName, CultureInfo cultureInfo);
|
||||||
|
}
|
|
@ -0,0 +1,17 @@
|
||||||
|
#nullable disable
|
||||||
|
using System.Globalization;
|
||||||
|
|
||||||
|
namespace EllieBot.Common;
|
||||||
|
|
||||||
|
public static class BotStringsExtensions
|
||||||
|
{
|
||||||
|
// this one is for pipe fun, see PipeExtensions.cs
|
||||||
|
public static string GetText(this IBotStrings strings, in LocStr str, in ulong guildId)
|
||||||
|
=> strings.GetText(str.Key, guildId, str.Params);
|
||||||
|
|
||||||
|
public static string GetText(this IBotStrings strings, in LocStr str, ulong? guildId = null)
|
||||||
|
=> strings.GetText(str.Key, guildId, str.Params);
|
||||||
|
|
||||||
|
public static string GetText(this IBotStrings strings, in LocStr str, CultureInfo culture)
|
||||||
|
=> strings.GetText(str.Key, culture, str.Params);
|
||||||
|
}
|
|
@ -0,0 +1,28 @@
|
||||||
|
#nullable disable
|
||||||
|
namespace EllieBot.Services;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Implemented by classes which provide localized strings in their own ways
|
||||||
|
/// </summary>
|
||||||
|
public interface IBotStringsProvider
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Gets localized string
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="localeName">Language name</param>
|
||||||
|
/// <param name="key">String key</param>
|
||||||
|
/// <returns>Localized string</returns>
|
||||||
|
string GetText(string localeName, string key);
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Reloads string cache
|
||||||
|
/// </summary>
|
||||||
|
void Reload();
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Gets command arg examples and description
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="localeName">Language name</param>
|
||||||
|
/// <param name="commandName">Command name</param>
|
||||||
|
CommandStrings GetCommandStrings(string localeName, string commandName);
|
||||||
|
}
|
17
src/EllieBot/_common/Abstractions/strings/IStringsSource.cs
Normal file
17
src/EllieBot/_common/Abstractions/strings/IStringsSource.cs
Normal file
|
@ -0,0 +1,17 @@
|
||||||
|
#nullable disable
|
||||||
|
|
||||||
|
namespace EllieBot.Services;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Basic interface used for classes implementing strings loading mechanism
|
||||||
|
/// </summary>
|
||||||
|
public interface IStringsSource
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Gets all response strings
|
||||||
|
/// </summary>
|
||||||
|
/// <returns>Dictionary(localename, Dictionary(key, response))</returns>
|
||||||
|
Dictionary<string, Dictionary<string, string>> GetResponseStrings();
|
||||||
|
|
||||||
|
Dictionary<string, Dictionary<string, CommandStrings>> GetCommandStrings();
|
||||||
|
}
|
13
src/EllieBot/_common/Abstractions/strings/LocStr.cs
Normal file
13
src/EllieBot/_common/Abstractions/strings/LocStr.cs
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
namespace EllieBot;
|
||||||
|
|
||||||
|
public readonly struct LocStr
|
||||||
|
{
|
||||||
|
public readonly string Key;
|
||||||
|
public readonly object[] Params;
|
||||||
|
|
||||||
|
public LocStr(string key, params object[] data)
|
||||||
|
{
|
||||||
|
Key = key;
|
||||||
|
Params = data;
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue