package com.atlassian.plugin.servlet.descriptors;
   
   import com.atlassian.plugin.Plugin;
   import com.atlassian.plugin.module.ModuleFactory;
   import com.atlassian.plugin.module.PrefixDelegatingModuleFactory;
   import com.atlassian.plugin.servlet.ObjectFactories;
   import com.atlassian.plugin.servlet.ObjectFactory;
   import com.atlassian.plugin.servlet.PluginBuilder;
   import com.atlassian.plugin.servlet.ServletModuleManager;
  
  import javax.servlet.http.HttpServlet;
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.HashMap;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  import java.util.Optional;
  
  import static java.util.Collections.emptySet;
  import static org.mockito.Mockito.mock;
  
  public class ServletModuleDescriptorBuilder {
      private Plugin plugin = new PluginBuilder().build();
      private String key = "test.servlet";
      private Optional<String> scopeKey = Optional.empty();
      private List<String> paths = new LinkedList<>();
      private ServletModuleManager servletModuleManager = mock(ServletModuleManager.class);
      private Map<String, String> initParams = new HashMap<>();
      private ObjectFactory<HttpServlet> servletFactory;
  
      public ServletModuleDescriptorBuilder with(Plugin plugin) {
          this.plugin = plugin;
          return this;
      }
  
      public ServletModuleDescriptorBuilder withKey(String key) {
          this.key = key;
          return this;
      }
  
      public ServletModuleDescriptorBuilder with(final HttpServlet servlet) {
          this.servletFactory = ObjectFactories.createSingleton(servlet);
          return this;
      }
  
      public ServletModuleDescriptorBuilder withFactory(ObjectFactory<HttpServlet> servlet) {
          this.servletFactory = servlet;
          return this;
      }
  
      public ServletModuleDescriptorBuilder withPath(String path) {
          paths.add(path);
          return this;
      }
  
      public ServletModuleDescriptorBuilder with(ServletModuleManager servletModuleManager) {
          this.servletModuleManager = servletModuleManager;
          return this;
      }
  
      public ServletModuleDescriptorBuilder withInitParam(String name, String value) {
          initParams.put(name, value);
          return this;
      }
  
      public ServletModuleDescriptorBuilder withScopeKey(Optional<String> scopeKey) {
          this.scopeKey = scopeKey;
          return this;
      }
  
  
      public ServletModuleDescriptor build() {
          Descriptor d = new Descriptor(plugin, key, scopeKey, servletFactory, immutableList(paths), immutableMap(initParams),
                  servletModuleManager, new PrefixDelegatingModuleFactory(emptySet()));
          plugin.addModuleDescriptor(d);
          return d;
      }
  
      static <K, V> Map<K, V> immutableMap(Map<K, V> initParams) {
          return Collections.unmodifiableMap(new HashMap<K, V>(initParams));
      }
  
      <T> List<T> immutableList(List<T> list) {
          List<T> copy = new ArrayList<T>(list.size());
          copy.addAll(list);
          return Collections.unmodifiableList(copy);
      }
  
      static final class Descriptor extends ServletModuleDescriptor {
          final String key;
          final ObjectFactory<HttpServlet> servletFactory;
          final List<String> paths;
          final ServletModuleManager servletModuleManager;
          final Map<String, String> initParams;
          final Optional<String> scopeKey;
  
          public Descriptor(
                  Plugin plugin,
                 String key,
                 Optional<String> scopeKey,
                 ObjectFactory<HttpServlet> servletFactory,
                 List<String> paths,
                 Map<String, String> initParams,
                 ServletModuleManager servletModuleManager,
                 ModuleFactory moduleFactory) {
             super(moduleFactory, servletModuleManager);
             this.plugin = plugin;
             this.key = key;
             this.scopeKey = scopeKey;
             this.servletFactory = servletFactory;
             this.paths = paths;
             this.initParams = initParams;
             this.servletModuleManager = servletModuleManager;
         }
 
         @Override
         public Plugin getPlugin() {
             return plugin;
         }
 
         @Override
         public String getCompleteKey() {
             return getPluginKey() + ":" + key;
         }
 
         @Override
         public String getKey() {
             return key;
         }
 
         @Override
         public HttpServlet getModule() {
             return servletFactory.create();
         }
 
 
         @Override
         public Optional<String> getScopeKey() {
             return scopeKey;
         }
 
         @Override
         public List<String> getPaths() {
             return paths;
         }
 
         @Override
         public Map<String, String> getInitParams() {
             return initParams;
         }
 
     }
 
 }