simplify a bit
[jSite2.git] / src / net / pterodactylus / jsite / gui / FileManager.java
index 8f10ebb..26b6e14 100644 (file)
 package net.pterodactylus.jsite.gui;
 
 import java.awt.BorderLayout;
+import java.awt.Color;
 import java.awt.Component;
+import java.awt.Dimension;
 import java.awt.FlowLayout;
+import java.awt.Font;
 import java.awt.GridBagConstraints;
 import java.awt.GridBagLayout;
 import java.awt.Insets;
 import java.awt.event.ActionEvent;
 import java.awt.event.ActionListener;
 import java.io.File;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -37,15 +41,16 @@ import javax.swing.JButton;
 import javax.swing.JCheckBox;
 import javax.swing.JComboBox;
 import javax.swing.JDialog;
-import javax.swing.JFrame;
+import javax.swing.JLabel;
+import javax.swing.JOptionPane;
 import javax.swing.JPanel;
 import javax.swing.JScrollPane;
 import javax.swing.JTree;
 import javax.swing.event.TreeSelectionEvent;
 import javax.swing.event.TreeSelectionListener;
-import javax.swing.tree.DefaultMutableTreeNode;
+import javax.swing.tree.DefaultTreeCellRenderer;
 import javax.swing.tree.DefaultTreeModel;
-import javax.swing.tree.TreeModel;
+import javax.swing.tree.TreeNode;
 
 import net.pterodactylus.jsite.i18n.I18n;
 import net.pterodactylus.jsite.i18n.I18nable;
@@ -54,6 +59,7 @@ import net.pterodactylus.jsite.i18n.gui.I18nLabel;
 import net.pterodactylus.jsite.project.Project;
 import net.pterodactylus.util.io.MimeTypes;
 import net.pterodactylus.util.logging.Logging;
+import net.pterodactylus.util.swing.SortableTreeNode;
 import net.pterodactylus.util.swing.SwingUtils;
 
 /**
@@ -72,8 +78,20 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
        /** The project whose files to manage. */
        private final Project project;
 
+       /** The root of the file tree. */
+       private final SortableTreeNode fileTreeRoot;
+
        /** The tree model for the project files. */
-       private final TreeModel fileTreeModel;
+       private final DefaultTreeModel fileTreeModel;
+
+       /** Files that are hidden as per {@link File#isHidden()}. */
+       private final List<String> hiddenFiles = new ArrayList<String>();
+
+       /** The tree cell renderer. */
+       private final FileCellRenderer fileCellRenderer;
+
+       /** The “rescan” action. */
+       private I18nAction rescanAction;
 
        /** The “close” action. */
        private I18nAction closeAction;
@@ -84,6 +102,9 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
        /** The tree that shows the files. */
        private JTree fileTree;
 
+       /** The scroll pane that holds the file tree. */
+       private JScrollPane fileScrollPane;
+
        /** The “insert” action. */
        private I18nAction insertAction;
 
@@ -112,7 +133,9 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
                logger.log(Level.FINEST, "project: " + project);
                this.swingInterface = swingInterface;
                this.project = project;
-               fileTreeModel = new DefaultTreeModel(new DefaultMutableTreeNode(File.separator));
+               fileTreeRoot = new SortableTreeNode(project.getName());
+               fileTreeModel = new DefaultTreeModel(fileTreeRoot);
+               fileCellRenderer = new FileCellRenderer();
                initActions();
                initComponents();
                pack();
@@ -131,7 +154,7 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
                if (visible) {
                        initiateFileScan();
                }
-           super.setVisible(visible);
+               super.setVisible(visible);
        }
 
        //
@@ -151,6 +174,16 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
                                setVisible(false);
                        }
                };
