FileTransferable.java

/*
 * Copyright 2008-2017 by Emeric Vernat
 *
 *     This file is part of Java Melody.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.bull.javamelody.swing;

import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;

/**
 * Implémentation de Transferable pour le drag and drop de fichiers.
 * @author Emeric Vernat
 */
class FileTransferable extends ArrayList<File> implements Transferable {
	private static final long serialVersionUID = 1L;
	private static final int FILE = 0;
	private static final int STRING = 1;
	private static final DataFlavor[] FLAVORS = { DataFlavor.javaFileListFlavor,
			DataFlavor.stringFlavor, };
	private final byte[] data;

	/**
	 * Constructeur.
	 * @param fileName String
	 * @param data byte[]
	 */
	FileTransferable(String fileName, byte[] data) {
		super();
		// instancie le fichier temporaire dans le répertoire temporaire,
		// il doit avoir le bon nom et il sera vraiment écrit au besoin dans getTransferData
		final File tmpFile = new File(System.getProperty("java.io.tmpdir"), fileName);
		// supprime un fichier de même nom éventuellement présent avant
		delete(tmpFile);
		add(tmpFile);
		this.data = data.clone();
	}

	private boolean delete(File tmpFile) {
		return tmpFile.delete();
	}

	/** {@inheritDoc} */
	@Override
	public DataFlavor[] getTransferDataFlavors() {
		return FLAVORS.clone();
	}

	/** {@inheritDoc} */
	@Override
	public boolean isDataFlavorSupported(DataFlavor flavor) {
		return flavor.equals(FLAVORS[FILE]) || flavor.equals(FLAVORS[STRING]);
	}

	/** {@inheritDoc} */
	@Override
	public Object getTransferData(DataFlavor flavor)
			throws UnsupportedFlavorException, IOException {
		if (flavor.equals(FLAVORS[FILE])) {
			final File tmpFile = get(0);
			if (!tmpFile.exists()) {
				try {
					try (OutputStream output = new BufferedOutputStream(
							new FileOutputStream(tmpFile))) {
						output.write(data);
					}
				} finally {
					// prévoit de supprimer le fichier temporaire à la fermeture de
					// l'application puisqu'on ne peut le faire dans exportDone
					tmpFile.deleteOnExit();
				}
			}
			return this;
		} else if (flavor.equals(FLAVORS[STRING])) {
			return get(0).getPath();
		} else {
			throw new UnsupportedFlavorException(flavor);
		}
	}
}