From 505b925ed81683215be5b2e0408f974cd5b5d752 Mon Sep 17 00:00:00 2001
From: tibbi <tibor@kaputa.sk>
Date: Tue, 2 Jul 2019 20:36:24 +0200
Subject: [PATCH] prompt folder unlocking at some operations

---
 .../gallery/pro/adapters/DirectoryAdapter.kt   | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/app/src/main/kotlin/com/simplemobiletools/gallery/pro/adapters/DirectoryAdapter.kt b/app/src/main/kotlin/com/simplemobiletools/gallery/pro/adapters/DirectoryAdapter.kt
index 2d83263cf..5a1e56edd 100644
--- a/app/src/main/kotlin/com/simplemobiletools/gallery/pro/adapters/DirectoryAdapter.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/gallery/pro/adapters/DirectoryAdapter.kt
@@ -122,10 +122,10 @@ class DirectoryAdapter(activity: BaseSimpleActivity, var dirs: ArrayList<Directo
             R.id.cab_copy_to -> copyMoveTo(true)
             R.id.cab_move_to -> moveFilesTo()
             R.id.cab_select_all -> selectAll()
-            R.id.cab_create_shortcut -> createShortcut()
+            R.id.cab_create_shortcut -> tryCreateShortcut()
             R.id.cab_delete -> askConfirmDelete()
-            R.id.cab_select_photo -> changeAlbumCover(false)
-            R.id.cab_use_default -> changeAlbumCover(true)
+            R.id.cab_select_photo -> tryChangeAlbumCover(false)
+            R.id.cab_use_default -> tryChangeAlbumCover(true)
         }
     }
 
@@ -407,6 +407,12 @@ class DirectoryAdapter(activity: BaseSimpleActivity, var dirs: ArrayList<Directo
         }
     }
 
+    private fun tryCreateShortcut() {
+        activity.handleLockedFolderOpening(getFirstSelectedItemPath() ?: "") {
+            createShortcut()
+        }
+    }
+
     @SuppressLint("NewApi")
     private fun createShortcut() {
         val manager = activity.getSystemService(ShortcutManager::class.java)
@@ -505,6 +511,12 @@ class DirectoryAdapter(activity: BaseSimpleActivity, var dirs: ArrayList<Directo
         }
     }
 
+    private fun tryChangeAlbumCover(useDefault: Boolean) {
+        activity.handleLockedFolderOpening(getFirstSelectedItemPath() ?: "") {
+            changeAlbumCover(useDefault)
+        }
+    }
+
     private fun changeAlbumCover(useDefault: Boolean) {
         if (selectedKeys.size != 1)
             return