com.litongjava.tio.utils.environment.EnvUtils Maven / Gradle / Ivy

package com.litongjava.tio.utils.environment;

import java.util.HashMap;
import java.util.Map;

import com.litongjava.tio.utils.hutool.ResourceUtil;

public class EnvUtils {
  private static String[] args;
  private static Map cmdArgsMap = new HashMap<>();
  private static Map appMap = new HashMap<>();

  public static String[] getArgs() {
    return args;
  }

  public Map getCmdArgsMap() {
    return cmdArgsMap;
  }

  public static Map buildCmdArgsMap(String[] args) {
    EnvUtils.args = args;
    Map result = new HashMap<>();
    for (String arg : args) {
      if (arg.startsWith("--")) {
        String[] parts = arg.substring(2).split("=", 2);
        if (parts.length == 2) {
          result.put(parts[0], parts[1]);
        }
      }
    }
    cmdArgsMap = result;
    return result;
  }

  public static String getStr(String key) {
    String value = appMap.get(key);
    if (value != null) {
      return value;
    }
    // comamdn line
    value = cmdArgsMap.get(key);

    if (value != null) {
      return value;
    }

    // java env
    value = System.getProperty(key);
    if (value != null) {
      return value;
    }
    // system env
    value = System.getenv(key);
    if (value != null) {
      return value;
    }

    value = System.getenv(key.replace(".", "_").toUpperCase());
    if (value != null) {
      return value;
    }
    // config file
    if (PropUtils.isLoad()) {
      value = PropUtils.get(key);
    }
    return value;
  }

  public static String get(String key) {
    return getStr(key);
  }

  public static Integer getInt(String key) {
    String value = getStr(key);
    if (value != null) {
      return Integer.valueOf(value);
    } else {
      return null;
    }

  }

  public static int getInt(String key, int defaultValue) {
    String value = get(key);
    if (value != null) {
      return Integer.parseInt(value);
    } else {
      return defaultValue;
    }
  }

  public static String get(String key, String defaultValue) {
    String value = get(key);
    if (value != null) {
      return value;
    } else {
      return defaultValue;
    }
  }

  public static boolean getBoolean(String key) {
    return Boolean.parseBoolean(get(key));
  }

  public static boolean getBoolean(String key, boolean defaultValue) {
    String value = get(key);
    if (value != null) {
      return Boolean.parseBoolean(value);
    } else {
      return defaultValue;
    }
  }

  public static boolean isDev() {
    return "dev".equals(getStr("app.env"));
  }

  public static boolean isTest() {
    return "test".equals(getStr("app.env"));
  }

  public static boolean isProd() {
    return "prod".equals(getStr("app.env"));
  }

  public static void load(String fileName) {
    PropUtils.use(fileName);
  }

  public static void load(String env, String filename) {
    PropUtils.use(filename, env);
  }

  public static void set(String key, String value) {
    appMap.put(key, value);
  }

  public static void load() {
    String env = EnvUtils.get("app.env");
    String defaultFilename = "app.properties";
    if (ResourceUtil.getResource(defaultFilename) != null) {
      PropUtils.use(defaultFilename, env);
    } else {
      if (env != null) {
        String fileName = "app-" + env + ".properties";
        PropUtils.use(fileName);
      } else {
        // crate file
        PropUtils.use(defaultFilename);

      }
    }

    if (ResourceUtil.getResource(".env") != null) {
      PropUtils.append(".env");
    }

  }
}