UnboundID Server SDK

Ping Identity
UnboundID Server SDK Documentation

ExampleScriptedTask.groovy

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at
 * docs/licenses/cddl.txt
 * or http://www.opensource.org/licenses/cddl1.php.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at
 * docs/licenses/cddl.txt.  If applicable,
 * add the following below this CDDL HEADER, with the fields enclosed
 * by brackets "[]" replaced with your own identifying information:
 *      Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 *
 *
 *      Copyright 2010-2020 Ping Identity Corporation
 */
package com.unboundid.directory.sdk.examples.groovy;



import java.io.File;
import java.util.Arrays;
import java.util.Collections;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import com.unboundid.directory.sdk.common.types.InternalConnection;
import com.unboundid.directory.sdk.common.types.LogSeverity;
import com.unboundid.directory.sdk.ds.scripting.ScriptedTask;
import com.unboundid.directory.sdk.ds.types.DirectoryServerContext;
import com.unboundid.directory.sdk.ds.types.TaskContext;
import com.unboundid.directory.sdk.ds.types.TaskReturnState;
import com.unboundid.ldap.sdk.Filter;
import com.unboundid.ldap.sdk.LDAPException;
import com.unboundid.ldap.sdk.LDAPSearchException;
import com.unboundid.ldap.sdk.ResultCode;
import com.unboundid.ldap.sdk.SearchRequest;
import com.unboundid.ldap.sdk.SearchResult;
import com.unboundid.ldap.sdk.SearchResultEntry;
import com.unboundid.ldap.sdk.SearchResultListener;
import com.unboundid.ldap.sdk.SearchResultReference;
import com.unboundid.ldap.sdk.SearchScope;
import com.unboundid.ldif.LDIFWriter;
import com.unboundid.util.StaticUtils;
import com.unboundid.util.args.ArgumentException;
import com.unboundid.util.args.ArgumentParser;
import com.unboundid.util.args.DNArgument;
import com.unboundid.util.args.FileArgument;
import com.unboundid.util.args.FilterArgument;
import com.unboundid.util.args.StringArgument;



/**
 * This class provides a simple example of a scripted task that will perform an
 * internal search and write the results to a file on the server filesystem.
 * The search criteria will be specified using the provided arguments:
 * <UL>
 *   <LI>base-dn -- The base DN for the search to perform.</LI>
 *   <LI>scope -- The scope for the search.  The value should be one of the
 *       following strings:  base, one, sub, or subordinate-subtree.</LI>
 *   <LI>filter -- The filter for the search.</LI>
 *   <LI>attribute -- An attribute to request.  This may be provided multiple
 *       times to request multiple attributes.</LI>
 *   <LI>output-file -- The path to the file to which the search results should
 *       be written.</LI>
 * </UL>
 */
