package net.sf.jabref.external;

 import java.io.File;

 import java.io.IOException;

 import java.util.Iterator;

 import java.util.List;

 import javax.swing.*;

 import javax.swing.event.ChangeEvent;

 import javax.swing.event.ChangeListener;

 import net.sf.jabref.*;

 import net.sf.jabref.gui.MainTable;

 import net.sf.jabref.gui.FileListTableModel;

 import net.sf.jabref.gui.FileListEntry;

 import net.sf.jabref.undo.NamedCompound;

 import net.sf.jabref.undo.UndoableFieldChange;

 import net.sf.jabref.undo.UndoableInsertEntry;

 import net.sf.jabref.util.XMPUtil;

 import com.jgoodies.forms.builder.DefaultFormBuilder;

 import com.jgoodies.forms.debug.FormDebugPanel;

 import com.jgoodies.forms.layout.CellConstraints;

 import com.jgoodies.forms.layout.FormLayout;

 /**

  * This class holds the functionality of autolinking to a file that's dropped

  * onto an entry.

  * 

  * Options for handling the files are:

  * 

  * 1) Link to the file in its current position (disabled if the file is remote)

  * 

  * 2) Copy the file to ??? directory, rename after bibtex key, and extension

  * 

  * 3) Move the file to ??? directory, rename after bibtex key, and extension

  */

 public class DroppedFileHandler {

     private JabRefFrame frame;

     private BasePanel panel;

     private JRadioButton linkInPlace = new JRadioButton(), copyRadioButton = new JRadioButton(),

         moveRadioButton = new JRadioButton();

     private JLabel destDirLabel = new JLabel();

     private JCheckBox renameCheckBox = new JCheckBox();

     private JTextField renameToTextBox = new JTextField(25);

     private JPanel optionsPanel = new JPanel();

     public DroppedFileHandler(JabRefFrame frame, BasePanel panel) {

         this.frame = frame;

         this.panel = panel;

         ButtonGroup grp = new ButtonGroup();

         grp.add(linkInPlace);

         grp.add(copyRadioButton);

         grp.add(moveRadioButton);

         copyRadioButton.setSelected(true);

         FormLayout layout = new FormLayout("left:15dlu,pref,pref,pref","bottom:14pt,pref,pref,pref,pref");

         layout.setRowGroups(new int[][]{{1, 2, 3, 4, 5}});

         DefaultFormBuilder builder = new DefaultFormBuilder(layout,        optionsPanel);

         builder.setDefaultDialogBorder();

         CellConstraints cc = new CellConstraints();

         builder.add(linkInPlace, cc.xyw(1, 1, 4));

         builder.add(destDirLabel, cc.xyw(1, 2, 4));

         builder.add(copyRadioButton, cc.xyw(2, 3, 3));        

         builder.add(moveRadioButton, cc.xyw(2, 4, 3));

         builder.add(renameCheckBox, cc.xyw(2, 5, 1));

         builder.add(renameToTextBox, cc.xyw(4, 5, 1));

     }

     /**

      * Offer copy/move/linking options for a dragged external file. Perform the

      * chosen operation, if any.

      * 

      * @param fileName

      *            The name of the dragged file.

      * @param fileType

      *            The FileType associated with the file.

      * @param localFile

      *            Indicate whether this is a local file, or a remote file copied

      *            to a local temporary file.

      * @param mainTable

      *            The MainTable the file was dragged to.

      * @param dropRow

      *            The row where the file was dropped.

      */

     public void handleDroppedfile(String fileName, ExternalFileType fileType, boolean localFile,

         MainTable mainTable, int dropRow) {

         BibtexEntry entry = mainTable.getEntryAt(dropRow);

         handleDroppedfile(fileName, fileType, localFile, entry);

     }

     /**

      *

      * @param fileName

      *        The name of the dragged file.

      * @param fileType

      *        The FileType associated with the file.

      * @param localFile

  *            Indicate whether this is a local file, or a remote file copied

  *            to a local temporary file.

      * @param entry

      *        The target entry for the drop.

      */

     public void handleDroppedfile(String fileName, ExternalFileType fileType, boolean localFile,

         BibtexEntry entry) {

         NamedCompound edits = new NamedCompound(Globals.lang("Drop %0", fileType.extension));

         if (tryXmpImport(fileName, fileType, localFile, edits)) {

             edits.end();

             panel.undoManager.addEdit(edits);

             return;

         }

         // Show dialog

         boolean newEntry = false;

         String citeKey = entry.getCiteKey();

         int reply = showLinkMoveCopyRenameDialog(fileName, fileType, citeKey, newEntry, false);

         if (reply != JOptionPane.OK_OPTION)

             return;

         /*

          * Ok, we're ready to go. See first if we need to do a file copy before

          * linking:

          */

         boolean success = true;

         String destFilename;

         if (linkInPlace.isSelected()) {

             destFilename = fileName;

         } else {

             destFilename = (renameCheckBox.isSelected() ? renameToTextBox.getText() : new File(fileName).getName());

             if (copyRadioButton.isSelected()) {

                 success = doCopy(fileName, fileType, destFilename, edits);

             } else if (moveRadioButton.isSelected()) {

                 success = doMove(fileName, fileType, destFilename, edits);

             }

         }

         if (success) {

             doLink(entry, fileType, destFilename, false, edits);

             panel.markBaseChanged();

             panel.updateEntryEditorIfShowing();

         }

         edits.end();

         panel.undoManager.addEdit(edits);

     }

     private boolean tryXmpImport(String fileName, ExternalFileType fileType, boolean localFile,

         NamedCompound edits) {

         if (!fileType.extension.equals("pdf")) {

             return false;

         }

         List<BibtexEntry> xmpEntriesInFile = null;

         try {

             xmpEntriesInFile = XMPUtil.readXMP(fileName);

         } catch (Exception e) {

             return false;

         }

         if ((xmpEntriesInFile == null) || (xmpEntriesInFile.size() == 0)) {

             return false;

         }

         JLabel confirmationMessage = new JLabel(

             Globals

                 .lang("The PDF contains one or several bibtex-records.\nDo you want to import these as new entries into the current database?"));

         int reply = JOptionPane.showConfirmDialog(frame, confirmationMessage, Globals.lang(

             "XMP metadata found in PDF: %0", fileName), JOptionPane.YES_NO_CANCEL_OPTION,

             JOptionPane.QUESTION_MESSAGE);

         if (reply == JOptionPane.CANCEL_OPTION) {

             return true; // The user canceled thus that we are done.

         }

         if (reply == JOptionPane.NO_OPTION) {

             return false;

         }

         // reply == JOptionPane.YES_OPTION)

         /*

          * TODO Extract Import functionality from ImportMenuItem then we could

          * do:

          * 

          * ImportMenuItem importer = new ImportMenuItem(frame, (mainTable ==

          * null), new PdfXmpImporter());

          * 

          * importer.automatedImport(new String[] { fileName });

          */

         boolean isSingle = xmpEntriesInFile.size() == 1;

         BibtexEntry single = (isSingle ? xmpEntriesInFile.get(0) : null);

         boolean success = true;

         String destFilename;

         if (linkInPlace.isSelected()) {

             destFilename = fileName;

         } else {

             if (renameCheckBox.isSelected()) {

                 destFilename = fileName;

             } else {

                 destFilename = single.getCiteKey() + "." + fileType.extension;

             }

             if (copyRadioButton.isSelected()) {

                 success = doCopy(fileName, fileType, destFilename, edits);

             } else if (moveRadioButton.isSelected()) {

                 success = doMove(fileName, fileType, destFilename, edits);

             }

         }

         if (success) {

             Iterator<BibtexEntry> it = xmpEntriesInFile.iterator();

             while (it.hasNext()) {

                 try {

                     BibtexEntry entry = it.next();

                     entry.setId(Util.createNeutralId());

                     edits.addEdit(new UndoableInsertEntry(panel.getDatabase(), entry, panel));

                     panel.getDatabase().insertEntry(entry);

                     doLink(entry, fileType, destFilename, true, edits);

                 } catch (KeyCollisionException ex) {

                 }

             }

             panel.markBaseChanged();

             panel.updateEntryEditorIfShowing();

         }

         return true;

     }

     public int showLinkMoveCopyRenameDialog(String linkFileName, ExternalFileType fileType,

         String citeKey, boolean newEntry, final boolean multipleEntries) {

             String dialogTitle = Globals.lang("Link to file %0", linkFileName);

         //String dir = panel.metaData().getFileDirectory(fileType.getFieldName());

         String dir = panel.metaData().getFileDirectory(GUIGlobals.FILE_FIELD);

         if ((dir == null) || !(new File(dir)).exists()) {

             destDirLabel.setText(Globals.lang("File directory is not set or does not exist."));

             copyRadioButton.setEnabled(false);

             moveRadioButton.setEnabled(false);

             renameToTextBox.setEnabled(false);

             renameCheckBox.setEnabled(false);

             linkInPlace.setSelected(true);

         } else {

             destDirLabel.setText(Globals.lang("File directory is '%0':", dir));

             copyRadioButton.setEnabled(true);

             moveRadioButton.setEnabled(true);

             renameToTextBox.setEnabled(true);

             renameCheckBox.setEnabled(true);

         }

         ChangeListener cl = new ChangeListener() {

                         public void stateChanged(ChangeEvent arg0) {

                                 renameCheckBox.setEnabled(!linkInPlace.isSelected()

                                                 &&  (!multipleEntries));

                                 renameToTextBox.setEnabled(!linkInPlace.isSelected()

                                                 &&  (!multipleEntries));

                                 if (multipleEntries) { renameToTextBox.setText("Multiple entries"); }

                         }

                 };

                 if (multipleEntries) {

                         linkInPlace.setText(Globals

                                         .lang("Leave files in their current directory."));

                         copyRadioButton.setText(Globals.lang("Copy files to file directory."));

                         moveRadioButton.setText(Globals.lang("Move files to file directory."));

                 } else {

                         linkInPlace.setText(Globals

                                         .lang("Leave file in its current directory."));

                         copyRadioButton.setText(Globals.lang("Copy file to file directory."));

                         moveRadioButton.setText(Globals.lang("Move file to file directory."));

                 }

         renameCheckBox.setText(Globals.lang("Rename file to") + ": ");

         renameToTextBox.setText(citeKey == null ? "default" : citeKey + "." + fileType.extension);

         linkInPlace.addChangeListener(cl);

         cl.stateChanged(new ChangeEvent(linkInPlace));

         try {

                 Object[] messages = {"How would you like to link to " + linkFileName + "?", optionsPanel}; 

             return JOptionPane.showConfirmDialog(frame, messages, dialogTitle,

                 JOptionPane.OK_CANCEL_OPTION, JOptionPane.QUESTION_MESSAGE);

         } finally {

             linkInPlace.removeChangeListener(cl);

         }

     }

     /**

      * Make a extension to the file.

      * 

      * @param entry

      *            The entry to extension from.

      * @param fileType

      *            The FileType associated with the file.

      * @param filename

      *            The path to the file.

      * @param edits

      *            An NamedCompound action this action is to be added to. If none

      *            is given, the edit is added to the panel's undoManager.

      */

     private void doLink(BibtexEntry entry, ExternalFileType fileType, String filename,

         boolean avoidDuplicate, NamedCompound edits) {

         String oldValue = entry.getField(GUIGlobals.FILE_FIELD);

         FileListTableModel tm = new FileListTableModel();

         if (oldValue != null)

             tm.setContent(oldValue);

         // If avoidDuplicate==true, we should check if this file is already linked:

         if (avoidDuplicate) {

             // For comparison, find the absolute filename:

             String fileDir = panel.metaData().getFileDirectory(GUIGlobals.FILE_FIELD);

             String absFilename = (!(new File(filename).isAbsolute()) && (fileDir != null)) ?

                     new File(fileDir, filename).getAbsolutePath() : filename;

             System.out.println("absFilename: "+absFilename);

             for (int i=0; i<tm.getRowCount(); i++) {

                 FileListEntry flEntry = tm.getEntry(i);

                 // Find the absolute filename for this existing link:

                 String absName = (!(new File(flEntry.getLink()).isAbsolute()) && (fileDir != null)) ?

                         new File(fileDir, flEntry.getLink()).getAbsolutePath() : flEntry.getLink();

                 System.out.println("absName: "+absName);

                 // If the filenames are equal, we don't need to link, so we simply return:

                 if (absFilename.equals(absName))

                     return;

             }

         }

         tm.addEntry(tm.getRowCount(), new FileListEntry("", filename, fileType));

         String newValue = tm.getStringRepresentation();

         UndoableFieldChange edit = new UndoableFieldChange(entry, GUIGlobals.FILE_FIELD,

                 oldValue, newValue);

         entry.setField(GUIGlobals.FILE_FIELD, newValue);

         if (edits == null) {

             panel.undoManager.addEdit(edit);

         } else {

             edits.addEdit(edit);

         }

     }

     /**

      * Move the given file to the base directory for its file type, and rename

      * it to the given filename.

      * 

      * @param fileName

      *            The name of the source file.

      * @param fileType

      *            The FileType associated with the file.

      * @param destFilename

      *            The destination filename.

      * @param edits

      *            TODO we should be able to undo this action

      * @return true if the operation succeeded.

      */

     private boolean doMove(String fileName, ExternalFileType fileType, String destFilename,

         NamedCompound edits) {

         String dir = panel.metaData().getFileDirectory(GUIGlobals.FILE_FIELD);

         if ((dir == null) || !(new File(dir)).exists()) {

             // OOps, we don't know which directory to put it in, or the given

             // dir doesn't exist....

             // This should not happen!!

             return false;

         }

         File fromFile = new File(fileName);

         File toFile = new File(dir + System.getProperty("file.separator") + destFilename);

         if (toFile.exists()) {

                 int answer = JOptionPane.showConfirmDialog(frame,

                                 toFile.getAbsolutePath() + " exists. Overwrite?", "Overwrite file?", 

                                 JOptionPane.YES_NO_OPTION);

                 if (answer == JOptionPane.NO_OPTION) {

                         return false;

                 }

         }

         if (!fromFile.renameTo(toFile)) {

                 JOptionPane.showMessageDialog(frame,

                                 "There was an error moving the file. Please move the file manually and link in place.",

                                 "Error moving file", JOptionPane.ERROR_MESSAGE);

                 return false;

         } else {

                 return true;

         }

     }

     /**

      * Copy the given file to the base directory for its file type, and give it

      * the given name.

      * 

      * @param fileName

      *            The name of the source file.

      * @param fileType

      *            The FileType associated with the file.

      * @param toFile

      *            The destination filename. An existing path-component will be removed.

      * @param edits

      *            TODO we should be able to undo this!

      * @return

      */

     private boolean doCopy(String fileName, ExternalFileType fileType, String toFile,

         NamedCompound edits) {

         String dir = panel.metaData().getFileDirectory(GUIGlobals.FILE_FIELD);

         if ((dir == null) || !(new File(dir)).exists()) {

             // OOps, we don't know which directory to put it in, or the given

             // dir doesn't exist....

             System.out.println("dir: " + dir + "\t ext: " + fileType.getExtension());

             return false;

         }

         toFile = new File(toFile).getName();

         File destFile = new File(new StringBuffer(dir).append(System.getProperty("file.separator"))

             .append(toFile).toString());

         if (destFile.equals(new File(fileName))){

             // File is already in the correct position. Don't override!

             return true;

         }

         if (destFile.exists()) {

             int answer = JOptionPane.showConfirmDialog(frame, "'" + destFile.getPath() + "' "

                 + Globals.lang("exists. Overwrite?"), Globals.lang("File exists"),

                 JOptionPane.YES_NO_OPTION, JOptionPane.QUESTION_MESSAGE);

             if (answer == JOptionPane.NO_OPTION)

                 return false;

         }

         try {

             Util.copyFile(new File(fileName), destFile, true);

         } catch (IOException e) {

             e.printStackTrace();

             return false;

         }

         return true;

     }

 }