/*
 * Copyright (C) 2016 Didier Clermonté <dclermonte@april.org>
 * Copyright (C) 2016 Christian Pierre Momon <christian.momon@devinsy.fr>
 *
 * This file is part of Siba.
 *
 * Siba is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.dclermonte.siba.model;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.net.URL;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

import org.apache.commons.compress.archivers.tar.TarArchiveEntry;
import org.apache.commons.compress.archivers.tar.TarArchiveOutputStream;
import org.apache.commons.io.IOUtils;

/**
 *
 * @author Didier Clermonté (dclermonte@april.org)
 *
 */
public class SibaUtils
{
	/**
	 *
	 * This method Generate the file with MD5.
	 *
	 * @param inputFile
	 * @param target
	 * @return
	 * @throws IOException
	 * @throws NoSuchAlgorithmException
	 */
	public static File createMD5File(final File inputFile, final String target)
			throws IOException, NoSuchAlgorithmException
	{
		File result;

		String md5 = SibaUtils.md5(inputFile);
		String fileName = inputFile.getName();
		result = createMD5File(md5, fileName, target);

		//
		return result;
	}

	/**
	 *
	 * @param md5
	 * @param fileName
	 * @param target
	 * @return
	 * @throws IOException
	 */
	public static File createMD5File(final String md5, final String fileName, final String target) throws IOException
	{
		File result;
		FileWriter out = null;
		try
		{
			result = new File(target);
			out = new FileWriter(result);
			out.write(md5);
			String newLine = System.getProperty("line.separator");
			out.append(" ");
			out.append(fileName);
			out.append(newLine);
			out.close();
		}
		finally
		{
			IOUtils.closeQuietly(out);
		}

		//
		return result;
	}

	/**
	 *
	 * @param choosenFile
	 * @return
	 * @throws IOException
	 */
	public static String loadMD5Sum(final File choosenFile) throws IOException
	{
		String result;
		FileReader in = null;
		try
		{
			if (choosenFile.exists() && (choosenFile.length() > 32))
			{
				in = new FileReader(choosenFile);
				BufferedReader bufferedReader = new BufferedReader(in);
				String line = bufferedReader.readLine();
				result = line.substring(0, 32);
				in.close();
			}
			else
			{
				result = "";
			}
		}
		finally
		{
			IOUtils.closeQuietly(in);
		}
		//
		return result;
	}

	/**
	 * This method calculate the MD5 itself.
	 *
	 * @param input
	 * @return
	 * @throws IOException
	 * @throws NoSuchAlgorithmException
	 */
	public static String md5(final File input) throws IOException, NoSuchAlgorithmException
	{
		String result;

		StringBuilder hashString = new StringBuilder();
		FileInputStream fileInputStream = null;
		try
		{
			byte[] byteInput = new byte[(int) input.length()];
			fileInputStream = new FileInputStream(input);
			for (int index1 = 0; index1 < input.length(); index1++)
			{
				byteInput[index1] = (byte) fileInputStream.read();
			}
			byte[] hash = null;
			MessageDigest messageDigest = MessageDigest.getInstance("MD5");
			hash = messageDigest.digest(byteInput);

			for (int index = 0; index < hash.length; index++)
			{
				String hex = Integer.toHexString(hash[index]);
				if (hex.length() == 1)
				{
					hashString.append('0');
					hashString.append(hex.charAt(hex.length() - 1));
				}
				else
				{
					hashString.append(hex.substring(hex.length() - 2));
				}
			}
			fileInputStream.close();

			result = hashString.toString();
		}
		finally
		{
			IOUtils.closeQuietly(fileInputStream);
		}

		//
		return result;
	}

	/**
	 *
	 * @param choosenFile
	 * @return
	 * @throws IOException
	 */
	public static File readFileNameToCheck(final File choosenFile) throws IOException
	{
		File result;
		FileReader in = null;
		try
		{
			if (choosenFile.exists() && (choosenFile.length() > 32))
			{
				in = new FileReader(choosenFile);
				BufferedReader bufferedReader = new BufferedReader(in);
				String line = bufferedReader.readLine();
				String fileNameToString;
				fileNameToString = line.substring(33);
				in.close();
				if (choosenFile.isAbsolute())
				{
					result = new File(choosenFile.getParent() + "/" + fileNameToString);
				}
				else
				{
					String path = System.getProperty("user.dir");
					result = new File(path + "/" + fileNameToString);
				}
			}
			else
			{
				result = null;
			}
		}
		finally
		{
			IOUtils.closeQuietly(in);
		}
		//
		return result;
	}

	/**
	 *
	 * @return
	 * @throws IOException
	 */
	public static String readResource(final String resource, final URL url) throws IOException
	{
		String result;

		result = IOUtils.toString(url, (String) null);

		return result;
	}

	/**
	 *
	 * @param fileToSave
	 * @param outputStream
	 * @throws IOException
	 */
	public static void tarDirectoryTree(final File fileToSave, final TarArchiveOutputStream outputStream)
			throws IOException
	{
		int pathLength = fileToSave.getParentFile().getAbsolutePath().length();
		tarDirectoryTree(fileToSave, outputStream, pathLength);
	}

	/**
	 *
	 * This method generates ArchiveEntry.
	 *
	 * @param directory
	 * @param outputStream
	 * @param pathLength
	 * @throws IOException
	 */
	public static void tarDirectoryTree(final File directory, final TarArchiveOutputStream outputStream,
			final int pathLength) throws IOException
	{
		for (File file : directory.listFiles())
		{
			if (file.isDirectory())
			{
				if (file.listFiles().length == 0)
				{
					TarArchiveEntry tarArchiveEntry = new TarArchiveEntry(file);
					String pathPartiel = file.getPath().substring(pathLength);
					tarArchiveEntry.setName(pathPartiel);
					outputStream.putArchiveEntry(tarArchiveEntry);
					outputStream.closeArchiveEntry();
				}
				else
				{
					tarDirectoryTree(file, outputStream, pathLength);
				}
			}
			else
			{
				TarArchiveEntry tarArchiveEntry = new TarArchiveEntry(file);
				String pathPartiel = file.getPath().substring(pathLength);
				tarArchiveEntry.setName(pathPartiel);
				outputStream.putArchiveEntry(tarArchiveEntry);
				IOUtils.copy(new FileInputStream(file), outputStream);
				outputStream.closeArchiveEntry();

			}
		}
	}
}