public final class ExampleScriptedTask
       extends ScriptedTask
       implements SearchResultListener
{
  /**
   * The name of the argument used to specify the search base DN.
   */
  private static final String ARG_NAME_BASE_DN = "base-dn";



  /**
   * The name of the argument used to specify the search scope.
   */
  private static final String ARG_NAME_SCOPE = "scope";



  /**
   * The name of the argument used to specify the filter.
   */
  private static final String ARG_NAME_FILTER = "filter";



  /**
   * The name of the argument used to specify the requested attributes.
   */
  private static final String ARG_NAME_ATTR = "attribute";



  /**
   * The name of the argument used to specify the path to the output file.
   */
  private static final String ARG_NAME_OUTPUT_FILE = "output-file";



  /**
   * The map of search scope names to values.
   */
  private static final Map<String,SearchScope> SCOPE_MAP;



  /**
   * The set of allowed search scope values.
   */
  private static final Set<String> ALLOWED_SCOPES;



  static
  {
    final LinkedHashMap<String,SearchScope> scopeMap =
         new LinkedHashMap<String,SearchScope>(4);
    scopeMap.put("base", SearchScope.BASE);
    scopeMap.put("one", SearchScope.ONE);
    scopeMap.put("sub", SearchScope.SUB);
    scopeMap.put("subordinate-subtree", SearchScope.SUBORDINATE_SUBTREE);

    SCOPE_MAP = Collections.unmodifiableMap(scopeMap);
    ALLOWED_SCOPES = Collections.unmodifiableSet(scopeMap.keySet());
  }



  /**
   * The serial version UID for this serializable class.
   */
  private static final long serialVersionUID = -7584765913920729867L;



  // The server context for the server in which this extension is running.
  private DirectoryServerContext serverContext;

  // The output file for the search results.
  private File outputFile;

  // The filter for the search.
  private Filter filter;

  // The LDIF writer to use when writing the results.
  private volatile LDIFWriter ldifWriter;

  // The list of requested attributes.
  private List<String> requestedAttributes;

  // The scope for the search.
  private SearchScope scope;

  // The base DN for the search.
  private String baseDN;

  // The task context for the task.
  private volatile TaskContext taskContext;

  // The return state for the task.
  private volatile TaskReturnState returnState;



  /**
   * Creates a new instance of this task.  All task implementations must
   * include a default constructor, but any initialization should generally be
   * done in the {@code initializeTask} method.
   */
  public ExampleScriptedTask()
  {
    // No implementation required.
  }



  /**
   * Updates the provided argument parser to define any configuration arguments
   * which may be used by this plugin.  The argument parser may also be updated
   * to define relationships between arguments (e.g., to specify required,
   * exclusive, or dependent argument sets).
   *
   * @param  parser  The argument parser to be updated with the configuration
   *                 arguments which may be used by this plugin.
   *
   * @throws  ArgumentException  If a problem is encountered while updating the
   *                             provided argument parser.
   */
  @Override()
  public void defineConfigArguments(final ArgumentParser parser)
         throws ArgumentException
  {
    // Add an argument that allows you to specify the search base DN.
    Character shortIdentifier = null;
    String    longIdentifier  = ARG_NAME_BASE_DN;
    boolean   required        = true;
    int       maxOccurrences  = 1;
    String    placeholder     = "{dn}";
    String    description     = "The base DN to use for the search.";

    parser.addArgument(new DNArgument(shortIdentifier, longIdentifier,
         required, maxOccurrences, placeholder, description));


    // Add an argument that allows you to specify the search scope.
    shortIdentifier = null;
    longIdentifier  = ARG_NAME_SCOPE;
    required        = true;
    maxOccurrences  = 1;
    placeholder     = "{base|one|sub|subordinate-subtree}";
    description     = "The scope to use for the search.";

    parser.addArgument(new StringArgument(shortIdentifier, longIdentifier,
         required, maxOccurrences, placeholder, description,
         ALLOWED_SCOPES));


    // Add an argument that allows you to specify the search filter.
    shortIdentifier = null;
    longIdentifier  = ARG_NAME_FILTER;
    required        = true;
    maxOccurrences  = 1;
    placeholder     = "{filter}";
    description     = "The filter to use for the search.";

    parser.addArgument(new FilterArgument(shortIdentifier, longIdentifier,
         required, maxOccurrences, placeholder, description));


    // Add an argument that allows you to specify the requested attributes.
    shortIdentifier = null;
    longIdentifier  = ARG_NAME_ATTR;
    required        = false;
    maxOccurrences  = 0; // No limit
    placeholder     = "{attr}";
    description     = "An attribute to include in matching entries.  This " +
         "argument may be provided multiple times to request multiple " +
         "attributes.  If no requested attributes are provided, then all " +
         "user attributes will be requested.";

    parser.addArgument(new StringArgument(shortIdentifier, longIdentifier,
         required, maxOccurrences, placeholder, description));


    // Add an argument that allows you to specify the output file.
    shortIdentifier = null;
    longIdentifier  = ARG_NAME_OUTPUT_FILE;
    required        = true;
    maxOccurrences  = 1;
    placeholder     = "{path}";
    description     = "The output file to which the search results should be " +
         "written.  Relative paths will be relative to the server root.";

    boolean fileMustExist   = false;
    boolean parentMustExist = true;
    boolean mustBeFile      = true;
    boolean mustBeDirectory = false;

    parser.addArgument(new FileArgument(shortIdentifier, longIdentifier,
         required, maxOccurrences, placeholder, description, fileMustExist,
         parentMustExist, mustBeFile, mustBeDirectory));
  }



  /**
   * Initializes this task.
   *
   * @param  serverContext  A handle to the server context for the server in
   *                        which this task is running.
   * @param  parser         The argument parser which has been initialized from
   *                        the configuration for this task.
   *
   * @throws  LDAPException  If a problem occurs while initializing this task.
   */
  @Override()
  public void initializeTask(final DirectoryServerContext serverContext,
                             final ArgumentParser parser)
         throws LDAPException
  {
    this.serverContext = serverContext;


    // Get the base DN.
    final DNArgument baseArg =
         (DNArgument) parser.getNamedArgument(ARG_NAME_BASE_DN);
    baseDN = baseArg.getValue().toString();


    // Get the scope.
    final StringArgument scopeArg =
         (StringArgument) parser.getNamedArgument(ARG_NAME_SCOPE);
    scope = SCOPE_MAP.get(StaticUtils.toLowerCase(scopeArg.getValue()));


    // Get the filter.
    final FilterArgument filterArg =
         (FilterArgument) parser.getNamedArgument(ARG_NAME_FILTER);
    filter = filterArg.getValue();


    // Get the list of requested attributes.
    final StringArgument attrsArg =
         (StringArgument) parser.getNamedArgument(ARG_NAME_ATTR);
    requestedAttributes = attrsArg.getValues();


    // Get the output file.
    final FileArgument outputFileArg =
         (FileArgument) parser.getNamedArgument(ARG_NAME_OUTPUT_FILE);
    outputFile = outputFileArg.getValue();
  }



  /**
   * Performs the appropriate processing for this task.
   *
   * @param  taskContext  Information about the task to be run.
   *
   * @return  Information about the state of the task after processing has
   *          completed.
   */
  @Override()
  public TaskReturnState runTask(final TaskContext taskContext)
  {
    this.taskContext = taskContext;

    // Create an LDIF writer to use to write the results.
    try
    {
      ldifWriter = new LDIFWriter(outputFile);
    }
    catch (final Exception e)
    {
      serverContext.debugCaught(e);
      serverContext.logMessage(LogSeverity.SEVERE_ERROR,
           "Unable to create an LDIF writer to write to file " +
                outputFile.getAbsolutePath() + " for task " +
                taskContext.getTaskEntryDN() + ":  " +
                StaticUtils.getExceptionMessage(e));
      return TaskReturnState.STOPPED_BY_ERROR;
    }

    try
    {
      // Append a header to the LDIF file with details of the search.
      try
      {
        ldifWriter.writeComment("Internal search initiated at " +
                  new Date().toString() + " for task " +
                  taskContext.getTaskEntryDN(),
             false, false);
        ldifWriter.writeComment("Base DN:  " + baseDN, false, false);
        ldifWriter.writeComment("Scope:  " + scope.getName(), false, false);
        ldifWriter.writeComment("Filter:  " + filter.toString(), false, false);
        ldifWriter.writeComment("Requested Attributes:  " + requestedAttributes,
             false, true);
      }
      catch (final Exception e)
      {
        serverContext.debugCaught(e);
        serverContext.logMessage(LogSeverity.SEVERE_ERROR,
             "Unable to write a header to LDIF file " +
                  outputFile.getAbsolutePath() + ":  " +
                  StaticUtils.getExceptionMessage(e));
        return TaskReturnState.STOPPED_BY_ERROR;
      }


      // Create the search request to issue.  We'll use a search result listener
      // since the search may return a large number of entries.
      final SearchRequest searchRequest = new SearchRequest(this, baseDN, scope,
           filter);
      searchRequest.setAttributes(requestedAttributes);


      // Get an internal connection and issue the search.
      returnState = TaskReturnState.COMPLETED_SUCCESSFULLY;
      SearchResult searchResult;
      try
      {
        final InternalConnection conn =
             serverContext.getInternalRootConnection();
        searchResult = conn.search(searchRequest);
      }
      catch (final LDAPSearchException lse)
      {
        serverContext.debugCaught(lse);

        searchResult = lse.getSearchResult();
        if (returnState == TaskReturnState.COMPLETED_SUCCESSFULLY)
        {
          returnState = TaskReturnState.COMPLETED_WITH_ERRORS;
        }

        serverContext.logMessage(LogSeverity.MILD_ERROR,
             "An error occurred while processing the search for task " +
                  taskContext.getTaskEntryDN() + ":  " +
                  searchResult.toString());
      }


      // Append a footer to the LDIF file with a summary of the results.
      try
      {
        ldifWriter.writeComment("Internal search completed at " +
                  new Date().toString(),
             true, false);
        ldifWriter.writeComment("Result Code:  " +
                  searchResult.getResultCode().toString(),
             false, false);

        final String diagnosticMessage = searchResult.getDiagnosticMessage();
        if (diagnosticMessage != null)
        {
          ldifWriter.writeComment("Diagnostic Message:  " + diagnosticMessage,
               false, false);
        }

        final String matchedDN = searchResult.getMatchedDN();
        if (matchedDN != null)
        {
          ldifWriter.writeComment("Matched DN:  " + matchedDN,
               false, false);
        }

        final String[] referralURLs = searchResult.getReferralURLs();
        if ((referralURLs != null) && (referralURLs.length > 0))
        {
          ldifWriter.writeComment("Referral URLs:  " +
                    Arrays.toString(referralURLs),
               false, false);
        }

        ldifWriter.writeComment("Entries Returned:  " +
                  searchResult.getEntryCount(),
             false, false);
        ldifWriter.writeComment("References Returned:  " +
                  searchResult.getReferenceCount(),
             false, false);
      }
      catch (final Exception e)
      {
        serverContext.debugCaught(e);
        serverContext.logMessage(LogSeverity.SEVERE_ERROR,
             "Unable to write a header to LDIF file " +
                  outputFile.getAbsolutePath() + ":  " +
                  StaticUtils.getExceptionMessage(e));
        return TaskReturnState.STOPPED_BY_ERROR;
      }

      return returnState;
    }
    finally
    {
      try
      {
        ldifWriter.close();
      }
      catch (final Exception e)
      {
        serverContext.debugCaught(e);
        serverContext.logMessage(LogSeverity.SEVERE_ERROR,
             "An error occurred while closing file " +
                  outputFile.getAbsolutePath() + " for task " +
                  taskContext.getTaskEntryDN() + ":  " +
                  StaticUtils.getExceptionMessage(e));
      }
    }
  }



  /**
   * Indicates that the provided search result entry has been returned by the
   * server and may be processed by this search result listener.
   *
   * @param  searchEntry  The search result entry that has been returned by the
   *                      server.
   */
  public void searchEntryReturned(final SearchResultEntry searchEntry)
  {
    try
    {
      ldifWriter.writeEntry(searchEntry);
    }
    catch (final Exception e)
    {
      serverContext.debugCaught(e);
      serverContext.logMessage(LogSeverity.SEVERE_ERROR,
           "An error occurred while attempting to write matching entry " +
                searchEntry.getDN() + " to file " +
                outputFile.getAbsolutePath() + " for task " +
                taskContext.getTaskEntryDN() + ":  " +
                StaticUtils.getExceptionMessage(e));
      returnState = TaskReturnState.COMPLETED_WITH_ERRORS;
    }
  }



  /**
   * Indicates that the provided search result reference has been returned by
   * the server and may be processed by this search result listener.
   *
   * @param  searchReference  The search result reference that has been returned
   *                          by the server.
   */
  public void searchReferenceReturned(
                   final SearchResultReference searchReference)
  {
    try
    {
      ldifWriter.writeComment(
           "Received search result reference with URLs " +
                Arrays.toString(searchReference.getReferralURLs()),
           true, true);
    }
    catch (final Exception e)
    {
      serverContext.debugCaught(e);
      serverContext.logMessage(LogSeverity.SEVERE_ERROR,
           "An error occurred while attempting to information about " +
                "search result reference " +
                Arrays.toString(searchReference.getReferralURLs()) +
                " to file " + outputFile.getAbsolutePath() + " for task " +
                taskContext.getTaskEntryDN() + ":  " +
                StaticUtils.getExceptionMessage(e));
      returnState = TaskReturnState.COMPLETED_WITH_ERRORS;
    }
  }
}