X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftext%2FPartContainer.java;h=75714f714270c5b16d5b37823056b5da5be7f164;hp=739985932c6b463ef360870d07e140e472d06295;hb=419098bcd6215125408b29e60bd888e60979d37b;hpb=e8411efc4eaff6418cad496eb34fd7bc260eb9cd

diff --git a/src/main/java/net/pterodactylus/sone/text/PartContainer.java b/src/main/java/net/pterodactylus/sone/text/PartContainer.java
index 7399859..75714f7 100644
--- a/src/main/java/net/pterodactylus/sone/text/PartContainer.java
+++ b/src/main/java/net/pterodactylus/sone/text/PartContainer.java
@@ -1,5 +1,5 @@
 /*
- * Sone - PartContainer.java - Copyright © 2010 David Roden
+ * Sone - PartContainer.java - Copyright © 2010–2015 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -17,10 +17,12 @@
 
 package net.pterodactylus.sone.text;
 
-import java.io.IOException;
-import java.io.Writer;
+import java.util.ArrayDeque;
 import java.util.ArrayList;
+import java.util.Deque;
+import java.util.Iterator;
 import java.util.List;
+import java.util.NoSuchElementException;
 
 /**
  * Part implementation that can contain an arbitrary amount of other parts.
@@ -29,7 +31,7 @@ import java.util.List;
  *
  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
  */
-public class PartContainer implements Part {
+public class PartContainer implements Part, Iterable<Part> {
 
 	/** The parts to render. */
 	private final List<Part> parts = new ArrayList<Part>();
@@ -48,6 +50,36 @@ public class PartContainer implements Part {
 		parts.add(part);
 	}
 
+	/**
+	 * Returns the part at the given index.
+	 *
+	 * @param index
+	 *            The index of the part
+	 * @return The part
+	 */
+	public Part getPart(int index) {
+		return parts.get(index);
+	}
+
+	/**
+	 * Removes the part at the given index.
+	 *
+	 * @param index
+	 *            The index of the part to remove
+	 */
+	public void removePart(int index) {
+		parts.remove(index);
+	}
+
+	/**
+	 * Returns the number of parts.
+	 *
+	 * @return The number of parts
+	 */
+	public int size() {
+		return parts.size();
+	}
+
 	//
 	// PART METHODS
 	//
@@ -56,10 +88,78 @@ public class PartContainer implements Part {
 	 * {@inheritDoc}
 	 */
 	@Override
-	public void render(Writer writer) throws IOException {
+	public String getText() {
+		StringBuilder partText = new StringBuilder();
 		for (Part part : parts) {
-			part.render(writer);
+			partText.append(part.getText());
 		}
+		return partText.toString();
+	}
+
+	//
+	// ITERABLE METHODS
+	//
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	@SuppressWarnings("synthetic-access")
+	public Iterator<Part> iterator() {
+		return new Iterator<Part>() {
+
+			private Deque<Iterator<Part>> partStack = new ArrayDeque<Iterator<Part>>();
+			private Part nextPart;
+			private boolean foundNextPart;
+			private boolean noNextPart;
+
+			{
+				partStack.push(parts.iterator());
+			}
+
+			private void findNext() {
+				if (foundNextPart) {
+					return;
+				}
+				noNextPart = true;
+				while (!partStack.isEmpty()) {
+					Iterator<Part> parts = partStack.pop();
+					if (parts.hasNext()) {
+						nextPart = parts.next();
+						partStack.push(parts);
+						if (nextPart instanceof PartContainer) {
+							partStack.push(((PartContainer) nextPart).iterator());
+						} else {
+							noNextPart = false;
+							break;
+						}
+					}
+				}
+				foundNextPart = true;
+			}
+
+			@Override
+			public boolean hasNext() {
+				findNext();
+				return !noNextPart;
+			}
+
+			@Override
+			public Part next() {
+				findNext();
+				if (noNextPart) {
+					throw new NoSuchElementException();
+				}
+				foundNextPart = false;
+				return nextPart;
+			}
+
+			@Override
+			public void remove() {
+				/* ignore. */
+			}
+
+		};
 	}
 
 }