feat: add isAllowedToSave option and improve error handling for image generation and saving

Author: yaroslav8765

custom/imageGenerationCarousel.vue

@@ -339,7 +339,7 @@ async function getGenerationPrompt() {
     if(!resp) {
       emit('error', {
         isError: true,
-        errorMessage: "Something went wrong. Check your internet connection and try again."
+        errorMessage: "Error getting generation prompts."
     });
     }
     return resp?.generationOptions || null;
@@ -400,7 +400,7 @@ async function generateImages() {
       });
     emit('error', {
       isError: true,
-      errorMessage: error
+      errorMessage: "Error re-generating images"
     });
     }
     return;

custom/visionAction.vue

@@ -429,10 +429,18 @@ async function saveData() {
       },
     });
 
-    if(res.ok) {
+    if(res.ok === true) {
       confirmDialog.value.close();
       props.updateList();
       props.clearCheckboxes();
+    } else if (res.ok === false) {
+      adminforth.alert({
+        message: 'You are not allowed to save.',
+        variant: 'danger',
+        timeout: 'unlimited',
+      });
+      isError.value = true;
+      errorMessage.value = `Failed to save data. You are not allowed to save.`;
     } else {
       console.error('Error saving data:', res);
       isError.value = true;

index.ts

@@ -339,45 +339,49 @@ export default class  BulkAiFlowPlugin extends AdminForthPlugin {
       method: 'POST',
       path: `/plugin/${this.pluginInstanceId}/update_fields`,
       handler: async ( body ) => {
-        const selectedIds = body.body.selectedIds || [];
-        const fieldsToUpdate = body.body.fields || {};
-        const outputImageFields = [];
-        if (this.options.generateImages) {
-          for (const [key, value] of Object.entries(this.options.generateImages)) {
-            outputImageFields.push(key);
+        if(this.options.isAllowedToSave !== false) {
+          const selectedIds = body.body.selectedIds || [];
+          const fieldsToUpdate = body.body.fields || {};
+          const outputImageFields = [];
+          if (this.options.generateImages) {
+            for (const [key, value] of Object.entries(this.options.generateImages)) {
+              outputImageFields.push(key);
+            }
           }
-        }
-        const primaryKeyColumn = this.resourceConfig.columns.find((col) => col.primaryKey);
-        const updates = selectedIds.map(async (ID, idx) => {
-          const oldRecord = await this.adminforth.resource(this.resourceConfig.resourceId).get( [Filters.EQ(primaryKeyColumn.name, ID)] );
-          for (const [key, value] of Object.entries(outputImageFields)) {
-            const columnPlugin = this.adminforth.activatedPlugins.find(p => 
-              p.resourceConfig!.resourceId === this.resourceConfig.resourceId &&
-              p.pluginOptions.pathColumnName === value
-            );
-            if (columnPlugin) {
-              if(columnPlugin.pluginOptions.storageAdapter.objectCanBeAccesedPublicly()) {
-                if (oldRecord[value]) {
-                  // put tag to delete old file
-                  try {
-                    await columnPlugin.pluginOptions.storageAdapter.markKeyForDeletation(oldRecord[value]);
-                  } catch (e) {
-                    // file might be e.g. already deleted, so we catch error
-                    console.error(`Error setting tag to true for object ${oldRecord[value]}. File will not be auto-cleaned up`, e);
+          const primaryKeyColumn = this.resourceConfig.columns.find((col) => col.primaryKey);
+          const updates = selectedIds.map(async (ID, idx) => {
+            const oldRecord = await this.adminforth.resource(this.resourceConfig.resourceId).get( [Filters.EQ(primaryKeyColumn.name, ID)] );
+            for (const [key, value] of Object.entries(outputImageFields)) {
+              const columnPlugin = this.adminforth.activatedPlugins.find(p => 
+                p.resourceConfig!.resourceId === this.resourceConfig.resourceId &&
+                p.pluginOptions.pathColumnName === value
+              );
+              if (columnPlugin) {
+                if(columnPlugin.pluginOptions.storageAdapter.objectCanBeAccesedPublicly()) {
+                  if (oldRecord[value]) {
+                    // put tag to delete old file
+                    try {
+                      await columnPlugin.pluginOptions.storageAdapter.markKeyForDeletation(oldRecord[value]);
+                    } catch (e) {
+                      // file might be e.g. already deleted, so we catch error
+                      console.error(`Error setting tag to true for object ${oldRecord[value]}. File will not be auto-cleaned up`, e);
+                    }
+                  }
+                  if (fieldsToUpdate[idx][key] !== null) {
+                  // remove tag from new file
+                  // in this case we let it crash if it fails: this is a new file which just was uploaded. 
+                    await  columnPlugin.pluginOptions.storageAdapter.markKeyForNotDeletation(fieldsToUpdate[idx][value]);
                   }
-                }
-                if (fieldsToUpdate[idx][key] !== null) {
-                // remove tag from new file
-                // in this case we let it crash if it fails: this is a new file which just was uploaded. 
-                  await  columnPlugin.pluginOptions.storageAdapter.markKeyForNotDeletation(fieldsToUpdate[idx][value]);
                 }
               }
             }
-          }
-          return this.adminforth.resource(this.resourceConfig.resourceId).update(ID, fieldsToUpdate[idx])
-        });
-        await Promise.all(updates);
-      return { ok: true };
+            return this.adminforth.resource(this.resourceConfig.resourceId).update(ID, fieldsToUpdate[idx])
+          });
+          await Promise.all(updates);
+          return { ok: true };
+        } else {
+          return { ok: false }
+        }
       }
     });
 

types.ts

@@ -47,4 +47,9 @@ export interface PluginOptions {
      * As rateLimit on generateImages, but applied to bulk generations  
      **/
     bulkGenerationRateLimit?: string, 
+
+    /**
+     * Whether the user is allowed to save the generated images
+     */
+    isAllowedToSave?: boolean
 }