codex-storage
/
cs-codex-dist-tests
mirror of https://github.com/codex-storage/cs-codex-dist-tests.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Correct chunking of long messages

This commit is contained in:
benbierens 2023-11-09 10:13:34 +01:00
parent 180bf3fd35
commit ef83441b2f
No known key found for this signature in database
GPG Key ID: FE44815D96D0A1AA
4 changed files with 169 additions and 73 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Tools/BiblioTech/BaseCommand.cs
View File

@ -28,8 +28,16 @@ namespace BiblioTech
await Invoke(context);
}
catch (Exception ex)
{
if (IsInAdminChannel(command))
{
var msg = "Failed with exception: " + ex;
await command.FollowupAsync(msg.Substring(0, Math.Min(1900, msg.Length)));
}
else
{
await command.FollowupAsync("Something failed while trying to do that...", ephemeral: true);
}
Console.WriteLine(ex);
}
}

66
Tools/BiblioTech/Commands/AdminCommand.cs
View File

@ -109,7 +109,7 @@ namespace BiblioTech.Commands
return;
}
var report = string.Join(Environment.NewLine, Program.UserRepo.GetInteractionReport(user));
var report = Program.UserRepo.GetInteractionReport(user);
await context.Followup(report);
}
}
@ -134,7 +134,7 @@ namespace BiblioTech.Commands
}
var nl = Environment.NewLine;
await context.Followup($"Deployments:{nl}{string.Join(nl, deployments.Select(FormatDeployment))}");
await context.Followup(deployments.Select(FormatDeployment).ToArray());
}
private string FormatDeployment(CodexDeployment deployment)
@ -249,23 +249,31 @@ namespace BiblioTech.Commands
this.ci = ci;
}
protected async Task OnDeployment(CommandContext context, Func<ICodexNodeGroup, string, Task> action)
protected async Task<T?> OnDeployment<T>(CommandContext context, Func<ICodexNodeGroup, string, T> action)
{
var deployment = Program.DeploymentFilesMonitor.GetDeployments().SingleOrDefault();
if (deployment == null)
{
await context.Followup("No deployment found.");
return;
return default;
}
try
{
var group = ci.WrapCodexContainers(deployment.CodexInstances.Select(i => i.Containers).ToArray());
await action(group, deployment.Metadata.Name);
var result = action(group, deployment.Metadata.Name);
return result;
}
catch (Exception ex)
{
await context.Followup("Failed to wrap nodes with exception: " + ex);
var message = new[]
{
"Failed to wrap nodes with exception: "
};
var exceptionMessage = ex.ToString().Split(Environment.NewLine);
await context.Followup(message.Concat(exceptionMessage).ToArray());
return default;
}
}
}
@ -280,13 +288,19 @@ namespace BiblioTech.Commands
protected override async Task onSubCommand(CommandContext context)
{
await OnDeployment(context, async (group, name) =>
var report = await OnDeployment(context, CreateNetInfoReport);
if (report != null && report.Any())
{
await context.Followup(report);
}
}
private string[] CreateNetInfoReport(ICodexNodeGroup group, string name)
{
var nl = Environment.NewLine;
var content = new List<string>
{
$"{DateTime.UtcNow.ToString("o")} - {group.Count()} Codex nodes.",
$"Deployment name: '{name}'"
$"Deployment name: '{name}'."
};
foreach (var node in group)
@ -295,8 +309,11 @@ namespace BiblioTech.Commands
{
var info = node.GetDebugInfo();
var json = JsonConvert.SerializeObject(info, Formatting.Indented);
var jsonInsert = $"{nl}```{nl}{json}{nl}```{nl}";
content.Add($"Node '{node.GetName()}' responded with {jsonInsert}");
var jsonLines = json.Split(Environment.NewLine);
content.Add($"Node '{node.GetName()}' responded with:");
content.Add("---");
content.AddRange(jsonLines);
content.Add("---");
}
catch (Exception ex)
{
@ -304,8 +321,7 @@ namespace BiblioTech.Commands
}
}
await context.Followup(string.Join(nl, content));
});
return content.ToArray();
}
}
@ -326,24 +342,33 @@ namespace BiblioTech.Commands
var peerId = await peerIdOption.Parse(context);
if (string.IsNullOrEmpty(peerId)) return;
await OnDeployment(context, async (group, name) =>
var report = await OnDeployment(context, (group, name) => CreateDebugPeerReport(group, name, peerId));
if (report != null && report.Any())
{
await context.Followup(report);
}
}
private string[] CreateDebugPeerReport(ICodexNodeGroup group, string name, string peerId)
{
var nl = Environment.NewLine;
var content = new List<string>
{
$"{DateTime.UtcNow.ToString("o")} - {group.Count()} Codex nodes.",
$"Deployment name: '{name}'"
$"Deployment name: '{name}'.",
$"Calling debug/peer for '{peerId}'."
};
content.Add($"Calling debug/peer for '{peerId}' on {group.Count()} Codex nodes.");
foreach (var node in group)
{
try
{
var info = node.GetDebugPeer(peerId);
var json = JsonConvert.SerializeObject(info, Formatting.Indented);
var jsonInsert = $"{nl}```{nl}{json}{nl}```{nl}";
content.Add($"Node '{node.GetName()}' responded with {jsonInsert}");
var jsonLines = json.Split(Environment.NewLine);
content.Add($"Node '{node.GetName()}' responded with:");
content.Add("---");
content.AddRange(jsonLines);
content.Add("---");
}
catch (Exception ex)
{
@ -351,8 +376,7 @@ namespace BiblioTech.Commands
}
}
await context.Followup(string.Join(nl, content));
});
return content.ToArray();
}
}
}

