package com.sunking.swing;

import java.io.*;
import java.util.*;

import java.awt.*;
import javax.swing.*;
import javax.swing.event.*;
import javax.swing.filechooser.*;
import javax.swing.tree.*;

/**
 * <p>Title: OpenSwing</p>
 * <p>Description: JFileTree 文件樹</p>
 * <p>Copyright: Copyright (c) 2004</p>
 * <p>Company: </p>
 * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
 * @version 1.0
 */

public class JFileTree extends JTree implements Serializable{
    public static final FileSystemView fileSystemView =
        FileSystemView.getFileSystemView();
    DefaultTreeModel treeModel;
    private java.io.FileFilter filter = null;

    /**
     * 只瀏覽目錄的目錄樹
     */
    public JFileTree(){
        this(new DirFilter());
    }

    /**
     * 指定過濾器的文件樹
     * @param filter FileFilter 指定過濾器
     */
    public JFileTree(java.io.FileFilter filter){
        FileNode root = new FileNode(fileSystemView.getRoots()[0], filter);
        treeModel = new DefaultTreeModel(root);
        root.explore();
        treeModel.nodeStructureChanged(root);
        this.setModel(treeModel);
        addTreeExpansionListener(new JFileTreeExpandsionListener());
        setCellRenderer(new JFileTreeCellRenderer());
    }

    /**
     * 取得當前選擇的節點
     * @return FileNode
     */
    public FileNode getSelectFileNode(){
        TreePath path = getSelectionPath();
        if(path == null || path.getLastPathComponent() == null){
            return null;
        }
        return(FileNode)path.getLastPathComponent();
    }

    /**
     * 設置當前選擇的節點
     * @param f FileNode
     * @throws Exception
     */
    public void setSelectFileNode(FileNode f) throws Exception{
        this.setSelectFile(f.getFile());
    }

    /**
     * 取得當前選擇的文件或目錄
     * @return File
     */
    public File getSelectFile(){
        FileNode node = getSelectFileNode();
        return node == null ? null : node.getFile();
    }

    /**
     * 設置當前選擇的文件或目錄
     * @param f File
     * @throws Exception
     */
    public void setSelectFile(File f) throws Exception{
        FileNode node = this.expandFile(f);
        TreePath path = new TreePath(node.getPath());
        this.scrollPathToVisible(path);
        this.setSelectionPath(path);
        this.repaint();
    }

    /**
     * 展開指定的文件或目錄
     * @param f File
     * @return FileNode
     * @throws Exception
     */
    public FileNode expandFile(File f) throws Exception{
        if(!f.exists()){
            throw new java.io.FileNotFoundException(f.getAbsolutePath());
        }
        Vector vTemp = new Vector();
        File fTemp = f;
        while(fTemp != null){
            vTemp.add(fTemp);
            fTemp = fileSystemView.getParentDirectory(fTemp);
        }

        FileNode nParent = (FileNode)treeModel.getRoot();
        for(int i = vTemp.size() - 1; i >= 0; i--){
            fTemp = (File)vTemp.get(i);
            nParent.explore();
            for(int j = 0; j < nParent.getChildCount(); j++){
                FileNode nChild = (FileNode)nParent.getChildAt(j);
                if(nChild.getFile().equals(fTemp)){
                    nParent = nChild;
                }
            }
        }
        return nParent;
    }

