Merge pull request #4 from leetcode-roulette/feature/get-submission-information

Feature/get submission information

Author: hicks8989

package.json

@@ -5,8 +5,8 @@
   "main": "index.js",
   "scripts": {
     "build": "tsc",
-    "dev": "nodemon ./src/app.ts",
-    "start": "node ./dist/app.js",
+    "dev": "nodemon ./src/index.ts",
+    "start": "node ./dist/index.js",
     "test": "jest --passWithNoTests"
   },
   "repository": {

src/app.ts

@@ -1,25 +1,12 @@
 import { config } from 'dotenv';
 import express, { Application, NextFunction, Request, Response } from 'express';
-import { PopulateProblems } from './db/problems';
-import { Database } from './db/db.config';
 import healthcheck from './healthcheck';
 
 config();
 
-const serve = async () : Promise<void> => {
+export const app = (() => {
     const app: Application = express();
-
-    await Database.connect();
-
-    const PORT = process.env.PORT || 3000;
-
     app.use("/", healthcheck);
 
-    app.listen(PORT, () => {
-        console.log(`Server is listening on port ${PORT}`);
-    });
-
-    await PopulateProblems.populate();
-}
-
-serve();
+    return app;
+})();

src/db/problems.ts

@@ -23,25 +23,45 @@ export class PopulateProblems {
     const rowAlreadyExists : boolean = await this.rowAlreadyExists(problem);
 
     if (rowAlreadyExists) {
+      await this.updateProblem(problem);
       return;
     }
 
+    await this.saveProblem(problem);
+  }
+
+  private static async rowAlreadyExists(problem : LeetcodeProblem) : Promise<boolean> {
+    return await Problems.exists({ problemId: problem.stat.question_id }) !== null;
+  }
+
+  private static async updateProblem(problem : LeetcodeProblem) : Promise<void> {
+    Problems.findOneAndUpdate({ problemId: problem.stat.question_id}, {
+      problemId: problem.stat.question_id,
+      title: problem.stat.question__title,
+      titleSlug: problem.stat.question__title_slug,
+      isPremium: problem.paid_only,
+      difficulty: problem.difficulty.level,
+      frontEndId: problem.stat.frontend_question_id,
+      numSubmitted: problem.stat.total_submitted,
+      numAccepted: problem.stat.total_acs
+    });
+  }
+
+  private static async saveProblem(problem : LeetcodeProblem) : Promise<void> {
     const p : IProblem = await Problems.create({
       problemId: problem.stat.question_id,
       title: problem.stat.question__title,
       titleSlug: problem.stat.question__title_slug,
       isPremium: problem.paid_only,
       difficulty: problem.difficulty.level,
-      frontEndId: problem.stat.frontend_question_id
+      frontEndId: problem.stat.frontend_question_id,
+      numSubmitted: problem.stat.total_submitted,
+      numAccepted: problem.stat.total_acs
     });
 
     await p.save();
   }
 
-  private static async rowAlreadyExists(problem : LeetcodeProblem) : Promise<boolean> {
-    return await Problems.exists({ problemId: problem.stat.question_id }) !== null;
-  }
-
   public static get problems() : Promise<LeetcodeProblem[] | null> | null {
     if (this.data === null) {
       this.data = ProblemExtractor.problems;

src/index.ts

@@ -0,0 +1,20 @@
+import { app } from "./app";
+import { Database } from "./db/db.config";
+import { PopulateProblems } from "./db/problems";
+
+const serve = async () : Promise<void> => {
+  const PORT : string | number = process.env.PORT || 3000;
+
+  try {
+    await Database.connect();
+    await PopulateProblems.populate();
+  } catch(e) {
+    console.log("Exception caught populating database: " + e);
+  }
+
+  app.listen(PORT, () => {
+      console.log(`Server is listening on port ${PORT}`);
+  });
+}
+
+serve();

src/models/problems.ts

@@ -6,7 +6,9 @@ export interface IProblem extends Document {
   titleSlug: string,
   isPremium: boolean,
   difficulty: number,
-  frontEndId: number
+  frontEndId: number,
+  numSubmitted: number,
+  numAccepted: number
 };
 
 const problemsSchema : Schema<IProblem> = new mongoose.Schema({
@@ -15,7 +17,9 @@ const problemsSchema : Schema<IProblem> = new mongoose.Schema({
   titleSlug: 'string',
   isPremium: 'boolean',
   difficulty: 'number',
-  frontEndId: 'number'
+  frontEndId: 'number',
+  numSubmitted: 'number',
+  numAccepted: 'number'
 });
 
 const Problems : Model<IProblem> = mongoose.model('problems', problemsSchema);