package com.vci.starter.web.util;

import com.vci.starter.web.exception.VciBaseException;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.boot.system.ApplicationHome;
import org.springframework.core.io.ClassPathResource;
import org.springframework.util.ResourceUtils;

import java.io.*;
import java.nio.channels.FileChannel;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.UUID;

/**
 * 本地文件操作类
 * @author weidy
 */
public class LocalFileUtil {

	/**
	 * 主类
	 */
	public static Class mainClass = null;
	/**
	 * 删除文件夹
	 * @param file 要删除的文件或者文件夹
	 */
	public static void deleteFile(File file) {
		if(file == null || !file.exists()){
			return;
		}
		if (file.isDirectory()) {
			File[] files = file.listFiles();
			for (int i = 0; i < files.length; i++) {
				deleteFile(files[i]);
			}
		}
		file.delete();
	}

	/**
	 * 拷贝文件
	 * @param source 源文件
	 * @param target 目标文件
	 * @throws VciBaseException 拷贝文件出错的时候会抛出异常
	 */
	public static void copyFile(File source, File target) throws VciBaseException {
		FileInputStream is = null;
		FileOutputStream os = null;
		try{
			is = new FileInputStream(source);
			os = new FileOutputStream(target);
			FileChannel sourceChannel = is.getChannel();
			FileChannel targetChannel = os.getChannel();
			int i=0;
			int length=2097152;
			while(true){
				if(sourceChannel.position()==sourceChannel.size()){
					sourceChannel.close();
					targetChannel.close();
					break;
				}
				if((sourceChannel.size()-sourceChannel.position())<20971520) {
					length = (int) (sourceChannel.size() - sourceChannel.position());
				}else {
					length = 20971520;
				}
				sourceChannel.transferTo(sourceChannel.position(),length,targetChannel);
				sourceChannel.position(sourceChannel.position()+length);
				i++;
			}
		}catch(Throwable e){
			throw new VciBaseException("拷贝文件出错",new String[0],e);
		}finally {
			IOUtils.closeQuietly(is);
			IOUtils.closeQuietly(os);
		}
	}

	/**
	 * 获取默认的临时文件夹
	 * @return 临时文件夹，默认为当前项目的运行路径下的tempFolder下的随机值
	 */
	public static String getDefaultTempFolder() throws VciBaseException{
		String path = getProjectFolder();
		path = path + File.separator + "tempFolder" + File.separator + UUID.randomUUID().toString();
		File folder = new File(path);
		if(!folder.exists()){
			folder.mkdirs();
		}
		return path;
	}

	/**
	 * 获取项目的路径
	 * @return 项目的路径
	 * @throws VciBaseException 获取路径出现了错误会抛出异常
	 */
	public static String getProjectFolder() throws VciBaseException{
		return getProjectFolder(LocalFileUtil.class);
	}

	/**
	 * 获取某个class所在项目的路径
	 * @param classObj 某个class
	 * @return 项目的路径
	 * @throws VciBaseException 获取路径出现了错误会抛出异常
	 */
	public static String getProjectFolder(Class classObj) throws VciBaseException{
		String path = "";
		try {
			ApplicationHome h = new ApplicationHome(classObj == null?(mainClass ==null?LocalFileUtil.class:mainClass):classObj);
			File jarF = h.getSource();
			if(jarF == null){
				//针对测试的时候
				jarF = h.getDir();
			}
			path =  jarF.getParentFile().toString();
			if(path.contains("!")){
				path = new File(path).getParentFile().getParent();
			}
		}catch (Throwable e){
			e.printStackTrace();
			throw new VciBaseException("获取当前服务所在的文件夹出现了错误",new String[0],e);
		}
		if(path.startsWith("file:\\")){
			path = path.substring(6);
		}
		if (System.getProperty("os.name").toLowerCase().startsWith("win") && path.startsWith("/")) {
			path = path.substring(path.indexOf("/") + 1);
		}
		System.out.println("项目的路径是:" + path);
		return path;
	}

