From a7cb381fbf506b5a741e9c622bf4a6655edc3de6 Mon Sep 17 00:00:00 2001
From: Mike Patnode <mpatnode@popsugar.com>
Date: Mon, 2 Dec 2019 17:47:22 -0800
Subject: [PATCH] Allow validateFilename to return a string or Parse Error
 (#6246)

---
 spec/FilesController.spec.js       | 17 ++++++++++++++++-
 src/Controllers/FilesController.js |  7 ++++++-
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/spec/FilesController.spec.js b/spec/FilesController.spec.js
index 2ad6f861..aef64481 100644
--- a/spec/FilesController.spec.js
+++ b/spec/FilesController.spec.js
@@ -16,7 +16,9 @@ const mockAdapter = {
   deleteFile: () => {},
   getFileData: () => {},
   getFileLocation: () => 'xyz',
-  validateFilename: () => {},
+  validateFilename: () => {
+    return null;
+  },
 };
 
 // Small additional tests to improve overall coverage
@@ -77,6 +79,19 @@ describe('FilesController', () => {
       });
   });
 
+  it('should create a parse error when a string is returned', done => {
+    const mock2 = mockAdapter;
+    mock2.validateFilename = () => {
+      return 'Bad file! No biscuit!';
+    };
+    const filesController = new FilesController(mockAdapter);
+    const error = filesController.validateFilename();
+    expect(typeof error).toBe('object');
+    expect(error.message.indexOf('biscuit')).toBe(13);
+    expect(error.code).toBe(Parse.Error.INVALID_FILE_NAME);
+    done();
+  });
+
   it('should add a unique hash to the file name when the preserveFileName option is false', done => {
     const config = Config.get(Parse.applicationId);
     const gridStoreAdapter = new GridFSBucketAdapter(
diff --git a/src/Controllers/FilesController.js b/src/Controllers/FilesController.js
index 4d2b60ed..461fa229 100644
--- a/src/Controllers/FilesController.js
+++ b/src/Controllers/FilesController.js
@@ -4,6 +4,7 @@ import AdaptableController from './AdaptableController';
 import { validateFilename, FilesAdapter } from '../Adapters/Files/FilesAdapter';
 import path from 'path';
 import mime from 'mime';
+const Parse = require('parse').Parse;
 
 const legacyFilesRegex = new RegExp(
   '^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}-.*'
@@ -98,7 +99,11 @@ export class FilesController extends AdaptableController {
 
   validateFilename(filename) {
     if (typeof this.adapter.validateFilename === 'function') {
-      return this.adapter.validateFilename(filename);
+      const error = this.adapter.validateFilename(filename);
+      if (typeof error !== 'string') {
+        return error;
+      }
+      return new Parse.Error(Parse.Error.INVALID_FILE_NAME, error);
     }
     return validateFilename(filename);
   }