    /**
     *
     * <p>Title: OpenSwing</p>
     *
     * <p>Description: 文件樹修飾器</p>
     *
     * <p>Copyright: Copyright (c) 2004</p>
     *
     * <p>Company: </p>
     *
     * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
     * @version 1.0
     */
    class JFileTreeCellRenderer extends DefaultTreeCellRenderer{
        public Component getTreeCellRendererComponent(JTree tree, Object value,
            boolean sel, boolean expanded,
            boolean leaf, int row,
            boolean hasFocus){
            super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf,
                                               row,
                                               hasFocus);
            try{
                FileNode node = (FileNode)value;
                closedIcon = fileSystemView.getSystemIcon(((FileNode)value).
                    getFile());
                openIcon = closedIcon;
                setIcon(closedIcon);
            } catch(Exception ex){
            }
            return this;
        }
    }

    /**
     *
     * <p>Title: OpenSwing</p>
     *
     * <p>Description: 文件樹展開事件監聽器</p>
     *
     * <p>Copyright: Copyright (c) 2004</p>
     *
     * <p>Company: </p>
     *
     * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
     * @version 1.0
     */
    class JFileTreeExpandsionListener implements TreeExpansionListener{
        public JFileTreeExpandsionListener(){}

        public void treeExpanded(TreeExpansionEvent event){
            TreePath path = event.getPath();
            if(path == null || path.getLastPathComponent() == null)
                return;
            setCursor(new Cursor(Cursor.WAIT_CURSOR));
            FileNode node = (FileNode)path.getLastPathComponent();
            node.explore();
            JTree tree = (JTree)event.getSource();
            DefaultTreeModel treeModel = (DefaultTreeModel)tree.getModel();
            treeModel.nodeStructureChanged(node);
            setCursor(new Cursor(Cursor.DEFAULT_CURSOR));
        }

        public void treeCollapsed(TreeExpansionEvent event){}
    }

    /**
     *
     * <p>Title: OpenSwing</p>
     *
     * <p>Description:文件節點 </p>
     *
     * <p>Copyright: Copyright (c) 2004</p>
     *
     * <p>Company: </p>
     *
     * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
     * @version 1.0
     */
    public static class FileNode extends DefaultMutableTreeNode{
        private boolean explored = false;
        private java.io.FileFilter filter = null;

        public FileNode(File file, java.io.FileFilter filter){
            if(filter == null){
                this.filter = new DirFilter();
            } else{
                this.filter = filter;
            }
            setUserObject(file);
        }

        public boolean getAllowsChildren(){
            return isDirectory();
        }

        public boolean isDirectory(){
            return!isLeaf();
        }

        public boolean isLeaf(){
            return getFile().isFile();
        }

        public File getFile(){
            return(File)getUserObject();
        }

        public boolean isExplored(){
            return explored;
        }

        public void setExplored(boolean b){
            explored = b;
        }

        public String toString(){
            if(getFile() instanceof File)
                return fileSystemView.getSystemDisplayName((File)getFile());
            else
                return getFile().toString();
        }

        /**
         * 展開節點
         */
        public void explore(){
            if(!explored){
                explored = true;
                File file = getFile();
                //如果這里使用 file.listFiles(filter) 有BUG
                File[] children = file.listFiles();
                if(children == null || children.length == 0){
                    return;
                }
                //過濾后排序,選加入排序后的目錄, 再加入排序后的文件
                ArrayList listDir = new ArrayList();
                ArrayList listFile = new ArrayList();
                for(int i = 0; i < children.length; ++i){
                    File f = children[i];
                    if(filter.accept(f)){
                        if(f.isDirectory()){
                            listDir.add(f);
                        }else{
                            listFile.add(f);
                        }
                    }
                }
                Collections.sort(listDir);
                Collections.sort(listFile);
                for(int i = 0; i < listDir.size(); i++){
                    add(new FileNode((File)listDir.get(i),filter));
                }
                for(int i = 0; i < listFile.size(); i++){
                    add(new FileNode((File)listFile.get(i),filter));
                }
            }
        }
    }
    /**
     *
     * <p>Title: OpenSwing</p>
     *
     * <p>Description:目錄過濾器 </p>
     *
     * <p>Copyright: Copyright (c) 2004</p>
     *
     * <p>Company: </p>
     *
     * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
     * @version 1.0
     */
    public static class DirFilter implements java.io.FileFilter{
         public boolean accept(File pathname){
             return pathname.isDirectory();
         }
    }
    /**
     *
     * <p>Title: OpenSwing</p>
     *
     * <p>Description: 所有文件過濾器</p>
     *
     * <p>Copyright: Copyright (c) 2004</p>
     *
     * <p>Company: </p>
     *
     * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
     * @version 1.0
     */
    public static class AllFileFilter implements java.io.FileFilter{
         public boolean accept(File pathname){
             return true;
         }
    }
    /**
     *
     * <p>Title: OpenSwing</p>
     *
     * <p>Description:擴展名過濾器 </p>
     *
     * <p>Copyright: Copyright (c) 2004</p>
     *
     * <p>Company: </p>
     *
     * @author <a href="mailto:sunkingxie@hotmail.com">SunKing</a>
     * @version 1.0
     */
    public static class ExtensionFilter implements java.io.FileFilter{
        String extensions[];
        public ExtensionFilter(String extensions[]){
            this.extensions = new String[extensions.length];
            for(int i = 0; i < extensions.length; i++){
                this.extensions[i] = extensions[i].toLowerCase();
            }
        }

        public boolean accept(File pathname){
            if(pathname.isDirectory()){
                return true;
            }
            String name = pathname.getName().toLowerCase();
            for(int i = 0; i < extensions.length; i++){
                if(name.endsWith(this.extensions[i])){
                    return true;
                }
            }
            return false;
        }
    }

    public static void main(String[] args){
        JFrame frame = OpenSwingUtil.createDemoFrame("JFileTree Demo");
        frame.setSize(600, 400);
        JFileTree tree = new JFileTree();
        try{
            tree.setSelectFile(new File("C:/"));
        } catch(Exception ex){
            ex.printStackTrace();
        }
        Container pContent = frame.getContentPane();
        pContent.setLayout(new BorderLayout());
        JPanel pNorth = new JPanel(new GridLayout(1,3));
        JPanel pCenter = new JPanel(new GridLayout(1,3));
        pNorth.add(new JLabel("Directory only"));
        pNorth.add(new JLabel("All file"));
        pNorth.add(new JLabel("Directory and *.doc,*.txt file"));
        pCenter.add(new JScrollPane(tree));
        pCenter.add(new JScrollPane(new JFileTree(new AllFileFilter())));
        pCenter.add(new JScrollPane(new JFileTree(
            new ExtensionFilter(new String[]{"doc", "txt"}))));
        pContent.add(pNorth, BorderLayout.NORTH);
        pContent.add(pCenter, BorderLayout.CENTER);
        frame.setVisible(true);
    }
}