org.nasdanika.html.impl.RowContainerImpl Maven / Gradle / Ivy

Go to download
package org.nasdanika.html.impl;

import java.util.ArrayList;
import java.util.List;

import org.nasdanika.html.HTMLFactory;
import org.nasdanika.html.RowContainer;

class RowContainerImpl> extends HTMLElementImpl implements RowContainer {

	RowContainerImpl(HTMLFactory factory, String tagName) {
		super(factory, tagName, false);
	}

	class RowImpl extends HTMLElementImpl implements Row {
		
		RowImpl() {
			super(RowContainerImpl.this.factory, "tr", false);
		}
		
		class CellImpl extends HTMLElementImpl implements Cell {
			
			CellImpl(boolean isHeader, Object... content) {
				super(RowContainerImpl.this.factory, isHeader ? "th" : "td", false);
				content(content);
			}

			@Override
			public Cell colspan(int colspan) {
				return attribute("colspan", String.valueOf(colspan));
			}

			@Override
			public Cell rowspan(int rowspan) {
				return attribute("rowspan", String.valueOf(rowspan));
			}

			@Override
			public Cell content(Object... content) {
				for (Object c: content) {
					this.content.add(c);
				}
				return this;
			}
			
		}

		@Override
		public Cell cell(Object... content) {
			CellImpl cell = new CellImpl(false, content);
			this.content.add(cell);
			return cell;
		}

		@Override
		public Cell header(Object... content) {
			CellImpl cell = new CellImpl(true, content);
			this.content.add(cell);
			return cell;
		}

		@Override
		public List cells() {
			List cells = new ArrayList<>();
			for (Object c: content) {
				if (c instanceof Cell) {
					cells.add((Cell) c);
				}
			}
			return cells;
		}

		@Override
		public Row content(Object... content) {
			for (Object c: content) {
				this.content.add(c);
			}
			return this;
		}
		
	}

	@Override
	public Row row(Object... cells) {
		Row row = new RowImpl();
		for (Object cell: cells) {
			row.cell(cell);
		}
		content.add(row);
		return row;
	}
	
	@Override
	public Row headerRow(Object... headers) {
		Row row = new RowImpl();
		for (Object header: headers) {
			row.header(header);
		}
		content.add(row);
		return row;
	}

	@Override
	public List rows() {
		List rows = new ArrayList<>();
		for (Object c: content) {
			if (c instanceof Row) {
				rows.add((Row) c);
			}
		}
		return rows;
	}

	@SuppressWarnings("unchecked")
	@Override
	public T content(Object... content) {
		for (Object c: content) {
			this.content.add(c);
		}
		return (T) this;
	}

}