Discord-History-Tracker/app/Desktop/Main/Pages/DebugPageModel.cs

#if DEBUG
using System;
using System.Collections.Immutable;
using System.Linq;
using System.Threading.Tasks;
using Avalonia.Controls;
using DHT.Desktop.Dialogs.Message;
using DHT.Desktop.Dialogs.Progress;
using DHT.Server;
using DHT.Server.Data;
using DHT.Server.Service;

namespace DHT.Desktop.Main.Pages;

sealed class DebugPageModel {
	public string GenerateChannels { get; set; } = "0";
	public string GenerateUsers { get; set; } = "0";
	public string GenerateMessages { get; set; } = "0";

	private readonly Window window;
	private readonly State state;

	[Obsolete("Designer")]
	public DebugPageModel() : this(null!, State.Dummy) {}

	public DebugPageModel(Window window, State state) {
		this.window = window;
		this.state = state;
	}

	public async void OnClickAddRandomDataToDatabase() {
		if (!int.TryParse(GenerateChannels, out int channels) || channels < 1) {
			await Dialog.ShowOk(window, "Generate Random Data", "Amount of channels must be at least 1!");
			return;
		}

		if (!int.TryParse(GenerateUsers, out int users) || users < 1) {
			await Dialog.ShowOk(window, "Generate Random Data", "Amount of users must be at least 1!");
			return;
		}

		if (!int.TryParse(GenerateMessages, out int messages) || messages < 1) {
			await Dialog.ShowOk(window, "Generate Random Data", "Amount of messages must be at least 1!");
			return;
		}

		await ProgressDialog.Show(window, "Generating Random Data", async (_, callback) => await GenerateRandomData(channels, users, messages, callback));
	}

	private const int BatchSize = 500;

	private async Task GenerateRandomData(int channelCount, int userCount, int messageCount, IProgressCallback callback) {
		int batchCount = (messageCount + BatchSize - 1) / BatchSize;
		await callback.Update("Adding messages in batches of " + BatchSize, 0, batchCount);

		var rand = new Random();
		var server = new DHT.Server.Data.Server {
			Id = RandomId(rand),
			Name = RandomName("s"),
			Type = ServerType.Server,
		};

		var channels = Enumerable.Range(0, channelCount).Select(i => new Channel {
			Id = RandomId(rand),
			Server = server.Id,
			Name = RandomName("c"),
			ParentId = null,
			Position = i,
			Topic = RandomText(rand, 10),
			Nsfw = rand.Next(4) == 0,
		}).ToArray();

		var users = Enumerable.Range(0, userCount).Select(_ => new User {
			Id = RandomId(rand),
			Name = RandomName("u"),
			DisplayName = RandomName("u"),
			AvatarUrl = null,
			Discriminator = rand.Next(0, 9999).ToString(),
		}).ToArray();

		await state.Db.Users.Add(users);
		await state.Db.Servers.Add([server]);
		await state.Db.Channels.Add(channels);

		var now = DateTimeOffset.Now;
		int batchIndex = 0;

		while (messageCount > 0) {
			int hourOffset = batchIndex;

			var messages = Enumerable.Range(0, Math.Min(messageCount, BatchSize)).Select(i => {
				DateTimeOffset time = now.AddHours(hourOffset).AddMinutes(i * 60.0 / BatchSize);
				DateTimeOffset? edit = rand.Next(100) == 0 ? time.AddSeconds(rand.Next(1, 60)) : null;

				var timeMillis = time.ToUnixTimeMilliseconds();
				var editMillis = edit?.ToUnixTimeMilliseconds();

				return new Message {
					Id = (ulong) timeMillis,
					Sender = RandomBiasedIndex(rand, users).Id,
					Channel = RandomBiasedIndex(rand, channels).Id,
					Text = RandomText(rand, 100),
					Timestamp = timeMillis,
					EditTimestamp = editMillis,
					RepliedToId = null,
					Attachments = ImmutableList<Attachment>.Empty,
					Embeds = ImmutableList<Embed>.Empty,
					Reactions = ImmutableList<Reaction>.Empty,
				};
			}).ToArray();

			await state.Db.Messages.Add(messages);

			messageCount -= BatchSize;
			await callback.Update("Adding messages in batches of " + BatchSize, ++batchIndex, batchCount);
		}
	}

	private static ulong RandomId(Random rand) {
		ulong h = unchecked((ulong) rand.Next());
		ulong l = unchecked((ulong) rand.Next());
		return (h << 32) | l;
	}

	private static string RandomName(string prefix) {
		return prefix + "-" + ServerUtils.GenerateRandomToken(5);
	}

	private static T RandomBiasedIndex<T>(Random rand, T[] options) {
		return options[(int) Math.Floor(options.Length * rand.NextDouble() * rand.NextDouble())];
	}

	private static readonly string[] RandomWords = [
		"apple", "apricot", "artichoke", "arugula", "asparagus", "avocado",
		"banana", "bean", "beechnut", "beet", "blackberry", "blackcurrant", "blueberry", "boysenberry", "bramble", "broccoli",
		"cabbage", "cacao", "cantaloupe", "caper", "carambola", "carrot", "cauliflower", "celery", "chard", "cherry", "chokeberry", "citron", "clementine", "coconut", "corn", "crabapple", "cranberry", "cucumber", "currant",
		"daikon", "date", "dewberry", "durian",
		"edamame", "eggplant", "elderberry", "endive",
		"fig",
		"garlic", "ginger", "gooseberry", "grape", "grapefruit", "guava",
		"honeysuckle", "horseradish", "huckleberry",
		"jackfruit", "jicama",
		"kale", "kiwi", "kohlrabi", "kumquat",
		"leek", "lemon", "lentil", "lettuce", "lime",
		"mandarin", "mango", "mushroom", "myrtle",
		"nectarine", "nut",
		"olive", "okra", "onion", "orange",
		"papaya", "parsnip", "pawpaw", "peach", "pear", "pea", "pepper", "persimmon", "pineapple", "plum", "plantain", "pomegranate", "pomelo", "potato", "prune", "pumpkin",
		"quandong", "quinoa",
		"radicchio", "radish", "raisin", "raspberry", "redcurrant", "rhubarb", "rutabaga",
		"spinach", "strawberry", "squash",
		"tamarind", "tangerine", "tomatillo", "tomato", "turnip",
		"vanilla",
		"watercress", "watermelon",
		"yam",
		"zucchini"
	];

	private static string RandomText(Random rand, int maxWords) {
		int wordCount = 1 + (int) Math.Floor(maxWords * Math.Pow(rand.NextDouble(), 3));
		return string.Join(' ', Enumerable.Range(0, wordCount).Select(_ => RandomWords[rand.Next(RandomWords.Length)]));
	}
}
#else
namespace DHT.Desktop.Main.Pages;

sealed class DebugPageModel {
	public string GenerateChannels { get; set; } = "0";
	public string GenerateUsers { get; set; } = "0";
	public string GenerateMessages { get; set; } = "0";

	public void OnClickAddRandomDataToDatabase() {}
}
#endif