fix(refresh): handle array in refresh context data

This is needed for pawcoding/astro-integration-pocketbase#10.

Author: pawcoding

src/pocketbase-loader.ts

@@ -5,6 +5,7 @@ import { generateSchema } from "./generate-schema";
 import { loadEntries } from "./load-entries";
 import type { PocketBaseLoaderOptions } from "./types/pocketbase-loader-options.type";
 import { getSuperuserToken } from "./utils/get-superuser-token";
+import { shouldRefresh } from "./utils/should-refresh";
 
 /**
  * Loader for collections stored in PocketBase.
@@ -15,13 +16,12 @@ export function pocketbaseLoader(options: PocketBaseLoaderOptions): Loader {
   return {
     name: "pocketbase-loader",
     load: async (context: LoaderContext): Promise<void> => {
-      if (
-        context.refreshContextData?.source === "astro-integration-pocketbase" &&
-        context.refreshContextData.collection &&
-        context.refreshContextData.collection !== options.collectionName
-      ) {
-        // Skip the refresh if the reload was triggered by the `astro-integration-pocketbase`
-        // and the collection name does not match the current collection.
+      // Check if the collection should be refreshed.
+      const refresh = shouldRefresh(
+        context.refreshContextData,
+        options.collectionName
+      );
+      if (!refresh) {
         return;
       }
 

src/utils/should-refresh.ts

@@ -0,0 +1,32 @@
+import type { LoaderContext } from "astro/loaders";
+
+/**
+ * Checks if the collection should be refreshed.
+ */
+export function shouldRefresh(
+  context: LoaderContext["refreshContextData"],
+  collectionName: string
+): boolean {
+  // Check if the refresh was triggered by the `astro-integration-pocketbase`
+  // and the correct metadata is provided.
+  if (
+    !context ||
+    context.source !== "astro-integration-pocketbase" ||
+    !context.collection
+  ) {
+    return true;
+  }
+
+  // Check if the collection name matches the current collection.
+  if (typeof context.collection === "string") {
+    return context.collection === collectionName;
+  }
+
+  // Check if the collection is included in the list of collections.
+  if (Array.isArray(context.collection)) {
+    return context.collection.includes(collectionName);
+  }
+
+  // Should not happen but return true to be safe.
+  return true;
+}