/*
    * Copyright 2019 the original author or authors.
    *
    * Licensed 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.bremersee.ldaptive.reactive;
  
  import java.util.Collection;
  import java.util.Collections;
  import java.util.Objects;
  import java.util.Optional;
  import org.bremersee.ldaptive.LdaptiveEntryMapper;
  import org.bremersee.ldaptive.LdaptiveErrorHandler;
  import org.ldaptive.AddRequest;
  import org.ldaptive.BindRequest;
  import org.ldaptive.CompareRequest;
  import org.ldaptive.ConnectionFactory;
  import org.ldaptive.DeleteRequest;
  import org.ldaptive.LdapEntry;
  import org.ldaptive.ModifyDnRequest;
  import org.ldaptive.ModifyRequest;
  import org.ldaptive.Result;
  import org.ldaptive.SearchRequest;
  import org.ldaptive.extended.ExtendedRequest;
  import org.ldaptive.extended.ExtendedResponse;
  import org.ldaptive.extended.PasswordModifyRequest;
  import org.ldaptive.extended.PasswordModifyResponseParser;
  import reactor.core.publisher.Flux;
  import reactor.core.publisher.Mono;
  
  /**
   * The reactive ldaptive operations.
   *
   * @author Christian Bremer
   */
  public interface ReactiveLdaptiveOperations {
  
    /**
     * Returns a new instance of this ldap operations with the same connection factory and error
     * handler.
     *
     * @return a new instance of this ldap operations
     */
    ReactiveLdaptiveOperations copy();
  
    /**
     * Returns a new instance of this ldap operations with the same connection factory and the given
     * error handler.
     *
     * @param errorHandler the new error handler
     * @return the new instance of this ldap operations
     */
    ReactiveLdaptiveOperations copy(LdaptiveErrorHandler errorHandler);
  
  
    /**
     * Gets connection factory.
     *
     * @return the connection factory
     */
    ConnectionFactory getConnectionFactory();
  
    /**
     * Executes add operation.
     *
     * @param addRequest the add request
     * @return the mono
     */
    Mono<Result> add(AddRequest addRequest);
  
    /**
     * Executes bind operation.
     *
     * @param bindRequest the bind request
     * @return the mono
     */
    Mono<Boolean> bind(BindRequest bindRequest);
  
    /**
     * Executes compare operation.
     *
     * @param compareRequest the compare request
     * @return the mono
     */
    Mono<Boolean> compare(CompareRequest compareRequest);
  
    /**
     * Executes delete operation.
    *
    * @param deleteRequest the delete request
    * @return the mono
    */
   Mono<Result> delete(DeleteRequest deleteRequest);
 
   /**
    * Executes extended request.
    *
    * @param request the request
    * @return the mono
    */
   Mono<ExtendedResponse> executeExtension(ExtendedRequest request);
 
   /**
    * Generate user password.
    *
    * @param dn the dn
    * @return the mono
    */
   default Mono<String> generateUserPassword(String dn) {
     return executeExtension(new PasswordModifyRequest(dn))
         .map(PasswordModifyResponseParser::parse);
   }
 
   /**
    * Executes modify operation.
    *
    * @param modifyRequest the modify request
    * @return the mono
    */
   Mono<Result> modify(ModifyRequest modifyRequest);
 
   /**
    * Executes modify DN operation.
    *
    * @param modifyDnRequest the modify dn request
    * @return the mono
    */
   Mono<Result> modifyDn(ModifyDnRequest modifyDnRequest);
 
   /**
    * Modify user password.
    *
    * @param dn the dn
    * @param oldPass the old pass
    * @param newPass the new pass
    * @return the mono
    */
   default Mono<ExtendedResponse> modifyUserPassword(
       String dn,
       String oldPass,
       String newPass) {
     return executeExtension(new PasswordModifyRequest(dn, oldPass, newPass));
   }
 
   /**
    * Find one.
    *
    * @param searchRequest the search request
    * @return the mono
    */
   Mono<LdapEntry> findOne(SearchRequest searchRequest);
 
   /**
    * Find one.
    *
    * @param <T> the type parameter
    * @param searchRequest the search request
    * @param entryMapper the entry mapper
    * @return the mono
    */
   default <T> Mono<T> findOne(SearchRequest searchRequest,
       LdaptiveEntryMapper<T> entryMapper) {
     return findOne(searchRequest)
         .map(ldapEntry -> Objects.requireNonNull(entryMapper.map(ldapEntry)));
   }
 
   /**
    * Find all.
    *
    * @param searchRequest the search request
    * @return the flux
    */
   Flux<LdapEntry> findAll(SearchRequest searchRequest);
 
   /**
    * Find all.
    *
    * @param <T> the type parameter
    * @param searchRequest the search request
    * @param entryMapper the entry mapper
    * @return the flux
    */
   default <T> Flux<T> findAll(SearchRequest searchRequest,
       LdaptiveEntryMapper<T> entryMapper) {
     return findAll(searchRequest)
         .map(ldapEntry -> Objects.requireNonNull(entryMapper.map(ldapEntry)));
   }
 
   /**
    * Exists.
    *
    * @param dn the dn
    * @return the mono
    */
   default Mono<Boolean> exists(String dn) {
     int index = dn.indexOf('=');
     if (index > 0) {
       String attr = dn.substring(0, index).trim();
       return findOne(SearchRequest.objectScopeSearchRequest(dn, new String[]{attr}))
           .map(ldapEntry -> true)
           .defaultIfEmpty(false);
     }
     return Mono.just(false);
   }
 
   /**
    * Exists.
    *
    * @param <T> the type parameter
    * @param domainObject the domain object
    * @param entryMapper the entry mapper
    * @return the mono
    */
   default <T> Mono<Boolean> exists(T domainObject,
       LdaptiveEntryMapper<T> entryMapper) {
     return exists(entryMapper.mapDn(domainObject));
   }
 
   /**
    * Save.
    *
    * @param <T> the type parameter
    * @param domainObject the domain object
    * @param entryMapper the entry mapper
    * @return the mono
    */
   <T> Mono<T> save(T domainObject, LdaptiveEntryMapper<T> entryMapper);
 
   /**
    * Save all.
    *
    * @param <T> the type parameter
    * @param domainObjects the domain objects
    * @param entryMapper the entry mapper
    * @return the flux
    */
   default <T> Flux<T> saveAll(Collection<T> domainObjects,
       LdaptiveEntryMapper<T> entryMapper) {
     return Flux.fromIterable(Optional.ofNullable(domainObjects).orElseGet(Collections::emptyList))
         .flatMap(domainObject -> save(domainObject, entryMapper));
   }
 
   /**
    * Remove.
    *
    * @param <T> the type parameter
    * @param domainObject the domain object
    * @param entryMapper the entry mapper
    * @return the mono
    */
   default <T> Mono<Result> remove(
       T domainObject,
       LdaptiveEntryMapper<T> entryMapper) {
     return delete(DeleteRequest.builder().dn(entryMapper.mapDn(domainObject)).build());
   }
 
   /**
    * Remove all.
    *
    * @param <T> the type parameter
    * @param domainObjects the domain objects
    * @param entryMapper the entry mapper
    * @return the mono
    */
   default <T> Mono<Long> removeAll(
       Collection<T> domainObjects,
       LdaptiveEntryMapper<T> entryMapper) {
     return Flux.fromIterable(Optional.ofNullable(domainObjects).orElseGet(Collections::emptyList))
         .flatMap(domainObject -> remove(domainObject, entryMapper))
         .count();
   }
 
 }