Bitbucket Server condition in Scala to only display a button if open tasks exist

HasOpenTasksCondition.scala

import java.util.{Map => JMap}
import ch.mibex.crossproduct.api.OpenTasksCounter
import com.atlassian.bitbucket.pull.PullRequest
import com.atlassian.plugin.web.Condition

class HasOpenTasksCondition(openTasksCounter: OpenTasksCounter[PullRequest]) extends Condition {
  
  override def init(context: JMap[String, String]): Unit = {}
  
  override def shouldDisplay(context: JMap[String, AnyRef]): Boolean = context.get("pullRequest") match {
    case pullRequest: PullRequest => openTasksCounter.countOpenTasks(pullRequest) > 0
    case _ => false
  }
  
}