	/**
	 * 删除临时文件
	 * @param tempFile 临时文件
	 * @param deleteRandomFolder 是否删除随机的那个文件夹
	 * @throws VciBaseException 删除出错的时候抛出异常
	 */
	public static void deleteTempFile(File tempFile,boolean deleteRandomFolder) throws VciBaseException{
		if(tempFile!=null){
			tempFile.delete();
			if(deleteRandomFolder){
				tempFile.getParentFile().delete();
			}
		}
	}


	/**
	 * 判断参数是否为空
	 * @param s 参数，第一个为参数，第二个为提示信息
	 * @throws VciBaseException 为空时抛出这个异常
	 */
	public static void alertNotNull(Object... s) throws VciBaseException{
		if(s!=null && s.length>0){
			for(int i = 0 ; i < s.length ; i ++){
				Object obj = s[i];
				if(obj==null ||StringUtils.isBlank(obj.toString())){
					String param = "";
					try{
						i++;
						param = s[i].toString();
					}catch(Exception e){

					}
					throw new VciBaseException("参数[" + param + "]不能为空");
				}
			}
		}
	}

	/**
	 * 获取文件的后缀名
	 * @param file 文件对象
	 * @return 后缀名，不包含.；如果文件不存在或者没有后缀名返回""
	 */
	public static String getFileExtension(File file){
		if(file!=null){
			return getFileExtensionByName(file.getName());
		}
		return "";
	}

	/**
	 * 获取文件的后缀名
	 * @param name 文件名称
	 * @return 后缀名，不包含.；没有后缀名返回""
	 */
	public static String getFileExtensionByName(String name){
		if(StringUtils.isNotBlank(name)){
			if(name.contains(".")){
				return name.substring(name.lastIndexOf(".")+1);
			}
		}
		return "";
	}

	/**
	 * 获取上传文件的名称，在IE浏览器上，上传会把全路径带过来
	 * @param name 文件的名称
	 * @return 替换后的值
	 */
	public static String getFileNameForIE(String name){
		if(StringUtils.isBlank(name)){
			return name;
		}
		if(name.contains(File.separator)){
			return name.substring(name.lastIndexOf(File.separator)+1);
		}
		if(name.contains("\\")){
			return name.substring(name.lastIndexOf("\\") + 1);
		}
		return name;
	}

	/**
	 * 从工程中读取文件的内容
	 * @param fileName 文件的名称
	 * @return 文件的内容
	 * @throws VciBaseException 读取出错的时候会抛出异常
	 */
	public static String readContentForFileInJar(String fileName) throws VciBaseException{
		if(StringUtils.isBlank(fileName)){
			throw new VciBaseException("文件名为空");
		}
		if(!fileName.startsWith("/")){
			fileName = "/" + fileName;
		}
		InputStream in = null;
		ByteArrayOutputStream os = null;
		try{
			in = LocalFileUtil.class.getResourceAsStream( fileName);
			if(in ==null){
				in=Thread.currentThread().getContextClassLoader().getResourceAsStream(fileName);
			}
			os = new ByteArrayOutputStream();
			IOUtils.copy(in,os);
			return new String(os.toByteArray(),"UTF-8");
		}catch (Throwable e){
			throw new VciBaseException("读取文件的内容出现了错误",new String[0],e);
		}finally {
			IOUtils.closeQuietly(in);
			IOUtils.closeQuietly(os);
		}
	}

	/**
	 * 读取文件的内容，文件不能是jar里的
	 * @param fileName 文件的名称
	 * @return 文件的内容，是UTF-8的字符集
	 * @throws VciBaseException 文件不存在，和读取出错的时候会抛出异常
	 */
	public static String readContentForFile(String fileName) throws VciBaseException{
		File file = new File(fileName);
		if(!file.exists()){
			throw new VciBaseException("文件未找到，{0}",new String[]{fileName});
		}
		InputStream in = null;
		ByteArrayOutputStream os = null;
		try{
			in = new FileInputStream( file);
			os = new ByteArrayOutputStream();
			IOUtils.copy(in,os);
			return new String(os.toByteArray(),"UTF-8");
		}catch (Throwable e){
			throw new VciBaseException("读取文件的内容出现了错误",new String[0],e);
		}finally {
			IOUtils.closeQuietly(in);
			IOUtils.closeQuietly(os);
		}
	}

