com.sandpolis.plugin.filesys.util.FileUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sandpolis-plugin-filesys Show documentation
The filesystem plugin
There is a newer version: 6.1.0
Show newest version
/******************************************************************************
 *                                                                            *
 *                    Copyright 2019 Subterranean Security                    *
 *                                                                            *
 *  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 com.sandpolis.plugin.filesys.util;

import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.util.Objects;

import javax.imageio.ImageIO;
import javax.swing.Icon;
import javax.swing.filechooser.FileSystemView;

import com.google.protobuf.ByteString;
import com.sandpolis.plugin.filesys.net.Filesys.RS_FileInfo;

public final class FileUtil {

	/**
	 * Get a file's default icon from the system.
	 * 
	 * @param file A file
	 * @return The file's default icon
	 * @throws IOException
	 */
	public static byte[] getFileIcon(File file) throws IOException {
		Objects.requireNonNull(file);

		Icon icon = FileSystemView.getFileSystemView().getSystemIcon(file);
		BufferedImage image = new BufferedImage(icon.getIconWidth(), icon.getIconHeight(), BufferedImage.TYPE_INT_RGB);
		Graphics g = image.createGraphics();
		icon.paintIcon(null, g, 0, 0);
		g.dispose();

		try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
			ImageIO.write(image, "png", out);
			return out.toByteArray();
		}
	}

	public static RS_FileInfo getFileInfo(String path) {
		File file = new File(Objects.requireNonNull(path));

		RS_FileInfo.Builder rs = RS_FileInfo.newBuilder();
		try {
			rs.setLocalIcon(ByteString.copyFrom(getFileIcon(file)));
		} catch (IOException e) {
			rs.clearLocalIcon();
		}
		rs.setName(file.getName());
		rs.setPath(file.getParent());
		rs.setSize(file.length());
		rs.setMtime(file.lastModified());

		return rs.build();
	}

	private FileUtil() {
	}
}