From c43ffd6dd31658fefc2749990f17b6d4a153386d Mon Sep 17 00:00:00 2001
From: Rahul Rudragoudar <rr83019@gmail.com>
Date: Tue, 12 Mar 2019 18:17:34 +0530
Subject: [PATCH] Added GET request support for media end point

---
 src/main/scala/lc/FilterChallenge.scala |  2 +-
 src/main/scala/lc/Main.scala            | 14 ++++++++++----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/main/scala/lc/FilterChallenge.scala b/src/main/scala/lc/FilterChallenge.scala
index 6f8e5f0..690ceab 100644
--- a/src/main/scala/lc/FilterChallenge.scala
+++ b/src/main/scala/lc/FilterChallenge.scala
@@ -7,7 +7,7 @@ import java.awt.Font
 import java.awt.Color
 
 class FilterChallenge extends ChallengeProvider {
-  def getId = "filter"
+  def getId = "FilterChallenge"
   def returnChallenge(): Challenge = {
     val filterTypes = List(new FilterType1, new FilterType2)
     val r = new scala.util.Random
diff --git a/src/main/scala/lc/Main.scala b/src/main/scala/lc/Main.scala
index 4755a52..92a2897 100644
--- a/src/main/scala/lc/Main.scala
+++ b/src/main/scala/lc/Main.scala
@@ -126,14 +126,20 @@ class Server(port: Int){
     },"POST")
 
     host.addContext("/v1/media",(req, resp) => {
-    	val body = req.getJson()
-    	val json = parse(body)
-    	val id = json.extract[Id]
+    	var id = Id(null)
+    	if ("GET" == req.getMethod()){
+    		val params = req.getParams()
+    		id = Id(params.get("id"))
+    	} else {
+    		val body = req.getJson()
+    		val json = parse(body)
+    		id = json.extract[Id]
+    	}
     	val image = captcha.getCaptcha(id)
     	resp.getHeaders().add("Content-Type","image/png")
     	resp.send(200, image)
     	0
-    },"POST")
+    },"POST", "GET")
 
     host.addContext("/v1/answer",(req, resp) => {
     	val body = req.getJson()