1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.core.technical.generator
21 
22 import eu.timepit.refined.auto._
23 import eu.timepit.refined.types.numeric.PosInt
24 import org.make.core.SlugHelper
25 import org.scalacheck.Gen
26 
27 import scala.io.Source
28 
29 object CustomGenerators {
30   object Color {
31     def gen: Gen[String] = Gen.listOfN(6, Gen.hexChar).map(_.mkString.toLowerCase.prependedAll("#"))
32   }
33 
34   object LoremIpsumGen {
35     private lazy val loremIpsumWords: Seq[String] = Source.fromResource("loremIpsum.csv").getLines().toSeq
36 
37     def word: Gen[String] = Gen.oneOf(loremIpsumWords)
38     def words: Gen[Seq[String]] = Gen.someOf(loremIpsumWords).map(_.toSeq)
39     def words(n: Int): Gen[Seq[String]] = Gen.listOfN(n, word)
40 
41     def sentence(maxLength: Option[PosInt] = None): Gen[String] =
42       Gen.atLeastOne(loremIpsumWords).map(_.toSeq).map { sen =>
43         val ret = sen.mkString("", " ", ".").capitalize
44         maxLength.collect {
45           case length if length < ret.length => ret.take(ret.take(length).lastIndexOf(" ")).appendedAll(".")
46         }.getOrElse(ret)
47       }
48     def sentences(n: Int, maxLength: Option[PosInt] = None): Gen[List[String]] = Gen.listOfN(n, sentence(maxLength))
49 
50     def slug(maxLength: Option[PosInt] = None): Gen[String] = sentence(maxLength).map(SlugHelper.apply)
51   }
52 
53   object ImageUrl {
54     def gen(width: Int, height: Int): Gen[String] =
55       Gen.chooseNum(0, 1084).map(picNum => s"https://i.picsum.photos/id/$picNum/$width/$height.jpg")
56   }
57 
58   object Mail {
59     def gen(prefix: Option[String] = None): Gen[String] = Gen.uuid.map { id =>
60       val tag: String = prefix.map(p => s"$p-$id").getOrElse(id.toString)
61       s"$tag@example.com"
62     }
63   }
64 
65   object PostalCode {
66     def gen: Gen[String] = Gen.listOfN(5, Gen.numChar).map(_.mkString)
67   }
68 
69   object FirstName {
70     private lazy val dictionary: Seq[String] = Source.fromResource("first_names.csv").getLines().toSeq
71 
72     lazy val gen: Gen[String] = {
73       Gen.oneOf(dictionary).map(_.trim.toLowerCase.capitalize)
74     }
75   }
76 
77   object URL {
78     def gen: Gen[String] = Gen.uuid.map(id => s"https://example.com/$id")
79   }
80 
81 }