refs skygragon#81: add session command.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

lib/commands/session.js

@@ -0,0 +1,123 @@
+'use strict';
+var prompt = require('prompt');
+var sprintf = require('sprintf-js').sprintf;
+
+var h = require('../helper');
+var chalk = require('../chalk');
+var log = require('../log');
+var core = require('../core');
+var session = require('../session');
+
+const cmd = {
+  command: 'session [keyword]',
+  desc:    'Manage sessions',
+  builder: function(yargs) {
+    return yargs
+      .option('c', {
+        alias:    'create',
+        type:     'boolean',
+        describe: 'Create session',
+        default:  false
+      })
+      .option('d', {
+        alias:    'delete',
+        type:     'boolean',
+        describe: 'Delete session',
+        default:  false
+      })
+      .option('e', {
+        alias:    'enable',
+        type:     'boolean',
+        describe: 'Enable/activate session',
+        default:  false
+      })
+      .positional('keyword', {
+        type:     'string',
+        describe: 'Session name or id',
+        default:  ''
+      })
+      .example(chalk.yellow('leetcode session'), 'Show all cache')
+      .example(chalk.yellow('leetcode session xxx'), 'Show session by keyword')
+      .example('', '')
+      .example(chalk.yellow('leetcode session -c xxx'), 'Create session with name')
+      .example(chalk.yellow('leetcode session -e xxx'), 'Enable session by keyword')
+      .example(chalk.yellow('leetcode session -d xxx'), 'Delete session by keyword');
+  }
+};
+
+function printSessions(e, sessions) {
+  if (e) return log.fail(e);
+
+  log.info(chalk.gray(sprintf(' %6s %5s %18s %28s %16s',
+    'Active', 'Id', 'Name', 'AC Questions', 'AC Submits')));
+  log.info(chalk.gray('-'.repeat(80)));
+
+  for (let s of sessions) {
+    let questionRate = 0;
+    let submissionRate = 0;
+    if (s.submitted_questions > 0)
+      questionRate = s.ac_questions * 100 / s.submitted_questions;
+    if (s.total_submitted > 0)
+      submissionRate = s.total_acs * 100 / s.total_submitted;
+
+    log.printf('   %s   %8d   %-26s %s (%6s %%) %s (%6s %%)',
+      s.is_active ? h.prettyState('ac') : ' ',
+      s.id,
+      s.name || 'Anonymous Session',
+      chalk.green(sprintf('%6s', s.ac_questions)),
+      sprintf('%.2f', questionRate),
+      chalk.green(sprintf('%6s', s.total_acs)),
+      sprintf('%.2f', submissionRate));
+  }
+}
+
+cmd.handler = function(argv) {
+  session.argv = argv;
+
+  if (argv.create)
+    return core.createSession(argv.keyword, printSessions);
+
+  core.getSessions(function(e, sessions) {
+    if (e) return log.fail(e);
+
+    if (argv.keyword) {
+      const key = Number(argv.keyword) || argv.keyword;
+      sessions = sessions.filter(x => x.name === key || x.id === key);
+      const session = sessions[0];
+      if (!session) return log.fail('Session not found!');
+
+      if (argv.enable && !session.is_active) {
+        core.activateSession(session, function(e, sessions) {
+          if (e) return log.fail(e);
+          require('../session').deleteCodingSession();
+          printSessions(e, sessions);
+        });
+        return;
+      }
+
+      if (argv.delete) {
+        log.info([
+          chalk.red.bold('CAREFUL! This action CANNOT be undone!'),
+          '\nThis will permanently delete all your submissions',
+          'and progress associated with this session.',
+          '\nAre you sure you want to delete this session?\n',
+          '\nPlease type in the session\'s',
+          chalk.yellow.bold('number of accepted submissions'),
+          'to confirm.\n'
+        ].join(' '));
+
+        prompt.colors = false;
+        prompt.message = '';
+        prompt.start();
+        prompt.get([{name: 'answer', type: 'integer', required: true}], function(e, x) {
+          if (x.answer !== session.total_acs) return;
+          return core.deleteSession(session, printSessions);
+        });
+        return;
+      }
+    }
+    printSessions(null, sessions);
+  });
+};
+
+module.exports = cmd;

lib/commands/stat.js

@@ -49,10 +49,11 @@ function printLine(key, done, all) {
   const n = 30;
   const percent = (all > 0) ? done / all : 0;
   const x = Math.ceil(n * percent);
-  log.printf(' %s\t%3d/%-3d (%.2f%%)  %s%s',
-      h.prettyLevel(key), done, all, 100 * percent,
-      chalk.green('█'.repeat(x)),
-      chalk.red('░'.repeat(n - x)));
+  log.printf(' %s\t%3d/%-3d (%6s %%)  %s%s',
+    h.prettyLevel(key), done, all,
+    sprintf('%.2f', 100 * percent),
+    chalk.green('█'.repeat(x)),
+    chalk.red('░'.repeat(n - x)));
 }
 
 function showProgress(problems) {

lib/config.js

@@ -35,6 +35,7 @@ const DEFAULT_CONFIG = {
       problem:         'https://leetcode.com/problems/$slug/description/',
       problem_detail:  'https://leetcode.com/graphql',
       test:            'https://leetcode.com/problems/$slug/interpret_solution/',
+      session:         'https://leetcode.com/session/',
       submit:          'https://leetcode.com/problems/$slug/submit/',
       submissions:     'https://leetcode.com/api/submissions/$slug',
       submission:      'https://leetcode.com/submissions/detail/$id/',

lib/plugins/cache.js

@@ -78,7 +78,7 @@ plugin.logout = function(user, purge) {
   if (!user) user = session.getUser();
   if (purge) session.deleteUser();
   // NOTE: need invalidate any user related cache
-  cache.del(h.KEYS.problems);
+  session.deleteCodingSession();
   return user;
 };
 

lib/plugins/leetcode.js

@@ -377,6 +377,43 @@ plugin.getFavorites = function(cb) {
   });
 };
 
+function runSession(method, data, cb) {
+  const opts = makeOpts(config.sys.urls.session);
+  opts.json = true;
+  opts.method = method;
+  opts.body = data;
+
+  const spin = h.spin('Waiting session result');
+  request(opts, function(e, resp, body) {
+    spin.stop();
+    e = checkError(e, resp, 200);
+    return cb(e, body.sessions);
+  });
+}
+
+plugin.getSessions = function(cb) {
+  log.debug('running leetcode.getSessions');
+  runSession('POST', {}, cb);
+};
+
+plugin.activateSession = function(session, cb) {
+  log.debug('running leetcode.activateSession');
+  const data = {func: 'activate', target: session.id};
+  runSession('PUT', data, cb);
+};
+
+plugin.createSession = function(name, cb) {
+  log.debug('running leetcode.createSession');
+  const data = {func: 'create', name: name};
+  runSession('PUT', data, cb);
+};
+
+plugin.deleteSession = function(session, cb) {
+  log.debug('running leetcode.deleteSession');
+  const data = {target: session.id};
+  runSession('DELETE', data, cb);
+};
+
 plugin.signin = function(user, cb) {
   log.debug('running leetcode.signin');
   const spin = h.spin('Signing in leetcode.com');

lib/session.js

@@ -30,6 +30,10 @@ session.deleteUser = function() {
   cache.del(h.KEYS.user);
 };
 
+session.deleteCodingSession = function() {
+  cache.del(h.KEYS.problems);
+};
+
 session.isLogin = function() {
   return this.getUser() !== null;
 };