ject.ja.text.Inflection.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ject-ja_2.13 Show documentation

ject-ja

The newest version!

package ject.ja.text

import ject.ja.text.inflection.*
import ject.ja.text.Transformation.Transform
import zio.NonEmptyChunk

object Inflection {

  def inflect(word: String, wordType: WordType, targetForm: Form): Either[String, NonEmptyChunk[String]] = {
    val inflections = inflectionsFor(wordType)

    for {
      transforms <- inflections.get(targetForm).toRight(s"Invalid inflection form: $targetForm")
      result     <- transforms(word)
    } yield result
  }

  def inflectAll(word: String, wordType: WordType): Map[Form, Either[String, NonEmptyChunk[String]]] = {
    val inflections = inflectionsFor(wordType)

    inflections.map { case (k, f) =>
      (k, f(word))
    }
  }

  private def inflectionsFor(wordType: WordType): Map[Form, Transform] =
    wordType match {
      case WordType.VerbIchidan => Ichidan.inflections
      case WordType.VerbGodan   => Godan.inflections
      case WordType.VerbSuru    => Suru.inflections
      case WordType.VerbAru     => Aru.inflections
      case WordType.VerbIku     => Iku.inflections
      case WordType.AdjectiveI  => AdjectiveI.inflections
      case wordType @ WordType.AdjectiveNa =>
        throw new NotImplementedError(s"Deinflections for $wordType not yet supported.")
    }
}