	/**
	 * 写入字符串到文件
	 * @param content 字符串
	 * @param fileName 文件的名称
	 * @throws VciBaseException 写入出错的时候会抛出异常
	 */
	public static void writeContentToFile(String content,String fileName) throws VciBaseException{
		File file = new File(fileName);
		if(!file.exists()){
			File parentFile = file.getParentFile();
			if(!parentFile.exists()){
				parentFile.mkdirs();
			}
			try {
				file.createNewFile();
			} catch (IOException e) {
				throw new VciBaseException("写入文件的内容时候出现了创建文件失败的问题,{}",new String[]{fileName},e);
			}
		}
		if(content == null){
			content = "";
		}
		OutputStream os = null;
		try{
			os = new FileOutputStream(file);
			os.write(content.getBytes(StandardCharsets.UTF_8));
			os.flush();
		}catch(Throwable e){
			throw new VciBaseException("写入文件的内容时候出现了创错误,{}",new String[]{fileName},e);
		}finally {
			IOUtils.closeQuietly(os);
		}
	}

	/**
	 * 从项目里或者jar的文件到目标文件
	 * @param fileInJar jar或者项目里的文件名称
	 * @param target 目标文件
	 * @throws VciBaseException 拷贝文件出现错误的时候会抛出异常
	 */
	public static void copyFileInJar(String fileInJar,String target) throws  VciBaseException{
		VciBaseUtil.alertNotNull(fileInJar,"源文件",target,"目标文件");
		File targetFile = new File(target);
		File folder = targetFile.getParentFile();
		if(!folder.exists()){
			folder.mkdirs();
		}
		if(!targetFile.exists()){
			try {
				targetFile.createNewFile();
			} catch (IOException e) {
				throw new VciBaseException("创建目标文件的出现了错误",new String[0],e);
			}
		}
		if(!fileInJar.startsWith("/")){
			fileInJar = "/" + fileInJar;
		}
		InputStream in = null;
		FileOutputStream os = null;
		try{
			in =  LocalFileUtil.class.getResourceAsStream( fileInJar);
			if(in ==null){
				in=Thread.currentThread().getContextClassLoader().getResourceAsStream(fileInJar);
			}
			os = new FileOutputStream(target);
			IOUtils.copy(in,os);
		}catch (Throwable e){
			throw new VciBaseException("拷贝文件的内容出现了错误.{0},{1}",new String[]{fileInJar,target},e);
		}finally {
			IOUtils.closeQuietly(in);
			IOUtils.closeQuietly(os);
		}
	}

	/**
	 * 文件转换为路径
	 * @param fileList 文件列表
	 * @return 路径列表
	 * @throws VciBaseException 参数为空和文件不存在会抛出异常
	 */
	public static List<String> fileToPatch(Collection<File> fileList) throws VciBaseException  {
		VciBaseUtil.alertNotNull("文件列表",fileList);
		List<String> filePath = new ArrayList<>();
		for(File file : fileList){
			if(!file.exists()){
				throw new VciBaseException(file.getAbsolutePath() + "文件没有找到",new String[0]);
			}
			filePath.add(file.getAbsolutePath());
		}
		return filePath;
	}

	/**
	 * 获取图片专用的临时路径
	 * @return 路径
	 * @throws VciBaseException 获取出错会报错
	 */
	public static String getDefaultPicTempFolder() throws VciBaseException{
		String path = LocalFileUtil.getProjectFolder();
		path = path + File.separator + "tempFolder" + File.separator + "picFolder" + File.separator ;
		File folder = new File(path);
		if(!folder.exists()){
			folder.mkdirs();
		}
		return path;
	}
}