adding export relation

Author: Jcarlosjunior

src/Routers/ExportRouter.js

@@ -7,32 +7,30 @@ import tmp             from 'tmp';
 import fs from 'fs';
 
 const DefaultExportExportProgressCollectionName = "_ExportProgress";
+const relationSchema = { fields: { relatedId: { type: 'String' }, owningId: { type: 'String' } } };
 
 export class ExportRouter extends PromiseRouter {
 
-  handleExportProgress(req) {
 
-    const databaseController = req.config.database;
 
-    let query = {
-      masterKey: req.info.masterKey,
-      applicationId: req.info.appId
-    }; 
+  exportClassPage(req, name, jsonFileStream, where, skip, limit) {
 
-    return databaseController.find(DefaultExportExportProgressCollectionName, query)
-    .then((response) => { 
-      return { response };
-    });
-  }
+    const databaseController = req.config.database;
 
-  handleExportPage(req, name, jsonFileStream, where, skip, limit) {
     let options =  {
       skip,
       limit
     };
 
-    return rest.find(req.config, req.auth, name,  where, options)
+    let findPromise = name.indexOf('_Join') === 0 ? 
+    databaseController.adapter.find(name, relationSchema, where, options)
+    : rest.find(req.config, req.auth, name,  where, options);
+
+    return findPromise 
     .then((data) => {
+      if (Array.isArray(data)) {
+        data = { results : data };
+      }
 
       if (skip && data.results.length) {
         jsonFileStream.write(',\n');
@@ -42,35 +40,26 @@ export class ExportRouter extends PromiseRouter {
     });
   }
 
-  handleExport(req) {
+  exportClass(req, data) {
 
-    const databaseController = req.config.database;
-    let data = req.body;
 
+    const databaseController = req.config.database;
     let tmpJsonFile = tmp.fileSync();
     let jsonFileStream = fs.createWriteStream(tmpJsonFile.name);
 
     jsonFileStream.write('{\n"results" : [\n');
 
-      let exportProgress = {
-        id: data.name,
-        masterKey: req.info.masterKey,
-        applicationId: req.info.appId
-      };
-
+      let findPromise = data.name.indexOf('_Join') === 0 ? 
+      databaseController.adapter.count(data.name, relationSchema, data.where)
+      : rest.find(req.config, req.auth, data.name,  data.where, { count: true, limit: 0 });
 
-      let emailControllerAdapter = req.config.emailControllerAdapter;
-
-      if (!emailControllerAdapter) {
-        return Promise.reject(new Error('You have to setup a Mail Adapter.'));
-      }
-
-      databaseController.create(DefaultExportExportProgressCollectionName, exportProgress)
-      .then(() => {
-        return rest.find(req.config, req.auth, data.name,  data.where, { count: true, limit: 0 });
-      })
+      return findPromise
       .then((result) => {
 
+        if (Number.isInteger(result)) {
+          result = { count: result };
+        }
+
         let i = 0;
         let pageLimit = 1000;
         let promise = Promise.resolve();
@@ -80,7 +69,7 @@ export class ExportRouter extends PromiseRouter {
 
           let skip = i;
           promise = promise.then(() => {
-            return this.handleExportPage(req, data.name, jsonFileStream, data.where, skip, pageLimit);
+            return this.exportClassPage(req, data.name, jsonFileStream, data.where, skip, pageLimit);
           });
         }
 
@@ -93,54 +82,118 @@ export class ExportRouter extends PromiseRouter {
         return new Promise((resolve, reject) => {
 
           jsonFileStream.on('close', () => {
+            tmpJsonFile._name = `${data.name.replace(':', '/')}.json`;
+
+            resolve(tmpJsonFile);
+          });
+        })
+      });
+  }
+
+  handleExportProgress(req) {
+
+    const databaseController = req.config.database;
+
+    let query = {
+      masterKey: req.info.masterKey,
+      applicationId: req.info.appId
+    }; 
 
-            let tmpZipFile = tmp.fileSync();
-            let tmpZipStream = fs.createWriteStream(tmpZipFile.name);
+    return databaseController.find(DefaultExportExportProgressCollectionName, query)
+    .then((response) => { 
+      return { response };
+    });
+  }
 
-            let zip  = archiver('zip', { store: true });
-            zip.pipe(tmpZipStream);
-            zip.append(fs.readFileSync(tmpJsonFile.name), { name: `${data.name}.json` });
-            zip.finalize();
+  handleExport(req) {
 
-            tmpJsonFile.removeCallback();
+    const databaseController = req.config.database;
 
-            tmpZipStream.on('close', () => {
+    let emailControllerAdapter = req.config.emailControllerAdapter;
 
-              let buf = fs.readFileSync(tmpZipFile.name);
-              tmpZipFile.removeCallback();
-              resolve(buf);
+    if (!emailControllerAdapter) {
+      return Promise.reject(new Error('You have to setup a Mail Adapter.'));
+    }
 
-            });
+    let exportProgress = {
+      id: req.body.name,
+      masterKey: req.info.masterKey,
+      applicationId: req.info.appId
+    };
 
-          });
-        })
-      })
-      .then((zippedFile) => {
-        const filesController = req.config.filesController;
-        return filesController.createFile(req.config, data.name, zippedFile, 'application/zip');
-      })
-      .then((fileData) => {
-
-        return emailControllerAdapter.sendMail({
-          text: `We have successfully exported your data from the class ${data.name}.\n
-          Please download from ${fileData.url}`,
-          link: fileData.url,
-          to: req.body.feedbackEmail,
-          subject: 'Export completed'
+    databaseController.create(DefaultExportExportProgressCollectionName, exportProgress)
+    .then(() => {
+      return databaseController.loadSchema({ clearCache: true});
+    })
+    .then(schemaController => schemaController.getOneSchema(req.body.name, true))
+    .then((schema) => {
+      let classNames = [ req.body.name ];
+      Object.keys(schema.fields).forEach((fieldName) => {
+        let field = schema.fields[fieldName];
+
+        if (field.type === 'Relation') {
+          classNames.push(`_Join:${fieldName}:${req.body.name}`);
+        }
+      });
+
+      let promisses = classNames.map((name) => {
+        return this.exportClass(req, { name });
+      });
+
+      return Promise.all(promisses)
+    })
+    .then((jsonFiles) => {
+
+      return new Promise((resolve) => {
+        let tmpZipFile = tmp.fileSync();
+        let tmpZipStream = fs.createWriteStream(tmpZipFile.name);
+
+        let zip  = archiver('zip', { store: true });
+        zip.pipe(tmpZipStream);
+
+        jsonFiles.forEach(tmpJsonFile => {
+          zip.append(fs.readFileSync(tmpJsonFile.name), { name:  tmpJsonFile._name });
+          tmpJsonFile.removeCallback();
         });
-      })
-      .catch((error) => {
-        return emailControllerAdapter.sendMail({
-          text: `We could not export your data to the class ${data.name}. Error: ${error}`,
-          to: req.body.feedbackEmail,
-          subject: 'Export failed'
+
+        zip.finalize();
+
+        tmpZipStream.on('close', () => {
+
+          let buf = fs.readFileSync(tmpZipFile.name);
+          tmpZipFile.removeCallback();
+          resolve(buf);
+
         });
-      })
-      .then(() => {
-        return databaseController.destroy(DefaultExportExportProgressCollectionName, exportProgress);
       });
 
-      return Promise.resolve({response: 'We are exporting your data. You will be notified by e-mail once it is completed.'});
+    })
+    .then((zippedFile) => {
+      const filesController = req.config.filesController;
+      return filesController.createFile(req.config, req.body.name, zippedFile, 'application/zip');
+    })
+    .then((fileData) => {
+
+      return emailControllerAdapter.sendMail({
+        text: `We have successfully exported your data from the class ${req.body.name}.\n
+        Please download from ${fileData.url}`,
+        link: fileData.url,
+        to: req.body.feedbackEmail,
+        subject: 'Export completed'
+      });
+    })
+    .catch((error) => {
+      return emailControllerAdapter.sendMail({
+        text: `We could not export your data to the class ${data.name}. Error: ${error}`,
+        to: req.body.feedbackEmail,
+        subject: 'Export failed'
+      });
+    })
+    .then(() => {
+      return databaseController.destroy(DefaultExportExportProgressCollectionName, exportProgress);
+    });
+
+    return Promise.resolve({response: 'We are exporting your data. You will be notified by e-mail once it is completed.'});
   }
 
   mountRoutes() {