+               rescanAction = new I18nAction("fileManager.button.rescan") {
+
+                       /**
+                        * {@inheritDoc}
+                        */
+                       @SuppressWarnings("synthetic-access")
+                       public void actionPerformed(ActionEvent actionEvent) {
+                               initiateFileScan();
+                       }
+               };
                insertAction = new I18nAction("fileManager.checkbox.insertFile") {
 
                        /**
@@ -196,21 +229,29 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
        private Component createFileManagerPanel() {
                JPanel fileManagerPanel = new JPanel(new BorderLayout(12, 12));
 
+               /* file tree panel */
                JPanel fileTreePanel = new JPanel(new BorderLayout(12, 12));
                fileManagerPanel.add(fileTreePanel, BorderLayout.LINE_START);
 
                fileTree = new JTree(fileTreeModel);
                fileTree.setShowsRootHandles(false);
                fileTree.addTreeSelectionListener(this);
-               fileTreePanel.add(new JScrollPane(fileTree), BorderLayout.CENTER);
+               fileTree.setCellRenderer(fileCellRenderer);
+               fileTreePanel.add(fileScrollPane = new JScrollPane(fileTree), BorderLayout.CENTER);
+               fileScrollPane.setPreferredSize(new Dimension(250, 400));
 
                projectFilesLabel = new I18nLabel("fileManager.label.projectFiles", fileTree);
                JPanel projectFilesLabelPanel = new JPanel(new FlowLayout(FlowLayout.CENTER, 0, 0));
                fileTreePanel.add(projectFilesLabelPanel, BorderLayout.NORTH);
                projectFilesLabelPanel.add(projectFilesLabel);
 
+               /* the right panel */
+               JPanel rightPanel = new JPanel(new BorderLayout(12, 12));
+               fileManagerPanel.add(rightPanel, BorderLayout.CENTER);
+
+               /* properties panel */
                JPanel propertiesPanel = new JPanel(new GridBagLayout());
-               fileManagerPanel.add(propertiesPanel, BorderLayout.CENTER);
+               rightPanel.add(propertiesPanel, BorderLayout.CENTER);
                propertiesPanel.setBorder(BorderFactory.createCompoundBorder(BorderFactory.createEtchedBorder(), BorderFactory.createEmptyBorder(12, 12, 12, 12)));
 
                insertCheckBox = new JCheckBox(insertAction);
@@ -226,6 +267,14 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
 
                propertiesPanel.add(new JPanel(), new GridBagConstraints(0, 2, 1, 1, 1.0, 1.0, GridBagConstraints.LINE_START, GridBagConstraints.BOTH, new Insets(0, 0, 0, 0), 0, 0));
 
+               /* action button panel */
+               JPanel actionButtonPanel = new JPanel(new FlowLayout(FlowLayout.LEADING, 12, 12));
+               rightPanel.add(actionButtonPanel, BorderLayout.PAGE_END);
+               actionButtonPanel.setBorder(BorderFactory.createEtchedBorder());
+
+               JButton rescanButton = new JButton(rescanAction);
+               actionButtonPanel.add(rescanButton);
+
                return fileManagerPanel;
        }
 
@@ -245,15 +294,50 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
                return buttonPanel;
        }
 
+       /**
+        * Initiates a file scan.
+        */
        private void initiateFileScan() {
                swingInterface.getThreadPool().execute(new Runnable() {
+
                        /**
                         * @see java.lang.Runnable#run()
                         */
+                       @SuppressWarnings("synthetic-access")
                        public void run() {
-                String basePath = project.getBasePath();
-                File basePathDirectory = new File(basePath);
+                               fileTree.setEnabled(false);
+                               rescanAction.setEnabled(false);
+                               String basePath = project.getBasePath();
+                               File basePathDirectory = new File(basePath);
+                               fileTreeRoot.removeAll();
+                               hiddenFiles.clear();
+                               if (!basePathDirectory.exists() || !basePathDirectory.isDirectory()) {
+                                       /* TODO - i18n */
+                                       JOptionPane.showMessageDialog(FileManager.this, I18n.get(""), I18n.get(""), JOptionPane.ERROR_MESSAGE);
+                               } else {
+                                       scanDirectory(fileTreeRoot, basePathDirectory, "", hiddenFiles);
+                               }
+                               fileTreeModel.reload();
+                               // fileScrollPane.revalidate();
+                               rescanAction.setEnabled(true);
+                               fileTree.setEnabled(true);
+                       }
+
+                       private void scanDirectory(SortableTreeNode rootNode, File directory, String currentDirectory, List<String> hiddenFiles) {
+                               for (File file : directory.listFiles()) {
+                                       String fileName = file.getName();
+                                       SortableTreeNode fileNode = new SortableTreeNode(fileName);
+                                       rootNode.add(fileNode);
+                                       if (file.isFile() && file.isHidden()) {
+                                               hiddenFiles.add((currentDirectory + File.separator + fileName).substring(1));
+                                       }
+                                       if (file.isDirectory()) {
+                                               scanDirectory(fileNode, file, currentDirectory + File.separator + fileName, hiddenFiles);
+                                       }
+                               }
+                               rootNode.sort();
                        }
+
                });
        }
 
