Merge pull request #44 from vaibhavsijaria/main

bearer token in header for judge0 api call

Author: RustyDev24

.env.example

@@ -13,6 +13,7 @@ REDIS_PORT=6379
 REDIS_PASSWORD=huehuehue
 
 JUDGE0_URI=https://google.com
+JUDGE0_TOKEN=sk_live_huehuehue
 
 JWT_KEY=secret
 CALLBACK_URL=https://google.com/callback

internal/controllers/runcode.go

@@ -1,7 +1,6 @@
 package controllers
 
 import (
-	"bytes"
 	"encoding/json"
 	"fmt"
 	"net/http"
@@ -49,15 +48,14 @@ func RunCode(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	judge0URL, err := url.Parse(JUDGE0_URI + "/submissions/")
+	judge0URL, err := url.Parse(JUDGE0_URI + "/submissions")
 	if err != nil {
 		httphelpers.WriteError(w, http.StatusInternalServerError, fmt.Sprintf("Error parsing Judge0 URL: %v", err))
 		return
 	}
 	params := url.Values{}
 	params.Add("base64_encoded", "true")
 	params.Add("wait", "true")
-	judge0URL.RawQuery = params.Encode()
 
 	var payload submission.Submission
 	response := resp{
@@ -86,7 +84,7 @@ func RunCode(w http.ResponseWriter, r *http.Request) {
 			return
 		}
 
-		result, err := http.Post(judge0URL.String(), "application/json", bytes.NewBuffer(payloadJSON))
+		result, err := submission.SendToJudge0(judge0URL, params, payloadJSON)
 		if err != nil {
 			logger.Errof("Error sending request to Judge0: %v", err)
 			httphelpers.WriteError(w, http.StatusInternalServerError, fmt.Sprintf("Error sending request to Judge0: %v", err))

internal/controllers/submission.go

@@ -1,7 +1,6 @@
 package controllers
 
 import (
-	"bytes"
 	"encoding/json"
 	"fmt"
 	"net/http"
@@ -75,8 +74,8 @@ func SubmitCode(w http.ResponseWriter, r *http.Request) {
 
 	params := url.Values{}
 	params.Add("base64_encoded", "true")
-	judge0URL.RawQuery = params.Encode()
-	resp, err := http.Post(judge0URL.String(), "application/json", bytes.NewBuffer(payload))
+
+	resp, err := submission.SendToJudge0(judge0URL, params, payload)
 	if err != nil {
 		logger.Errof("Error sending request to Judge0: %v", err)
 		httphelpers.WriteError(w, http.StatusInternalServerError, fmt.Sprintf("Error sending request to Judge0: %v", err))

internal/helpers/submission/common.go

@@ -1,11 +1,17 @@
 package submission
 
 import (
+	"bytes"
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
+	"net/http"
+	"net/url"
+	"os"
 )
 
+var bearer = os.Getenv("JUDGE0_TOKEN")
+
 type Submission struct {
 	LanguageID int     `json:"language_id"`
 	SourceCode string  `json:"source_code"`
@@ -60,3 +66,22 @@ func RuntimeMut(language_id int) (int, error) {
 	}
 	return runtime_mut, nil
 }
+
+func SendToJudge0(judge0Url *url.URL, params url.Values, payload []byte) (*http.Response, error) {
+	judge0Url.RawQuery = params.Encode()
+	judgereq, err := http.NewRequest("POST", judge0Url.String(), bytes.NewReader(payload))
+	if err != nil {
+		return nil, fmt.Errorf("error creating request to Judge0: %v", err)
+	}
+
+	judgereq.Header.Add("Content-Type", "application/json")
+	judgereq.Header.Add("Accept", "application/json")
+	judgereq.Header.Add("Authorization", fmt.Sprintf("Bearer %v", bearer))
+
+	resp, err := http.DefaultClient.Do(judgereq)
+	if err != nil {
+		return nil, fmt.Errorf("error sending request to Judge0: %v", err)
+	}
+
+	return resp, nil
+}