org.springframework.shell.core.Completion Java Examples

public boolean getAllPossibleValues(final List<Completion> completions, 
    final Class<?> requiredType, final String existingData, 
    final String optionContext, final MethodTarget target) {
	Class<Enum> enumClass = (Class<Enum>) requiredType;
	for (Enum enumValue : enumClass.getEnumConstants()) {
		String candidate = enumValue.name();
     // GemFire/gfsh addition - check 'existingData == null'. GfshParser can 
		// pass existingData as null  
     if ("".equals(existingData) || existingData == null
         || candidate.startsWith(existingData)
         || existingData.startsWith(candidate)
         || candidate.toUpperCase().startsWith(existingData.toUpperCase())
         || existingData.toUpperCase().startsWith(candidate.toUpperCase())) {
			completions.add(new Completion(candidate));
		}
	}
	return true;
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions, Class<?> targetType, String existingData, String optionContext, MethodTarget target) {
	try {
		String kind = determineKind(optionContext).get();
		switch (kind) {
			case "stream":
				streamOperations().list().forEach(sdr -> completions.add(new Completion(sdr.getName())));
				break;
			case "task":
				taskOperations().list().forEach(tdf -> completions.add(new Completion(tdf.getName())));
				break;
			case "job":
				jobOperations().executionList().forEach(jer -> {if (!"?".equals(jer.getName())) {
					completions.add(new Completion(jer.getName()));
				}});
				break;
			default:
				throw new AssertionError("Unsupported entity kind: " + kind);
		}
	} // Protect from exceptions in non-command code, as it would crash the whole shell
	catch (Exception e) {
	}
	return false;
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String[] existingData, String context,
    MethodTarget target) {
  String adjustedUserInput = convertUserInputIntoAFullyQualifiedPath((existingData != null) ? existingData[existingData.length - 1]
      : "");

  String directoryData = adjustedUserInput.substring(0,
      adjustedUserInput.lastIndexOf(File.separator) + 1);
  adjustedUserInput = adjustedUserInput.substring(adjustedUserInput
      .lastIndexOf(File.separator) + 1);

  populate(completions, adjustedUserInput,
      ((existingData != null) ? existingData[existingData.length - 1] : ""),
      directoryData);

  return true;
}
 

public boolean getAllPossibleValues(final List<Completion> completions, 
    final Class<?> requiredType, final String existingData, 
    final String optionContext, final MethodTarget target) {
	Class<Enum> enumClass = (Class<Enum>) requiredType;
	for (Enum enumValue : enumClass.getEnumConstants()) {
		String candidate = enumValue.name();
     // GemFire/gfsh addition - check 'existingData == null'. GfshParser can 
		// pass existingData as null  
     if ("".equals(existingData) || existingData == null
         || candidate.startsWith(existingData)
         || existingData.startsWith(candidate)
         || candidate.toUpperCase().startsWith(existingData.toUpperCase())
         || existingData.toUpperCase().startsWith(candidate.toUpperCase())) {
			completions.add(new Completion(candidate));
		}
	}
	return true;
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String[] existingData, String context,
    MethodTarget target) {
  String adjustedUserInput = convertUserInputIntoAFullyQualifiedPath((existingData != null) ? existingData[existingData.length - 1]
      : "");

  String directoryData = adjustedUserInput.substring(0,
      adjustedUserInput.lastIndexOf(File.separator) + 1);
  adjustedUserInput = adjustedUserInput.substring(adjustedUserInput
      .lastIndexOf(File.separator) + 1);

  populate(completions, adjustedUserInput,
      ((existingData != null) ? existingData[existingData.length - 1] : ""),
      directoryData);

  return true;
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.DISKSTORE_ALL.equals(optionContext)) {
    Set<String> diskStoreNames = getDiskStoreNames();
    
    for (String diskStoreName : diskStoreNames) {
      if (existingData != null) {
        if (diskStoreName.startsWith(existingData)) {
          completions.add(new Completion(diskStoreName));
        }
      } else {
        completions.add(new Completion(diskStoreName));
      }
    }
  }
  
  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.DISKSTORE_ALL.equals(optionContext)) {
    Set<String> diskStoreNames = getDiskStoreNames();
    
    for (String diskStoreName : diskStoreNames) {
      if (existingData != null) {
        if (diskStoreName.startsWith(existingData)) {
          completions.add(new Completion(diskStoreName));
        }
      } else {
        completions.add(new Completion(diskStoreName));
      }
    }
  }
  
  return !completions.isEmpty();
}
 

