mirror of
https://github.com/dmolony/DiskBrowser.git
synced 2024-09-29 22:54:45 +00:00
314 lines
9.4 KiB
Java
314 lines
9.4 KiB
Java
|
package com.bytezone.diskbrowser.gui;
|
||
|
|
||
|
import static javax.swing.ScrollPaneConstants.HORIZONTAL_SCROLLBAR_NEVER;
|
||
|
import static javax.swing.ScrollPaneConstants.VERTICAL_SCROLLBAR_ALWAYS;
|
||
|
|
||
|
import java.awt.BorderLayout;
|
||
|
import java.awt.Dimension;
|
||
|
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.Map;
|
||
|
|
||
|
import javax.swing.*;
|
||
|
|
||
|
import com.bytezone.common.DefaultAction;
|
||
|
import com.bytezone.input.SpringUtilities;
|
||
|
|
||
|
public class DuplicateAction extends DefaultAction
|
||
|
{
|
||
|
Map<String, List<DiskDetails>> duplicateDisks;
|
||
|
int rootFolderLength;
|
||
|
File rootFolder;
|
||
|
DuplicateWindow window;
|
||
|
|
||
|
public DuplicateAction ()
|
||
|
{
|
||
|
super ("Check for duplicates...", "Check for duplicate disks",
|
||
|
"/com/bytezone/diskbrowser/icons/");
|
||
|
|
||
|
setIcon (Action.SMALL_ICON, "save_delete_16.png");
|
||
|
setIcon (Action.LARGE_ICON_KEY, "save_delete_32.png");
|
||
|
}
|
||
|
|
||
|
public void setDuplicates (File rootFolder, Map<String, List<DiskDetails>> duplicateDisks)
|
||
|
{
|
||
|
this.duplicateDisks = duplicateDisks;
|
||
|
this.rootFolderLength = rootFolder.getAbsolutePath ().length ();
|
||
|
this.rootFolder = rootFolder;
|
||
|
setEnabled (duplicateDisks.size () > 0);
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void actionPerformed (ActionEvent arg0)
|
||
|
{
|
||
|
if (duplicateDisks == null)
|
||
|
{
|
||
|
System.out.println ("No duplicate disks found");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (window != null)
|
||
|
{
|
||
|
window.setVisible (true);
|
||
|
return;
|
||
|
}
|
||
|
window = new DuplicateWindow ();
|
||
|
for (List<DiskDetails> diskList : duplicateDisks.values ())
|
||
|
new DuplicateWorker (diskList, window).execute ();
|
||
|
}
|
||
|
|
||
|
class DuplicateWindow extends JFrame
|
||
|
{
|
||
|
int unfinishedWorkers;
|
||
|
int folderNameLength;
|
||
|
|
||
|
JButton buttonDelete = new JButton ("Delete selected");
|
||
|
JButton buttonCancel = new JButton ("Cancel");
|
||
|
JButton buttonAll = new JButton ("Select all duplicates");
|
||
|
JButton buttonClear = new JButton ("Clear all");
|
||
|
JPanel mainPanel = new JPanel ();
|
||
|
|
||
|
List<DiskDetails> disksSelected = new ArrayList<DiskDetails> ();
|
||
|
List<DuplicatePanel> duplicatePanels = new ArrayList<DuplicateAction.DuplicatePanel> ();
|
||
|
|
||
|
public DuplicateWindow ()
|
||
|
{
|
||
|
super ("Duplicate Disk Detection - " + rootFolder.getAbsolutePath ());
|
||
|
unfinishedWorkers = duplicateDisks.size ();
|
||
|
folderNameLength = rootFolder.getAbsolutePath ().length ();
|
||
|
|
||
|
mainPanel.setLayout (new BoxLayout (mainPanel, BoxLayout.PAGE_AXIS));
|
||
|
|
||
|
JScrollPane sp =
|
||
|
new JScrollPane (mainPanel, VERTICAL_SCROLLBAR_ALWAYS, HORIZONTAL_SCROLLBAR_NEVER);
|
||
|
sp.getVerticalScrollBar ().setUnitIncrement (100);
|
||
|
add (sp, BorderLayout.CENTER);
|
||
|
|
||
|
JPanel panel = new JPanel ();
|
||
|
panel.add (buttonClear);
|
||
|
panel.add (buttonAll);
|
||
|
panel.add (buttonDelete);
|
||
|
panel.add (buttonCancel);
|
||
|
add (panel, BorderLayout.SOUTH);
|
||
|
|
||
|
buttonClear.setEnabled (false);
|
||
|
buttonAll.setEnabled (false);
|
||
|
buttonDelete.setEnabled (false);
|
||
|
buttonCancel.setEnabled (false);
|
||
|
|
||
|
buttonAll.addActionListener (new ActionListener ()
|
||
|
{
|
||
|
@Override
|
||
|
public void actionPerformed (ActionEvent e)
|
||
|
{
|
||
|
for (DuplicatePanel dp : duplicatePanels)
|
||
|
{
|
||
|
int count = 0;
|
||
|
for (JCheckBox cb : dp.checkBoxes)
|
||
|
{
|
||
|
if (count > 0 && dp.duplicateDisks.get (count).duplicate)
|
||
|
if (!cb.isSelected ())
|
||
|
{
|
||
|
cb.setSelected (true); // doesn't fire the actionListener!
|
||
|
disksSelected.add (dp.duplicateDisks.get (count));
|
||
|
}
|
||
|
++count;
|
||
|
}
|
||
|
}
|
||
|
buttonDelete.setEnabled (disksSelected.size () > 0);
|
||
|
buttonClear.setEnabled (disksSelected.size () > 0);
|
||
|
}
|
||
|
});
|
||
|
|
||
|
buttonClear.addActionListener (new ActionListener ()
|
||
|
{
|
||
|
@Override
|
||
|
public void actionPerformed (ActionEvent e)
|
||
|
{
|
||
|
for (DuplicatePanel dp : duplicatePanels)
|
||
|
for (JCheckBox cb : dp.checkBoxes)
|
||
|
cb.setSelected (false); // doesn't fire the actionListener!
|
||
|
disksSelected.clear ();
|
||
|
buttonDelete.setEnabled (false);
|
||
|
buttonClear.setEnabled (false);
|
||
|
}
|
||
|
});
|
||
|
|
||
|
buttonCancel.addActionListener (new ActionListener ()
|
||
|
{
|
||
|
@Override
|
||
|
public void actionPerformed (ActionEvent e)
|
||
|
{
|
||
|
DuplicateWindow.this.setVisible (false);
|
||
|
}
|
||
|
});
|
||
|
|
||
|
buttonDelete.addActionListener (new ActionListener ()
|
||
|
{
|
||
|
@Override
|
||
|
public void actionPerformed (ActionEvent e)
|
||
|
{
|
||
|
int totalDeleted = 0;
|
||
|
int totalFailed = 0;
|
||
|
for (DuplicatePanel dp : duplicatePanels)
|
||
|
{
|
||
|
int count = 0;
|
||
|
for (JCheckBox cb : dp.checkBoxes)
|
||
|
{
|
||
|
if (cb.isSelected () && false)
|
||
|
{
|
||
|
DiskDetails dd = dp.duplicateDisks.get (count);
|
||
|
if (dd.delete ())
|
||
|
{
|
||
|
++totalDeleted;
|
||
|
System.out.println ("Deleted : " + dd);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
++totalFailed;
|
||
|
System.out.println ("Failed : " + dd);
|
||
|
}
|
||
|
}
|
||
|
++count;
|
||
|
}
|
||
|
}
|
||
|
System.out.printf ("Deleted : %d, Failed : %d%n", totalDeleted, totalFailed);
|
||
|
}
|
||
|
});
|
||
|
|
||
|
setSize (600, 700);
|
||
|
setLocationRelativeTo (null);
|
||
|
setDefaultCloseOperation (HIDE_ON_CLOSE);
|
||
|
setVisible (true);
|
||
|
}
|
||
|
|
||
|
// create a DuplicatePanel based on the updated DiskDetails
|
||
|
public synchronized void addResult (List<DiskDetails> duplicateDisks)
|
||
|
{
|
||
|
// create panel and add it to the window
|
||
|
DuplicatePanel dp =
|
||
|
new DuplicatePanel (duplicateDisks, folderNameLength, disksSelected, buttonDelete,
|
||
|
buttonClear);
|
||
|
mainPanel.add (dp);
|
||
|
duplicatePanels.add (dp);
|
||
|
|
||
|
validate ();
|
||
|
|
||
|
if (--unfinishedWorkers == 0)
|
||
|
{
|
||
|
buttonAll.setEnabled (true);
|
||
|
buttonCancel.setEnabled (true);
|
||
|
}
|
||
|
else
|
||
|
mainPanel.add (Box.createRigidArea (new Dimension (0, 20)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
class DuplicatePanel extends JPanel
|
||
|
{
|
||
|
List<JCheckBox> checkBoxes = new ArrayList<JCheckBox> ();
|
||
|
List<DiskDetails> duplicateDisks;
|
||
|
|
||
|
public DuplicatePanel (List<DiskDetails> duplicateDisks, int folderNameLength,
|
||
|
List<DiskDetails> disksSelected, JButton deleteButton, JButton clearButton)
|
||
|
{
|
||
|
this.duplicateDisks = duplicateDisks;
|
||
|
setLayout (new SpringLayout ());
|
||
|
setAlignmentX (LEFT_ALIGNMENT);
|
||
|
|
||
|
int count = 0;
|
||
|
for (DiskDetails dd : duplicateDisks)
|
||
|
{
|
||
|
JCheckBox cb = new JCheckBox ();
|
||
|
checkBoxes.add (cb);
|
||
|
|
||
|
cb.addActionListener (new CheckBoxActionListener (dd, disksSelected, deleteButton,
|
||
|
clearButton));
|
||
|
add (cb);
|
||
|
if (++count == 1)
|
||
|
add (new JLabel ("Source disk"));
|
||
|
else
|
||
|
{
|
||
|
String text = dd.duplicate ? "Duplicate" : "OK";
|
||
|
add (new JLabel (text));
|
||
|
}
|
||
|
String checksum =
|
||
|
dd.duplicate || count == 1 ? "" : " (checksum = " + dd.getChecksum () + ")";
|
||
|
add (new JLabel (dd.getAbsolutePath ().substring (folderNameLength) + checksum));
|
||
|
}
|
||
|
SpringUtilities.makeCompactGrid (this, duplicateDisks.size (), 3, //rows, cols
|
||
|
10, 0, //initX, initY
|
||
|
10, 0); //xPad, yPad
|
||
|
}
|
||
|
}
|
||
|
|
||
|
class CheckBoxActionListener implements ActionListener
|
||
|
{
|
||
|
DiskDetails diskDetails;
|
||
|
List<DiskDetails> disksSelected;
|
||
|
JButton deleteButton;
|
||
|
JButton clearButton;
|
||
|
|
||
|
public CheckBoxActionListener (DiskDetails diskDetails, List<DiskDetails> disksSelected,
|
||
|
JButton deleteButton, JButton clearButton)
|
||
|
{
|
||
|
this.diskDetails = diskDetails;
|
||
|
this.disksSelected = disksSelected;
|
||
|
this.deleteButton = deleteButton;
|
||
|
this.clearButton = clearButton;
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void actionPerformed (ActionEvent e)
|
||
|
{
|
||
|
if (((JCheckBox) e.getSource ()).isSelected ())
|
||
|
disksSelected.add (diskDetails);
|
||
|
else
|
||
|
disksSelected.remove (diskDetails);
|
||
|
deleteButton.setEnabled (disksSelected.size () > 0);
|
||
|
clearButton.setEnabled (disksSelected.size () > 0);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
class DuplicateWorker extends SwingWorker<List<DiskDetails>, Void>
|
||
|
{
|
||
|
List<DiskDetails> duplicateDisks;
|
||
|
DuplicateWindow owner;
|
||
|
|
||
|
public DuplicateWorker (List<DiskDetails> duplicateDisks, DuplicateWindow owner)
|
||
|
{
|
||
|
this.duplicateDisks = duplicateDisks;
|
||
|
this.owner = owner;
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
protected void done ()
|
||
|
{
|
||
|
try
|
||
|
{
|
||
|
owner.addResult (get ());
|
||
|
}
|
||
|
catch (Exception e)
|
||
|
{
|
||
|
e.printStackTrace ();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
protected List<DiskDetails> doInBackground () throws Exception
|
||
|
{
|
||
|
long firstChecksum = -1;
|
||
|
for (DiskDetails dd : duplicateDisks)
|
||
|
{
|
||
|
if (firstChecksum < 0)
|
||
|
firstChecksum = dd.getChecksum ();
|
||
|
else
|
||
|
dd.duplicate = (dd.getChecksum () == firstChecksum);
|
||
|
}
|
||
|
return duplicateDisks;
|
||
|
}
|
||
|
}
|
||
|
}
|