backend-manager 3.2.113 → 3.2.114

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "backend-manager",
-  "version": "3.2.113",
+  "version": "3.2.114",
   "description": "Quick tools for developing Firebase functions",
   "main": "src/manager/index.js",
   "bin": {

package/src/manager/functions/core/cron/daily/reset-usage.js

@@ -1,31 +1,20 @@
-const _ = require('lodash');
+// Libraries
 const fetch = require('wonderful-fetch');
 
 function Module() {
 
 }
 
-Module.prototype.init = function (Manager, data) {
+Module.prototype.main = function (assistant, context) {
   const self = this;
-  self.Manager = Manager;
 
-  self.libraries = Manager.libraries;
-  self.assistant = Manager.Assistant();
-
-  self.storage = Manager.storage({name: 'usage', temporary: true, clear: false, log: false});
-
-  self.context = data.context;
-  return self;
-}
-
-Module.prototype.main = function() {
-  const self = this;
-  const Manager = self.Manager;
-  const libraries = self.libraries;
-  const assistant = self.assistant;
-  const context = self.context;
+  // Shortcuts
+  const Manager = assistant.Manager;
+  const libraries = Manager.libraries;
 
   return new Promise(async function(resolve, reject) {
+    self.storage = Manager.storage({name: 'usage', temporary: true, clear: false, log: false});
+
     assistant.log(`cron/daily/reset-usage(): Starting main...`);
 
     // Clear local
@@ -40,6 +29,8 @@ Module.prototype.main = function() {
 
 Module.prototype.clearLocal = function() {
   const self = this;
+
+  // Shortcuts
   const Manager = self.Manager;
   const libraries = self.libraries;
   const assistant = self.assistant;
@@ -127,7 +118,7 @@ Module.prototype.clearFirestore = function() {
             data.usage[metric].total = data.usage[metric].total || 0;
             data.usage[metric].last = data.usage[metric].last || {};
 
-            // Bail if its 0
+            // Yeet if its 0
             if (data.usage[metric].period <= 0) {
               continue;
             }

package/src/manager/functions/core/cron/daily.js

@@ -8,6 +8,7 @@ function Module() {
 Module.prototype.init = function (Manager, data) {
   const self = this;
 
+  // Shortcuts
   self.Manager = Manager;
   self.libraries = Manager.libraries;
   self.assistant = Manager.Assistant();
@@ -19,42 +20,88 @@ Module.prototype.init = function (Manager, data) {
 
 Module.prototype.main = function() {
   const self = this;
+
+  // Shortcuts
   const Manager = self.Manager;
   const libraries = self.libraries;
   const assistant = self.assistant;
   const context = self.context;
 
   return new Promise(async function(resolve, reject) {
-    assistant.log(`cron/daily(): Starting...`);
+    // Set log prefix
+    assistant.setLogPrefix('cron/daily()');
+
+    // Log
+    assistant.log(`Starting...`);
+
+    // Setup error
+    let error;
+
+    // Load BEM jobs
+    await loadAndExecuteJobs(`${__dirname}/daily`, Manager, context).catch((e) => error = e);
+
+    // Load custom jobs
+    await loadAndExecuteJobs(`${process.cwd()}/hooks/cron/daily`, Manager, context).catch((e) => error = e);
+
+    // If there was an error, reject
+    if (error) {
+      return reject(error);
+    }
+
+    // Return
+    return resolve();
+  });
+}
 
-    const jobsPath = `${__dirname}/daily`;
-    const jobs = jetpack.list(jobsPath);
+function loadAndExecuteJobs(jobsPath, Manager, context) {
+  const assistant = Manager.assistant;
+
+  return new Promise(async function(resolve, reject) {
+    const jobs = jetpack.list(jobsPath) || [];
     let caught;
 
-    // For of loop for jobs, load the job, execute it, and log the result
+    // Log
+    assistant.log(`Located ${jobs.length} jobs @ ${jobsPath}...`);
+
     for (let i = 0; i < jobs.length; i++) {
+      // Create new assistant for each job
+      const assistant = Manager.Assistant();
+
+      // Load job
       const job = jobs[i];
       const jobName = job.replace('.js', '');
 
+      // Set log prefix
+      assistant.setLogPrefix(`cron/daily/${jobName}()`);
+
       // Log
-      assistant.log(`cron/daily(): Job ${jobName} starting...`);
+      assistant.log(`Starting...`);
 
-      // Load the job
+      // Load job
       const Job = require(`${jobsPath}/${job}`);
       const jobInstance = new Job();
-      jobInstance.init(Manager, { context: context, });
-
-      // Execute the job
-      await jobInstance.main()
-      .then(res => {
-        assistant.log(`cron/daily(): Job ${jobName} completed...`);
-      })
-      .catch(e => {
-        assistant.errorify(`Error executing ${jobName}: ${e}`, {code: 500, sentry: true});
-        caught = e;
-      })
+
+      // Setup
+      jobInstance.Manager = Manager;
+      jobInstance.assistant = assistant;
+      jobInstance.context = context;
+      jobInstance.libraries = Manager.libraries;
+
+      // Execute job
+      await jobInstance.main(assistant, context)
+        .then(res => {
+          assistant.log(`Completed!`);
+        })
+        .catch(e => {
+          assistant.errorify(`Error executing: ${e}`, {
+            code: 500,
+            sentry: true
+          });
+          caught = e;
+        })
     }
 
+    // If there was an error, reject
     if (caught) {
       return reject(caught);
     }

package/src/manager/helpers/assistant.js

@@ -81,6 +81,9 @@ BackendAssistant.prototype.init = function (ref, options) {
   // Set tag
   self.tag = `${self.meta.name}/${self.id}`;
 
+  // Set logger prefix
+  self.logPrefix = '';
+
   // Set stuff about request
   self.request = {};
   self.request.referrer = (self.ref.req.headers || {}).referrer || (self.ref.req.headers || {}).referer || '';
@@ -255,8 +258,15 @@ BackendAssistant.prototype._log = function () {
   // 1. Convert args to a normal array
   const logs = [...Array.prototype.slice.call(arguments)];
 
+  // Add log prefix
+  const prefix = self.logPrefix
+    ? ` ${self.logPrefix}:`
+    : ':';
+
   // 2. Prepend log prefix log string
-  logs.unshift(`[${self.tag} @ ${new Date().toISOString()}]:`);
+  logs.unshift(
+    `[${self.tag} @ ${new Date().toISOString()}]${prefix}`
+  );
 
   // 3. Pass along arguments to console.log
   if (logs[1] === 'error') {
@@ -309,6 +319,30 @@ BackendAssistant.prototype._log = function () {
   }
 }
 
+BackendAssistant.prototype.setLogPrefix = function (s) {
+  const self = this;
+
+  // Set logger prefix
+  self.logPrefix = s
+
+  return self;
+};
+
+BackendAssistant.prototype.clearLogPrefix = function () {
+  const self = this;
+
+  // Set logger prefix
+  self.logPrefix = '';
+
+  return self;
+};
+
+BackendAssistant.prototype.getLogPrefix = function () {
+  const self = this;
+
+  return self.logPrefix;
+};
+
 BackendAssistant.prototype.getUser = function () {
   const self = this;
 

package/src/manager/index.js

@@ -826,7 +826,7 @@ Manager.prototype.setupFunctions = function (exporter, options) {
   // Setup cron jobs
   exporter.bm_cronDaily =
   self.libraries.functions
-  .runWith({ memory: '256MB', timeoutSeconds: 120 })
+  .runWith({ memory: '256MB', timeoutSeconds: 190 })
   .pubsub.schedule('every 24 hours')
   .onRun(async (context) => self._process((new (require(`${core}/cron/daily.js`))()).init(self, { context: context, })));
 };