/*
    * 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 org.apache.commons.net.examples.mail;
  
  import java.io.IOException;
  import java.net.URI;
  
  import org.apache.commons.net.ProtocolCommandListener;
  import org.apache.commons.net.imap.IMAPClient;
  import org.apache.commons.net.imap.IMAPSClient;
  
  /**
   * Utility class for shared IMAP utilities
   */
  class IMAPUtils {
  
      /**
       * Parses the URI and use the details to connect to the IMAP(S) server and login.
       *
       * @param uri            the URI to use, e.g. imaps://user:pass@imap.mail.yahoo.com/folder or imaps://user:pass@imap.googlemail.com/folder
       * @param defaultTimeout initial timeout (in milliseconds)
       * @param listener       for tracing protocol IO (may be null)
       * @return the IMAP client - connected and logged in
       * @throws IOException if any problems occur
       */
      static IMAPClient imapLogin(final URI uri, final int defaultTimeout, final ProtocolCommandListener listener) throws IOException {
          final String userInfo = uri.getUserInfo();
          if (userInfo == null) {
              throw new IllegalArgumentException("Missing userInfo details");
          }
  
          final String[] userPassword = userInfo.split(":");
          if (userPassword.length != 2) {
              throw new IllegalArgumentException("Invalid userInfo details: '" + userInfo + "'");
          }
  
          final String user = userPassword[0];
          String password = userPassword[1];
          // prompt for the password if necessary
          password = Utils.getPassword(user, password);
  
          final IMAPClient imap;
  
          final String scheme = uri.getScheme();
          if ("imaps".equalsIgnoreCase(scheme)) {
              System.out.println("Using secure protocol");
              imap = new IMAPSClient(true); // implicit
          } else if ("imap".equalsIgnoreCase(scheme)) {
              imap = new IMAPClient();
          } else {
              throw new IllegalArgumentException("Invalid protocol: " + scheme);
          }
          final int port = uri.getPort();
          if (port != -1) {
              imap.setDefaultPort(port);
          }
  
          imap.setDefaultTimeout(defaultTimeout);
  
          if (listener != null) {
              imap.addProtocolCommandListener(listener);
          }
  
          final String server = uri.getHost();
          System.out.println("Connecting to server " + server + " on " + imap.getDefaultPort());
  
          try {
              imap.connect(server);
              System.out.println("Successfully connected");
          } catch (final IOException e) {
              throw new IOException("Could not connect to server.", e);
          }
  
          if (!imap.login(user, password)) {
              imap.disconnect();
              throw new IOException("Could not login to server. Check login details.");
          }
  
          return imap;
      }
  }