FolderCompareFrame with good layout

21ca發表於2016-12-16
Go

點選(此處)摺疊或開啟

  1. import java.awt.BorderLayout;
  2. import java.awt.Dimension;
  3. import java.awt.event.ActionEvent;
  4. import java.awt.event.ActionListener;
  5. import java.awt.event.MouseEvent;
  6. import java.awt.event.MouseListener;
  7. import java.io.File;
  8. import java.io.FileInputStream;
  9. import java.io.FileOutputStream;
  10. import java.io.IOException;
  11. import java.security.MessageDigest;
  12. import java.security.NoSuchAlgorithmException;
  13. import java.util.ArrayList;
  14. import java.util.Arrays;
  15. import java.util.Collections;
  16. import java.util.LinkedHashSet;
  17. import java.util.List;
  18. import java.util.Properties;
  19. import java.util.Set;

  21. import javax.swing.DefaultListModel;
  22. import javax.swing.JButton;
  23. import javax.swing.JComponent;
  24. import javax.swing.JFileChooser;
  25. import javax.swing.JFrame;
  26. import javax.swing.JLabel;
  27. import javax.swing.JList;
  28. import javax.swing.JPanel;
  29. import javax.swing.JScrollPane;
  30. import javax.swing.JTextField;
  31. import javax.swing.plaf.basic.BasicBorders;


  34. public class FolderCompareFrame2 extends JFrame implements ActionListener, MouseListener {

  36.     private static final String[] SUFFIX_FILTER = new String[] {".java", ".properties", ".xml", ".jsp", ".sql", ".jar", ".php" };
  37.     private static final String[] EXCLUDE_FILTER = new String[] {"/test/", "/integration/", "/target/"};
  38.     
  39.     private static final long serialVersionUID = 3278358999317656288L;
  40.     private static final String notepadPlusCmd = "C:/Program Files (x86)/Notepad++/notepad++.exe";
  41.     
  42.     private static final String CMD_SELECT_SOURCE = "source";
  43.     private static final String CMD_SELECT_TARGET = "target";
  44.     private static final String CMD_COMPARE = "compare";

  46.     private JTextField sourceInput;
  47.     private JTextField targetInput;

  49.     private JList<String> resultList;
  50.     private DefaultListModel<String> resultListModel = new DefaultListModel<String>();
  51.     
  52.     private String sourceFolder;
  53.     private String targetFolder;
  54.     
  55.     private JFileChooser fileChooser = new JFileChooser();
  56.     
  57.     public FolderCompareFrame2() {
  58.         setSize(800, 600);
  59.         fileChooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
  60.         
  61.         setDefaultCloseOperation(EXIT_ON_CLOSE);
  62.         setTitle("Folder Compare");
  63.         setLayout(new BorderLayout());
  64.         addTopPanel();
  65.         addResultList();
  66.         loadSettings();
  67.         setVisible(true);

  69.     }
  70.     
  71.     private void addTopPanel() {
  72.         JPanel topPanel = new JPanel(new BorderLayout(10, 0));
  73.         
  74.         JPanel inputPanel = new JPanel(new BorderLayout());
  75.         addSourceFolderInput(inputPanel);
  76.         addTargetFolderInput(inputPanel);
  77.         JButton compareButton = new JButton("Compare");
  78.         compareButton.setActionCommand(CMD_COMPARE);
  79.         compareButton.addActionListener(this);

  81.         topPanel.add(inputPanel, BorderLayout.CENTER);
  82.         topPanel.add(compareButton, BorderLayout.EAST);
  83.         
  84.         add(topPanel, BorderLayout.NORTH);
  85.         
  86.     }

  88.     private void loadSettings() {
  89.         Properties props = new Properties();
  90.         try {
  91.             props.load(new FileInputStream(this.getClass().getSimpleName() + ".config"));
  92.             sourceInput.setText(props.getProperty("folder1"));
  93.             targetInput.setText(props.getProperty("folder2"));
  94.         } catch (Exception e) {
  95.             e.printStackTrace();
  96.         }
  97.     }
  98.     
  99.     private void saveSettings() {
  100.         Properties props = new Properties();
  101.         try {
  102.             props.put("folder1", sourceInput.getText());
  103.             props.put("folder2", targetInput.getText());
  104.             props.store(new FileOutputStream(this.getClass().getSimpleName() + ".config"), null);
  105.         } catch (Exception e) {
  106.             e.printStackTrace();
  107.         }
  108.     }

  110.     private void addResultList() {
  111.         resultList = new JList<String>();
  112.         JScrollPane pane = new JScrollPane(resultList);
  113.         pane.setBorder(BasicBorders.getTextFieldBorder());
  114.         add(pane, BorderLayout.CENTER);
  115.         resultList.addMouseListener(this);
  116.         resultList.setModel(resultListModel);
  117.     }

  119.     private void addTargetFolderInput(JComponent p) {
  120.         JPanel panel = new JPanel();
  121.         p.add(panel, BorderLayout.SOUTH);
  122.         panel.setLayout(new BorderLayout());
  123.         
  124.         JLabel label = new JLabel("Target:");
  125.         label.setPreferredSize(new Dimension(60, 0));
  126.         panel.add(label, BorderLayout.WEST);
  127.         
  128.         targetInput = new JTextField();
  129.         panel.add(targetInput, BorderLayout.CENTER);
  130.         
  131.         JButton button = new JButton("...");
  132.         button.setActionCommand(CMD_SELECT_TARGET);
  133.         button.addActionListener(this);
  134.         panel.add(button, BorderLayout.EAST);
  135.     }

  137.     private void addSourceFolderInput(JComponent p) {
  138.         JPanel panel = new JPanel();
  139.         p.add(panel, BorderLayout.NORTH);
  140.         panel.setLayout(new BorderLayout());

  142.         JLabel label = new JLabel("Source:");
  143.         label.setPreferredSize(new Dimension(60, 0));
  144.         panel.add(label, BorderLayout.WEST);
  145.         
  146.         sourceInput = new JTextField();
  147.         panel.add(sourceInput, BorderLayout.CENTER);
  148.         
  149.         JButton button = new JButton("...");
  150.         button.setActionCommand(CMD_SELECT_SOURCE);
  151.         button.addActionListener(this);
  152.         panel.add(button, BorderLayout.EAST);
  153.     }

  155.     public static void main(String[] args) {
  156.         new FolderCompareFrame2();
  157.     }

  159.     @Override
  160.     public void actionPerformed(ActionEvent e) {
  161.         if (e.getActionCommand() == CMD_SELECT_SOURCE) {
  162.             fileChooser.setCurrentDirectory(new File(sourceInput.getText()));
  163.             int result = fileChooser.showOpenDialog(FolderCompareFrame2.this);
  164.             if(result == JFileChooser.APPROVE_OPTION) {
  165.                 sourceInput.setText(fileChooser.getSelectedFile().getAbsolutePath());
  166.             }
  167.         } else if (e.getActionCommand() == CMD_SELECT_TARGET) {
  168.             fileChooser.setCurrentDirectory(new File(targetInput.getText()));
  169.             int result = fileChooser.showOpenDialog(FolderCompareFrame2.this);
  170.             if(result == JFileChooser.APPROVE_OPTION) {
  171.                 targetInput.setText(fileChooser.getSelectedFile().getAbsolutePath());
  172.             }
  173.         } else if (e.getActionCommand() == CMD_COMPARE) {
  174.             sourceFolder = sourceInput.getText().replaceAll("\\\\", "/");
  175.             if (!sourceFolder.endsWith("/")) {
  176.                 sourceFolder = sourceFolder + "/";
  177.             }
  178.             targetFolder = targetInput.getText().replaceAll("\\\\", "/");
  179.             if (!targetFolder.endsWith("/")) {
  180.                 targetFolder = targetFolder + "/";
  181.             }
  182.             saveSettings();
  183.             resultListModel.clear();;
  184.             compare(sourceFolder, targetFolder);
  185.         }
  186.     }
  187.     
  188.     /**
  189.      * Start NotePad++
  190.      * @param sourceFile
  191.      * @param targetFile
  192.      */
  193.     private void compareFile(String sourceFile, String targetFile) {
  194.         ProcessBuilder pb = new ProcessBuilder(notepadPlusCmd, sourceFile, targetFile);
  195.         try {
  196.             pb.start();
  197.         } catch (Exception e) {
  198.             e.printStackTrace();
  199.         }
  200.     }
  201.     
  202.     private boolean endWith(String file, String[] suffix) {
  203.         for (String s : suffix) {
  204.             if (file.endsWith(s)) {
  205.                 return true;
  206.             }
  207.         }
  208.         return false;
  209.     }
  210.     
  211.     private boolean containString(String file, String[] containString) {
  212.         for (String s : containString) {
  213.             if (file.contains(s)) {
  214.                 return true;
  215.             }
  216.         }
  217.         return false;
  218.     }
  219.     
  220.     private List<String> list(File folder) {
  221.         if (folder.isFile()) {
  222.             return Collections.singletonList(folder.getAbsolutePath().replace("\\", "/"));
  223.         }
  224.         List<String> list = new ArrayList<String>();
  225.         for (File f : folder.listFiles()) {
  226.             list.addAll(list(f));
  227.         }
  228.         return list;
  229.     }

  231.     private void compare(String source, String target) {
  232.         System.out.println("Compare File: " + source + " with " + target);
  233.         File sourceFile = new File(source);
  234.         File targetFile = new File(target);
  235.         
  236.         if (!sourceFile.exists() || !targetFile.exists()) {
  237.             //Source file or target file not exist.
  238.             if ((sourceFile.isFile() || targetFile.isFile())
  239.                     && endWith(source, SUFFIX_FILTER) && !containString(source, EXCLUDE_FILTER)) {
  240.                 resultListModel.addElement(source.substring(sourceFolder.length()));
  241.             }
  242.             
  243.             //Source file is a folder, target folder not exist.
  244.             if (sourceFile.isDirectory()) {
  245.                 for (String f : list(sourceFile)) {
  246.                     if (endWith(f, SUFFIX_FILTER) && !containString(f, EXCLUDE_FILTER)) {
  247.                         resultListModel.addElement(f.substring(sourceFolder.length()));
  248.                     }
  249.                 }
  250.             }
  251.             
  252.             if (targetFile.isDirectory()) {
  253.                 for (String f : list(targetFile)) {
  254.                     if (endWith(f, SUFFIX_FILTER) && !containString(f, EXCLUDE_FILTER)) {
  255.                         resultListModel.addElement(f.substring(targetFolder.length()));
  256.                     }
  257.                 }
  258.             }
  259.             
  260.         } else if (sourceFile.isFile() && targetFile.isFile() && endWith(source, SUFFIX_FILTER)
  261.                 && !containString(source, EXCLUDE_FILTER) && !fileMatch(sourceFile, targetFile)) {
  262.             resultListModel.addElement(source.substring(sourceFolder.length()));
  263.         } else if (sourceFile.isDirectory() && targetFile.isDirectory()) {
  264.             if (!source.endsWith("/")) {
  265.                 source = source + "/";
  266.             }
  267.             if (!target.endsWith("/")) {
  268.                 target = target + "/";
  269.             }
  270.             
  271.             Set<String> subFolders = new LinkedHashSet<String>();
  272.             subFolders.addAll(Arrays.asList(sourceFile.list()));
  273.             subFolders.addAll(Arrays.asList(targetFile.list()));
  274.             for (String subFolder : subFolders) {
  275.                 compare(source + subFolder, target + subFolder);
  276.             }
  277.         }
  278.     }

  280.     private boolean fileMatch(File sourceFile, File targetFile) {
  281.         if (sourceFile.length() != targetFile.length()) {
  282.             return false;
  283.         }
  284.         
  285.         try {
  286.             if (!Arrays.equals(md5(readFileToByteArray(sourceFile)), md5(readFileToByteArray(targetFile)))) {
  287.                 return false;
  288.             }
  289.         } catch (IOException e) {
  290.             e.printStackTrace();
  291.             return false;
  292.         }
  293.         
  294.         return true;
  295.     }

  297.     private byte[] md5(byte[] buf) {
  298.         try {
  299.             MessageDigest md5 = MessageDigest.getInstance("MD5");
  300.             return md5.digest(buf);
  301.         } catch (NoSuchAlgorithmException e) {
  302.             throw new IllegalArgumentException();
  303.         }
  304.     }

  306.     private byte[] readFileToByteArray(File f) throws IOException {
  307.         FileInputStream input = new FileInputStream(f);
  308.         try {
  309.             if (f.isFile() && f.canRead()) {
  310.                 int size = (int)f.length();
  311.                 byte[] data = new byte[size];
  312.                 
  313.                 int offset = 0;
  314.          int readed;
  315.     
  316.          while (offset < size && (readed = input.read(data, offset, size - offset)) != -1) {
  317.          offset += readed;
  318.          }
  319.                 return data;
  320.             } else {
  321.                 return null;
  322.             }
  323.         } finally {
  324.             input.close();
  325.         }
  326.     }

  328.     @Override
  329.     public void mouseClicked(MouseEvent e) {
  330.         if (e.getClickCount() == 2 && resultList.getSelectedIndex() >= 0) {
  331.             String file = (String) resultList.getSelectedValue();
  332.             compareFile(sourceFolder + file, targetFolder + file);
  333.         }
  334.     }
  335.     public void mousePressed(MouseEvent e) {}    
  336.     public void mouseReleased(MouseEvent e) {}
  337.     public void mouseEntered(MouseEvent e) {}
  338.     public void mouseExited(MouseEvent e) {}
  339. }


來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/10742815/viewspace-2130753/，如需轉載，請註明出處，否則將追究法律責任。

相關文章