package org.jelly.util;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import org.jelly.code.IndexCode;
import org.jelly.exception.ExecutetimeException;
/**
 * 文件常用的操作的工具类
 * @since 1.0.0
 */
public class FileUtil {

	private FileUtil(){}
	
	private static final String LOCK = "_LOCK";
	private static final int BUFFER_SIZE = 1024 * 1024 / 2;
	
	/**
	 * <des> 文件拷贝, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
	 * @param in InputStream
	 * @param out OutputStream
	 * @since 1.0.0
	 */
	public static void copyFile(InputStream in, OutputStream out){
		try {
			byte[] buffer = new byte[BUFFER_SIZE];
			int read;
			while((read = in.read(buffer)) != IndexCode.EOF.toCode()){
				out.write(buffer, 0, read);
			}
		} catch (Throwable e) {
			throw new ExecutetimeException(e);
		} finally {
			closeStream(in, out);
		}
	}
	
	/**
	 * <des> 写出, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
	 * @since 1.6.0
	 */
	public static void write(InputStream in, OutputStream out){
		copyFile(in, out);
	}

	/**
	 * <des> 字符串内容写出, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
	 * @since 1.6.0
	 */
	public static void write(String text, OutputStream out){
		try {
			out.write(text.getBytes());
		} catch (Throwable e) {
			throw new ExecutetimeException(e);
		} finally {
			closeStream(out);
		}
	}

	/**
	 * <des> 字符串内容写出到文件, 方法调用结束或抛出异常, 都将关闭输入输出流, 输出流flush </des>
	 * @since 1.6.0
	 */
	public static void write(String text, File file){
		BufferedReader reader = null;
		BufferedWriter writer = null;
		try {
			reader = new BufferedReader(new StringReader(text));
			writer = new BufferedWriter(new FileWriter(file));
			char[] buffer = new char[BUFFER_SIZE];
			int read;
			while((read = reader.read(buffer)) != IndexCode.EOF.toCode()){
				writer.write(buffer, 0, read);
			}
		} catch (Throwable e) {
			throw new ExecutetimeException(e);
		} finally {
			closeStream(reader, writer);
		}
	}
	
	/**
	 * <des> 读取获得文件内容 </des>
	 * @param pathname 文件全路径名称
	 * @return 读取获得的文件内容
	 * @since 1.0.0
	 */
	public static String readFile(String pathname){
		return readFile(new File(pathname));
	}
	
	/**
	 * <des> 读取获得文件内容 </des>
	 * @param file 文件对象
	 * @return 读取获得的文件内容
	 * @since 1.0.0
	 */
	public static String readFile(File file){
		String[] contents = readLineFile(file);
		StringBuilder builder = new StringBuilder();
		for(String content : contents){
			builder.append(content);
		}
		return builder.toString();
	}
	
	/**
	 * <des> 按行读取文件, 文件的每一行作为数组的一个元素 </des>
	 * @param pathname 文件全路径名称
	 * @return 数组
	 * @since 1.0.0
	 */
	public static String[] readLineFile(String pathname){
		return readLineFile(new File(pathname));
	}
	
	/**
	 * <des> 按行读取文件, 文件的每一行作为数组的一个元素 </des>
	 * @param file 文件对象
	 * @return 数组
	 * @since 1.0.0
	 */
	public static String[] readLineFile(File file){
		BufferedReader reader = null;
		try {
			reader = new BufferedReader(new FileReader(file));
			LinkedList<String> list = new LinkedList<String>();
			String read;
			while((read = reader.readLine()) != null){
				list.add(read);
			}
			return ArrayUtil.asArray(list);
		} catch (Throwable e) {
			throw new ExecutetimeException(e);
		} finally {
			closeStream(reader);
		}
	}
	
	/**
	 * <des> 获取目录下所有的文件对象 </des>
	 * @param directory 文件目录对象
	 * @return 文件对象列表
	 * @since 1.0.0
	 */
	public static List<File> listFiles(File directory){
		return listFiles(directory, null);
	}
	
