-
-
Notifications
You must be signed in to change notification settings - Fork 25
/
CsvRow.scala
51 lines (42 loc) · 1.96 KB
/
CsvRow.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
/*
* Copyright 2023 Lucas Satabin
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package fs2.data.csv
import cats.data.NonEmptyList
import cats.syntax.all._
object CsvRow {
/** Constructs a [[CsvRow]] and checks that the size of values and headers match. */
def apply[Header](values: NonEmptyList[String],
headers: NonEmptyList[Header],
line: Option[Long] = None): Either[CsvException, CsvRow[Header]] =
if (values.length =!= headers.length)
Left(new HeaderSizeError(headers.length, values.length, line))
else
Right(new CsvRow(values, Some(headers), line))
def unsafe[Header](values: NonEmptyList[String], headers: NonEmptyList[Header]): CsvRow[Header] =
apply(values, headers).fold(throw _, identity)
def liftRow[Header](headers: NonEmptyList[Header])(row: Row): Either[CsvException, CsvRow[Header]] =
apply(row.values, headers, row.line)
def fromListHeaders[Header](l: List[(Header, String)]): Option[CsvRow[Header]] = {
val (hs, vs) = l.unzip
(NonEmptyList.fromList(vs), NonEmptyList.fromList(hs)).mapN((v, h) => new CsvRow(v, Some(h)))
}
def fromNelHeaders[Header](nel: NonEmptyList[(Header, String)]): CsvRow[Header] = {
val (hs, vs) = nel.toList.unzip
new CsvRow(NonEmptyList.fromListUnsafe(vs), Some(NonEmptyList.fromListUnsafe(hs)))
}
def unapply[Header](arg: CsvRow[Header]): Some[(NonEmptyList[String], NonEmptyList[Header])] = Some(
(arg.values, arg.headers.get))
}