Измените if- else-construct в более функциональном стиле?

можно ли изменить эту конструкцию if-else в более функциональный стиль Scala?

def getMIMEType(document: String): String = {

  if (document.endsWith(".pdf")) {
    return "application/pdf"
  } else if (document.endsWith(".dxf")) {
    return "application/dxf"
  } else if (document.endsWith(".jpg")) {
    return "image/jpeg"
  } else return "application/octet-stream"
}

Я пытался использовать сопоставление с образцом, но это не сработало. Так что мне было бы любопытно найти хорошее решение.

Заранее спасибо!

Понго