Protected Folder Fragment: Add remove Shortcut from list.

Change-Id: I0e9f141bc34bcfa955d7f5f3535d9d437e94b1da

1 files changed, 14 insertions, 0 deletions

diff --git a/src/com/android/launcher3/Folder.java b/src/com/android/launcher3/Folder.java
index 7f3f47f87..dc33552ff 100644
--- a/src/com/android/launcher3/Folder.java
+++ b/src/com/android/launcher3/Folder.java
@@ -1482,6 +1482,20 @@ public class Folder extends LinearLayout implements DragSource, View.OnClickList
         return mItemsInReadingOrder;
     }
 
+    public ShortcutInfo getShortcutForComponent(ComponentName componentName) {
+        for (View v : mItemsInReadingOrder) {
+            Object tag = v.getTag();
+            if (tag instanceof ShortcutInfo) {
+                ComponentName cName = ((ShortcutInfo) tag).getIntent().getComponent();
+                if (cName.equals(componentName)) {
+                    return (ShortcutInfo) tag;
+                }
+            }
+        }
+
+        return null;
+    }
+
     public void getLocationInDragLayer(int[] loc) {
         mLauncher.getDragLayer().getLocationInDragLayer(this, loc);
     }