Merge pull request #70 from prabalsingh24/project-submission

Project submission

Author: jatinkatyal13

src/rabbitmq/jobqueue.ts

@@ -26,7 +26,17 @@ export interface RunJob {
   scenario: string
 }
 
-export type JudgeJob = RunJob | SubmissionJob
+export interface ProjectJob {
+  id: number
+  source: string,
+  lang: string,
+  problem: string,
+  timelimit: number,
+  scenario: string,
+  submissionDirs: string
+}
+
+export type JudgeJob = RunJob | SubmissionJob | ProjectJob
 
 let jobQ = 'job_queue'
 let successQ = 'success_queue'
@@ -50,8 +60,17 @@ amqp.connect(`amqp://${config.AMQP.USER}:${config.AMQP.PASS}@${config.AMQP.HOST}
       jobChannel.consume(successQ, (msg) => {
         debug(`SUCCESS:CONSUME: msg.content = ${msg.content.toString()}`)
 
-        const payload = JSON.parse(msg.content.toString())        
-        const eventName = payload.testcases ? 'submit_result' : 'run_result'
+        const payload = JSON.parse(msg.content.toString())
+        let eventName;
+        if (payload.testcases) {
+          eventName = 'submit_result'
+        }
+        else if (payload.code && payload.score) {
+          eventName = 'project_result'
+        }
+        else {
+          eventName = 'run_result'
+        }
 
         successListener.emit(eventName, payload)
         jobChannel.ack(msg)

src/routes/api/project/controller.ts

@@ -0,0 +1,67 @@
+import { Request, Response } from 'express'
+import {ProjectJob, queueJob} from 'rabbitmq/jobqueue'
+import DB from 'models'
+import axios from 'axios'
+
+type ProjectResponse = {
+    id: number,
+    stdout: string,
+    stderr: string,
+    time: number,
+    code: number,
+    score: number
+}
+type RunPoolElement = {
+    res: Response
+}
+
+const RunPool: { [x: number]: RunPoolElement } = {}
+
+export default {
+    async runPOST(req: Request, res: Response) {
+        const mode = req.body.mode || 'sync'
+        const job = await DB.submissions.create({
+            lang: req.body.lang,
+            start_time: new Date(),
+            mode,
+            callback: req.body.callback
+        })
+
+        await queueJob(<ProjectJob>{
+            id: job.id,
+            source: req.body.submission,
+            problem: req.body.problem,
+            lang: req.body.lang,
+            submissionDirs: req.body.submissionDirs,
+            timelimit: req.body.timelimit,
+            scenario: 'project'
+        })
+
+
+        if (['callback', 'poll'].includes(mode)) {
+            return res.json({
+                id: job.id
+            })
+        }
+
+        // if mode === 'sync'
+        RunPool[job.id] = {
+            res
+        }
+    },
+
+    async onSuccess(result: ProjectResponse) {
+        const job = await DB.submissions.findById(result.id)
+        job.results = result
+        await job.save()
+
+        switch (job.mode) {
+            case 'callback':
+                await axios.post(job.callback, result)
+                break
+            case 'sync':
+                RunPool[job.id].res.json(result)
+                break
+        }
+    }
+}

src/routes/api/project/index.ts

@@ -0,0 +1,12 @@
+import { Router } from 'express'
+import Controller from './controller'
+import Validator from './validators'
+import { successListener } from 'rabbitmq/jobqueue'
+
+const router: Router = Router()
+const validator = new Validator()
+
+router.post('/', validator.POST, Controller.runPOST)
+successListener.on('project_result', Controller.onSuccess)
+
+export default router

src/routes/api/project/validators.ts

@@ -0,0 +1,35 @@
+import { Handler } from 'express'
+import BaseValidator from 'validators/baseValidator'
+import * as Joi from '@hapi/joi'
+
+export default class ProjectValidator extends BaseValidator {
+    POST: Handler
+
+    constructor() {
+        super()
+        this.POST = this.requestValidator(this.POSTSchema)
+    }
+
+    POSTSchema = Joi.object({
+        problem: Joi
+            .string()
+            .uri()
+            .required(),
+        submission: Joi
+            .string()
+            .uri()
+            .required(),
+        submissionDirs: Joi
+            .string()
+            .required(),
+        mode: Joi
+            .string()
+            .valid('sync', 'callback', 'poll'),
+        callback: Joi
+            .string()
+            .uri()
+            .when('mode', { is: 'callback', then: Joi.required() }),
+        timelimit: Joi
+            .number(),
+    })
+}