org.sitoolkit.wt.gui.infra.fx.FileTreeItem Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sit-wt-app Show documentation
The newest version!
package org.sitoolkit.wt.gui.infra.fx;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import javafx.scene.control.CheckBoxTreeItem;
import javafx.scene.control.TreeItem;
import javafx.scene.image.ImageView;

public final class FileTreeItem extends CheckBoxTreeItem {
    private boolean isLeaf;
    private boolean isFirstTimeLeaf = true;
    private boolean isSelectable = true;

    public FileTreeItem(File file) {
        super(new FileWrapper(file));

        String url = file.isDirectory() ? "/icon/ic_folder_open_black_18dp_1x.png"
                : "/icon/ic_description_black_18dp_1x.png";
        setGraphic(new ImageView(url));
    }

    public List getSelectedFiles() {
        List selectedFiles = new ArrayList<>();

        if (isSelected()) {
            selectedFiles.add(getValue().getFile());
        }

        for (TreeItem child : getChildren()) {
            selectedFiles.addAll(((FileTreeItem) child).getSelectedFiles());
        }

        return selectedFiles;
    }

    @Override
    public boolean isLeaf() {
        if (isFirstTimeLeaf) {
            isFirstTimeLeaf = false;
            FileWrapper file = getValue();
            isLeaf = file.getFile().isFile();
        }

        return isLeaf;
    }

    public void buildChildren() {
        FileWrapper file = getValue();

        if (file != null && file.getFile().isDirectory()) {

            File[] files = file.getFile().listFiles();

            if (files != null) {

                for (File childFile : files) {
                    addChild(childFile);
                }

            }
        }
    }

    public boolean isSelectable() {
        return isSelectable;
    }

    public FileTreeItem addChild(File childFile) {
        FileTreeItem child = new FileTreeItem(childFile);
        child.isSelectable = isSelectable;
        child.buildChildren();
        getChildren().add(child);
        return child;
    }

}