@@ -276,6 +360,7 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
         * {@inheritDoc}
         */
        public void valueChanged(TreeSelectionEvent treeSelectionEvent) {
+               /* TODO */
        }
 
        //
@@ -286,6 +371,68 @@ public class FileManager extends JDialog implements I18nable, ActionListener, Tr
         * {@inheritDoc}
         */
        public void actionPerformed(ActionEvent actionEvent) {
+               /* TODO */
+       }
+
+       /**
+        * Tree cell renderer that takes care of certain display properties for
+        * project-specific stuff.
+        *
+        * @author David ‘Bombe’ Roden &lt;bombe@freenetproject.org&gt;
+        */
+       private class FileCellRenderer extends DefaultTreeCellRenderer {
+
+               /**
+                * @see javax.swing.tree.TreeCellRenderer#getTreeCellRendererComponent(javax.swing.JTree,
+                *      java.lang.Object, boolean, boolean, boolean, int, boolean)
+                */
+               @Override
+               public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {
+                       Component superCellRenderer = super.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus);
+                       if (!(superCellRenderer instanceof JLabel)) {
+                               logger.log(Level.WARNING, "superCellRenderer is not a JLabel!");
+                               return superCellRenderer;
+                       }
+                       if (!(value instanceof SortableTreeNode)) {
+                               logger.log(Level.WARNING, "value is not a SortableTreeNode!");
+                               return superCellRenderer;
+                       }
+                       SortableTreeNode node = (SortableTreeNode) value;
+                       TreeNode[] pathToRoot = fileTreeModel.getPathToRoot(node);
+                       if (pathToRoot.length > 1) {
+                               StringBuilder completePathBuilder = new StringBuilder();
+                               boolean first = true;
+                               for (TreeNode rootPathNode : pathToRoot) {
+                                       if (first) {
+                                               first = false;
+                                               continue;
+                                       }
+                                       if (!(rootPathNode instanceof SortableTreeNode)) {
+                                               logger.log(Level.WARNING, "rootPathNode is not a SortableTreeNode!");
+                                               continue;
+                                       }
+                                       completePathBuilder.append(File.separatorChar).append(((SortableTreeNode) rootPathNode).getUserObject());
+                               }
+                               String completePath = completePathBuilder.substring(1);
+                               if (project.getDefaultFile().equals(completePath)) {
+                                       superCellRenderer.setFont(superCellRenderer.getFont().deriveFont(Font.BOLD));
+                               } else {
+                                       if (hiddenFiles.contains(completePath)) {
+                                               /* fade hidden files’ font */
+                                               Color foreground = superCellRenderer.getForeground();
+                                               Color background = selected ? getBackgroundSelectionColor() : getBackgroundNonSelectionColor();
+                                               Color averageColor = new Color((foreground.getRed() + background.getRed()) / 2, (foreground.getGreen() + background.getGreen()) / 2, (foreground.getBlue() + background.getBlue()) / 2);
+                                               superCellRenderer.setForeground(averageColor);
+                                       } else {
+                                               superCellRenderer.setFont(superCellRenderer.getFont().deriveFont(Font.PLAIN));
+                                       }
+                               }
+                       } else {
+                               superCellRenderer.setFont(superCellRenderer.getFont().deriveFont(Font.BOLD));
+                       }
+                       return superCellRenderer;
+               }
+
        }
 
 }