sheepit-shadow-nabber/src/com/sheepit/client/Utils.java

/*
 * Copyright (C) 2010-2014 Laurent CLOUET
 * Author Laurent CLOUET <laurent.clouet@nopnop.net>
 *
 * This program is free software; you can redistribute it and/or 
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; version 2
 * of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

package com.sheepit.client;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Calendar;
import java.util.Date;
import java.util.TimeZone;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;

import javax.xml.bind.DatatypeConverter;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

import com.sheepit.client.Error.ServerCode;
import com.sheepit.client.exception.FermeExceptionNoSpaceLeftOnDevice;

public class Utils {
	public static int unzipFileIntoDirectory(String zipFileName_, String jiniHomeParentDirName_) throws FermeExceptionNoSpaceLeftOnDevice {
		File rootdir = new File(jiniHomeParentDirName_);
		try {
			ZipInputStream zis = new ZipInputStream(new FileInputStream(zipFileName_));
			byte[] buffer = new byte[4096];
			ZipEntry ze;
			while ((ze = zis.getNextEntry()) != null) {
				FileOutputStream fos = null;
				try {
					File f = new File(rootdir.getAbsolutePath() + File.separator + ze.getName());
					if (ze.isDirectory()) {
						f.mkdirs();
						continue;
					}
					else {
						f.getParentFile().mkdirs();
						f.createNewFile();
						try {
							f.setExecutable(true);
						}
						catch (NoSuchMethodError e2) {
							// do nothing it's related to the filesystem
						}
					}
					
					fos = new FileOutputStream(f);
					int numBytes;
					while ((numBytes = zis.read(buffer, 0, buffer.length)) != -1) {
						fos.write(buffer, 0, numBytes);
					}
					fos.close();
				}
				catch (IOException e) {
					if (noFreeSpaceOnDisk(jiniHomeParentDirName_)) {
						throw new FermeExceptionNoSpaceLeftOnDevice();
					}
					
					Log logger = Log.getInstance(null); // might not print the log since the config is null
					logger.error("Utils::unzipFileIntoDirectory(" + zipFileName_ + "," + jiniHomeParentDirName_ + ") exception " + e);
					return -3;
				}
				catch (Exception e1) {
					e1.printStackTrace();
				}
				if (fos != null) {
					try {
						fos.close();
					}
					catch (IOException e) {
					}
				}
				zis.closeEntry();
			}
		}
		catch (FermeExceptionNoSpaceLeftOnDevice e) {
			throw e;
		}
		catch (IllegalArgumentException e) {
			Log logger = Log.getInstance(null); // might not print the log since the config is null
			logger.error("Utils::unzipFileIntoDirectory(" + zipFileName_ + "," + jiniHomeParentDirName_ + ") exception " + e);
			return -2;
		}
		catch (Exception e) {
			Log logger = Log.getInstance(null); // might not print the log since the config is null
			logger.error("Utils::unzipFileIntoDirectory(" + zipFileName_ + "," + jiniHomeParentDirName_ + ") exception " + e);
			return -1;
		}
		return 0;
	}
	
	public static String md5(String path_of_file_) {
		try {
			MessageDigest md = MessageDigest.getInstance("MD5");
			InputStream is = Files.newInputStream(Paths.get(path_of_file_));
			DigestInputStream dis = new DigestInputStream(is, md);
			byte[] buffer = new byte[8192];
			while (dis.read(buffer) > 0)
				; // process the entire file
			String data = DatatypeConverter.printHexBinary(md.digest()).toLowerCase();
			dis.close();
			is.close();
			return data;
		}
		catch (NoSuchAlgorithmException | IOException e) {
			return "";
		}
	}
	
	public static double lastModificationTime(File directory_) {
		double max = 0.0;
		if (directory_.isDirectory()) {
			File[] list = directory_.listFiles();
			if (list != null) {
				for (File aFile : list) {
					double max1 = lastModificationTime(aFile);
					if (max1 > max) {
						max = max1;
					}
				}
			}
		}
		else if (directory_.isFile()) {
			return directory_.lastModified();
		}
		
		return max;
	}
	
	public static ServerCode statusIsOK(Document document_, String rootname_) {
		if (document_ == null) {
			return Error.ServerCode.UNKNOWN;
		}
		NodeList ns = document_.getElementsByTagName(rootname_);
		if (ns.getLength() == 0) {
			return Error.ServerCode.ERROR_NO_ROOT;
		}
		Element a_node = (Element) ns.item(0);
		if (a_node.hasAttribute("status")) {
			return Error.ServerCode.fromInt(Integer.parseInt(a_node.getAttribute("status")));
		}
		return Error.ServerCode.UNKNOWN;
	}
	
	/**
	 * Will recursively delete a directory
	 */
	public static void delete(File file) {
		if (file == null) {
			return;
		}
		if (file.isDirectory()) {
			String files[] = file.list();
			if (files != null) {
				if (files.length != 0) {
					for (String temp : files) {
						File fileDelete = new File(file, temp);
						delete(fileDelete);
					}
				}
			}
		}
		file.delete();
	}
	
	public static long parseNumber(String in) {
		in = in.trim();
		in = in.replaceAll(",", ".");
		try {
			return Long.parseLong(in);
		}
		catch (NumberFormatException e) {
		}
		final Matcher m = Pattern.compile("([\\d.,]+)\\s*(\\w)").matcher(in);
		m.find();
		int scale = 1;
		switch (m.group(2).charAt(0)) {
			case 'G':
				scale *= 1000;
			case 'g':
				scale *= 1000;
			case 'M':
				scale *= 1000;
			case 'm':
				scale *= 1000;
			case 'K':
				break;
			default:
				return 0;
		}
		return Math.round(Double.parseDouble(m.group(1)) * scale);
	}
	
	public static String humanDuration(Date date) {
		Calendar calendar = Calendar.getInstance(TimeZone.getTimeZone("GMT"));
		calendar.setTime(date);
		
		int hours = (calendar.get(Calendar.DAY_OF_MONTH) - 1) * 24 + calendar.get(Calendar.HOUR_OF_DAY);
		int minutes = calendar.get(Calendar.MINUTE);
		int seconds = calendar.get(Calendar.SECOND);
		
		String output = "";
		if (hours > 0) {
			output += hours + "h";
		}
		if (minutes > 0) {
			output += minutes + "min";
		}
		if (seconds > 0) {
			output += seconds + "s";
		}
		return output;
	}
	
	public static boolean noFreeSpaceOnDisk(String destination_) {
		try {
			File file = new File(destination_);
			return (file.getUsableSpace() < 512 * 1024); // at least the same amount as Server.HTTPGetFile
		}
		catch (SecurityException e) {
		}
		
		return false;
	}
}