discord-bot/Tests/ZalgoTests.cs

using System;
using System.Globalization;
using System.IO;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using CompatBot.Database;
using CompatBot.EventHandlers;
using NUnit.Framework;
using File = System.IO.File;

namespace Tests;

[TestFixture]
public class ZalgoTests
{
    [OneTimeSetUp]
    public async Task SetupAsync()
    {
        var result = await DbImporter.UpgradeAsync(CancellationToken.None).ConfigureAwait(false);
        Assert.That(result, Is.True);
    }
        
    [Test, Explicit("Requires external data")]
    public async Task ZalgoAuditTestAsync()
    {
        var samplePath = @"C:/Users/13xforever/Downloads/names.txt";
        var resultPath = Path.Combine(Path.GetDirectoryName(samplePath), "zalgo.txt");

        var names = await File.ReadAllLinesAsync(samplePath, Encoding.UTF8);
        await using var r = File.Open(resultPath, FileMode.Create, FileAccess.Write, FileShare.Read);
        await using var w = new StreamWriter(r, new UTF8Encoding(false));
        foreach (var line in names)
        {
            var user = UserInfo.Parse(line);
            var isZalgo = UsernameZalgoMonitor.NeedsRename(user.DisplayName);
            if (isZalgo)
                await w.WriteLineAsync(user.DisplayName).ConfigureAwait(false);
        }
    }

    [Test, Explicit("Requires external data")]
    public async Task RoleSortTestAsync()
    {
        var samplePath = @"C:/Users/13xforever/Downloads/names.txt";
        var resultPath = Path.Combine(Path.GetDirectoryName(samplePath), "role_count.txt");

        var stats = new int[10];
        var names = await File.ReadAllLinesAsync(samplePath, Encoding.UTF8);
        await using var r = File.Open(resultPath, FileMode.Create, FileAccess.Write, FileShare.Read);
        await using var w = new StreamWriter(r, new UTF8Encoding(false));
        foreach (var line in names)
        {
            var user = UserInfo.Parse(line);
            var roleCount = user.Roles?.Length ?? 0;
            stats[roleCount]++;
            w.Write(roleCount);
            await w.WriteAsync('\t').ConfigureAwait(false);
            await w.WriteLineAsync(user.DisplayName).ConfigureAwait(false);
        }

        for (var i = 0; i < stats.Length && stats[i] > 0; i++)
        {
            Console.WriteLine($"{i:#0} roles: {stats[i]} members");
        }
    }

    [TestCase("ᵇᶦᵒˢʰᵒᶜᵏ96", false)]
    [TestCase("GodPan กับยูนิตแขนที่หายไป", false)]
    [TestCase("⛧Bζ͜͡annerBomb⛧", false)]
    [TestCase("(_A_Y_A_Z_)  (͡๏̯͡๏)", false)]
    [TestCase("🥛🥛", false)]
    [TestCase("🎮P̷͙͋a̵̛̳k̶̫̀o̸̿͜ỏ̸̝🎮", true)]
    [TestCase("Cindellด้้้", true)]
    [TestCase("󠂪󠂪󠂪󠂪 󠂪󠂪󠂪󠂪󠂪󠂪󠂪󠂪 󠂪󠂪󠂪", true)]
    [TestCase("󠀀󠀀", true)]
    [TestCase("꧁꧂🥴🥴🥴HOJU🥴🥴🥴╲⎝⧹", true)]
    [TestCase("", true)]
    [TestCase("᲼᲼᲼", true, "Reserved block")]
    [TestCase("𒁃𒃋𒋬𒑭𒕃", true, "Cuneiform block")]
    [TestCase("𐎄𐎝𐎪𐎼", true, "Ugaritic and Old Persian blocks")]
    [TestCase("͔", true, "Combining marks")]
    public void ZalgoDetectionTest(string name, bool isBad, string comment = null)
    {
        Assert.That(UsernameZalgoMonitor.NeedsRename(name), Is.EqualTo(isBad), comment);
    }
}

internal class UserInfo
{
    public string Username { get; private set; }
    public string Nickname { get; private set; }
    public DateTime JoinDate { get; private set; }
    public string[] Roles { get; private set; }

    public string DisplayName => string.IsNullOrEmpty(Nickname) ? Username : Nickname;

    public static UserInfo Parse(string line)
    {
        var parts = line.Split('\t');
        if (parts.Length != 4)
            throw new FormatException("Invalid user info line: " + line);

        return new()
        {
            Username = parts[0],
            Nickname = parts[1],
            JoinDate = DateTime.Parse(parts[2], CultureInfo.InvariantCulture),
            Roles = parts[3]?.Split(',', StringSplitOptions.RemoveEmptyEntries) ?? Array.Empty<string>(),
        };
    }
}