org.eclipse.egerrit.model/src/org/eclipse/egerrit/internal/model/ModifiedFileInfoImpl.java

package org.eclipse.egerrit.internal.model;

import org.eclipse.egerrit.internal.model.impl.FileInfoImpl;
import org.eclipse.egerrit.internal.model.impl.StringToFileInfoImpl;
import org.eclipse.egerrit.internal.model.impl.StringToRevisionInfoImpl;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.common.util.BasicEList;
import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.util.EContentAdapter;

public class ModifiedFileInfoImpl extends FileInfoImpl {
	@Override
	public String getPath() {
		 StringToFileInfoImpl container = (StringToFileInfoImpl) this.eContainer();
		 if (container != null) {
				return container.getKey();
		 } else {
			 return null;
		 }
	}

	@Override
	public RevisionInfo getRevision() {
		return ((StringToRevisionInfoImpl) this.eContainer().eContainer().eContainer()).getValue();
	}

	@Override
	public EList<CommentInfo> getAllComments() {
		EList<CommentInfo> allComments = new BasicEList<>();
		allComments.addAll(getComments());
		allComments.addAll(getDraftComments());
		return allComments;
	}

	@Override
	public int getCommentsCount() {
		return getComments().size();
	}
	
	@Override
	public int getDraftsCount() {
		return getDraftComments().size();
	}
}