list.js

var _ = require('underscore');
var sprintf = require('sprintf-js').sprintf;

var h = require('../helper');
var chalk = require('../chalk');
var icon = require('../icon');
var log = require('../log');
var core = require('../core');
var session = require('../session');

var cmd = {
  command: 'list [keyword]',
  desc:    'list problems',
  builder: {
    keyword: {
      type:     'string',
      default:  '',
      describe: 'Filter problems by keyword'
    },
    query: {
      alias:    'q',
      type:     'string',
      default:  '',
      describe: 'Filter problems by conditions:\n' +
                'e(easy),m(medium),h(hard),d(done),l(locked),s(starred)\n' +
                'Uppercase means negative, e.g. D(not done)'
    },
    stat: {
      alias:    's',
      type:     'boolean',
      default:  false,
      describe: 'Show problems statistics'
    },
    tag: {
      alias:    't',
      type:     'string',
      default:  '',
      describe: 'Filter problems by tags'
    }
  }
};

function byLevel(x, q) {
  return x.level[0].toLowerCase() === q.toLowerCase();
}

function byStateAC(x, q) {
  return x.state === 'ac';
}

function byLocked(x, q) {
  return x.locked;
}

function byStarred(x, q) {
  return x.starred;
}

var QUERY_HANDLERS = {
  e: byLevel,
  E: _.negate(byLevel),
  m: byLevel,
  M: _.negate(byLevel),
  h: byLevel,
  H: _.negate(byLevel),
  l: byLocked,
  L: _.negate(byLocked),
  d: byStateAC,
  D: _.negate(byStateAC),
  s: byStarred,
  S: _.negate(byStarred)
};

cmd.handler = function(argv) {
  session.argv = argv;
  core.getProblems(function(e, problems) {
    if (e) return log.fail(e);

    var all = problems.length;

    if (argv.query) {
      argv.query.split('').forEach(function(q) {
        var f = QUERY_HANDLERS[q];
        if (!f) return;

        problems = _.filter(problems, _.partial(f, _, q));
      });
    }

    if (argv.tag) {
      var tag = argv.tag.toLowerCase();
      // TODO: fill company/tags in problems
      problems = _.filter(problems, function(x) {
        return x.category === tag ||
          (_.isArray(x.companies) && x.companies.indexOf(tag) !== -1) ||
          (_.isArray(x.tags) && x.tags.indexOf(tag) !== -1);
      });
    }

    var word = String(argv.keyword).toLowerCase();
    if (word) {
      if (word.endsWith(word.substr(-1).repeat(6))) {
        log.warn('Hmmm...you might need a new keyboard?');
      }
      problems = _.filter(problems, function(x) {
        return x.name.toLowerCase().indexOf(word) !== -1;
      });
    }

    var stat = {locked: 0, starred: 0};
    problems.forEach(function(problem) {
      stat[problem.level] = (stat[problem.level] || 0) + 1;
      stat[problem.state] = (stat[problem.state] || 0) + 1;
      if (problem.locked) ++stat.locked;
      if (problem.starred) ++stat.starred;

      log.info(sprintf('%s %s %s [%3d] %-60s %-6s (%.2f %%)',
            (problem.starred ? chalk.yellow(icon.like) : icon.none),
            (problem.locked ? chalk.red(icon.lock) : icon.none),
            h.prettyState(problem.state),
            problem.id,
            problem.name,
            problem.level,
            problem.percent));
    });

    if (argv.stat) {
      log.info();
      log.info(sprintf('  All:    %-9d Listed:  %-9d', all, problems.length));
      log.info(sprintf('  Locked: %-9d Starred: %-9d', stat.locked, stat.starred));
      log.info(sprintf('  Accept: %-9d Not-AC:  %-9d New:  %-9d',
            (stat.ac || 0), (stat.notac || 0), (stat.None || 0)));
      log.info(sprintf('  Easy:   %-9d Medium:  %-9d Hard: %-9d',
            (stat.Easy || 0), (stat.Medium || 0), (stat.Hard || 0)));
    }
  });
};

module.exports = cmd;