/
SeqShapedRowDecoder.scala
88 lines (75 loc) · 3.16 KB
/
SeqShapedRowDecoder.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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*
* 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.generic.internal
import cats.data.NonEmptyList
import cats.syntax.all._
import fs2.data.csv._
import shapeless._
trait SeqShapedRowDecoder[Repr] extends RowDecoder[Repr]
object SeqShapedRowDecoder extends LowPrioritySeqShapedRowDecoder1 {
implicit def hnilOptionDecoder[Head](implicit Head: CellDecoder[Head]): SeqShapedRowDecoder[Option[Head] :: HNil] =
new SeqShapedRowDecoder[Option[Head] :: HNil] {
def apply(row: Row): DecoderResult[Option[Head] :: HNil] =
row.values match {
case NonEmptyList(c, Nil) =>
if (c.isEmpty)
Right(None :: HNil)
else
Head(c).bimap(_.withLine(row.line), Some(_) :: HNil)
case _ =>
Left(new DecoderError(s"expected 1 element but got ${row.values.size}", row.line))
}
}
implicit def hconsOptionDecoder[Head, Tail <: HList](implicit
Head: CellDecoder[Head],
Tail: Lazy[SeqShapedRowDecoder[Tail]]): SeqShapedRowDecoder[Option[Head] :: Tail] =
new SeqShapedRowDecoder[Option[Head] :: Tail] {
def apply(row: Row): DecoderResult[Option[Head] :: Tail] =
for {
tail <- NonEmptyList
.fromList(row.values.tail)
.liftTo[DecoderResult](new DecoderError("unexpected end of row", row.line))
head <-
if (row.values.head.isEmpty) Right(None) else Head(row.values.head).bimap(_.withLine(row.line), Some(_))
tail <- Tail.value(Row(tail))
} yield head :: tail
}
}
private[generic] trait LowPrioritySeqShapedRowDecoder1 {
implicit def hnilDecoder[Head](implicit Head: CellDecoder[Head]): SeqShapedRowDecoder[Head :: HNil] =
new SeqShapedRowDecoder[Head :: HNil] {
def apply(row: Row): DecoderResult[Head :: HNil] =
row.values match {
case NonEmptyList(c, Nil) =>
Head(c).bimap(_.withLine(row.line), _ :: HNil)
case _ =>
Left(new DecoderError(s"expected 1 element but got ${row.values.size}", row.line))
}
}
implicit def hconsDecoder[Head, Tail <: HList](implicit
Head: CellDecoder[Head],
Tail: Lazy[SeqShapedRowDecoder[Tail]]): SeqShapedRowDecoder[Head :: Tail] =
new SeqShapedRowDecoder[Head :: Tail] {
def apply(row: Row): DecoderResult[Head :: Tail] =
for {
tail <- NonEmptyList
.fromList(row.values.tail)
.liftTo[DecoderResult](new DecoderError("unexpect end of row", row.line))
head <- Head(row.values.head).leftMap(_.withLine(row.line))
tail <- Tail.value(Row(tail))
} yield head :: tail
}
}