solution.discuss.js

var request = require('request');

var log = require('../log');
var chalk = require('../chalk');
var Plugin = require('../plugin');
var session = require('../session');

//
// [Usage]
//
// https://github.com/skygragon/leetcode-cli-plugins/blob/master/docs/solution.discuss.md
//
var plugin = new Plugin(200, 'solution.discuss', '2019.02.03',
    'Plugin to fetch most voted solution in discussions.');

var URL_DISCUSSES = 'https://leetcode.com/graphql';
var URL_DISCUSS = 'https://leetcode.com/problems/$slug/discuss/$id';

function setupHeaders(opts, user) {
  opts.headers.Cookie = 'LEETCODE_SESSION=' + user.sessionId +
                        ';csrftoken=' + user.sessionCSRF + ';';
  opts.headers['X-CSRFToken'] = user.sessionCSRF;
  opts.headers['x-csrftoken'] = user.sessionCSRF;
  opts.headers['X-Requested-With'] = 'XMLHttpRequest';

  opts.headers['User-Agent'] = 'Mozilla/5.0 (X11; Linux x86_64; rv:123.0) Gecko/20100101 Firefox/123.0';
  opts.headers['Referer'] = 'https://leetcode.com';
  opts.headers['Origin'] = 'https://leetcode.com/';
  opts.headers['Host'] = 'leetcode.com';
  opts.headers['Content-Type'] = 'application/json';
  opts.headers['Accept'] = 'application/json';
}

function getSolution(problem, lang, cb) {
  if (!problem) return cb();

  if (lang === 'python3') lang = 'python';

  var opts = {
    url:  URL_DISCUSSES,
    json: true,
    body: {
      query: [
        'query questionTopicsList($questionId: String!, $orderBy: TopicSortingOption, $skip: Int, $query: String, $first: Int!, $tags: [String!]) {',
        '  questionTopicsList(questionId: $questionId, orderBy: $orderBy, skip: $skip, query: $query, first: $first, tags: $tags) {',
        '    ...TopicsList',
        '  }',
        '}',
        'fragment TopicsList on TopicConnection {',
        '  totalNum',
        '  edges {',
        '    node {',
        '      id',
        '      title',
        '      post {',
        '        content',
        '        voteCount',
        '        author {',
        '          username',
        '        }',
        '      }',
        '    }',
        '  }',
        '}'
      ].join('\n'),

      operationName: 'questionTopicsList',
      variables:     JSON.stringify({
        query:      '',
        first:      1,
        skip:       0,
        orderBy:    'most_votes',
        questionId: '' + problem.id,
        tags:       [lang]
      })
    }
  };

  opts.headers = {};
  
  setupHeaders(opts, session.getUser());

  request(opts, function(e, resp, body) {
    if (e) return cb(e);
    if (resp.statusCode !== 200)
      return cb({msg: 'http error', statusCode: resp.statusCode});

    const solutions = body.data.questionTopicsList.edges;
    const solution = solutions.length > 0 ? solutions[0].node : null;
    return cb(null, solution);
  });
}

plugin.getProblem = function(problem, needTranslation, cb) {
  plugin.next.getProblem(problem, needTranslation, function(e, problem) {
    if (e || !session.argv.solution) return cb(e, problem);

    var lang = session.argv.lang;
    getSolution(problem, lang, function(e, solution) {
      if (e) return cb(e);
      if (!solution) return log.error('Solution not found for ' + lang);

      var link = URL_DISCUSS.replace('$slug', problem.slug).replace('$id', solution.id);
      var content = solution.post.content.replace(/\\n/g, '\n').replace(/\\t/g, '\t');

      log.info();
      log.info(problem.name);
      log.info();
      log.info(solution.title);
      log.info();
      log.info(chalk.underline(link));
      log.info();
      log.info('* Lang:    ' + lang);
      log.info('* Author:  ' + solution.post.author.username);
      log.info('* Votes:   ' + solution.post.voteCount);
      log.info();
      log.info(content);
    });
  });
};

module.exports = plugin;