'use strict';

var Benchmark = require('benchmark');

var strings = [
/* 0 */ 'lastname',
/* 1 */ 'firstname123@gmail.com',
/* 2 */ 'line\nbreak',
/* 3 */ '"supercalifra\r\ngilisticexpialidocious"',
/* 4 */ 'Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.',
/* 5 */ 'Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.\nUt enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.\nDuis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.\nExcepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.',
/* 6 */ JSON.stringify({name: 'John Brown', id: 123456789, now: new Date(), settings: {on: true, value: 5, defaultView: 'activity'}}),
];
var string = strings[process.argv[2] || 0];

var chars_g = /[\0\b\t\n\r\x1a\"\'\\]/g;

function charReplacer(s) {
  switch (s) {
    case '\0': return '\\0';
    case '\n': return '\\n';
    case '\r': return '\\r';
    case '\b': return '\\b';
    case '\t': return '\\t';
    case '\x1a': return '\\Z';
    default: return '\\' + s;
  }
}

function escapeString0(val) {
  return "'" + val.replace(chars_g, charReplacer) + "'";
}

function escapeString1(val) {
  var escapedVal = '';
  var chunkIndex = 0;

  for (var i = 0; i < val.length; i++) {
    switch (val.charCodeAt(i)) {
      case 0: // '\0'
        escapedVal += val.slice(chunkIndex, i) + '\\0';
        break;
      case 8: // '\b'
        escapedVal += val.slice(chunkIndex, i) + '\\b';
        break;
      case 9: // '\t'
        escapedVal += val.slice(chunkIndex, i) + '\\t';
        break;
      case 10: // '\n'
        escapedVal += val.slice(chunkIndex, i) + '\\n';
        break;
      case 13: // '\r'
        escapedVal += val.slice(chunkIndex, i) + '\\r';
        break;
      case 26: // '\x1a'
        escapedVal += val.slice(chunkIndex, i) + '\\Z';
        break;
      case 34: // '"'
      case 39: // '\''
      case 92: // '\\'
        escapedVal += val.slice(chunkIndex, i) + '\\' + val[i];
        break;
      default:
        continue;
    }
    chunkIndex = i + 1;
  }

  if (chunkIndex === 0) { // Nothing was escaped
    return "'" + val + "'";
  }
  if (chunkIndex < val.length) {
    return "'" + escapedVal + val.slice(chunkIndex) + "'";
  }
  return "'" + escapedVal + "'";
}

var chars = /[\0\b\t\n\r\x1a\"\'\\]/;
function escapeString2(val) {
  if (!chars.test(val)) {
    return "'" + val + "'";
  }

  var escapedVal = '';
  var chunkIndex = 0;

  for (var i = 0; i < val.length; i++) {
    switch (val.charCodeAt(i)) {
      case 0: // '\0'
        escapedVal += val.slice(chunkIndex, i) + '\\0';
        break;
      case 8: // '\b'
        escapedVal += val.slice(chunkIndex, i) + '\\b';
        break;
      case 9: // '\t'
        escapedVal += val.slice(chunkIndex, i) + '\\t';
        break;
      case 10: // '\n'
        escapedVal += val.slice(chunkIndex, i) + '\\n';
        break;
      case 13: // '\r'
        escapedVal += val.slice(chunkIndex, i) + '\\r';
        break;
      case 26: // '\x1a'
        escapedVal += val.slice(chunkIndex, i) + '\\Z';
        break;
      case 34: // '"'
      case 39: // '\''
      case 92: // '\\'
        escapedVal += val.slice(chunkIndex, i) + '\\' + val[i];
        break;
      default:
        continue;
    }
    chunkIndex = i + 1;
  }

  if (chunkIndex < val.length) {
    return "'" + escapedVal + val.slice(chunkIndex) + "'";
  }
  return "'" + escapedVal + "'";
}

function escapeString3(val) {
  var chunkIndex = chars_g.lastIndex = 0;
  var escapedVal = '';
  var match;

  while ((match = chars_g.exec(val))) {
    switch (match[0]) {
      case '\0':
        escapedVal += val.slice(chunkIndex, match.index) + '\\0';
        break;
      case '\b':
        escapedVal += val.slice(chunkIndex, match.index) + '\\b';
        break;
      case '\t':
        escapedVal += val.slice(chunkIndex, match.index) + '\\t';
        break;
      case '\n':
        escapedVal += val.slice(chunkIndex, match.index) + '\\n';
        break;
      case '\r':
        escapedVal += val.slice(chunkIndex, match.index) + '\\r';
        break;
      case '\x1a':
        escapedVal += val.slice(chunkIndex, match.index) + '\\Z';
        break;
      default:
        escapedVal += val.slice(chunkIndex, match.index) + '\\' + match[0];
        break;
    }
    chunkIndex = chars_g.lastIndex;
  }

  if (chunkIndex === 0) { // Nothing was escaped
    return "'" + val + "'";
  }
  if (chunkIndex < val.length) {
    return "'" + escapedVal + val.slice(chunkIndex) + "'";
  }
  return "'" + escapedVal + "'";
}

new Benchmark.Suite()
  .add('escapeString0', () => {
    escapeString0(string);
  })
  .add('escapeString1', () => {
    escapeString1(string);
  })
  .add('escapeString2', () => {
    escapeString2(string);
  })
  .add('escapeString3', () => {
    escapeString3(string);
  })
  .on('cycle', function(event) {
    console.log(String(event.target));
  })
  .on('complete', function() {
    console.log('Fastest is ' + this.filter('fastest').map('name'));
  })
  .run();