add: api for predictions

Author: SysSn13

background.js

@@ -13,7 +13,6 @@ const { addQueue, removeQueue, setQueues, replaceQueues } = createBullBoard({
 
 serverAdapter.setBasePath("/bull-board");
 
-console.log("background is running...");
 const initScheduler = async () => {
     await jobScheduler.add({});
     await jobScheduler.add({}, { repeat: { cron: "0 0 * * *" } });

controllers/predictionsController.js

@@ -0,0 +1,48 @@
+const Contest = require("../models/contest");
+
+exports.get = function (req, res) {
+    if (!req.query.contestId || !req.query.handles) {
+        res.status(400).send("Invalid query params");
+        return;
+    }
+    const contestId = req.query.contestId;
+    const handles = req.query.handles
+        .split(";")
+        .map((handle) => {
+            return handle.trim();
+        })
+        .filter((handle) => handle != "");
+    handles.length = Math.min(handles.length, 25);
+
+    Contest.aggregate(
+        [
+            {
+                $project: {
+                    contest_id: 1,
+                    _id: 1,
+                    "rankings._id": 1,
+                    "rankings.delta": 1,
+                    "rankings.data_region": 1,
+                },
+            },
+            { $match: { _id: contestId } },
+            { $unwind: "$rankings" },
+            { $match: { "rankings._id": { $in: handles } } },
+        ],
+        function (err, result) {
+            let resp = {};
+            if (err) {
+                resp.status = "FAILED";
+                res.status(500).send(err);
+            } else {
+                resp.status = "OK";
+                resp.meta = {
+                    total_count: result.length,
+                    contest_id: contestId,
+                };
+                resp.items = result.map((item) => item.rankings);
+                res.send(resp);
+            }
+        }
+    );
+};

main.js

@@ -4,34 +4,45 @@ if (process.env.NODE_ENV !== "production") {
 const express = require("express");
 const expressLayouts = require("express-ejs-layouts");
 
+// database
 const mongoose = require("mongoose");
 mongoose.connect(process.env.DATABASE_URL, {
     useNewUrlParser: true,
     useUnifiedTopology: true,
 });
 mongoose.set("useFindAndModify", false);
-
 const db = mongoose.connection;
-
-const app = express();
-
 db.on("error", (error) => console.error(error));
 db.once("open", () => {
-    console.log("Connected to Mongoose");
+    console.info("Connected to Mongoose");
 });
 
+const app = express();
+
+// background
 if (process.env.BACKGROUND == true) {
     const { bullBoardServerAdapter } = require("./background");
     app.use("/bull-board", bullBoardServerAdapter.getRouter());
+    console.info("BACKGROUND is up.");
 }
+
+// web
 if (process.env.WEB == true) {
-    const webRouter = require("./server");
+    const webRouter = require("./web");
     app.set("view engine", "ejs");
     app.set("views", __dirname + "/views");
     app.set("layout", "layouts/layout");
     app.use(expressLayouts);
     app.use(express.static("public"));
     app.use("/", webRouter);
+    console.info("WEB is up.");
+}
+
+// api
+if (!process.env.API_DISABLED) {
+    const apiRoutes = require("./routes/api");
+    app.use("/api/v1/", apiRoutes);
+    console.info("API is up.");
 }
 
 const port = process.env.PORT || 8080;
@@ -41,5 +52,5 @@ app.listen(port, (err) => {
         console.error(err);
         return;
     }
-    console.log("Listening on " + port);
+    console.info("Listening on " + port);
 });

package.json

@@ -8,7 +8,7 @@
   "main": "server.js",
   "scripts": {
     "start": "node main.js --max_old_space_size=512",
-    "dev": "nodemon main.js --max_old_space_size=1024",
+    "dev": "nodemon main.js --max_old_space_size=1024  --trace-warnings",
     "buildAddon": "npm run build --prefix ./services/predict-addon/"
   },
   "keywords": [],

routes/api.js

@@ -0,0 +1,11 @@
+const router = require("express").Router();
+const predictionsController = require("../controllers/predictionsController.js");
+router.get("/ping", function (req, res) {
+    res.json({
+        status: "OK",
+        message: "pong",
+    });
+});
+
+router.route("/predictions").get(predictionsController.get);
+module.exports = router;

routes/index.js

@@ -3,10 +3,6 @@ const Contest = require("../models/contest");
 const { fetchContest, fetchContestRankings } = require("../services/contests");
 const router = express.Router();
 
-//fetchContest()
-//fetchContestRankings('weekly-contest-242')
-//fetchContestRankings('weekly-contest-240')
-
 router.get("/", async (req, res) => {
     try {
         let contests = await Contest.find({}, { rankings: 0 }).sort({

services/contests.js

@@ -1,7 +1,5 @@
 const fetch = require("node-fetch");
-
 const Contest = require("../models/contest");
-let halfHour = 1000 * 60 * 30;
 
 // async function pushRankingsToContest(
 //     contest_id,

services/helpers.js

@@ -0,0 +1,3 @@
+exports.getUserId = function (username, dataRegion = "US") {
+    return dataRegion + "/" + username.trim().toLowerCase();
+};

services/users.js

@@ -4,10 +4,7 @@ const fetch = require("node-fetch");
 
 const BASE_URL = "https://leetcode.com";
 const BASE_CN_URL = "https://leetcode-cn.com";
-
-function getUserId(username, dataRegion = "US") {
-    return dataRegion + "/" + username.trim().toLowerCase();
-}
+const { getUserId } = require("./helpers");
 async function getLastContestStartTime(user_id) {
     try {
         let lastStartTime = 0;