private void assertAdvancedCompletionValues(List<Completion> expected,
    List<Completion> actual) {
  // TODO Auto-generated method stub
  assertEquals("Check size", expected.size(), actual.size());
  for (int i = 0; i < expected.size(); i++) {
    assertEquals("Check completion value no." + i+". Expected("+expected.get(i)+") & Actual("+actual.get(i)+").",
        expected.get(i).getValue(), actual.get(i).getValue());
    if (expected.get(i).getFormattedValue() != null) {
      assertEquals("Check completion formatted value no." + i +". Expected("+expected.get(i).getFormattedValue()+") & Actual("+actual.get(i).getFormattedValue()+").", expected
          .get(i).getFormattedValue(), actual.get(i).getFormattedValue());
    }
  }
}
 

protected void populate(final List<Completion> completions,
    final String adjustedUserInput, final String originalUserInput,
    final String directoryData) {
  File directory = new File(directoryData);

  if (!directory.isDirectory()) {
    return;
  }

  for (File file : directory.listFiles()) {
    if (adjustedUserInput == null
        || adjustedUserInput.length() == 0
        || file.getName().toLowerCase()
            .startsWith(adjustedUserInput.toLowerCase())) {

      String completion = "";
      if (directoryData.length() > 0)
        completion += directoryData;
      completion += file.getName();

      completion = convertCompletionBackIntoUserInputStyle(originalUserInput,
          completion);

      if (file.isDirectory()) {
        completions.add(new Completion(completion + File.separator));
      }
    }
  }
}
 

private void modifyCompletionCandidates(
    List<Completion> completionCandidates, String prefix, boolean endsWithValueSeparator,
    String... existingData) {
  List<Completion> temp = new ArrayList<Completion>();
  while (completionCandidates.size() > 0) {
    temp.add(completionCandidates.remove(0));
  }
  for (Completion completion : temp) {
    boolean includeCompletion = true;
    String value = completion.getValue();
    if (existingData != null) {
      for (String string : existingData) {
        if (string != null) {
          // Check whether that value matches any of the
          // existingData
          // If it matches any one of existing data then we do not
          // need to include it in the list of completion
          // candidates
          if (value.equals(string)) {
            includeCompletion = false;
          }
        }
      }
      if (includeCompletion) {
        if (existingData[existingData.length - 1] != null
            && (!value.startsWith(existingData[existingData.length - 1])
                && !endsWithValueSeparator)) {
          includeCompletion = false;
        }
      }
    }
    if (includeCompletion) {
      // Also we only need to check with the last string of
      // existingData
      // whether the completion value starts with it.
      completionCandidates.add(new Completion(prefix + completion.getValue(),
          completion.getValue(), "", 0));
    }
  }
}
 

private boolean perfectMatch(List<Completion> completionCandidates,
    String... argumentValue) {
  // Here only the last value should match one of the
  // completionCandidates
  if (argumentValue.length > 0) {
    for (Completion completion : completionCandidates) {
      if (completion.getValue().equals(
          argumentValue[argumentValue.length - 1])) {
        return true;
      }
    }
  }
  return false;
}
 

@SuppressWarnings("unused")
private void updateCompletionCandidates(
    List<Completion> completionCandidates, String buffer, int position) {
  List<Completion> temp = new ArrayList<Completion>();
  while (completionCandidates.size() > 0) {
    temp.add(completionCandidates.remove(0));
  }
  for (Completion completion : temp) {
    completionCandidates.add(new Completion(buffer.substring(0, position)
        + completion.getValue(), completion.getFormattedValue(), completion
        .getHeading(), completion.getOrder()));
  }
}
 

/**
 * Populates a list of completion candidates. See
 * {@link Parser#complete(String, int, List)} for details.
 *
 * @param buffer
 * @param cursor
 * @param completionCandidates
 * @return new cursor position
 */
