From 0c520d25adc0b6393a95005605996ecddb371659 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?E=CC=81ric=20Gaspar?=
 <46165813+ericgaspar@users.noreply.github.com>
Date: Sun, 21 May 2023 11:34:57 +0200
Subject: [PATCH] Update restore

---
 scripts/restore | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/scripts/restore b/scripts/restore
index 71ec97b1..154d47ca 100644
--- a/scripts/restore
+++ b/scripts/restore
@@ -47,7 +47,18 @@ ynh_restore_file --origin_path="/etc/nginx/conf.d/$domain.d/$app.conf"
 # RESTORE THE DATABASE
 #=================================================
 
-ynh_mysql_connect_as --user=$db_user --password=$db_pwd --database=$db_name < ./db.sql
+if [ $database = "mysql" ]; then
+	ynh_script_progression --message="Restoring  MySQL database..." --weight=1
+	db_pwd=$(ynh_app_setting_get --app=$app --key=mysqlpwd)
+	ynh_mysql_setup_db --db_user=$db_user --db_name=$db_name --db_pwd=$db_pwd
+	ynh_mysql_connect_as --user=$db_user --password=$db_pwd --database=$db_name < ./db.sql
+elif [ $database = "postgresql" ]; then
+	ynh_script_progression --message="Restoring the PostgreSQL database..." --weight=1
+	db_pwd=$(ynh_app_setting_get --app=$app --key=psqlpwd)
+	ynh_psql_test_if_first_run
+	ynh_psql_setup_db --db_user=$db_user --db_name=$db_name --db_pwd=$db_pwd
+	ynh_psql_execute_file_as_root --file="./db.sql" --database="$db_name"
+fi
 
 #=================================================
 # RESTORE VARIOUS FILES