ojscraper/test/index.js

"A module that scraps user information from various OJ"

const express = require('express');
const app = express();
const server = require('http').createServer(app);

const ojscraper = require('../index');

app.set('port', 4000);

/* Basic Routes */
app.get('/userinfo/:ojname/:username', function(req, res, next) {
  const { ojname, username } = req.params;
  const subojname = req.query.suboj;

  let userId;
  let password;
  if (ojname === 'loj') {
    userId = req.query.userId;
    password = req.query.password;
  }

  ojscraper.getUserInfo({ojname, username, subojname, credential: {userId, password}})
    .then(function(stats) {
      return res.send(stats);
    })
    .catch(next);
});

app.get('/probleminfo/:ojname/:problemID', function(req, res, next) {
  const { ojname, problemID } = req.params;

  let userId;
  let password;
  if (ojname === 'loj') {
    userId = req.query.userId;
    password = req.query.password;
  }

  ojscraper.getProblemInfo({ojname, problemID, credential: {userId, password}})
    .then(function(stats) {
      return res.send(stats);
    })
    .catch(next);
});

app.use(function(err, req, res, next) {
  console.error(err.stack);
  res.status(500).send(err.message);
});

app.get('*', function(req, res) {
  return res.status(404).send('Page not found\n');
});

if (require.main === module) {
  server.listen(app.get('port'), function() {
    console.log(`Server running at port ${app.get('port')}`);
  });
} else {
  module.exports = {
    server,
    app,
  };
}