nightwatch/lib/utils/colors.js

Easy to use Node.js based end-to-end testing solution for web applications using the W3C WebDriver API.

const Settings = {
  enabled: process.env.COLORS !== '0'
};

function Background() {
  return this;
}

function ConsoleColor() {
  this.background = new Background();

  const foregroundColors = {
    black: '0;30',
    dark_gray: '1;30',
    blue: '0;34',
    light_blue: '1;34',
    green: '0;32',
    light_green: '1;32',
    cyan: '0;36',
    light_cyan: '1;36',
    red: '0;31',
    light_red: '1;31',
    purple: '0;35',
    light_purple: '1;35',
    brown: '0;33',
    yellow: '1;33',
    light_gray: '0;37',
    white: '1;37',
    stack_trace: '0;90'
  };

  const backgroundColors = {
    black: '40',
    red: '41',
    green: '42',
    yellow: '43',
    blue: '44',
    magenta: '45',
    cyan: '46',
    light_gray: '47'
  };

  Object.keys(foregroundColors).forEach(k => {
    ConsoleColor.prototype[k.toLowerCase()] = (text, background) => {
      if (!Settings.enabled) {
        return text;
      }

      let string = `\u{1b}[${foregroundColors[k.toLowerCase()]}m`;
      if (background !== undefined) {
        string += background();
      }

      string += `${text}\u{1b}[0m`;

      return string;
    };
  });

  Object.keys(backgroundColors).forEach(k => {
    Background.prototype[k.toLowerCase()] = (text) => {
      return `\u{1b}[${backgroundColors[k.toLowerCase()]}m`;
    };
  });

  return this;
}

module.exports = new (function() {
  let instance;

  const disable = function() {
    Settings.enabled = false;

    Object.keys(ConsoleColor.prototype).forEach(function (color) {
      ConsoleColor.prototype[color] = function (text) {
        return text;
      };
    });
  };

  const setup = function() {
    instance = new ConsoleColor();
  };

  const initialize = function() {
    if (!Settings.enabled) {
      disable();
    }

    setup();
  };

  ////////////////////////////////////////////////////////////
  // Public methods
  ////////////////////////////////////////////////////////////
  const exported = {
    disableColors() {
      disable();
      setup();
    }
  };

  Object.defineProperty(exported, 'colors', {
    configurable: true,
    get: function() {
      return instance;
    }
  });

  Object.defineProperty(exported, 'colorsEnabled', {
    configurable: true,
    get: function() {
      return Settings.enabled;
    }
  });

  initialize();

  return exported;
});