87
Tools/BiblioTech/Options/CommandContext.cs
View File

@ -4,36 +4,99 @@ namespace BiblioTech.Options
{
public class CommandContext
{
private const string AttachmentFolder = "attachments";
public CommandContext(SocketSlashCommand command, IReadOnlyCollection<SocketSlashCommandDataOption> options)
{
Command = command;
Options = options;
var attachmentPath = Path.Combine(Program.Config.DataPath, AttachmentFolder);
if (Directory.Exists(attachmentPath))
{
Directory.CreateDirectory(attachmentPath);
}
}
public SocketSlashCommand Command { get; }
public IReadOnlyCollection<SocketSlashCommandDataOption> Options { get; }
public async Task Followup(string message)
public async Task Followup(string line)
{
if (message.Length > 1900)
{
await FollowupWithAttachement("codexdiscordbot_file.txt", message);
return;
var array = line.Split(Environment.NewLine);
await Followup(array);
}
public async Task Followup(string[] lines)
{
var chunker = new LineChunker(lines);
var chunks = chunker.GetChunks();
if (!chunks.Any()) return;
// First chunk is a modification of the original message.
// Everything after that, we must create a new message.
var first = chunks.First();
chunks.RemoveAt(0);
await Command.ModifyOriginalResponseAsync(m =>
{
m.Content = message;
m.Content = FormatChunk(first);
});
}
private async Task FollowupWithAttachement(string filename, string content)
foreach (var remaining in chunks)
{
using var fileStream = new MemoryStream();
using var streamWriter = new StreamWriter(fileStream);
await streamWriter.WriteAsync(content);
await Command.FollowupAsync(FormatChunk(remaining));
}
}
await Command.FollowupWithFileAsync(fileStream, filename);
private string FormatChunk(string[] chunk)
{
return string.Join(Environment.NewLine, chunk);
}
}
public class LineChunker
{
private readonly List<string> input;
private readonly int maxCharacters;
public LineChunker(string[] input, int maxCharacters = 1950)
{
this.input = input.ToList();
this.maxCharacters = maxCharacters;
}
public List<string[]> GetChunks()
{
var result = new List<string[]>();
while (input.Any())
{
result.Add(GetChunk());
}
return result;
}
private string[] GetChunk()
{
var totalLength = 0;
var result = new List<string>();
while (input.Any())
{
var nextLine = input[0];
var nextLength = totalLength + nextLine.Length;
if (nextLength > maxCharacters)
{
return result.ToArray();
}
input.RemoveAt(0);
result.Add(nextLine);
totalLength += nextLine.Length;
}
return result.ToArray();
}
}
}

23
Tools/BiblioTech/UserRepo.cs
View File

@ -76,17 +76,17 @@ namespace BiblioTech
return result.ToArray();
}
public string GetUserReport(IUser user)
public string[] GetUserReport(IUser user)
{
var userData = GetUserData(user);
if (userData == null) return "User has not joined the test net.";
if (userData == null) return new[] { "User has not joined the test net." };
return userData.CreateOverview();
}
public string GetUserReport(EthAddress ethAddress)
public string[] GetUserReport(EthAddress ethAddress)
{
var userData = GetUserDataForAddress(ethAddress);
if (userData == null) return "No user is using this eth address.";
if (userData == null) return new[] { "No user is using this eth address." };
return userData.CreateOverview();
}
@ -196,14 +196,15 @@ namespace BiblioTech
public List<UserAssociateAddressEvent> AssociateEvents { get; }
public List<UserMintEvent> MintEvents { get; }
public string CreateOverview()
public string[] CreateOverview()
{
var nl = Environment.NewLine;
return
$"name: '{Name}' - id:{DiscordId}{nl}" +
$"joined: {CreatedUtc.ToString("o")}{nl}" +
$"current address: {CurrentAddress}{nl}" +
$"{AssociateEvents.Count + MintEvents.Count} total bot events.";
return new[]
{
$"name: '{Name}' - id:{DiscordId}",
$"joined: {CreatedUtc.ToString("o")}",
$"current address: {CurrentAddress}",
$"{AssociateEvents.Count + MintEvents.Count} total bot events."
};
}
}