Fixes issues with persistent mode

- adds mode to the persistent file
- adds filenames for share and website mode in persistent file

1 files changed, 7 insertions, 1 deletions

diff --git a/cli/onionshare_cli/__init__.py b/cli/onionshare_cli/__init__.py
index 8ba0aac2..72212760 100644
--- a/cli/onionshare_cli/__init__.py
+++ b/cli/onionshare_cli/__init__.py
@@ -267,6 +267,7 @@ def main(cwd=None):
     if persistent_filename:
         mode_settings = ModeSettings(common, persistent_filename)
         mode_settings.set("persistent", "enabled", True)
+        mode_settings.set("persistent", "mode", mode)
     else:
         mode_settings = ModeSettings(common)
 
@@ -292,7 +293,7 @@ def main(cwd=None):
     if mode == "share" or mode == "website":
         # Unless you passed in a persistent filename, in which case get the filenames from
         # the mode settings
-        if persistent_filename and not mode_settings.just_created:
+        if persistent_filename and not mode_settings.just_created and len(filenames) != 0:
             filenames = mode_settings.get(mode, "filenames")
 
         else:
@@ -316,6 +317,11 @@ def main(cwd=None):
             if not valid:
                 sys.exit()
 
+        # Save the filenames in persistent file
+        if persistent_filename:
+            mode_settings.set(mode, "filenames", filenames)
+
+
     # Create the Web object
     web = Web(common, False, mode_settings, mode)