1 files changed, 21 insertions, 0 deletions
diff --git a/src/com/android/dreams/phototable/AlbumSettings.java b/src/com/android/dreams/phototable/AlbumSettings.java
index 23dda46..069948b 100644
--- a/src/com/android/dreams/phototable/AlbumSettings.java
+++ b/src/com/android/dreams/phototable/AlbumSettings.java
@@ -56,6 +56,12 @@ public class AlbumSettings {
         }
     }
 
+    public boolean areAllEnabled(Collection<String> validAlbums) {
+        synchronized (mEnabledAlbums) {
+            return mEnabledAlbums.containsAll(validAlbums);
+        }
+    }
+
     public void setAlbumEnabled(String albumId, boolean enabled) {
         if (isAlbumEnabled(albumId) != enabled) {
             synchronized (mEnabledAlbums) {
@@ -70,6 +76,21 @@ public class AlbumSettings {
         }
     }
 
+    public void disableAllAlbums() {
+        synchronized (mEnabledAlbums) {
+            mEnabledAlbums.clear();
+            writeEnabledAlbumsLocked();
+        }
+    }
+
+    public void enableAllAlbums(Collection<String> validAlbums) {
+        synchronized (mEnabledAlbums) {
+            mEnabledAlbums.clear();
+            mEnabledAlbums.addAll(validAlbums);
+            writeEnabledAlbumsLocked();
+        }
+    }
+
     public void pruneObsoleteSettings(Collection<String> validAlbums) {
         if (!validAlbums.containsAll(mEnabledAlbums)) {
             synchronized (mEnabledAlbums) {