}
@Override
public void execute(final Control<BootstrapContext> control) {
final BootstrapContext context = control.getContext();
final ModelNode operation = new ModelNode();
operation.get(OP).set(COMPOSITE);
operation.get(ADDRESS).setEmptyList();
ModelNode step;
List<ModelNode> steps = new ArrayList<ModelNode>();
// exec type
step = new ModelNode();
step.get(OP).set(READ_ATTRIBUTE_OPERATION);
step.get(NAME).set("process-type");
step.get(ADDRESS).setEmptyList();
steps.add(step);
// product name
step = new ModelNode();
step.get(OP).set(READ_ATTRIBUTE_OPERATION);
step.get(NAME).set("product-name");
step.get(ADDRESS).setEmptyList();
steps.add(step);
// release codename
step = new ModelNode();
step.get(OP).set(READ_ATTRIBUTE_OPERATION);
step.get(NAME).set("release-codename");
step.get(ADDRESS).setEmptyList();
steps.add(step);
// product version
step = new ModelNode();
step.get(OP).set(READ_ATTRIBUTE_OPERATION);
step.get(NAME).set("product-version");
step.get(ADDRESS).setEmptyList();
steps.add(step);
// release version
step = new ModelNode();
step.get(OP).set(READ_ATTRIBUTE_OPERATION);
step.get(NAME).set("release-version");
step.get(ADDRESS).setEmptyList();
steps.add(step);
// whoami
step = new ModelNode();
step.get(OP).set("whoami");
step.get(ADDRESS).setEmptyList();
step.get("verbose").set(true);
steps.add(step);
operation.get(STEPS).set(steps);
dispatcher.execute(new DMRAction(operation), new AsyncCallback<DMRResponse>() {
@Override
public void onFailure(Throwable caught) {
context.setlastError(caught);
Log.error(caught.getMessage());
control.abort();
}
@Override
public void onSuccess(DMRResponse result) {
ModelNode response = result.get();
if(response.isFailure()) {
context.setlastError(new RuntimeException(response.getFailureDescription()));
control.abort();
} else {
// capture exec mode
ModelNode execMode = response.get(RESULT).get("step-1");
boolean isServer = execMode.get(RESULT).asString().equals("Server");
context.setProperty(BootstrapContext.STANDALONE, Boolean.valueOf(isServer).toString());
// product name, release codename
ModelNode productName = response.get(RESULT).get("step-2");
ModelNode releaseCodename = response.get(RESULT).get("step-3");
if (productName.get(RESULT).isDefined()) {
context.setProductName(productName.get(RESULT).asString());
} else if (releaseCodename.get(RESULT).isDefined()) {
context.setProductName(releaseCodename.get(RESULT).asString());
}
// product version, release version
ModelNode productVersion = response.get(RESULT).get("step-4");
ModelNode releaseVersion = response.get(RESULT).get("step-5");
if (productVersion.get(RESULT).isDefined()) {
context.setProductVersion(productVersion.get(RESULT).asString());
} else if (releaseVersion.get(RESULT).isDefined()) {
context.setProductVersion(releaseVersion.get(RESULT).asString());
}
ModelNode whoami = response.get(RESULT).get("step-6");
ModelNode whoamiResult = whoami.get(RESULT);
String username = whoamiResult.get("identity").get("username").asString();
context.setPrincipal(username);
Set<String> mappedRoles = new HashSet<String>();
if(whoamiResult.hasDefined("mapped-roles"))
{
List<ModelNode> roles = whoamiResult.get("mapped-roles").asList();
for(ModelNode role : roles)
{
final String roleName = role.asString();
mappedRoles.add(roleName);
}
}
context.setRoles(mappedRoles);
if(context.isSuperUser() && Preferences.has(Preferences.Key.RUN_AS_ROLE))
{
String runAsRole = Preferences.get(Preferences.Key.RUN_AS_ROLE);
dispatcher.setProperty("run_as", runAsRole);
context.setRunAs(runAsRole);
}
Preferences.clear(Preferences.Key.RUN_AS_ROLE);
control.proceed();
}