/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
    * with the License.  You may obtain a copy of the License at
    *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  package com.icegreen.greenmail.util;
  
  import com.icegreen.greenmail.Managers;
  import com.icegreen.greenmail.imap.ImapServer;
  import com.icegreen.greenmail.pop3.Pop3Server;
  import com.icegreen.greenmail.server.AbstractServer;
  import com.icegreen.greenmail.smtp.InterruptableSmtpServer;
  import java.util.HashMap;
  import java.util.Map;
  
  public class InterruptableGreenMail extends GreenMail {
  
      public InterruptableGreenMail(final ServerSetup[] config) {
          super(config);
      }
  
      @Override
      protected Map<String, AbstractServer> createServices(final ServerSetup[] config, final Managers mgr) {
          Map<String, AbstractServer> srvc = new HashMap<>();
          for (ServerSetup setup : config) {
              if (srvc.containsKey(setup.getProtocol())) {
                  throw new IllegalArgumentException("Server '" + setup.getProtocol()
                          + "' was found at least twice in the array");
              }
              final String protocol = setup.getProtocol();
              if (protocol.startsWith(ServerSetup.PROTOCOL_SMTP)) {
                  srvc.put(protocol, new InterruptableSmtpServer(setup, mgr));
              } else if (protocol.startsWith(ServerSetup.PROTOCOL_POP3)) {
                  srvc.put(protocol, new Pop3Server(setup, mgr));
              } else if (protocol.startsWith(ServerSetup.PROTOCOL_IMAP)) {
                  srvc.put(protocol, new ImapServer(setup, mgr));
              }
          }
          return srvc;
      }
  }