org.jboss.elemento.ElementsBag Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of theme-processor Show documentation
There is a newer version: 1.0.139
/*
 * Copyright 2015-2016 Red Hat, Inc, and individual contributors.
 *
 * 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 org.jboss.elemento;

import java.util.Iterator;

import elemental2.core.JsArray;
import elemental2.dom.HTMLElement;

/** Class to collect {@link HTMLElement}s */
public class ElementsBag implements TypedBuilder, ElementsBag> {

    private final IterableImpl iterable;

    public ElementsBag() {
        iterable = new IterableImpl();
    }

    @Override
    public ElementsBag that() {
        return this;
    }

    // ------------------------------------------------------ mirror add() methods from HtmlContent

    /** @return the elements in this bag. */
    public Iterable elements() {
        return iterable;
    }

    /** Adds the given element. */
    public ElementsBag add(HTMLElement element) {
        iterable.elements.push(element);
        return that();
    }

    /** Adds the given element by calling {@code element.element()}. */
    public ElementsBag add(IsElement element) {
        if (element != null) {
            return add(element.element());
        }
        return that();
    }

    /** Adds all elements. */
    public ElementsBag addAll(HTMLElement... elements) {
        for (HTMLElement element : elements) {
            add(element);
        }
        return that();
    }

    /** Adds all elements. */
    public ElementsBag addAll(IsElement... elements) {
        for (IsElement element : elements) {
            if (element != null) {
                add(element.element());
            }
        }
        return that();
    }

    /** Adds all elements. */
    public ElementsBag addAll(Iterable elements) {
        for (Object element : elements) {
            if (element instanceof HTMLElement) {
                add(((HTMLElement) element));
            } else if (element instanceof IsElement) {
                //noinspection rawtypes
                add(((IsElement) element).element());
            }
        }
        return that();
    }

    // ------------------------------------------------------ inner classes

    private static class IterableImpl implements Iterable {

        final JsArray elements;

        private IterableImpl() {
            elements = new JsArray<>();
        }

        @Override
        public Iterator iterator() {
            return Elements.iterator(elements);
        }
    }
}