package org.apache.maven.wagon.providers.ssh.knownhost;
   
   /*
    * 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.
   */
  
  import java.io.BufferedReader;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.StringReader;
  import java.util.HashSet;
  import java.util.Set;
  
  import org.codehaus.plexus.util.IOUtil;
  import org.codehaus.plexus.util.StringOutputStream;
  import org.codehaus.plexus.util.StringUtils;
  
  /**
   * Provides known hosts from a file
   *
   * @author Juan F. Codagnone
   * @since Sep 12, 2005
   */
  public class StreamKnownHostsProvider
      extends AbstractKnownHostsProvider
  {
  
      public StreamKnownHostsProvider( InputStream stream )
          throws IOException
      {
          try
          {
              StringOutputStream stringOutputStream = new StringOutputStream();
              IOUtil.copy( stream, stringOutputStream );
              this.contents = stringOutputStream.toString();
              
              this.knownHosts = this.loadKnownHosts( this.contents );
          }
          finally
          {
              IOUtil.close( stream );
          }
      }
      
      protected Set<KnownHostEntry> loadKnownHosts( String contents )
          throws IOException
      {
          Set<KnownHostEntry> hosts = new HashSet<KnownHostEntry>();
          
          BufferedReader br = new BufferedReader( new StringReader( contents ) );
          
          String line = null;
          
          do 
          {
              line = br.readLine();
              if ( line != null )
              {
                  String tokens[] = StringUtils.split( line );
                  if ( tokens.length == 3 )
                  {
                      hosts.add( new KnownHostEntry( tokens[0], tokens[1], tokens[2] ) );
                  }
              }
              
          }
          while ( line != null );
          
          return hosts;
      }
  }