From 25782b644b73902fef6890ea6625049d0f8a4933 Mon Sep 17 00:00:00 2001
From: Voomra <dimon550@gmail.com>
Date: Sun, 19 Jan 2025 19:54:11 +0300
Subject: [PATCH] =?UTF-8?q?fix:=20=D1=85=D0=BE=D1=81=D1=82=20=D0=BF=D0=BE-?=
 =?UTF-8?q?=D1=83=D0=BC=D0=BE=D0=BB=D1=87=D0=B0=D0=BD=D0=B8=D1=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/restore.go | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/cmd/restore.go b/cmd/restore.go
index 270e090..af65b83 100644
--- a/cmd/restore.go
+++ b/cmd/restore.go
@@ -72,5 +72,16 @@ func restoreRunE(_ *cobra.Command, args []string) error {
 	spaceWorker := space_worker.NewSpaceWorker(logger.LogVerbose(flagVerbose), workDir)
 	spaceWorker.AnsibleBin = flagAnsibleBin
 	spaceWorker.AnsibleVerbose = flagAnsibleVerbose
-	return spaceWorker.Restore(flagTargetHost, flagRestoreTimestamp, flagRestoreInventory, args...)
+
+	var restoreInventory string
+	if flagRestoreInventory != "" {
+		restoreInventory = flagRestoreInventory
+	} else {
+		restoreInventory, err = spaceWorker.GetDefaultHost()
+		if err != nil {
+			return err
+		}
+	}
+
+	return spaceWorker.Restore(flagTargetHost, flagRestoreTimestamp, restoreInventory, args...)
 }