org.databene.commons.file.FileTreeModel Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2004-2014 Volker Bergmann ([email protected]).
 * All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.databene.commons.file;

import org.databene.commons.ArrayUtil;
import org.databene.commons.Element;
import org.databene.commons.TreeModel;
import org.databene.commons.Visitor;

import java.io.File;
import java.util.Arrays;
import java.util.Comparator;

/**
 * databene TreeModel implementation for files.
 * Created: 08.05.2007 17:55:54
 * @author Volker Bergmann
 */
public class FileTreeModel implements TreeModel, Element {

    private File root;
    private Comparator fileComparator;

    public FileTreeModel(File root) {
        this(root, new FilenameComparator());
    }

    public FileTreeModel(File root, Comparator fileComparator) {
        if (!root.exists())
            throw new IllegalArgumentException("File/Directory not found: " + root);
        this.root = root;
        this.fileComparator = fileComparator;
    }

    @Override
	public File getRoot() {
        return root;
    }

    @Override
	public File getParent(File child) {
        return child.getParentFile();
    }

    @Override
	public File getChild(File parent, int index) {
        return listFiles(parent)[index];
    }

    @Override
	public int getChildCount(File parent) {
        if (parent.isFile())
            return 0;
        else
            return listFiles(parent).length;
    }

    @Override
	public boolean isLeaf(File node) {
        return (node).isFile();
    }

    @Override
	public int getIndexOfChild(File parent, File child) {
        File[] files = listFiles(parent);
        return ArrayUtil.indexOf(child, files);
    }

    private File[] listFiles(File parent) {
        File[] files = parent.listFiles();
        if (files != null)
        	Arrays.sort(files, fileComparator);
        else
        	files = new File[0];
        return files;
    }

	@Override
	public void accept(Visitor visitor) {
		accept(visitor, root);
	}

	private void accept(Visitor visitor, File file) {
		visitor.visit(file);
		if (file.isDirectory()) {
			for (File child : file.listFiles())
				accept(visitor, child);
		}
	}

}