From f38756dd76f198048b5622001faaed35e434b51f Mon Sep 17 00:00:00 2001
From: Felipe <41008398+StrawberryMaster@users.noreply.github.com>
Date: Fri, 30 May 2025 14:00:32 +0000
Subject: [PATCH] Correction for downloaded data folder if you downloaded
 content from example.org/*, it would be listed in a folder titled * instead
 of the sitename. See #6 (and thanks to elidickinson for pointing it out!)

---
 lib/wayback_machine_downloader.rb | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/lib/wayback_machine_downloader.rb b/lib/wayback_machine_downloader.rb
index 26ecac1..cf0a0b2 100644
--- a/lib/wayback_machine_downloader.rb
+++ b/lib/wayback_machine_downloader.rb
@@ -154,10 +154,12 @@ class WaybackMachineDownloader
   end
 
   def backup_name
-    if @base_url.include? '//'
-      @base_url.split('/')[2]
+    url_to_process = @base_url.end_with?('/*') ? @base_url.chomp('/*') : @base_url
+    
+    if url_to_process.include? '//'
+      url_to_process.split('/')[2]
     else
-      @base_url
+      url_to_process
     end
   end