package com.wazn.learn.util;

import java.io.File;
 /**
  * 刪除檔案工具類
  * @author yangzhenyu
  *
  */
public class DeleteFileUtil {

    /**
     * 刪除檔案，可以是檔案或資料夾
     *
     * @param fileName
     *            要刪除的檔名
     * @return 刪除成功返回true，否則返回false
     */
    public static boolean delete(String fileName) {
        File file = new File(fileName);
        if (!file.exists()) {
            System.out.println("刪除檔案失敗:" + fileName + "不存在！");
            return false;
        } else {
            if (file.isFile())
                return deleteFile(fileName);
            else
                return deleteDirectory(fileName);
        }
    }

    /**
     * 刪除單個檔案
     *
     * @param fileName
     *            要刪除的檔案的檔名
     * @return 單個檔案刪除成功返回true，否則返回false
     */
    public static boolean deleteFile(String fileName) {
        File file = new File(fileName);
        // 如果檔案路徑所對應的檔案存在，並且是一個檔案，則直接刪除
        if (file.exists() && file.isFile()) {
            if (file.delete()) {
                System.out.println("刪除單個檔案" + fileName + "成功！");
                return true;
            } else {
                System.out.println("刪除單個檔案" + fileName + "失敗！");
                return false;
            }
        } else {
            System.out.println("刪除單個檔案失敗：" + fileName + "不存在！");
            return false;
        }
    }

    /**
     * 刪除目錄及目錄下的檔案
     *
     * @param dir
     *            要刪除的目錄的檔案路徑
     * @return 目錄刪除成功返回true，否則返回false
     */
    public static boolean deleteDirectory(String dir) {
        // 如果dir不以檔案分隔符結尾，自動新增檔案分隔符
        if (!dir.endsWith(File.separator))
            dir = dir + File.separator;
        File dirFile = new File(dir);
        // 如果dir對應的檔案不存在，或者不是一個目錄，則退出
        if ((!dirFile.exists()) || (!dirFile.isDirectory())) {
            System.out.println("刪除目錄失敗：" + dir + "不存在！");
            return false;
        }
        boolean flag = true;
        // 刪除資料夾中的所有檔案包括子目錄
        File[] files = dirFile.listFiles();
        for (int i = 0; i < files.length; i++) {
            // 刪除子檔案
            if (files[i].isFile()) {
                flag = DeleteFileUtil.deleteFile(files[i].getAbsolutePath());
                if (!flag)
                    break;
            }
            // 刪除子目錄
            else if (files[i].isDirectory()) {
                flag = DeleteFileUtil.deleteDirectory(files[i]
                        .getAbsolutePath());
                if (!flag)
                    break;
            }
        }
        if (!flag) {
            System.out.println("刪除目錄失敗！");
            return false;
        }
        // 刪除當前目錄
        if (dirFile.delete()) {
            System.out.println("刪除目錄" + dir + "成功！");
            return true;
        } else {
            return false;
        }
    }

    
}