path: root/src/main/scala/com/drivergrp/core/generators.scala

package com.drivergrp.core

import java.math.MathContext
import com.drivergrp.core.time.{Time, TimeRange}

import scala.reflect.ClassTag
import scala.util.Random

object generators {

  private val random = new Random
  import random._

  private val DefaultMaxLength = 100

  def nextId[T](): Id[T] = Id[T](nextLong())

  def nextName[T](maxLength: Int = DefaultMaxLength): Name[T] = Name[T](nextString(maxLength))

  def nextString(maxLength: Int = DefaultMaxLength) = random.nextString(maxLength)

  def nextOption[T](value: => T): Option[T] = if (nextBoolean) Option(value) else None

  def nextPair[L, R](left: => L, right: => R): (L, R) = (left, right)

  def nextTriad[F, S, T](first: => F, second: => S, third: => T): (F, S, T) = (first, second, third)

  def nextTime(): Time = Time(math.abs(nextLong() % System.currentTimeMillis))

  def nextTimeRange(): TimeRange = TimeRange(nextTime(), nextTime())

  def nextBigDecimal(multiplier: Double = 1000000.00, precision: Int = 2): BigDecimal =
    BigDecimal(multiplier * nextDouble, new MathContext(precision))

  def oneOf[T](items: Seq[T]): T = items(nextInt(items.size))

  def arrayOf[T: ClassTag](generator: => T, maxLength: Int = DefaultMaxLength): Array[T] =
    Array.fill(maxLength)(generator)

  def seqOf[T](generator: => T, maxLength: Int = DefaultMaxLength): Seq[T] = Seq.fill(maxLength)(generator)

  def vectorOf[T](generator: => T, maxLength: Int = DefaultMaxLength): Vector[T] = Vector.fill(maxLength)(generator)

  def listOf[T](generator: => T, maxLength: Int = DefaultMaxLength): List[T] = List.fill(maxLength)(generator)

  def setOf[T](generator: => T, maxLength: Int = DefaultMaxLength): Set[T] = seqOf(generator, maxLength).toSet

  def mapOf[K, V](maxLength: Int, keyGenerator: => K, valueGenerator: => V): Map[K, V] =
    seqOf(nextPair(keyGenerator, valueGenerator), maxLength).toMap
}