package de.juplo.httpresources;
   
   
   import java.nio.file.Files;
   import java.nio.file.Path;
   import java.nio.file.Paths;
   import java.util.LinkedHashMap;
   import java.util.LinkedList;
   import java.util.List;
  import java.util.Map;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.testcontainers.containers.BindMode;
  import org.testcontainers.containers.NginxContainer;
  
  
  /**
   * A configurable subclass of the {@link NginxContainer}.
   * @author Kai Moritz
   */
  public class ConfigurableNginxContainer extends NginxContainer
  {
    private final static Logger LOG =
        LoggerFactory.getLogger(ConfigurableNginxContainer.class);
  
  
    public final static int DEFAULT_NGINX_PORT = 80;
  
    private final Map<String, String> directives;
  
  
    /**
     * Creates an instance with a default-configuration, that matches the
     * static configuration of the base-class {@link NginxContainer}
     */
    public ConfigurableNginxContainer()
    {
      directives = new LinkedHashMap<>();
      directives.put("daemon", "off");
    }
  
  
    /**
     * Adds a global directive to the configuration-map.
     * <p>
     * This method can only be used to add <em>global</em> directives.
     * Global directives are all directives, that are marked with the context
     * <code>Main</code> in the {@link http://nginx.org/en/docs/ngx_core_module.html
     * Nginx-documentation}, for example <code>daemon</code>, <code>user</code>
     * and <code>env</code>.
     * <p>
     * If you need to modify contexts, like <code>server</code> or
     * <code>http</code>, you have to
     * {@link #withConfiguration(String) pass in a complete configuration}.
     * @param key The key of the directive
     * @param value The value of the directive (the <code>;</code> is added automatically)
     * @return <code>this</code> instance, to enable a fluid API.
     */
    public ConfigurableNginxContainer withGlobalDirective(String key, String value)
    {
      directives.put(key, value);
      return this;
    }
  
    public Map<String, String> getGlobalDirectives()
    {
      return directives;
    }
  
    public ConfigurableNginxContainer withConfiguration(String path)
    {
      Path config = Paths.get(path);
  
      if (Files.isRegularFile(config))
      {
        LOG.info("Binding file {} as /etc/nginx/nginx.conf", config.toAbsolutePath());
        addFileSystemBind(path, "/etc/nginx/nginx.conf", BindMode.READ_ONLY);
        return this;
      }
  
      if (Files.isDirectory(config))
      {
        LOG.info("Binding directory {} as /etc/nginx/", config.toAbsolutePath());
        addFileSystemBind(path, "/etc/nginx/nginx.conf", BindMode.READ_ONLY);
        return this;
      }
  
      throw new IllegalArgumentException(
            config.toAbsolutePath() +
            " has to represent either a directory, or a regular file!");
    }
  
    @Override
    protected void configure()
    {
      addExposedPort(DEFAULT_NGINX_PORT);
      List<String> commands = new LinkedList<>();
      commands.add("nginx");
      directives.entrySet().forEach((directive) ->
     {
       commands.add("-g");
       commands.add(directive.getKey() + " " + directive.getValue() + ";");
     });
     LOG.info("CMD: {}", commands);
     setCommand(commands.toArray(new String[commands.size()]));
   }
 }