Added DeclareQueue and RemoveQueue commands to Tapeti.Cmd

Tapeti.Cmd/Program.cs

@@ -2,9 +2,11 @@
 using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO;
+using System.Linq;
 using System.Text;
 using CommandLine;
 using RabbitMQ.Client;
+using RabbitMQ.Client.Exceptions;
 using Tapeti.Cmd.Commands;
 using Tapeti.Cmd.Mock;
 using Tapeti.Cmd.RateLimiter;
@@ -85,6 +87,12 @@ namespace Tapeti.Cmd
         }
 
 
+        [Verb("example", HelpText = "Output an example SingleFileJSON formatted message.")]
+        public class ExampleOptions
+        {
+        }
+
+
         [Verb("shovel", HelpText = "Reads messages from a queue and publishes them to another queue, optionally to another RabbitMQ server.")]
         public class ShovelOptions : CommonOptions
         {
@@ -131,22 +139,43 @@ namespace Tapeti.Cmd
         }
 
 
-        [Verb("example", HelpText = "Output an example SingleFileJSON formatted message.")]
+        [Verb("declarequeue", HelpText = "Declares a durable queue without arguments, compatible with Tapeti.")]
-        public class ExampleOptions
+        public class DeclareQueueOptions : CommonOptions
         {
+            [Option('q', "queue", Required = true, HelpText = "The name of the queue to declare.")]
+            public string QueueName { get; set; }
+            
+            [Option('b', "bindings", Required = false, HelpText = "One or more bindings to add to the queue. Format: <exchange>:<routingKey>")]
+            public IEnumerable<string> Bindings { get; set; }
+        }
+
+
+        [Verb("removequeue", HelpText = "Removes a durable queue.")]
+        public class RemoveQueueOptions : CommonOptions
+        {
+            [Option('q', "queue", Required = true, HelpText = "The name of the queue to declare.")]
+            public string QueueName { get; set; }
+            
+            [Option("confirm", HelpText = "Confirms the removal of the specified queue. If not provided, an interactive prompt will ask for confirmation.", Default = false)]
+            public bool Confirm { get; set; }
+            
+            [Option("confirmpurge", HelpText = "Confirms the removal of the specified queue even if there still are messages in the queue. If not provided, an interactive prompt will ask for confirmation.", Default = false)]
+            public bool ConfirmPurge { get; set; }
         }
 
 
 
         public static int Main(string[] args)
         {
-            return Parser.Default.ParseArguments<ExportOptions, ImportOptions, ShovelOptions, PurgeOptions, ExampleOptions>(args)
+            return Parser.Default.ParseArguments<ExportOptions, ImportOptions, ShovelOptions, PurgeOptions, ExampleOptions, DeclareQueueOptions, RemoveQueueOptions>(args)
                 .MapResult(
                     (ExportOptions o) => ExecuteVerb(o, RunExport),
                     (ImportOptions o) => ExecuteVerb(o, RunImport),
+                    (ExampleOptions o) => ExecuteVerb(o, RunExample),
                     (ShovelOptions o) => ExecuteVerb(o, RunShovel),
                     (PurgeOptions o) => ExecuteVerb(o, RunPurge),
-                    (ExampleOptions o) => ExecuteVerb(o, RunExample),
+                    (DeclareQueueOptions o) => ExecuteVerb(o, RunDeclareQueue),
+                    (RemoveQueueOptions o) => ExecuteVerb(o, RunRemoveQueue),
                     errs =>
                     {
                         if (!Debugger.IsAttached) 
@@ -305,6 +334,33 @@ namespace Tapeti.Cmd
         }
 
 
+        private static void RunExample(ExampleOptions options)
+        {
+            using (var messageSerializer = new SingleFileJSONMessageSerializer(Console.OpenStandardOutput(), false, new UTF8Encoding(false)))
+            {
+                messageSerializer.Serialize(new Message
+                {
+                    Exchange = "example",
+                    Queue = "example.queue",
+                    RoutingKey = "example.routing.key",
+                    DeliveryTag = 42,
+                    Properties = new MockBasicProperties
+                    {
+                        ContentType = "application/json",
+                        DeliveryMode = 2,
+                        Headers = new Dictionary<string, object>
+                        {
+                            { "classType", Encoding.UTF8.GetBytes("Tapeti.Cmd.Example:Tapeti.Cmd") }
+                        },
+                        ReplyTo = "reply.queue",
+                        Timestamp = new AmqpTimestamp(new DateTimeOffset(DateTime.UtcNow).ToUnixTimeSeconds())
+                    },
+                    Body = Encoding.UTF8.GetBytes("{ \"Hello\": \"world!\" }")
+                });
+            }
+        }
+
+        
         private static void RunShovel(ShovelOptions options)
         {
             int messageCount;
@@ -403,34 +459,83 @@ namespace Tapeti.Cmd
             }
 
             Console.WriteLine($"{messageCount} message{(messageCount != 1 ? "s" : "")} purged from '{options.QueueName}'.");
-
         }
 
 
-        private static void RunExample(ExampleOptions options)
+        private static void RunDeclareQueue(DeclareQueueOptions options)
         {
-            using (var messageSerializer = new SingleFileJSONMessageSerializer(Console.OpenStandardOutput(), false, new UTF8Encoding(false)))
+            // Parse early to fail early
-            {
+            var bindings = options.Bindings
-                messageSerializer.Serialize(new Message
+                .Select(b =>
                 {
-                    Exchange = "example",
+                    var parts = b.Split(':');
-                    Queue = "example.queue",
+                    if (parts.Length != 2)
-                    RoutingKey = "example.routing.key",
+                        throw new InvalidOperationException($"Invalid binding format: {b}");
-                    DeliveryTag = 42,
+
-                    Properties = new MockBasicProperties
+                    return new Tuple<string, string>(parts[0], parts[1]);
-                    {
+                })
-                        ContentType = "application/json",
+                .ToArray();
-                        DeliveryMode = 2,
+            
-                        Headers = new Dictionary<string, object>
+            using (var connection = GetConnection(options))
-                        {
+            using (var channel = connection.CreateModel())
-                            { "classType", Encoding.UTF8.GetBytes("Tapeti.Cmd.Example:Tapeti.Cmd") }
+            {
-                        },
+                channel.QueueDeclare(options.QueueName, true, false, false);
-                        ReplyTo = "reply.queue",
+
-                        Timestamp = new AmqpTimestamp(new DateTimeOffset(DateTime.UtcNow).ToUnixTimeSeconds())
+                foreach (var (exchange, routingKey) in bindings)
-                    },
+                    channel.QueueBind(options.QueueName, exchange, routingKey);
-                    Body = Encoding.UTF8.GetBytes("{ \"Hello\": \"world!\" }")
-                });
             }
+
+            Console.WriteLine($"Queue {options.QueueName} declared with {bindings.Length} binding{(bindings.Length != 1 ? "s" : "")}.");
+        }
+
+
+        private static void RunRemoveQueue(RemoveQueueOptions options)
+        {
+            if (!options.Confirm)
+            {
+                Console.Write($"Do you want to remove the queue '{options.QueueName}'? (Y/N) ");
+                var answer = Console.ReadLine();
+
+                if (string.IsNullOrEmpty(answer) || !answer.Equals("Y", StringComparison.CurrentCultureIgnoreCase))
+                    return;
+            }
+
+            uint messageCount;
+
+            try
+            {
+                using (var connection = GetConnection(options))
+                using (var channel = connection.CreateModel())
+                {
+                    messageCount = channel.QueueDelete(options.QueueName, true, true);
+                }
+            }
+            catch (OperationInterruptedException e)
+            {
+                if (e.ShutdownReason.ReplyCode == 406)
+                {
+                    if (!options.ConfirmPurge)
+                    {
+                        Console.Write($"There are messages remaining. Do you want to purge the queue '{options.QueueName}'? (Y/N) ");
+                        var answer = Console.ReadLine();
+
+                        if (string.IsNullOrEmpty(answer) || !answer.Equals("Y", StringComparison.CurrentCultureIgnoreCase))
+                            return;
+                    }
+
+                    using (var connection = GetConnection(options))
+                    using (var channel = connection.CreateModel())
+                    {
+                        messageCount = channel.QueueDelete(options.QueueName, true, false);
+                    }
+                }
+                else
+                    throw;
+            }
+
+            Console.WriteLine(messageCount == 0 
+                ? $"Empty or non-existent queue '{options.QueueName}' removed." 
+                : $"{messageCount} message{(messageCount != 1 ? "s" : "")} purged while removing '{options.QueueName}'.");
         }
     }
 }