net.binis.codegen.collection.CodeSetImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of code-generator-core Show documentation
Binis Code Generation Core Classes
There is a newer version: 1.2.22
package net.binis.codegen.collection;

/*-
 * #%L
 * code-generator-core
 * %%
 * Copyright (C) 2021 - 2024 Binis Belev
 * %%
 * 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.
 * #L%
 */

import java.util.Collection;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.function.Consumer;

import static java.util.Objects.nonNull;

public class CodeSetImpl implements CodeSet {

    private final R parent;
    private final Set set;
    private final Consumer validator;

    public CodeSetImpl(R parent, Set set) {
        this.parent = parent;
        this.set = set;
        this.validator = null;
    }

    public CodeSetImpl(R parent, Set set, Consumer validator) {
        this.parent = parent;
        this.set = set;
        this.validator = validator;
    }

    @Override
    public CodeSet add(T value) {
        validate(value);
        set.add(value);
        return this;
    }

    @Override
    public CodeSet remove(Object o) {
        set.remove(o);
        return this;
    }

    @Override
    public CodeSet addAll(Collection c) {
        c.forEach(this::validate);
        set.addAll(c);
        return this;
    }

    @Override
    public CodeSet retainAll(Collection c) {
        set.retainAll(c);
        return this;
    }

    @Override
    public CodeSet removeAll(Collection c) {
        set.removeAll(c);
        return this;
    }

    @Override
    public CodeSet clear() {
        set.clear();
        return this;
    }

    @Override
    public R done() {
        return parent;
    }

    protected void validate(T value) {
        if (nonNull(validator)) {
            validator.accept(value);
        }
    }

    @Override
    public CodeSet _if(boolean condition, Consumer> consumer) {
        if (condition) {
            consumer.accept(this);
        }
        return this;
    }

    @Override
    public CodeSet _if(boolean condition, BiConsumer, R> consumer) {
        if (condition) {
            consumer.accept(this, parent);
        }
        return this;
    }

    @Override
    public CodeSet _if(boolean condition, Consumer> consumer, Consumer> elseConsumer) {
        if (condition) {
            consumer.accept(this);
        } else {
            elseConsumer.accept(this);
        }
        return this;
    }

    @Override
    public CodeSet _if(boolean condition, BiConsumer, R> consumer, BiConsumer, R> elseConsumer) {
        if (condition) {
            consumer.accept(this, parent);
        } else {
            elseConsumer.accept(this, parent);
        }
        return this;
    }

    @Override
    public CodeSet _self(BiConsumer, R> consumer) {
        consumer.accept(this, parent);
        return this;
    }

    @Override
    public CodeSet _map(Object source) {
        throw new UnsupportedOperationException("Not implemented yet!");
        //TODO: Implement handling for collections.
    }
    
}