added selecting new level set if dropped as zip file into window

[rocksndiamonds.git] / src / events.c

diff --git a/src/events.c b/src/events.c
index 90f6615a20f61ee0647a61f58f03b0b586d17791..05d4b351e8b868b9307917dc4fcba7483fd0b3eb 100644 (file)
--- a/src/events.c
+++ b/src/events.c
@@ -1543,7 +1543,27 @@ static void HandleDropFileEventExt(char *filename)
       return;
     }
 
-    ExtractZipFileIntoDirectory(filename, directory, tree_type);
+    char *top_dir = ExtractZipFileIntoDirectory(filename, directory, tree_type);
+
+    if (top_dir != NULL)
+    {
+      AddUserTreeSetToTreeInfo(top_dir, tree_type);
+
+      // when adding new level set, select it as the new current level set
+      if (tree_type == TREE_TYPE_LEVEL_DIR)
+      {
+       // change current level set to newly added level set from zip file
+       leveldir_current = getTreeInfoFromIdentifier(leveldir_first, top_dir);
+
+       // change current level number to first level of newly added level set
+       level_nr = leveldir_current->first_level;
+
+       // when in main menu, redraw screen to reflect changed level set
+       if (game_status == GAME_MODE_MAIN)
+         DrawMainMenu();
+      }
+    }
+  }
 }
 
 static void HandleDropTextEventExt(char *text)