/*
    *  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.directory.server.dns.store.jndi;
  
  
  import java.util.Map;
  import java.util.Set;
  
  import javax.naming.NamingException;
  import javax.naming.directory.DirContext;
  import javax.naming.ldap.LdapName;
  
  import org.apache.directory.api.ldap.model.exception.LdapNoSuchObjectException;
  import org.apache.directory.server.core.api.CoreSession;
  import org.apache.directory.server.core.api.DirectoryService;
  import org.apache.directory.server.core.jndi.ServerLdapContext;
  import org.apache.directory.server.dns.DnsException;
  import org.apache.directory.server.dns.messages.QuestionRecord;
  import org.apache.directory.server.dns.messages.ResourceRecord;
  import org.apache.directory.server.dns.messages.ResponseCode;
  import org.apache.directory.server.dns.store.jndi.operations.GetRecords;
  import org.apache.directory.server.i18n.I18n;
  import org.apache.directory.server.protocol.shared.ServiceConfigurationException;
  import org.apache.directory.server.protocol.shared.catalog.Catalog;
  import org.apache.directory.server.protocol.shared.catalog.GetCatalog;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  
  /**
   * A JNDI-backed search strategy implementation.  This search strategy builds a catalog
   * from directory configuration to determine where zones are to search for
   * resource records.
   * 
   * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
   */
  public class MultiBaseSearch implements SearchStrategy
  {
      /** the LOG for this class */
      private static final Logger LOG = LoggerFactory.getLogger( MultiBaseSearch.class );
  
      private final Catalog catalog;
      private final DirectoryService directoryService;
  
  
      MultiBaseSearch( String catalogBaseDn, DirectoryService directoryService )
      {
          this.directoryService = directoryService;
          try
          {
              CoreSession session = directoryService.getSession();
              catalog = new DnsCatalog( ( Map<String, Object> ) new GetCatalog().execute( session, null ) );
          }
          catch ( Exception e )
          {
              LOG.error( e.getLocalizedMessage(), e );
              String message = I18n.err( I18n.ERR_156, catalogBaseDn );
              throw new ServiceConfigurationException( message, e );
          }
      }
  
  
      public Set<ResourceRecord> getRecords( QuestionRecord question ) throws DnsException
      {
          try
          {
              GetRecords/dns/store/jndi/operations/GetRecords.html#GetRecords">GetRecords getRecords = new GetRecords( question );
              String baseDn = catalog.getBaseDn( question.getDomainName() );
              CoreSession session = directoryService.getSession();
              DirContext dirContext = new ServerLdapContext( directoryService, session, new LdapName( baseDn ) );
              return getRecords.execute( dirContext, null );
          }
          catch ( LdapNoSuchObjectException lnnfe )
          {
              LOG.debug( "Name for DNS record search does not exist.", lnnfe );
  
              throw new DnsException( ResponseCode.NAME_ERROR );
          }
          catch ( NamingException ne )
          {
              LOG.error( ne.getLocalizedMessage(), ne );
              String message = I18n.err( I18n.ERR_157, question.getDomainName() );
             throw new ServiceConfigurationException( message, ne );
         }
         catch ( Exception e )
         {
             LOG.debug( "Unexpected error retrieving DNS records.", e );
             throw new DnsException( ResponseCode.SERVER_FAILURE );
         }
 
     }
 }