	/**
	 * <des> 获取目录下所有由参数指定类型的文件对象 </des>
	 * @param source 文件目录对象
	 * @param filter 期望得到的文件的后缀名称
	 * @return 文件对象列表
	 * @since 1.0.0
	 */
	public static List<File> listFiles(File source, String filter){
		File[] fileList = source.listFiles();
		List<File> list = new ArrayList<File>();
		filter = filter == null ? null : filter.toLowerCase();
		if(fileList != null && fileList.length > 0){
			for(File file : fileList){
				if(file.isFile()){
					add(list, file, filter);
				}else if(file.isDirectory()){
					list.addAll(listFiles(file, filter));
				}
			}
		}else if(source.isFile()){
			add(list, source, filter);
		}
		return list;
	}
	
	/**
	 * <des> 在参数指定的目录中查找文件 </des>
	 * @param dirpath 文件目录全路径名称
	 * @param filename 查找的文件名称
	 * @return 若查找的到, 则返还该文件对象, 若查找不到, 则返还null
	 * @since 1.0.0
	 */
	public static File findFile(String dirpath, String filename){
		return findFile(new File(dirpath), filename);
	}
	
	/**
	 * <des> 在参数指定的目录中查找文件 </des>
	 * @param directory 文件目录对象
	 * @param filename 查找的文件名称
	 * @return 若查找的到, 则返还该文件对象, 若查找不到, 则返还null
	 * @since 1.0.0
	 */
	public static File findFile(File directory, String filename){
		String filter = StringUtil.substringAfterLastWith(filename, ".");
		List<File> files = listFiles(directory, filter);
		if(files.size() == 0) return null;
		for(File file : files){
			if(file.getAbsolutePath().endsWith(filename)){
				return file;
			}
		}
		return null;
	}
	
	/**
	 * <des> 删除文件或目录 </des>
	 * @param file 文件或文件目录对象
	 * @since 1.0.0
	 */
	public static void delete(File file){
		synchronized (LOCK) {
			if(file.isFile()){
				file.delete();
			}else if(file.isDirectory()){
				File[] files = file.listFiles();
				for(File item : files){
					delete(item);
				}
			}
			file.delete();
		}
	}
	
	/**
	 * <des> 创建目录 </des>
	 * @param pathname 目录名称
	 * @return 若目录不存在则创建, 若存在则直接返还true
	 * @since 1.0.0
	 */
	public static boolean createDir(String pathname){
		return createDir(new File(pathname));
	}
	
	/**
	 * <des> 创建目录 </des>
	 * @param dir 目录对象
	 * @return 若目录不存在则创建, 若存在则直接返还true
	 * @since 1.0.0
	 */
	public static boolean createDir(File dir){
		synchronized (LOCK) {
			if(!dir.exists()){
				return dir.mkdirs();
			}
			return true;
		}
	}
	
	
	/**
	 * <des> 获取FileInputStream实例 </des>
	 * @since 1.0.0
	 */
	public static FileInputStream getFileInputStream(String pathname){
		return getFileInputStream(new File(pathname));
	}
	
	/**
	 * <p><des> 获取FileInputStream实例 </des></p>
	 * @since 1.0.0
	 */
	public static FileInputStream getFileInputStream(File file){
		try {
			return new FileInputStream(file);
		} catch (FileNotFoundException e) {
			throw new ExecutetimeException(e);
		}
	}
	
	/**
	 * <p><des> 获取FileOutputStream实例 </des></p>
	 * @since 1.0.0
	 */
	public static FileOutputStream getFileOutputStream(String pathname){
		return getFileOutputStream(new File(pathname), false);
	}
	
	/**
	 * <p><des> 获取FileOutputStream实例 </des></p>
	 * @since 1.0.0
	 */
	public static FileOutputStream getFileOutputStream(String pathname, boolean append){
		return getFileOutputStream(new File(pathname), append);
	}
	
	/**
	 * <p><des> 获取FileOutputStream实例 </des></p>
	 * @since 1.0.0
	 */
	public static FileOutputStream getFileOutputStream(File file){
		return getFileOutputStream(file, false);
	}
	
	/**
	 * <p><des> 获取FileOutputStream实例 </des></p>
	 * @since 1.0.0
	 */
	public static FileOutputStream getFileOutputStream(File file, boolean append){
		try {
			return new FileOutputStream(file, append);
		} catch (FileNotFoundException e) {
			throw new ExecutetimeException(e);
		}
	}
	
	// 添加文件到列表
	private static void add(List<File> list, File file, String filter){
		if(filter == null){
			list.add(file);
		}else if(file.getAbsolutePath().toLowerCase().endsWith(filter)){
			list.add(file);
		}
	}
	
}

java File文件操作共用方法整理