import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.File;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;
import javax.swing.JComponent;
import javax.swing.JFrame;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.TransferHandler;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreeNode;
public class FSTest extends JFrame {
public FSTest() {
super("FSTree Component Test");
setSize(300, 300);
setDefaultCloseOperation(EXIT_ON_CLOSE);
FSTree fst = new FSTree(new DefaultMutableTreeNode("Starter"));
getContentPane().add(new JScrollPane(fst));
setVisible(true);
}
public static void main(String args[]) {
new FSTest();
}
}
class FSTree extends JTree {
public FSTree() {
super();
init();
}
public FSTree(TreeModel newModel) {
super(newModel);
init();
}
public FSTree(TreeNode root) {
super(root);
init();
}
public FSTree(TreeNode root, boolean asks) {
super(root, asks);
init();
}
private void init() {
setDragEnabled(false);
setTransferHandler(new FSTransfer());
}
public class FSTransfer extends TransferHandler {
public boolean importData(JComponent comp, Transferable t) {
if (!(comp instanceof FSTree)) {
return false;
}
if (!t.isDataFlavorSupported(DataFlavor.javaFileListFlavor)) {
return false;
}
FSTree tree = (FSTree) comp;
DefaultTreeModel model = (DefaultTreeModel) tree.getModel();
DefaultMutableTreeNode root = (DefaultMutableTreeNode) model.getRoot();
try {
List data = (List) t.getTransferData(DataFlavor.javaFileListFlavor);
Iterator i = data.iterator();
while (i.hasNext()) {
File f = (File) i.next();
root.add(new DefaultMutableTreeNode(f.getName()));
}
model.reload();
return true;
} catch (UnsupportedFlavorException ufe) {
System.err.println("Ack! we should not be here.\nBad Flavor.");
} catch (IOException ioe) {
System.out.println("Something failed during import:\n" + ioe);
}
return false;
}
// We only support file lists on FSTrees...
public boolean canImport(JComponent comp, DataFlavor[] transferFlavors) {
if (comp instanceof FSTree) {
for (int i = 0; i < transferFlavors.length; i++) {
if (!transferFlavors[i].equals(DataFlavor.javaFileListFlavor)) {
return false;
}
}
return true;
}
return false;
}
}
}