1 files changed, 125 insertions, 0 deletions
diff --git a/org.eclipse.egit.ui/src/org/eclipse/egit/ui/internal/staging/StagingFolderEntry.java b/org.eclipse.egit.ui/src/org/eclipse/egit/ui/internal/staging/StagingFolderEntry.java
new file mode 100644
index 0000000000..1f6dd43b98
--- /dev/null
+++ b/org.eclipse.egit.ui/src/org/eclipse/egit/ui/internal/staging/StagingFolderEntry.java
@@ -0,0 +1,125 @@
+/*******************************************************************************
+ * Copyright (C) 2013, Stephen Elsemore <selsemore@collab.net> and others.
+ *
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *******************************************************************************/
+package org.eclipse.egit.ui.internal.staging;
+
+import org.eclipse.core.resources.IContainer;
+import org.eclipse.core.resources.IMarker;
+import org.eclipse.core.resources.IResource;
+import org.eclipse.core.runtime.CoreException;
+import org.eclipse.core.runtime.IAdaptable;
+import org.eclipse.core.runtime.IPath;
+import org.eclipse.egit.core.internal.util.ResourceUtil;
+import org.eclipse.egit.ui.internal.decorators.IProblemDecoratable;
+
+/**
+ * A staged/unstaged folder entry in the tree
+ */
+public class StagingFolderEntry implements IAdaptable, IProblemDecoratable {
+	private final IPath repoLocation;
+	private final IPath repoRelativePath;
+	private final IPath nodePath;
+
+	private StagingFolderEntry parent;
+
+	/**
+	 * @param repoLocation
+	 * @param repoRelativePath
+	 * @param nodePath
+	 */
+	public StagingFolderEntry(IPath repoLocation, IPath repoRelativePath,
+			IPath nodePath) {
+		this.repoLocation = repoLocation;
+		this.repoRelativePath = repoRelativePath;
+		this.nodePath = nodePath;
+	}
+
+	/**
+	 * @return the container corresponding to the entry, if it exists in the
+	 *         workspace, null otherwise.
+	 */
+	public IContainer getContainer() {
+		return ResourceUtil.getContainerForLocation(getLocation());
+	}
+
+	public int getProblemSeverity() {
+		IContainer container = getContainer();
+		if (container == null)
+			return SEVERITY_NONE;
+
+		try {
+			return container.findMaxProblemSeverity(IMarker.PROBLEM, true,
+					IResource.DEPTH_ONE);
+		} catch (CoreException e) {
+			return SEVERITY_NONE;
+		}
+	}
+
+	public Object getAdapter(Class adapter) {
+		if (adapter == IResource.class || adapter == IContainer.class)
+			return getContainer();
+		else if (adapter == IPath.class)
+			return getLocation();
+		return null;
+	}
+
+	/**
+	 * @return the repo-relative path of this folder
+	 */
+	public IPath getPath() {
+		return repoRelativePath;
+	}
+
+	/**
+	 * @return the absolute path corresponding to the folder entry
+	 */
+	public IPath getLocation() {
+		return repoLocation.append(repoRelativePath);
+	}
+
+	/**
+	 * @return the repo-relative path of the parent folder entry
+	 */
+	public IPath getParentPath() {
+		return repoRelativePath.removeLastSegments(nodePath.segmentCount());
+	}
+
+	/**
+	 * @return the path of the node, relative to its parent
+	 */
+	public IPath getNodePath() {
+		return nodePath;
+	}
+
+	/**
+	 * @return the parent folder entry
+	 */
+	public StagingFolderEntry getParent() {
+		return parent;
+	}
+
+	/**
+	 * @param parent
+	 */
+	public void setParent(StagingFolderEntry parent) {
+		this.parent = parent;
+	}
+
+	@Override
+	public boolean equals(Object obj) {
+		if (obj instanceof StagingFolderEntry)
+			return ((StagingFolderEntry) obj).getLocation().equals(getLocation());
+		return super.equals(obj);
+	}
+
+	@Override
+	public int hashCode() {
+		return getLocation().hashCode();
+	}
+
+}
+\ No newline at end of file