public int complete(String buffer, int cursor,
    List<String> completionCandidates) {
  final List<Completion> candidates = new ArrayList<Completion>();
  final int result = completeAdvanced(buffer, cursor, candidates);
  for (final Completion completion : candidates) {
    completionCandidates.add(completion.getValue());
  }
  return result;
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.LOCATOR_MEMBER_IDNAME.equals(optionContext)) {
    Set<String> locatorIdsAndNames = getLocatorIdAndNames();

    for (String string : locatorIdsAndNames) {
      completions.add(new Completion(string));
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.ALL_MEMBER_IDNAME.equals(optionContext)) {
    Set<String> memberIdAndNames = getMemberIdAndNames();

    for (String string : memberIdAndNames) {
      completions.add(new Completion(string));
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions, Class<?> targetType, String existingData,
		String optionContext, MethodTarget target) {
	for (AppRegistrationResource app : dataFlowShell.getDataFlowOperations().appRegistryOperations().list()) {
		String value = app.getType() + ":" + app.getName();
		completions.add(new Completion(value, app.getName(), pretty(app.getType()), 0));
	}
	return true;

}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.GATEWAY_RECEIVER_ID.equals(optionContext)) {
    Set<String> gatewaySenderIds = getGatewayRecieverIds();

    for (String gatewaySenderId : gatewaySenderIds) {
      completions.add(new Completion(gatewaySenderId));
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.LOCATOR_DISCOVERY_CONFIG.equals(optionContext)) {
    Set<String> locatorIdsAndNames = getLocatorIdAndNames();

    for (String string : locatorIdsAndNames) {
      completions.add(new Completion(string));
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.GATEWAY_SENDER_ID.equals(optionContext)) {
    Set<String> gatewaySenderIds = getGatewaySenderIds();
    
    for (String gatewaySenderId : gatewaySenderIds) {
      completions.add(new Completion(gatewaySenderId));
    }
  }
  
  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.HINTTOPIC.equals(optionContext)) {
    CommandManager commandManager = CommandManager.getExisting();
    if (commandManager != null) {
      Set<String> topicNames = commandManager.getTopicNames();
      
      for (String topicName : topicNames) {
        if (existingData != null && !existingData.isEmpty()) {
          if (topicName.startsWith(existingData)) { // match exact case
            completions.add(new Completion(topicName));
          } else if (topicName.toLowerCase().startsWith(existingData.toLowerCase())) {  // match case insensitive
            String completionStr = existingData + topicName.substring(existingData.length());
            
            completions.add(new Completion(completionStr));
          }
        } else {
          completions.add(new Completion(topicName));
        }
      }
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  completions.addAll(logLevels);
  return !completions.isEmpty();
}
 

public LogLevelConverter() {
  logLevels = new LinkedHashSet<Completion>();
  int[] alllevels = LogWriterImpl.allLevels;
  for (int level : alllevels) {
    logLevels.add(new Completion(LogWriterImpl.levelToString(level)));
  }
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.MEMBERIDNAME.equals(optionContext)) {
    Set<String> memberIdAndNames = getMemberIdAndNames();

    for (String string : memberIdAndNames) {
      completions.add(new Completion(string));
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (ConnectionEndpoint.JMXMANAGER_OPTION_CONTEXT.equals(optionContext)) {
    completions.add(new Completion(DEFAULT_JMX_ENDPOINTS));
  } else if (ConnectionEndpoint.LOCATOR_OPTION_CONTEXT.equals(optionContext)) {
    completions.add(new Completion(DEFAULT_LOCATOR_ENDPOINTS));
  }
  
  return completions.size() > 0;
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  
  if (String.class.equals(targetType) && ConverterHint.INDEX_TYPE.equals(optionContext)) {
    completions.add(new Completion("range"));
    completions.add(new Completion("key"));
    completions.add(new Completion("hash"));
  }
  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions,
    Class<?> targetType, String existingData, String optionContext,
    MethodTarget target) {
  if (String.class.equals(targetType) && ConverterHint.REGIONPATH.equals(optionContext)) {
    Set<String> regionPathSet = getAllRegionPaths();
    Gfsh gfsh = Gfsh.getCurrentInstance();
    String currentContextPath = "";
    if (gfsh != null) {
      currentContextPath = gfsh.getEnvProperty(Gfsh.ENV_APP_CONTEXT_PATH);
      if (currentContextPath != null && !com.gemstone.gemfire.management.internal.cli.converters.RegionPathConverter.DEFAULT_APP_CONTEXT_PATH.equals(currentContextPath)) {
        regionPathSet.remove(currentContextPath);
        regionPathSet.add(com.gemstone.gemfire.management.internal.cli.converters.RegionPathConverter.DEFAULT_APP_CONTEXT_PATH);
      }
    }

    for (String regionPath : regionPathSet) {
      if (existingData != null) {
        if (regionPath.startsWith(existingData)) {
          completions.add(new Completion(regionPath));
        }
      } else {
        completions.add(new Completion(regionPath));
      }
    }
  }

  return !completions.isEmpty();
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions, Class<?> targetType, String existingData,
		String optionContext, MethodTarget target) {
	String start = (existingData.startsWith("'") || existingData.startsWith("\"")) ? existingData.substring(1)
			: existingData;

	try {
		int successiveInvocations = determineNumberOfInvocations(optionContext);
		String kind = completionKind(optionContext);
		CompletionProposalsResource candidates;
		switch (kind) {
		case "stream":
			candidates = completionOperations().streamCompletions(start, successiveInvocations);
			break;
		case "task":
			candidates = completionOperations().taskCompletions(start, successiveInvocations);
			break;
		default:
			throw new IllegalArgumentException("Unsupported completion kind: " + kind);
		}
		for (CompletionProposalsResource.Proposal candidate : candidates.getProposals()) {
			completions.add(new Completion(candidate.getText()));
		}
		return false;
	}
	// Protect from exception in non-command code
	catch (Exception e) {
		return false;
	}
}
 

@Override
public boolean getAllPossibleValues(List<Completion> completions, Class<?> targetType, String existingData,
		String optionContext, MethodTarget target) {
	for (AppRegistrationResource app : dataFlowShell.getDataFlowOperations().appRegistryOperations().list()) {
		String value = app.getType() + ":" + app.getName();
		completions.add(new Completion(value, app.getName(), pretty(app.getType()), 0));
	}
	return true;

}
 

private void assertAdvancedCompletionValues(List<Completion> expected,
    List<Completion> actual) {
  // TODO Auto-generated method stub
  assertEquals("Check size", expected.size(), actual.size());
  for (int i = 0; i < expected.size(); i++) {
    assertEquals("Check completion value no." + i+". Expected("+expected.get(i)+") & Actual("+actual.get(i)+").",
        expected.get(i).getValue(), actual.get(i).getValue());
    if (expected.get(i).getFormattedValue() != null) {
      assertEquals("Check completion formatted value no." + i +". Expected("+expected.get(i).getFormattedValue()+") & Actual("+actual.get(i).getFormattedValue()+").", expected
          .get(i).getFormattedValue(), actual.get(i).getFormattedValue());
    }
  }
}
 

private void modifyCompletionCandidates(
    List<Completion> completionCandidates, String prefix, boolean endsWithValueSeparator,
    String... existingData) {
  List<Completion> temp = new ArrayList<Completion>();
  while (completionCandidates.size() > 0) {
    temp.add(completionCandidates.remove(0));
  }
  for (Completion completion : temp) {
    boolean includeCompletion = true;
    String value = completion.getValue();
    if (existingData != null) {
      for (String string : existingData) {
        if (string != null) {
          // Check whether that value matches any of the
          // existingData
          // If it matches any one of existing data then we do not
          // need to include it in the list of completion
          // candidates
          if (value.equals(string)) {
            includeCompletion = false;
          }
        }
      }
      if (includeCompletion) {
        if (existingData[existingData.length - 1] != null
            && (!value.startsWith(existingData[existingData.length - 1])
                && !endsWithValueSeparator)) {
          includeCompletion = false;
        }
      }
    }
    if (includeCompletion) {
      // Also we only need to check with the last string of
      // existingData
      // whether the completion value starts with it.
      completionCandidates.add(new Completion(prefix + completion.getValue(),
          completion.getValue(), "", 0));
    }
  }
}