org.wartremover.warts.CollectHeadOption.scala Maven / Gradle / Ivy
package org.wartremover
package warts
object CollectHeadOption extends WartTraverser {
def apply(u: WartUniverse): u.Traverser = {
import u.universe._
val iterable = rootMirror.staticClass("scala.collection.Iterable").toTypeConstructor
new u.Traverser {
private[this] def message = "you can use collectFirst instead of collect.headOption"
override def traverse(tree: Tree): Unit = {
tree match {
case _ if hasWartAnnotation(u)(tree) =>
case Select(
Apply(Apply(TypeApply(Select(left, TermName("collect")), _ :: _ :: Nil), pf :: Nil), canBuildFrom),
TermName("headOption")
) if left.tpe.typeConstructor <:< iterable =>
error(u)(tree.pos, message)
case _ =>
super.traverse(tree)
}
}
}
}
}