Initial commit for the framework

build.sbt

@@ -16,3 +16,5 @@ lazy val root = (project in file(".")).
     libraryDependencies += "com.sksamuel.scrimage" %% "scrimage-filters" % "2.1.8"
 
 )
+
+fork in run := true

src/main/scala/FilterCaptcha.scala → src/main/scala/lc/FilterChallenge.scala

@@ -1,36 +1,34 @@
+package lc
+
 import com.sksamuel.scrimage._
 import com.sksamuel.scrimage.filter._
 import java.awt.image.BufferedImage
-import java.awt.{Graphics2D,Color,Font}
+import java.awt.Font
+import java.awt.Color
 
-class FilterCaptcha extends CaptchaProvider {
-  val tokenAnswer = scala.collection.mutable.Map[String, String]()
-  def getChallenge(): Challenge = {
+class FilterChallenge extends ChallengeProvider {
+  val id = "filter"
+  def returnChallenge(): (Image, String) = {
     val filterTypes = List(new FilterType1, new FilterType2)
     val r = new scala.util.Random
     val alphabet = "abcdefghijklmnopqrstuvwxyz"
     val n = 8
-    val answer = Stream.continually(r.nextInt(alphabet.size)).map(alphabet).take(n).mkString
-    val token = scala.util.Random.nextInt(10000).toString
-    synchronized {
-      tokenAnswer += token -> answer
-    }
+    val secret = Stream.continually(r.nextInt(alphabet.size)).map(alphabet).take(n).mkString
     val canvas = new BufferedImage(225, 50, BufferedImage.TYPE_INT_RGB)
     val g = canvas.createGraphics()
     g.setColor(Color.WHITE)
     g.fillRect(0, 0, canvas.getWidth, canvas.getHeight)
     g.setColor(Color.BLACK)
     g.setFont(new Font("Serif", Font.PLAIN, 30))
-    g.drawString(answer, 5, 30)
+    g.drawString(secret, 5, 30)
     g.dispose()
     var image = new Image(canvas, ImageMetadata.empty)
     val s = scala.util.Random.nextInt(2)
     image = filterTypes(s).applyFilter(image)
-    val challenge = new Challenge(token, image)
-    challenge
+    (image, secret)
   }
-  def checkAnswer(token: String, input: String): Boolean = {
-    tokenAnswer(token) == input
+  def checkAnswer(secret: String, answer: String): Boolean = {
+    secret == answer
   }
 }
 
@@ -61,3 +59,4 @@ class FilterType2 extends FilterType {
     image
   }
 }
+

src/main/scala/lc/Main.scala

@@ -0,0 +1,79 @@
+package lc
+
+import com.sksamuel.scrimage._
+import java.sql._
+import java.io._
+
+trait ChallengeProvider {
+  val id: String
+  def returnChallenge(): (Image, String)
+  def checkAnswer(secret: String, answer: String): Boolean
+  //TODO: def configure(): Unit
+}
+
+class Captcha {
+  val con: Connection = DriverManager.getConnection("jdbc:h2:./captcha", "sa", "")
+  val stmt: Statement = con.createStatement();
+  stmt.execute("CREATE TABLE IF NOT EXISTS challenge(token varchar, id varchar, secret varchar, image blob)");
+
+  def getCaptcha(): Boolean = {
+    val provider = new FilterChallenge
+    val (token, image) = this.getChallenge(provider)
+    val stmt: Statement = con.createStatement();
+    image.output(new File("Captcha.png"))
+    println(s"Token: ${token}")
+    println("Enter your answer: ")
+    val answer = scala.io.StdIn.readLine()
+    this.getAnswer(token, answer, provider)
+  }
+
+  def getChallenge(provider: ChallengeProvider): (String, Image) = {
+    val (image, secret) = provider.returnChallenge()
+    val token = scala.util.Random.nextInt(10000).toString
+    var pstmt: PreparedStatement = null
+    pstmt = con.prepareStatement("INSERT INTO challenge(token, id, secret) VALUES (?, ?, ?)")
+    pstmt.setString(1, token)
+    pstmt.setString(2, provider.id)
+    pstmt.setString(3, secret)
+    //TODO: insert image into database
+    pstmt.executeUpdate()
+    (token, image)
+  }
+
+  def getAnswer(token: String, answer: String, provider: ChallengeProvider): Boolean = {
+    val stmt: Statement  = con.createStatement();
+    val rs: ResultSet = stmt.executeQuery("SELECT secret FROM challenge WHERE token = "+token);
+    rs.next()
+    val secret = rs.getString("secret")
+    provider.checkAnswer(secret, answer)
+  }
+
+  def display(): Unit = {
+    val stmt: Statement = con.createStatement();
+		val rs: ResultSet = stmt.executeQuery("SELECT * FROM challenge");
+
+		println("token\t\tid\t\tsecret\t\timage");
+		while(rs.next()) {
+			val token = rs.getString("token");
+			val id = rs.getString("id");
+			val secret = rs.getString("secret");
+			val image = rs.getString("image");
+			println(s"${token}\t\t${id}\t\t${secret}\t\t${image}")
+		}
+  }
+
+  def closeConnection(): Unit = {
+	  con.close()
+  } 
+}
+
+object LCFramework{
+  def main(args: scala.Array[String]) {
+    val captcha = new Captcha
+    val result = captcha.getCaptcha()
+    println(result)
+    captcha.display()
+    captcha.closeConnection()
+  } 
+}
+