org.shaneking.roc.zero.net.InetAddress0Config Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.shaneking.roc.zero Show documentation
Zero with Spring for ShaneKing
The newest version!
package org.shaneking.roc.zero.net;

import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import org.shaneking.ling.zero.lang.String0;
import org.shaneking.ling.zero.lang.ZeroException;
import org.shaneking.ling.zero.net.InetAddress0;
import org.shaneking.ling.zero.util.Map0;
import org.shaneking.ling.zero.util.Regex0;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.nio.file.Files;
import java.util.Map;
import java.util.PropertyResourceBundle;
import java.util.ResourceBundle;

@Component
@ConfigurationProperties(prefix = InetAddress0Config.PREFIX, ignoreInvalidFields = true)
@Slf4j
public class InetAddress0Config {
  public static final String PREFIX = "sk.roc.zero.net.address";
  @Getter
  @Setter
  private boolean enabled = true;
  @Getter
  @Setter
  private boolean quietly = true;
  @Getter
  @Setter
  private String extHostsPath;
  @Getter
  @Setter
  private String extPropsPath;
  @Getter
  @Setter
  private Map vhosts = Map0.newHashMap();

  @PostConstruct
  public void postConstruct() {
    if (this.isEnabled()) {
      if (!String0.isNullOrEmpty(this.getExtHostsPath())) {
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        try {
          Resource[] resources = resolver.getResources(this.getExtHostsPath());
          for (Resource resource : resources) {
            Files.lines(resource.getFile().toPath()).forEachOrdered(line -> {
              if (!String0.isNullOrEmpty(line) && !line.startsWith(String0.POUND)) {
                String[] parts = line.split(Regex0.BLACKS);
                if (parts.length > 1) {
                  for (int i = 1; i < parts.length; i++) {
                    InetAddress0.putCustomHost(parts[i], parts[0]);
                  }
                }
              }
            });
          }
        } catch (Exception e) {
          log.error(this.getExtPropsPath(), e);
          if (!quietly) {
            throw new ZeroException(this.getExtHostsPath(), e);
          }
        }
      }
      if (!String0.isNullOrEmpty(this.getExtPropsPath())) {
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        try {
          Resource[] resources = resolver.getResources(this.getExtPropsPath());
          for (Resource resource : resources) {
            ResourceBundle resourceBundle = new PropertyResourceBundle(resource.getInputStream());
            resourceBundle.keySet().forEach(k -> InetAddress0.putCustomHost(k, String0.nullToEmpty(resourceBundle.getString(k))));
          }
        } catch (Exception e) {
          log.error(this.getExtPropsPath(), e);
          if (!quietly) {
            throw new ZeroException(this.getExtPropsPath(), e);
          }
        }
      }
      this.getVhosts().forEach(InetAddress0::putCustomHost);
    }
  }
}