+ /**
+ * Model for the tree of files.
+ *
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ */
+ private class FileTreeModel implements TreeModel, PropertyChangeListener {
+
+ /** Tree model listeners. */
+ private final List<TreeModelListener> treeModelListeners = new ArrayList<TreeModelListener>();
+
+ /** The tree of files. */
+ private final Tree<FileTreePath> fileTreePathTree = new Tree<FileTreePath>();
+
+ /**
+ * Creates a new file tree model.
+ */
+ FileTreeModel() {
+ buildTree();
+ }
+
+ //
+ // EVENT MANAGEMENT
+ //
+
+ /**
+ * {@inheritDoc}
+ */
+ public void addTreeModelListener(TreeModelListener treeModelListener) {
+ treeModelListeners.add(treeModelListener);
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ public void removeTreeModelListener(TreeModelListener treeModelListener) {
+ treeModelListeners.remove(treeModelListener);
+ }
+
+ //
+ // ACCESSORS
+ //
+
+ /**
+ * {@inheritDoc}
+ */
+ @SuppressWarnings("synthetic-access")
+ public Object getChild(Object parent, int index) {
+ logger.log(Level.FINEST, "getChild(" + parent + ", " + index + ")");
+ Node<FileTreePath> parentNode = findNode(parent);
+ if (parentNode != null) {
+ return parentNode.getChild(index).getElement();
+ }
+ return null;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @SuppressWarnings("synthetic-access")
+ public int getChildCount(Object parent) {
+ logger.log(Level.FINEST, "getChildCount(" + parent + ")");
+ Node<FileTreePath> parentNode = findNode(parent);
+ if (parentNode != null) {
+ logger.log(Level.FINEST, "getChildCount(" + parent + "): " + parentNode.size());
+ return parentNode.size();
+ }
+ return -1;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @SuppressWarnings("synthetic-access")
+ public int getIndexOfChild(Object parent, Object child) {
+ logger.log(Level.FINEST, "getIndexOfChild(" + parent + ", " + child + ")");
+ Node<FileTreePath> parentNode = findNode(parent);
+ if (parentNode != null) {
+ return parentNode.getIndexOfChild((FileTreePath) child);
+ }
+ return -1;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @SuppressWarnings("synthetic-access")
+ public Object getRoot() {
+ logger.log(Level.FINEST, "getRoot()");
+ return fileTreePathTree.getRootNode().getChild(0).getElement();
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @SuppressWarnings("synthetic-access")
+ public boolean isLeaf(Object node) {
+ logger.log(Level.FINEST, "isLeaf(" + node + ")");
+ Node<FileTreePath> parentNode = findNode(node);
+ if (parentNode != null) {
+ return parentNode.size() == 0;
+ }
+ return true;
+ }
+
+ //
+ // ACTIONS
+ //
+
+ /**
+ * {@inheritDoc}
+ */
+ public void valueForPathChanged(TreePath path, Object newValue) {
+ /* TODO - implement */
+ }
+
+ //
+ // PRIVATE METHODS
+ //
+
+ /**
+ * Finds the node for the given object. This method is quite necessary
+ * because the element for the root node of the JTree is
+ * <code>null</code>
+ *
+ * @param node
+ * The element whose node to return
+ * @return The node, or <code>null</code> if no node could be found
+ */
+ private Node<FileTreePath> findNode(Object node) {
+ if (node == null) {
+ return fileTreePathTree.getRootNode().getChild(0);
+ }
+ return fileTreePathTree.getRootNode().getChild(0).findChild((FileTreePath) node);
+ }
+
+ /**
+ * Builds the tree from the project’s file entries.
+ */
+ @SuppressWarnings("synthetic-access")
+ private void buildTree() {
+ Tree<String> pathTree = new Tree<String>();
+ Node<String> pathRootNode = pathTree.getRootNode().addChild(File.separator);
+ logger.log(Level.FINEST, "project: " + project);
+ buildTree(pathRootNode, project.getBasePathEntries());
+ buildTree(pathRootNode, project.getVirtualEntries());
+ /* now convert to a tree suitable for the JTree. */
+ Node<FileTreePath> fileTreePathRootNode = fileTreePathTree.getRootNode();
+ fileTreePathRootNode.removeAllChildren();
+ convertTree(File.separator, pathRootNode, fileTreePathRootNode.addChild(new FileTreePath(File.separator, project.getName())));
+ /* TODO - now add entries to all file tree path tree nodes. */
+ }
+
+ /**
+ * Traverses the tree of path nodes and converts all paths to
+ * {@link FileTreePath} objects, suitable for the JTree.
+ *
+ * @param completePath
+ * The base path of the current root node
+ * @param pathRootNode
+ * The root node of the path tree
+ * @param fileTreePathRootNode
+ * The root node of the file tree path tree.
+ */
+ private void convertTree(String completePath, Node<String> pathRootNode, Node<FileTreePath> fileTreePathRootNode) {
+ for (Node<String> pathChild: pathRootNode) {
+ String currentFilePath = completePath + pathChild.getElement();
+ Node<FileTreePath> newNode = fileTreePathRootNode.addChild(new FileTreePath(currentFilePath));
+ convertTree(currentFilePath, pathChild, newNode);
+ }
+ fileTreePathRootNode.sortChildren();
+ }
+
+ /**
+ * Builds a tree matching the directory structure of the given entries.
+ *
+ * @param pathRootNode
+ * The root node of the tree
+ * @param entries
+ * The entries
+ */
+ private void buildTree(Node<String> pathRootNode, List<Entry> entries) {
+ for (Entry basePathEntry: entries) {
+ String entryName = basePathEntry.getName();
+ String[] directories = entryName.split("\\" + File.separator);
+ Node<String> currentPathNode = pathRootNode;
+ for (String directory: directories) {
+ if (!currentPathNode.hasChild(directory)) {
+ currentPathNode = currentPathNode.addChild(directory);
+ } else {
+ currentPathNode = currentPathNode.getChild(directory);
+ }
+ }
+ }
+ }
+
+ //
+ // INTERFACE PropertyChangeListener
+ //
+
+ /**
+ * {@inheritDoc}
+ */
+ public void propertyChange(PropertyChangeEvent propertyChangeEvent) {
+ if (propertyChangeEvent.getSource() instanceof Project) {
+ if (propertyChangeEvent.getPropertyName().equals(Project.PROPERTY_BASE_PATH_ENTRIES)) {
+ buildTree();
+ }
+ }
+ }
+
+ }
+
+ private static class FileTreePath implements Comparable<FileTreePath> {
+
+ private final String filePath;
+ private final String fileName;
+ private Entry fileEntry;
+
+ public FileTreePath(String filePath) {
+ this(filePath, null);
+ }
+
+ public FileTreePath(String filePath, String fileName) {
+ this.filePath = filePath;
+ if (fileName == null) {
+ if (filePath.indexOf(File.separatorChar) != -1) {
+ this.fileName = filePath.substring(filePath.lastIndexOf(File.separatorChar) + 1);
+ } else {
+ this.fileName = filePath;
+ }
+ } else {
+ this.fileName = fileName;
+ }
+ }
+
+ /**
+ * Returns the complete file path.
+ *
+ * @return The file path
+ */
+ public String getFilePath() {
+ return filePath;
+ }
+
+ /**
+ * Returns the file name, i.e. everything after the last
+ * {@link File#separatorChar}.
+ *
+ * @return The file name
+ */
+ public String getFileName() {
+ return fileName;
+ }
+
+ /**
+ * Returns the file entry associated with this path, if any.
+ *
+ * @return The file entry associated with this path, or
+ * <code>null</code> if this path denotes a directory
+ */
+ public Entry getFileEntry() {
+ return fileEntry;
+ }
+
+ /**
+ * Sets the entry associated with this path.
+ *
+ * @param fileEntry
+ * The entry
+ */
+ public void setFileEntry(Entry fileEntry) {
+ this.fileEntry = fileEntry;
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public boolean equals(Object object) {
+ if ((object == null) || !(object instanceof FileTreePath)) {
+ return false;
+ }
+ FileTreePath fileTreePath = (FileTreePath) object;
+ return fileTreePath.filePath.equals(filePath);
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public int hashCode() {
+ return filePath.hashCode();
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public String toString() {
+ return fileName;
+ }
+
+ //
+ // INTERFACE Comparable
+ //
+
+ /**
+ * {@inheritDoc}
+ */
+ public int compareTo(FileTreePath otherFileTreePath) {
+ return filePath.compareTo(otherFileTreePath.filePath);
+ }
+
+ }
+