class Boot{
  LiftRules.dispatch.append {
    case Req( "captcha" :: Nil, _, _ ) => Captcha.challenge _
  }
}

import com.octo.captcha.service.image._
import com.octo.captcha.service.captchastore._
import com.octo.captcha.engine.image.gimpy._
import com.sun.image.codec.jpeg._
import java.io._
import java.util.Locale._
import net.liftweb.common._
import net.liftweb.http._

object Captcha {

  lazy val service =
    new DefaultManageableImageCaptchaService(
      new FastHashMapCaptchaStore, new GmailEngine,
      180,100,75)

  def getCaptcha:Box[ Array[Byte] ] = S.session match {
   case Full(session)=>
    val jpeg = new ByteArrayOutputStream
    val id = session.uniqueId
    val challenge = service.getImageChallengeForID(id, JAPAN)
    JPEGCodec.createJPEGEncoder(jpeg).encode(challenge)
    Full( jpeg.toByteArray )
   case _ => Empty
  }

  //Captchaを生成してレスポンス
  def challenge: Box[LiftResponse] = {
   getCaptcha match{
    case Full( capture ) =>
     val headers = ("Content-type" -> "image/jpeg") ::
      ("Cache-Control" -> "no-store") ::
      ("Pragma", "no-cache") ::
      ("Expires", "0" ) ::
      ("Content-length" -> capture.length.toString) :: Nil

     Full(StreamingResponse( new java.io.ByteArrayInputStream(capture),
      () => {}, capture.length, headers, Nil, 200) )
    case _ => Empty
   }
  }

  //ユーザー入力が正しいかチェック
  def validate(response:String):Boolean = S.session match{
   case Full(session) =>
    service.validateResponseForID(session.uniqueId,response)
      .asInstanceOf[Boolean]
   case _ => false
  }

}