package istlab.KisoJikken;

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

import javax.swing.event.TreeModelListener;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;

class MyFile extends File {
    public MyFile(String pathname) {
        super(pathname);
    }

    public String toString() {
        return super.toString().replaceAll(App.userhome_escaped, "~");
    }
}


public class FileTreeModel implements TreeModel {

    // We specify the root directory when we create the model.
    protected File root;
    boolean isExcludeClassFiles = false;

    public FileTreeModel(File root, boolean exCls) {
        this.root = root;
        isExcludeClassFiles = exCls;
    }

    // The model knows how to return the root object of the tree
    public Object getRoot() {
        return root;
    }

    // Tell JTree whether an object in the tree is a leaf or not
    public boolean isLeaf(Object node) {
        return ((File) node).isFile();
    }

    private File[] getChildFiles(File parent) {
        if (isExcludeClassFiles) {
            return parent.listFiles(path -> !path.toString().endsWith(".class"));
        } else {
            return parent.listFiles();
        }
    }

    // Tell JTree how many children a node has
    public int getChildCount(Object parent) {
        File[] children = getChildFiles((File) parent);
        if (children == null)
            return 0;
        return children.length;
    }

    // Fetch any numbered child of a node for the JTree.
    // Our model returns File objects for all nodes in the tree. The
    // JTree displays these by calling the File.toString() method.
    public Object getChild(Object parent, int index) {
        File[] children = getChildFiles((File) parent);
        // String[] children = ((File) parent).list();
        Arrays.sort(children);
        if ((children == null) || (index >= children.length))
            return null;
        // return new File((File) parent, children[index]);
        return new MyFile(children[index].getAbsolutePath());
    }

    // Figure out a child's position in its parent node.
    public int getIndexOfChild(Object parent, Object child) {
        File[] children = getChildFiles((File) parent);
        // String[] children = ((File) parent).list();
        Arrays.sort(children);
        if (children == null)
            return -1;
        String childname = ((File) child).getName();
        // if (childname.startsWith(".")) return -1;
        for (int i = 0; i < children.length; i++) {
            if (childname.equals(children[i].getName()))
                return i;
        }
        return -1;
    }

    // This method is only invoked by the JTree for editable trees.
    // This TreeModel does not allow editing, so we do not implement
    // this method. The JTree editable property is false by default.
    public void valueForPathChanged(TreePath path, Object newvalue) {
    }

    // Since this is not an editable tree model, we never fire any events,
    // so we don't actually have to keep track of interested listeners.
    public void addTreeModelListener(TreeModelListener l) {
    }

    public void removeTreeModelListener(TreeModelListener l) {
    }
}