org.zodiac.boot.file.PlatformFileUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zodiac-boot Show documentation
Zodiac Boot
The newest version!
package org.zodiac.boot.file;

import java.util.Comparator;
import java.util.Hashtable;
import java.util.Map;

import org.zodiac.commons.util.Colls;
import org.zodiac.commons.util.lang.Strings;
import org.zodiac.sdk.toolkit.constants.StringPool;
import org.zodiac.sdk.toolkit.util.collection.CollUtil;
import org.zodiac.sdk.toolkit.util.lang.StrUtil;

/**
 * 文件工具类。
 *
 */
public abstract class PlatformFileUtil {

    /**
     * 定义允许上传的文件扩展名。
     */
    private static final Map EXT_MAP = CollUtil.map();
    private static final String IS_DIR = "is_dir";
    private static final String FILE_NAME = "fileName";
    private static final String FILE_SIZE = "filesize";

    /**
     * 图片扩展名。
     */
    private static final String[] FILE_TYPES = new String[] {"gif", "jpg", "jpeg", "png", "bmp"};

    static {
        EXT_MAP.put("image", ".gif,.jpg,.jpeg,.png,.bmp,.JPG,.JPEG,.PNG");
        EXT_MAP.put("flash", ".swf,.flv");
        EXT_MAP.put("media", ".swf,.flv,.mp3,.mp4,.wav,.wma,.wmv,.mid,.avi,.mpg,.asf,.rm,.rmvb");
        EXT_MAP.put("file", ".doc,.docx,.xls,.xlsx,.ppt,.htm,.html,.txt,.zip,.rar,.gz,.bz2");
        EXT_MAP.put("allfile",
            ".gif,.jpg,.jpeg,.png,.bmp,.swf,.flv,.mp3,.mp4,.wav,.wma,.wmv,.mid,.avi,.mpg,.asf,.rm,.rmvb,.doc,.docx,.xls,.xlsx,.ppt,.htm,.html,.txt,.zip,.rar,.gz,.bz2");
    }

    /**
     * 获取文件后缀。
     *
     * @param fileName 文件名
     * @return String 返回后缀
     */
    public static String getFileExt(String fileName) {
        return fileName.substring(fileName.lastIndexOf(StringPool.DOT));
    }

    /**
     * 测试文件后缀 只让指定后缀的文件上传，像jsp,war,sh等危险的后缀禁止。
     *
     * @param dir 目录
     * @param fileName 文件名
     * @return 返回成功与否
     */
    public static boolean testExt(String dir, String fileName) {
        String fileExt = getFileExt(fileName);
        String ext = EXT_MAP.get(dir);
        return StrUtil.isNotBlank(ext)
            && (ext.contains(fileExt.toLowerCase()) || ext.contains(fileExt.toUpperCase()));
    }

    /**
     * 文件管理排序。
     */
    public enum FileSort {

        /**
         * 大小
         */
        size,

        /**
         * 类型
         */
        type,

        /**
         * 名称
         */
        name;

        /**
         * 文本排序转换成枚举。
         *
         * @param sort 排序名称
         * @return {@code FileSort}
         */
        public static FileSort of(String sort) {
            try {
                return FileSort.valueOf(sort);
            } catch (Exception e) {
                return FileSort.name;
            }
        }
    }

    public static class NameComparator implements Comparator {
        @Override
        public int compare(Object a, Object b) {
            Hashtable hashA = (Hashtable)a;
            Hashtable hashB = (Hashtable)b;
            if (((Boolean)hashA.get(IS_DIR)) && !((Boolean)hashB.get(IS_DIR))) {
                return -1;
            } else if (!((Boolean)hashA.get(IS_DIR)) && ((Boolean)hashB.get(IS_DIR))) {
                return 1;
            } else {
                return ((String)hashA.get(FILE_NAME)).compareTo((String)hashB.get(FILE_NAME));
            }
        }
    }

    public static class SizeComparator implements Comparator {
        @Override
        public int compare(Object a, Object b) {
            Hashtable hashA = (Hashtable)a;
            Hashtable hashB = (Hashtable)b;
            if (((Boolean)hashA.get(IS_DIR)) && !((Boolean)hashB.get(IS_DIR))) {
                return -1;
            } else if (!((Boolean)hashA.get(IS_DIR)) && ((Boolean)hashB.get(IS_DIR))) {
                return 1;
            } else {
                if (((Long)hashA.get(FILE_SIZE)) > ((Long)hashB.get(FILE_SIZE))) {
                    return 1;
                } else if (((Long)hashA.get(FILE_SIZE)) < ((Long)hashB.get(FILE_SIZE))) {
                    return -1;
                } else {
                    return 0;
                }
            }
        }
    }

    public static class TypeComparator implements Comparator {
        @Override
        public int compare(Object a, Object b) {
            Hashtable hashA = (Hashtable)a;
            Hashtable hashB = (Hashtable)b;
            if (((Boolean)hashA.get(IS_DIR)) && !((Boolean)hashB.get(IS_DIR))) {
                return -1;
            } else if (!((Boolean)hashA.get(IS_DIR)) && ((Boolean)hashB.get(IS_DIR))) {
                return 1;
            } else {
                return ((String)hashA.get("filetype")).compareTo((String)hashB.get("filetype"));
            }
        }
    }

    public static String formatUrl(String url) {
        return url.replaceAll("\\\\", "/");
    }

}