Dépôt officiel du code source de l'ERP OpenConcerto

/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 2011 OpenConcerto, by ILM Informatique. All rights reserved.

 *

 * The contents of this file are subject to the terms of the GNU General Public License Version 3

 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a

 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific

 * language governing permissions and limitations under the License.

 *

 * When distributing the software, include this License Header Notice in each file.

 */

 package org.openconcerto.erp.core.edm;

import org.openconcerto.erp.core.common.ui.ScrollablePanel;

import org.openconcerto.sql.model.SQLRowAccessor;

import org.openconcerto.sql.model.SQLRowValues;

import org.openconcerto.sql.model.SQLRowValuesListFetcher;

import org.openconcerto.sql.model.SQLTable;

import org.openconcerto.sql.model.Where;

import org.openconcerto.ui.DefaultGridBagConstraints;

import org.openconcerto.ui.SwingThreadUtils;

import org.openconcerto.utils.ExceptionHandler;

import org.openconcerto.utils.FileUtils;

import java.awt.Color;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.FileDialog;

import java.awt.FlowLayout;

import java.awt.Frame;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.datatransfer.DataFlavor;

import java.awt.datatransfer.Transferable;

import java.awt.dnd.DnDConstants;

import java.awt.dnd.DropTarget;

import java.awt.dnd.DropTargetDropEvent;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JButton;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JProgressBar;

import javax.swing.JScrollPane;

import javax.swing.SwingUtilities;

import javax.swing.event.ListDataEvent;

import javax.swing.event.ListDataListener;

public class AttachmentPanel extends JPanel {

    private final SQLRowAccessor rowSource;

    private List<ListDataListener> listeners = new ArrayList<ListDataListener>();

    public AttachmentPanel(SQLRowAccessor rowSource) {

        super();

        this.rowSource = rowSource;

        this.setLayout(new GridBagLayout());

        initUI();

        setFocusable(true);

    }

    public void addListener(ListDataListener l) {

        this.listeners.add(l);

    }

    public void removeListener(ListDataListener l) {

        this.listeners.remove(l);

    }

    public void fireDataChanged() {

        for (ListDataListener listDataListener : listeners) {

            listDataListener.contentsChanged(new ListDataEvent(this, ListDataEvent.CONTENTS_CHANGED, 0, 0));

        }

    }

