retrieveLogs.js

'use strict';

/* eslint no-use-before-define: 0 */

const BbPromise = require('bluebird');
const chalk = require('chalk');

module.exports = {
  retrieveLogs() {
    return BbPromise.bind(this).then(this.getLogs).then(this.printLogs);
  },

  getLogs() {
    const project = this.serverless.service.provider.project;
    let func = this.options.function;
    const pageSize = parseInt(this.options.count, 10) || 10;

    func = getGoogleCloudFunctionName(this.serverless.service.functions, func);

    return this.provider.request('logging', 'entries', 'list', {
      filter: `resource.labels.function_name="${func}" AND NOT textPayload=""`,
      orderBy: 'timestamp desc',
      resourceNames: [`projects/${project}`],
      pageSize,
    });
  },

  printLogs(logs) {
    if (!logs.entries || !logs.entries.length) {
      logs = {
        entries: [
          {
            timestamp: new Date().toISOString().slice(0, 10),
            textPayload: 'There is no log data to show...',
          },
        ],
      };
    }

    let output = logs.entries.reduce((p, c) => {
      let message = '';
      message += `${chalk.grey(`${c.timestamp}:`)}\t`;
      message += c.labels && c.labels.execution_id ? `[${c.labels.execution_id}]\t` : '';
      message += `${c.textPayload}\n`;
      return p + message;
    }, '');

    output = `Displaying the ${logs.entries.length} most recent log(s):\n\n${output}`; // prettify output
    output = output.slice(0, output.length - 1); // remove "\n---\n\n" for the last log entry

    this.serverless.cli.log(output);

    return BbPromise.resolve();
  },
};

const getGoogleCloudFunctionName = (serviceFunctions, func) => {
  if (!serviceFunctions[func]) {
    const errorMessage = [
      `Function "${func}" not found. `,
      'Please check your "serverless.yml" file for the correct function name.',
    ].join('');
    throw new Error(errorMessage);
  }
  return serviceFunctions[func].name;
};