Fix pr #123: Fix issue #121: Allow resetting base tag

kevinboss · Dec 16, 2024 · 85e551b · 85e551b
1 parent d964a2c
commit 85e551b
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 12 deletions.
diff --git a/src/Commands/Reset/ResetCliCommand.cs b/src/Commands/Reset/ResetCliCommand.cs
@@ -11,30 +11,49 @@ internal class ResetCliCommand : AsyncCommand<ResetSettings>
     private readonly ICreateContainerCommand _createContainerCommand;
     private readonly IRunContainerCommand _runContainerCommand;
     private readonly IContainerNamePrompt _containerNamePrompt;
+    private readonly IImageIdentifierPrompt _imageIdentifierPrompt;
+    private readonly IGetImageQuery _getImageQuery;
+    private readonly ICreateImageCommand _createImageCommand;
     private readonly ListCliCommand _listCliCommand;
 
-    public ResetCliCommand(IGetRunningContainersQuery getRunningContainersQuery,
-        IStopAndRemoveContainerCommand stopAndRemoveContainerCommand, ICreateContainerCommand createContainerCommand,
-        IRunContainerCommand runContainerCommand, IContainerNamePrompt containerNamePrompt,
+    public ResetCliCommand(
+        IGetRunningContainersQuery getRunningContainersQuery,
+        IStopAndRemoveContainerCommand stopAndRemoveContainerCommand,
+        ICreateContainerCommand createContainerCommand,
+        IRunContainerCommand runContainerCommand,
+        IContainerNamePrompt containerNamePrompt,
+        IImageIdentifierPrompt imageIdentifierPrompt,
+        IGetImageQuery getImageQuery,
+        ICreateImageCommand createImageCommand,
         ListCliCommand listCliCommand)
     {
         _getRunningContainersQuery = getRunningContainersQuery;
         _stopAndRemoveContainerCommand = stopAndRemoveContainerCommand;
         _createContainerCommand = createContainerCommand;
         _runContainerCommand = runContainerCommand;
         _containerNamePrompt = containerNamePrompt;
+        _imageIdentifierPrompt = imageIdentifierPrompt;
+        _getImageQuery = getImageQuery;
+        _createImageCommand = createImageCommand;
         _listCliCommand = listCliCommand;
     }
 
     public override async Task<int> ExecuteAsync(CommandContext context, ResetSettings settings)
     {
-        var container = await GetContainerAsync(settings);
-        if (container == null)
+        if (settings.IsImage)
         {
-            throw new InvalidOperationException("No running container found");
+            await ResetImageBaseTagAsync(settings);
         }
+        else
+        {
+            var container = await GetContainerAsync(settings);
+            if (container == null)
+            {
+                throw new InvalidOperationException("No running container found");
+            }
 
-        await ResetContainerAsync(container);
+            await ResetContainerAsync(container);
+        }
 
         await _listCliCommand.ExecuteAsync();
 
@@ -53,7 +72,6 @@ public override async Task<int> ExecuteAsync(CommandContext context, ResetSettin
         return containers.SingleOrDefault(c => c.ContainerName == identifier);
     }
 
-
     private async Task ResetContainerAsync(Container container)
     {
         await Spinner.StartAsync(
@@ -65,4 +83,44 @@ await Spinner.StartAsync(
                     await _runContainerCommand.ExecuteAsync(container);
                 });
     }
-}
+
+    private async Task ResetImageBaseTagAsync(ResetSettings settings)
+    {
+        var image = await GetImageAsync(settings);
+        if (image == null)
+        {
+            throw new InvalidOperationException("Image not found");
+        }
+
+        var tag = settings.Tag ?? image.Tag;
+        if (tag == null)
+        {
+            throw new InvalidOperationException("No tag specified and image has no tag");
+        }
+
+        await Spinner.StartAsync(
+            $"Resetting base tag for image '{image.Name}' to '{tag}'",
+            async _ =>
+            {
+                // Create a new image with the same content but new base tag
+                var labels = new Dictionary<string, string>
+                {
+                    { Constants.BaseTag, tag }
+                };
+
+                await _createImageCommand.ExecuteAsync(image.Name, tag);
+            });
+    }
+
+    private async Task<Image?> GetImageAsync(IImageIdentifierSettings settings)
+    {
+        if (settings.ImageIdentifier != null)
+        {
+            return await _getImageQuery.QueryAsync(settings.ImageIdentifier);
+        }
+
+        var images = await _getImageQuery.QueryAllAsync().ToListAsync();
+        var identifier = _imageIdentifierPrompt.GetIdentifierOfImageFromUser(images, "reset base tag");
+        return await _getImageQuery.QueryAsync(identifier);
+    }
+}
diff --git a/src/Commands/Reset/ResetSettings.cs b/src/Commands/Reset/ResetSettings.cs
@@ -2,8 +2,16 @@
 
 namespace port.Commands.Reset;
 
-internal class ResetSettings : CommandSettings, IContainerIdentifierSettings
+internal class ResetSettings : CommandSettings, IContainerIdentifierSettings, IImageIdentifierSettings
 {
-    [CommandArgument(0, "[ContainerIdentifier]")]
+    [CommandArgument(0, "[Identifier]")]
     public string? ContainerIdentifier { get; set; }
-}
+
+    [CommandOption("-i|--image")]
+    public bool IsImage { get; set; }
+
+    [CommandOption("-t|--tag")]
+    public string? Tag { get; set; }
+
+    string? IImageIdentifierSettings.ImageIdentifier => ContainerIdentifier;
+}