Source/BladeX-Tool/blade-core-tool/src/main/java/org/springblade/core/tool/utils/PathUtil.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,64 @@
package org.springblade.core.tool.utils;

import org.springframework.lang.Nullable;

import java.io.File;
import java.net.URL;

/**
 * ç”¨æ¥èŽ·å–各种目录
 *
 * @author L.cm
 */
public class PathUtil {
   public static final String FILE_PROTOCOL = "file";
   public static final String JAR_PROTOCOL = "jar";
   public static final String ZIP_PROTOCOL = "zip";
   public static final String FILE_PROTOCOL_PREFIX = "file:";
   public static final String JAR_FILE_SEPARATOR = "!/";

   /**
    * èŽ·å–jar包运行时的当前目录
    *
    * @return {String}
    */
   @Nullable
   public static String getJarPath() {
      try {
         URL url = PathUtil.class.getResource(StringPool.SLASH).toURI().toURL();
         return PathUtil.toFilePath(url);
      } catch (Exception e) {
         String path = PathUtil.class.getResource(StringPool.EMPTY).getPath();
         return new File(path).getParentFile().getParentFile().getAbsolutePath();
      }
   }

   /**
    * è½¬æ¢ä¸ºæ–‡ä»¶è·¯å¾„
    *
    * @param url è·¯å¾„
    * @return {String}
    */
   @Nullable
   public static String toFilePath(@Nullable URL url) {
      if (url == null) {
         return null;
      }
      String protocol = url.getProtocol();
      String file = UrlUtil.decode(url.getPath(), Charsets.UTF_8);
      if (FILE_PROTOCOL.equals(protocol)) {
         return new File(file).getParentFile().getParentFile().getAbsolutePath();
      } else if (JAR_PROTOCOL.equals(protocol) || ZIP_PROTOCOL.equals(protocol)) {
         int ipos = file.indexOf(JAR_FILE_SEPARATOR);
         if (ipos > 0) {
            file = file.substring(0, ipos);
         }
         if (file.startsWith(FILE_PROTOCOL_PREFIX)) {
            file = file.substring(FILE_PROTOCOL_PREFIX.length());
         }
         return new File(file).getParentFile().getAbsolutePath();
      }
      return file;
   }

}