From 80b3263290ba6f45e98f83f2f7267d95570acbcb Mon Sep 17 00:00:00 2001
From: Lala Sabathil <aiko@aitsys.dev>
Date: Thu, 9 Nov 2023 00:59:56 +0100
Subject: [PATCH] feat: personality generation

---
 .../DisCatSharpExtensions.cs                  | 11 +++++++++
 .../DiscordApiClientHook.cs                   | 24 ++++++++++++++++++-
 .../Payloads/ClydeSettingsPayloads.cs         | 12 ++++++++++
 DisCatSharp/Net/Rest/Endpoints.cs             |  1 +
 4 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/DisCatSharp.Experimental/DisCatSharpExtensions.cs b/DisCatSharp.Experimental/DisCatSharpExtensions.cs
index bc1067dadc..6f42bd2dec 100644
--- a/DisCatSharp.Experimental/DisCatSharpExtensions.cs
+++ b/DisCatSharp.Experimental/DisCatSharpExtensions.cs
@@ -72,4 +72,15 @@ public static async Task<ClydeSettings> ModifyClydeSettingsAsync(
 				? Optional.FromNullable<List<int>?>(null)
 				: Optional.None);
 	}
+
+	/// <summary>
+	/// Generates a basePersonality for clyde based on the given <paramref name="basePersonality"/>.
+	/// </summary>
+	/// <param name="client">The discord client.</param>
+	/// <param name="basePersonality">The base base personality to generate a new one from.</param>
+	public static async Task<string> GenerateClydePersonalityAsync(this DiscordClient client, string? basePersonality = null)
+	{
+		DiscordApiClientHook hook = new(client.ApiClient);
+		return await hook.GenerateClydePersonalityAsync(basePersonality);
+	}
 }
diff --git a/DisCatSharp.Experimental/DiscordApiClientHook.cs b/DisCatSharp.Experimental/DiscordApiClientHook.cs
index 666a0c2af5..6d7964ba84 100644
--- a/DisCatSharp.Experimental/DiscordApiClientHook.cs
+++ b/DisCatSharp.Experimental/DiscordApiClientHook.cs
@@ -101,7 +101,7 @@ internal async Task<ClydeSettings> ModifyClydeSettingsAsync(ulong guildId, ulong
 	/// </summary>
 	/// <param name="guildId">The guild id to modify clyde's settings for.</param>
 	/// <param name="name">The new name.</param>
-	/// <param name="personality">The new personality.</param>
+	/// <param name="personality">The new basePersonality.</param>
 	/// <param name="avatarBase64">The new avatar.</param>
 	/// <param name="bannerBase64">The new banner.</param>
 	/// <param name="themeColors">The new theme colors.</param>
@@ -137,4 +137,26 @@ internal async Task<ClydeSettings> ModifyClydeSettingsAsync(
 		var settings = DiscordJson.DeserializeObject<ClydeSettings>(settingsString, this.ApiClient.Discord);
 		return settings;
 	}
+
+	/// <summary>
+	/// Generates a basePersonality for clyde based on the given <paramref name="basePersonality"/>.
+	/// </summary>
+	/// <param name="basePersonality">The base base personality to generate a new one from.</param>
+	internal async Task<string> GenerateClydePersonalityAsync(string? basePersonality = null)
+	{
+		PersonalityGenerationPayload pld = new()
+		{
+			Personality = basePersonality ?? string.Empty
+		};
+
+		var route = $"{Endpoints.CLYDE_PROFILES}{Endpoints.GENERATE_PERSONALITY}";
+		var bucket = this.ApiClient.Rest.GetBucket(RestRequestMethod.POST, route, new
+			{ }, out var path);
+
+		var url = Utilities.GetApiUriFor(path, this.ApiClient.Discord.Configuration);
+		var res = await this.ApiClient.DoRequestAsync(this.ApiClient.Discord, bucket, url, RestRequestMethod.POST, route, payload: DiscordJson.SerializeObject(pld)).ConfigureAwait(false);
+
+		var generatedPersonality = DiscordJson.DeserializeObject<PersonalityGenerationPayload>(res.Response, this.ApiClient.Discord);
+		return generatedPersonality.Personality;
+	}
 }
diff --git a/DisCatSharp.Experimental/Payloads/ClydeSettingsPayloads.cs b/DisCatSharp.Experimental/Payloads/ClydeSettingsPayloads.cs
index 75c6bff856..29cda0cfa4 100644
--- a/DisCatSharp.Experimental/Payloads/ClydeSettingsPayloads.cs
+++ b/DisCatSharp.Experimental/Payloads/ClydeSettingsPayloads.cs
@@ -69,3 +69,15 @@ internal sealed class ClydeSettingsProfileUpdatePayload : ObservableApiObject
 	[JsonProperty("theme_colors", NullValueHandling = NullValueHandling.Include)]
 	internal Optional<List<int>?> ThemeColors { get; set; }
 }
+
+/// <summary>
+/// Represents a personality generation payload.
+/// </summary>
+internal sealed class PersonalityGenerationPayload : ObservableApiObject
+{
+	/// <summary>
+	/// Sets the base personality to generate a new one from.
+	/// </summary>
+	[JsonProperty("personality", NullValueHandling = NullValueHandling.Ignore)]
+	internal string Personality { get; set; } = string.Empty;
+}
diff --git a/DisCatSharp/Net/Rest/Endpoints.cs b/DisCatSharp/Net/Rest/Endpoints.cs
index cfa3c96e5d..1c9d548337 100644
--- a/DisCatSharp/Net/Rest/Endpoints.cs
+++ b/DisCatSharp/Net/Rest/Endpoints.cs
@@ -555,4 +555,5 @@ public static class Endpoints
 	public const string INVENTORY = "/inventory";
 	public const string CLYDE_SETTINGS = "/clyde-settings";
 	public const string CLYDE_PROFILES = "/clyde-profiles";
+	public const string GENERATE_PERSONALITY = "/generate-personality";
 }