    public void initUI() {

        this.invalidate();

        this.removeAll();

        GridBagConstraints c = new DefaultGridBagConstraints();

        // Recupération de la liste des fichiers

        // TODO requete dans un SwingWorker

        final SQLTable tableAttachment = rowSource.getTable().getTable("ATTACHMENT");

        SQLRowValues rowVals = new SQLRowValues(tableAttachment);

        rowVals.putNulls(tableAttachment.getFieldsName());

        SQLRowValuesListFetcher fetcher = SQLRowValuesListFetcher.create(rowVals);

        Where where = new Where(tableAttachment.getField("SOURCE_TABLE"), "=", this.rowSource.getTable().getName());

        where = where.and(new Where(tableAttachment.getField("SOURCE_ID"), "=", this.rowSource.getID()));

        // TODO en premier les dossier, puis trier par nom

        List<SQLRowValues> attachments = fetcher.fetch(where);

        // AJout d'un fichier

        final JButton addButton = new JButton("Ajouter un fichier");

        this.add(addButton, c);

        c.gridx++;

        c.weightx = 1;

        c.fill = GridBagConstraints.HORIZONTAL;

        final JProgressBar progressBar = new JProgressBar(0, 100);

        progressBar.setValue(100);

        progressBar.setStringPainted(true);

        progressBar.setVisible(false);

        this.add(progressBar, c);

        c.fill = GridBagConstraints.BOTH;

        c.gridx = 0;

        c.gridy++;

        addButton.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                final Frame frame = SwingThreadUtils.getAncestorOrSelf(Frame.class, (Component) e.getSource());

                final FileDialog fd = new FileDialog(frame, "Ajouter un fichier", FileDialog.LOAD);

                fd.setVisible(true);

                final String fileName = fd.getFile();

                if (fileName != null) {

                    File inFile = new File(fd.getDirectory(), fileName);

                    AttachmentUtils utils = new AttachmentUtils();

                    utils.uploadFile(inFile, rowSource);

                    initUI();

                }

            }

        });

        ScrollablePanel files = new ScrollablePanel() {

            @Override

            public Dimension getPreferredSize() {

                int w = getSize().width;

                int nbPerRow = (w - 5) / (FilePanel.WIDTH + 5);

                if (nbPerRow < 1) {

                    nbPerRow = 1;

                }

                int nbRow = 1 + (getComponentCount() / nbPerRow);

                if (nbRow < 1) {

                    nbRow = 1;

                }

                return new Dimension(w, 5 + nbRow * (FilePanel.HEIGHT + 5));

            }

        };

        files.setOpaque(true);

        files.setBackground(Color.WHITE);

        files.setScrollableWidth(ScrollablePanel.ScrollableSizeHint.FIT);

        files.setScrollableHeight(ScrollablePanel.ScrollableSizeHint.NONE);

        files.setLayout(new FlowLayout(FlowLayout.LEFT, 5, 5));

        // Liste des fichiers

        for (final SQLRowValues sqlRowValues : attachments) {

            final FilePanel filePanel = new FilePanel(sqlRowValues, this);

            filePanel.addMouseListener(new MouseAdapter() {

                @Override

                public void mousePressed(MouseEvent e) {

                    if (e.getClickCount() == 2) {

                        Thread t = new Thread() {

                            @Override

                            public void run() {

                                AttachmentUtils utils = new AttachmentUtils();

                                File f = utils.getFile(sqlRowValues);

                                if (f == null) {

                                    SwingUtilities.invokeLater(new Runnable() {

                                        public void run() {

                                            JOptionPane.showMessageDialog(null, "Impossible de récupérer le fichier.", "Erreur", JOptionPane.ERROR_MESSAGE);

                                        }

                                    });

                                    System.err.println("Impossible de récupérer le fichier.");

                                } else {

                                    try {

                                        FileUtils.openFile(f);

                                    } catch (IOException e1) {

                                        ExceptionHandler.handle("Erreur lors de l'ouverture du fichier.", e1);

                                    }

                                }

                            }

                        };

                        t.start();

                    }

                }

            });

            files.add(filePanel);

        }

        c.gridwidth = 2;

        c.gridx = 0;

        c.gridy++;

        c.weightx = 1;

        c.weighty = 1;

        JScrollPane scroll = new JScrollPane(files);

        scroll.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_NEVER);

        scroll.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_ALWAYS);

        scroll.setMinimumSize(new Dimension((int) (400 * 1.618), 400));

        scroll.setPreferredSize(new Dimension((int) (400 * 1.618), 400));

        scroll.setBackground(Color.WHITE);

        scroll.getViewport().setBackground(Color.WHITE);

        this.add(scroll, c);

        this.validate();

        this.repaint();

        DropTarget dt = new DropTarget() {

            @Override

            public synchronized void drop(DropTargetDropEvent dtde) {

                dtde.acceptDrop(DnDConstants.ACTION_COPY_OR_MOVE);

                Transferable t = dtde.getTransferable();

                try {

                    if (t.isDataFlavorSupported(DataFlavor.javaFileListFlavor)) {

                        @SuppressWarnings("unchecked")

                        List<File> fileList = (List<File>) t.getTransferData(DataFlavor.javaFileListFlavor);

                        // TODO faire en arriere plan, mettre une jauge à droite du bouton ajouter

                        // et mettre un bouton d'annulation

                        AttachmentUtils utils = new AttachmentUtils();

                        boolean cancelledByUser = false;

                        for (File f : fileList) {

                            if (cancelledByUser) {

                                break;

                            }

                            utils.uploadFile(f, rowSource);

                        }

                        initUI();

                    }

                } catch (Exception e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

            }

        };

        files.setDropTarget(dt);

        scroll.getViewport().setDropTarget(dt);

        fireDataChanged();

    }

}