Added Repeat and listed necessary tests
This commit is contained in:
@ -342,3 +342,30 @@ func Pipe[In, Ignore, Through any](p Parser[In, Ignore]) func(Through) Parser[In
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// Repeat applies p until p fails, and returns the collected outputs.
|
||||
// It succeeds if and only if p succeeds at least minCount times.
|
||||
// It consumes if and only if at least one of the applications of p consumes.
|
||||
func Repeat[In, Out any](minCount int, p Parser[In, Out]) Parser[In, []Out] {
|
||||
return func(s State[In]) (Result[In, []Out], error) {
|
||||
var values []Out
|
||||
var consumed bool
|
||||
next := s
|
||||
for {
|
||||
result, err := p(s)
|
||||
if err != nil {
|
||||
return Result[In, []Out]{}, fmt.Errorf("AtLeastN: %w", err)
|
||||
}
|
||||
consumed = consumed || result.Consumed()
|
||||
if failed, _, msg := result.Failed(); failed {
|
||||
if len(values) >= minCount {
|
||||
return Succeed(consumed, values, next, MessageOK(s.Pos())), nil
|
||||
}
|
||||
return Fail[In, []Out](consumed, msg), nil
|
||||
}
|
||||
var value Out
|
||||
_, _, value, next, _ = result.Succeeded()
|
||||
values = append(values, value)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user