Implement list to enable remote state storage.

perfectsense · Jul 9, 2020 · f216449 · f216449
1 parent 0cf7b2e
commit f216449
Showing 1 changed file with 40 additions and 3 deletions.
diff --git a/src/main/java/gyro/google/GoogleStorageFileBackend.java b/src/main/java/gyro/google/GoogleStorageFileBackend.java
@@ -19,15 +19,17 @@
 import java.io.ByteArrayOutputStream;
 import java.io.InputStream;
 import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.stream.Stream;
 
 import com.google.api.client.http.ByteArrayContent;
 import com.google.api.services.storage.Storage;
+import com.google.api.services.storage.model.Objects;
 import com.google.api.services.storage.model.StorageObject;
-
 import com.psddev.dari.util.ObjectUtils;
-
 import gyro.core.FileBackend;
+import gyro.core.GyroCore;
 import gyro.core.GyroException;
 import gyro.core.Type;
 import gyro.core.auth.CredentialsSettings;
@@ -69,7 +71,34 @@ public String getCredentials() {
 
     @Override
     public Stream<String> list() throws Exception {
-        return null;
+        if (this.equals(GyroCore.getStateBackend(getName()))) {
+            List<StorageObject> storageObjects = new ArrayList<>();
+            Storage client = client();
+            String nextPageToken = null;
+            int count = 0;
+
+            do {
+                Objects objects = client.objects()
+                    .list(getBucket())
+                    .setPrefix(prefixed(""))
+                    .setPageToken(nextPageToken)
+                    .execute();
+
+                if (objects.getItems() != null) {
+                    storageObjects.addAll(objects.getItems());
+                }
+
+                nextPageToken = objects.getNextPageToken();
+                count++;
+            } while (nextPageToken != null || count < 10);
+
+            return storageObjects.stream()
+                .map(StorageObject::getName)
+                .filter(f -> f.endsWith(".gyro"))
+                .map(this::removePrefix);
+        }
+
+        return Stream.empty();
     }
 
     @Override
@@ -109,4 +138,12 @@ private String prefixed(String file) {
         return getPrefix() != null ? getPrefix() + '/' + file : file;
     }
 
+    private String removePrefix(String file) {
+        if (getPrefix() != null && file.startsWith(getPrefix() + "/")) {
+            return file.substring(getPrefix().length() + 1);
+        }
+
+        return file;
+    }
+
 }