jsonvalues.OpFilterArrElems Maven / Gradle / Ivy

Go to download
package jsonvalues;

import java.util.function.Predicate;

import static jsonvalues.MatchExp.ifJsonElse;
import static jsonvalues.Trampoline.more;

final class OpFilterArrElems extends OpFilterElems {


    OpFilterArrElems(final JsArray a) {
        super(a);
    }

    @Override
    Trampoline filterAll(final JsPath startingPath,
                                  final Predicate predicate
                                 ) {
        return json.ifEmptyElse(Trampoline.done(json),
                                (head, tail) ->
                                {

                                    final JsPath headPath = startingPath.inc();

                                    final Trampoline tailCall =
                                            Trampoline.more(() -> new OpFilterArrElems(tail).filterAll(headPath,
                                                                                                       predicate
                                                                                                      ));
                                    return ifJsonElse(headObj -> more(() -> tailCall).flatMap(tailResult -> new OpFilterObjElems(headObj).filterAll(headPath,
                                                                                                                                                    predicate
                                                                                                                                                   )
                                                                                                                                         .map(tailResult::prepend)
                                                                                             ),
                                                      headArray -> more(() -> tailCall).flatMap(tailResult -> new OpFilterArrElems(headArray).filterAll(headPath.index(-1),
                                                                                                                                                        predicate
                                                                                                                                                       )
                                                                                                                                             .map(tailResult::prepend)
                                                                                               ),
                                                      headElem -> predicate.test(JsPair.of(headPath,
                                                                                           headElem
                                                                                          )) ?
                                                                  more(() -> tailCall).map(tailResult -> tailResult.prepend(headElem)) :
                                                                  tailCall
                                                     ).apply(head);
                                }
                               );
    }

    @Override
    Trampoline filter(final JsPath startingPath,
                               final Predicate predicate
                              ) {
        return json.ifEmptyElse(Trampoline.done(json),
                                (head, tail) ->
                                {

                                    final JsPath headPath = startingPath.inc();

                                    final Trampoline tailCall = Trampoline.more(() -> new OpFilterArrElems(tail).filter(headPath,
                                                                                                                                 predicate
                                                                                                                                ));
                                    return ifJsonElse(elem -> more(() -> tailCall).map(it -> it.prepend(elem)),
                                                      elem -> predicate.test(JsPair.of(headPath,
                                                                                       elem
                                                                                      )) ?
                                                              more(() -> tailCall).map(it -> it.prepend(elem)) :
                                                              tailCall

                                                     )
                                            .apply(head);
                                }
                               );
    }
}