Add lift-quarantine subcommand to agents CLI

Introduced a new lift-quarantine subcommand for the agents command, enabling users to lift quarantine mode for an AI agent by agent ID. Updated Vectra.Client.Net to v1.1.0 to support this feature. Added corresponding unit tests and updated command creation tests to cover the new subcommand.

#45

Author: ziagham

src/VectraCtl/Commands/AgentsCommand.cs

@@ -15,6 +15,7 @@ public static Command Create(IServiceProvider serviceProvider)
         command.Subcommands.Add(CreateListCommand(serviceProvider));
         command.Subcommands.Add(CreateRegisterCommand(serviceProvider));
         command.Subcommands.Add(CreateAssignPolicyCommand(serviceProvider));
+        command.Subcommands.Add(CreateLiftQuarantineCommand(serviceProvider));
         command.Subcommands.Add(CreateDeleteCommand(serviceProvider));
 
         return command;
@@ -45,6 +46,25 @@ private static Command CreateListCommand(IServiceProvider serviceProvider)
         return cmd;
     }
 
+    private static Command CreateLiftQuarantineCommand(IServiceProvider serviceProvider)
+    {
+        var agentIdOption = new Option<Guid>("--agent-id") { Description = "Agent ID (GUID)", Required = true };
+
+        var cmd = new Command("lift-quarantine", "Lift quarantine mode for an AI agent")
+        {
+            agentIdOption
+        };
+
+        cmd.SetAction((parseResult, ct) => CommandHelpers.ExecuteAsync(serviceProvider, async (logger, sp) =>
+        {
+            var client = sp.GetRequiredService<IVectraClient>();
+            await client.Agents.LiftQuarantineAsync(parseResult.GetValue(agentIdOption)!, ct);
+            logger.Write("Quarantine lifted successfully.");
+        }));
+
+        return cmd;
+    }
+
     private static Command CreateRegisterCommand(IServiceProvider serviceProvider)
     {
         var nameOption = new Option<string>("--name") { Description = "Agent name", Required = true };

src/VectraCtl/VectraCtl.csproj

@@ -30,7 +30,7 @@
   <ItemGroup>
     <PackageReference Include="Spectre.Console" Version="0.55.2" />
     <PackageReference Include="System.CommandLine" Version="2.0.8" />
-    <PackageReference Include="Vectra.Client.Net" Version="1.0.0" />
+    <PackageReference Include="Vectra.Client.Net" Version="1.1.0" />
     <PackageReference Include="YamlDotNet" Version="17.1.0" />
   </ItemGroup>
 

tests/VectraCtl.UnitTests/Commands/ApiCommandActionTests.cs

@@ -187,6 +187,44 @@ public async Task AgentsCommand_Delete_Exception_WritesError()
         provider.GetRequiredService<IVectraCtlLogger>().Received().WriteError(Arg.Any<string>());
     }
 
+    [Fact]
+    public async Task AgentsCommand_LiftQuarantine_Success_WritesSuccess()
+    {
+        var provider = BuildProvider(s =>
+        {
+            var client = Substitute.For<IVectraClient>();
+            var agentClient = Substitute.For<IVectraAgentClient>();
+            agentClient.LiftQuarantineAsync(Arg.Any<Guid>(), Arg.Any<CancellationToken>())
+                       .Returns(Task.CompletedTask);
+            client.Agents.Returns(agentClient);
+            s.AddSingleton(client);
+        });
+
+        var cmd = AgentsCommand.Create(provider);
+        await InvokeAsync(cmd, ["lift-quarantine", "--agent-id", Guid.NewGuid().ToString()]);
+
+        provider.GetRequiredService<IVectraCtlLogger>().Received().Write(Arg.Is<string>(s => s.Contains("Quarantine lifted")));
+    }
+
+    [Fact]
+    public async Task AgentsCommand_LiftQuarantine_Exception_WritesError()
+    {
+        var provider = BuildProvider(s =>
+        {
+            var client = Substitute.For<IVectraClient>();
+            var agentClient = Substitute.For<IVectraAgentClient>();
+            agentClient.LiftQuarantineAsync(Arg.Any<Guid>(), Arg.Any<CancellationToken>())
+                       .Throws(new Exception("bad request"));
+            client.Agents.Returns(agentClient);
+            s.AddSingleton(client);
+        });
+
+        var cmd = AgentsCommand.Create(provider);
+        await InvokeAsync(cmd, ["lift-quarantine", "--agent-id", Guid.NewGuid().ToString()]);
+
+        provider.GetRequiredService<IVectraCtlLogger>().Received().WriteError(Arg.Any<string>());
+    }
+
     // ── HitlCommand ──────────────────────────────────────────────────────────
 
     [Fact]

tests/VectraCtl.UnitTests/Commands/CommandCreationTests.cs

@@ -69,6 +69,7 @@ public void AgentsCommand_Create_ReturnsCommandWithCorrectNameAndSubcommands()
         cmd.Subcommands.Should().Contain(s => s.Name == "list");
         cmd.Subcommands.Should().Contain(s => s.Name == "register");
         cmd.Subcommands.Should().Contain(s => s.Name == "assign-policy");
+        cmd.Subcommands.Should().Contain(s => s.Name == "lift-quarantine");
         cmd.Subcommands.Should().Contain(s => s.Name == "delete");
     }