Scala Version of Cookie Jar

cookie-jar.scala

/* This version of Cookie Jar was written by Jan Friderici (https://github.com/fride) */

sealed trait CookieJar {
    def takeCookies(cookies: Int) : CookieJar
}
case object EmptyCookieJar extends CookieJar  {
    def takeCookies(cookies: Int) : CookieJar = this
}
case class FilledCookieJar(cookies: Int) extends CookieJar  {
    def takeCookies(cookies: Int) : CookieJar =
        if (this.cookies <= cookies) EmptyCookieJar
        else this.copy(cookies = this.cookies - cookies)
}
type CookieJarAction = CookieJar => CookieJar
object CookieJar {
     
    def create: CookieJar = FilledCookieJar(10)
     
    def cat: CookieJarAction =  {
        case EmptyCookieJar => EmptyCookieJar
        case c @ FilledCookieJar(_) => c.takeCookies(1)
      }
     
    def monster: CookieJarAction =
      jar => EmptyCookieJar
     
    def grandma: CookieJarAction = {
        case EmptyCookieJar => FilledCookieJar(20)      
        case FilledCookieJar(cookies) => FilledCookieJar(cookies + 20)
         
    }
    def logged(action: CookieJarAction) : CookieJarAction= {
      jar => {
        println(s"cookieJar: $jar, action: $action")
        action(jar)
      }
    }
}
import CookieJar._
val actions = grandma compose cat compose monster compose cat
println(actions(create))