From 6c0d35f5b9f998e7d9f7bdb797840b89c279d67b Mon Sep 17 00:00:00 2001
From: yalh76 <yalh@yahoo.com>
Date: Mon, 24 Oct 2022 02:05:26 +0200
Subject: [PATCH] Improving ynh_lxc_snapshot_load

---
 lib/ynh_lxd | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/lib/ynh_lxd b/lib/ynh_lxd
index b8b0275..5d31a00 100644
--- a/lib/ynh_lxd
+++ b/lib/ynh_lxd
@@ -456,15 +456,20 @@ ynh_lxc_snapshot_load () {
 	# Manage arguments with getopts
 	ynh_handle_getopts_args "$@"
 
-	log_debug "Loading snapshot $snapname ..."
+	if ynh_lxc_snapshot_exists --name="$name" --snapname="$snapname"
+	then
+		log_debug "Loading snapshot $snapname ..."
 
-	# Remove swap files before restoring the snapshot.
-	ynh_lxc_swapfiles_clean --name=$name
+		# Remove swap files before restoring the snapshot.
+		ynh_lxc_swapfiles_clean --name=$name
 
-	ynh_lxc_stop --name=$name
+		ynh_lxc_stop --name=$name
 
-	lxc restore $name $snapname
-	_ynh_lxc_start_and_wait --name=$name
+		lxc restore $name $snapname
+		_ynh_lxc_start_and_wait --name=$name
+	else
+		return 1
+	fi
 }
 
 # Reset an LXC container