/*
|
* 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
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE
|
* or https://OpenDS.dev.java.net/OpenDS.LICENSE.
|
* 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
|
* trunk/opends/resource/legal-notices/OpenDS.LICENSE. 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
|
*
|
*
|
* Portions Copyright 2006-2007 Sun Microsystems, Inc.
|
*/
|
package org.opends.server.admin.client.cli;
|
|
import static org.opends.server.messages.AdminMessages.*;
|
import static org.opends.server.tools.ToolConstants.*;
|
|
import java.io.IOException;
|
import java.io.OutputStream;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.HashSet;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
|
import javax.naming.NamingException;
|
import javax.naming.ldap.InitialLdapContext;
|
import javax.naming.ldap.Rdn;
|
|
|
import org.opends.admin.ads.ADSContext;
|
import org.opends.admin.ads.ADSContextException;
|
import org.opends.admin.ads.ADSContext.ServerGroupProperty;
|
import org.opends.admin.ads.ADSContext.ServerProperty;
|
import org.opends.server.admin.client.cli.DsFrameworkCliReturnCode.ReturnCode;
|
import org.opends.server.util.args.ArgumentException;
|
import org.opends.server.util.args.BooleanArgument;
|
import org.opends.server.util.args.StringArgument;
|
import org.opends.server.util.args.SubCommand;
|
|
/**
|
* This class is handling server group CLI.
|
*/
|
public class DsFrameworkCliServerGroup implements DsFrameworkCliSubCommandGroup
|
{
|
|
/**
|
* End Of Line.
|
*/
|
private String EOL = System.getProperty("line.separator");
|
|
/**
|
* The subcommand Parser.
|
*/
|
DsFrameworkCliParser argParser ;
|
|
/**
|
* The verbose argument.
|
*/
|
BooleanArgument verboseArg ;
|
|
/**
|
* The enumeration containing the different subCommand names.
|
*/
|
private enum SubCommandNameEnum
|
{
|
/**
|
* The create-group subcommand.
|
*/
|
CREATE_GROUP("create-group"),
|
|
/**
|
* The delete-group subcommand.
|
*/
|
DELETE_GROUP("delete-group"),
|
|
/**
|
* The modify-group subcommand.
|
*/
|
MODIFY_GROUP("modify-group"),
|
|
/**
|
* The list-groups subcommand.
|
*/
|
LIST_GROUPS("list-groups"),
|
|
/**
|
* The list-members subcommand.
|
*/
|
LIST_MEMBERS("list-members"),
|
|
/**
|
* The list-membership subcommand.
|
*/
|
LIST_MEMBERSHIP("list-membership"),
|
|
/**
|
* The add-to-group subcommand.
|
*/
|
ADD_TO_GROUP("add-to-group"),
|
|
/**
|
* The remove-from-group subcommand.
|
*/
|
REMOVE_FROM_GROUP("remove-from-group");
|
|
// String representation of the value.
|
private final String name;
|
|
// Private constructor.
|
private SubCommandNameEnum(String name)
|
{
|
this.name = name;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
public String toString()
|
{
|
return name;
|
}
|
|
// A lookup table for resolving a unit from its name.
|
private static final List<String> nameToSubCmdName ;
|
static
|
{
|
nameToSubCmdName = new ArrayList<String>();
|
|
for (SubCommandNameEnum subCmd : SubCommandNameEnum.values())
|
{
|
nameToSubCmdName.add(subCmd.toString());
|
}
|
}
|
public static boolean isSubCommand(String name)
|
{
|
return nameToSubCmdName.contains(name);
|
}
|
}
|
|
/**
|
* The 'create-group' subcommand.
|
*/
|
public SubCommand createGroupSubCmd;
|
|
/**
|
* The 'description' argument of the 'create-group' subcommand.
|
*/
|
private StringArgument createGroupDescriptionArg;
|
|
/**
|
* The 'group-name' argument of the 'create-group' subcommand.
|
*/
|
private StringArgument createGroupGroupNameArg;
|
|
/**
|
* The 'modify-group' subcommand.
|
*/
|
private SubCommand modifyGroupSubCmd;
|
|
/**
|
* The 'description' argument of the 'modify-group' subcommand.
|
*/
|
private StringArgument modifyGroupDescriptionArg;
|
|
/**
|
* The 'group-id' argument of the 'modify-group' subcommand.
|
*/
|
private StringArgument modifyGroupGroupIdArg;
|
|
/**
|
* The 'group-name' argument of the 'modify-group' subcommand.
|
*/
|
private StringArgument modifyGroupGroupNameArg;
|
|
/**
|
* The 'delete-group' subcommand.
|
*/
|
private SubCommand deleteGroupSubCmd;
|
|
/**
|
* The 'group-name' argument of the 'delete-group' subcommand.
|
*/
|
private StringArgument deleteGroupGroupNameArg;
|
|
/**
|
* The 'list-group' subcommand.
|
*/
|
private SubCommand listGroupSubCmd;
|
|
/**
|
* The 'add-to-group' subcommand.
|
*/
|
private SubCommand addToGroupSubCmd;
|
|
/**
|
* The 'group-name' argument of the 'add-to-group' subcommand.
|
*/
|
private StringArgument addToGroupGroupNameArg;
|
|
/**
|
* The 'member-id' argument of the 'add-to-group' subcommand.
|
*/
|
private StringArgument addToGoupMemberNameArg;
|
|
/**
|
* The 'remove-from-group' subcommand.
|
*/
|
private SubCommand removeFromGroupSubCmd;
|
|
/**
|
* The 'group-name' argument of the 'remove-from-group' subcommand.
|
*/
|
private StringArgument removeFromGroupGroupNameArg;
|
|
/**
|
* The 'member-id' argument of the 'remove-from-group' subcommand.
|
*/
|
private StringArgument removeFromGoupMemberNameArg;
|
|
/**
|
* The 'list-members' subcommand.
|
*/
|
private SubCommand listMembersSubCmd;
|
|
/**
|
* The 'group-name' argument of the 'list-members' subcommand.
|
*/
|
private StringArgument listMembersGroupNameArg;
|
|
/**
|
* The 'mlist-membership' subcommand.
|
*/
|
private SubCommand listMembershipSubCmd;
|
|
/**
|
* The 'meber-name' argument of the 'list-membership' subcommand.
|
*/
|
private StringArgument listMembershipMemberNameArg;
|
|
|
/**
|
* Association between ADSContext enum and display field.
|
*/
|
private HashMap<ServerGroupProperty, String> attributeDisplayName;
|
|
/**
|
* The subcommand list.
|
*/
|
private HashSet<SubCommand> subCommands = new HashSet<SubCommand>();
|
|
/**
|
* Indicates whether this subCommand should be hidden in the usage
|
* information.
|
*/
|
private boolean isHidden;
|
|
/**
|
* The subcommand group name.
|
*/
|
private String groupName;
|
|
/**
|
* Get the display attribute name for a given attribute.
|
* @param prop The server prperty
|
* @return the display attribute name for a given attribute
|
*/
|
public String getAttributeDisplayName(ServerGroupProperty prop)
|
{
|
return attributeDisplayName.get(prop);
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
public Set<SubCommand> getSubCommands()
|
{
|
return subCommands;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
public boolean isHidden()
|
{
|
return isHidden;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
public String getGroupName()
|
{
|
return groupName ;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
public void initializeCliGroup(DsFrameworkCliParser argParser,
|
BooleanArgument verboseArg)
|
throws ArgumentException
|
{
|
this.verboseArg = verboseArg ;
|
isHidden = false ;
|
groupName = "server-group";
|
this.argParser = argParser;
|
|
|
// Create-group subcommand
|
createGroupSubCmd = new SubCommand(argParser,
|
SubCommandNameEnum.CREATE_GROUP.toString(),
|
MSGID_ADMIN_SUBCMD_CREATE_GROUP_DESCRIPTION);
|
subCommands.add(createGroupSubCmd);
|
|
createGroupDescriptionArg = new StringArgument("description",
|
OPTION_SHORT_DESCRIPTION, OPTION_LONG_DESCRIPTION, false, false,
|
true, OPTION_VALUE_DESCRIPTION, "", null,
|
MSGID_ADMIN_ARG_DESCRIPTION_DESCRIPTION);
|
createGroupSubCmd.addArgument(createGroupDescriptionArg);
|
|
createGroupGroupNameArg = new StringArgument("groupName",
|
OPTION_SHORT_GROUPNAME, OPTION_LONG_GROUPNAME, true, true,
|
OPTION_VALUE_GROUPNAME,
|
MSGID_ADMIN_ARG_CREATE_GROUP_GROUPNAME_DESCRIPTION);
|
createGroupSubCmd.addArgument(createGroupGroupNameArg);
|
|
// modify-group
|
modifyGroupSubCmd = new SubCommand(argParser,
|
SubCommandNameEnum.MODIFY_GROUP.toString(),
|
MSGID_ADMIN_SUBCMD_MODIFY_GROUP_DESCRIPTION);
|
subCommands.add(modifyGroupSubCmd);
|
|
modifyGroupDescriptionArg = new StringArgument("new-description",
|
OPTION_SHORT_DESCRIPTION, OPTION_LONG_DESCRIPTION, false, false,
|
true, OPTION_VALUE_DESCRIPTION, "", null,
|
MSGID_ADMIN_ARG_NEW_DESCRIPTION_DESCRIPTION);
|
modifyGroupSubCmd.addArgument(modifyGroupDescriptionArg);
|
|
modifyGroupGroupIdArg = new StringArgument("new-groupName",
|
OPTION_SHORT_NEWGROUPNAME, OPTION_LONG_NEWGROUPNAME, false, false, true,
|
OPTION_VALUE_GROUPNAME, "", null,
|
MSGID_ADMIN_ARG_NEW_GROUPNAME_DESCRIPTION);
|
modifyGroupSubCmd.addArgument(modifyGroupGroupIdArg);
|
|
modifyGroupGroupNameArg = new StringArgument("groupName",
|
OPTION_SHORT_GROUPNAME, OPTION_LONG_GROUPNAME, true, true,
|
OPTION_VALUE_GROUPNAME,
|
MSGID_ADMIN_ARG_GROUPNAME_DESCRIPTION);
|
modifyGroupSubCmd.addArgument(modifyGroupGroupNameArg);
|
|
// delete-group
|
deleteGroupSubCmd = new SubCommand(argParser,SubCommandNameEnum.DELETE_GROUP
|
.toString(), MSGID_ADMIN_SUBCMD_DELETE_GROUP_DESCRIPTION);
|
subCommands.add(deleteGroupSubCmd);
|
|
deleteGroupGroupNameArg = new StringArgument("groupName",
|
OPTION_SHORT_GROUPNAME, OPTION_LONG_GROUPNAME, true, true,
|
OPTION_VALUE_GROUPNAME,
|
MSGID_ADMIN_ARG_GROUPNAME_DESCRIPTION);
|
deleteGroupSubCmd.addArgument(deleteGroupGroupNameArg);
|
|
// list-groups
|
listGroupSubCmd = new SubCommand(argParser, "list-groups",
|
MSGID_ADMIN_SUBCMD_LIST_GROUPS_DESCRIPTION);
|
subCommands.add(listGroupSubCmd);
|
|
// add-to-group
|
addToGroupSubCmd = new SubCommand(argParser,
|
SubCommandNameEnum.ADD_TO_GROUP.toString(),
|
MSGID_ADMIN_SUBCMD_ADD_TO_GROUP_DESCRIPTION);
|
subCommands.add(addToGroupSubCmd);
|
|
addToGoupMemberNameArg = new StringArgument("memberName",
|
OPTION_SHORT_MEMBERNAME, OPTION_LONG_MEMBERNAME, true, true,
|
OPTION_VALUE_MEMBERNAME,
|
MSGID_ADMIN_ARG_ADD_MEMBERNAME_DESCRIPTION);
|
addToGroupSubCmd.addArgument(addToGoupMemberNameArg);
|
|
addToGroupGroupNameArg = new StringArgument("groupName",
|
OPTION_SHORT_GROUPNAME, OPTION_LONG_GROUPNAME, true, true,
|
OPTION_VALUE_GROUPNAME,
|
MSGID_ADMIN_ARG_GROUPNAME_DESCRIPTION);
|
addToGroupSubCmd.addArgument(addToGroupGroupNameArg);
|
|
// remove-from-group
|
removeFromGroupSubCmd = new SubCommand(argParser,
|
SubCommandNameEnum.REMOVE_FROM_GROUP.toString(),
|
MSGID_ADMIN_SUBCMD_REMOVE_FROM_GROUP_DESCRIPTION);
|
subCommands.add(removeFromGroupSubCmd);
|
|
removeFromGoupMemberNameArg = new StringArgument("memberName",
|
OPTION_SHORT_MEMBERNAME, OPTION_LONG_MEMBERNAME, true, true,
|
OPTION_VALUE_MEMBERNAME,
|
MSGID_ADMIN_ARG_REMOVE_MEMBERNAME_DESCRIPTION);
|
removeFromGroupSubCmd.addArgument(removeFromGoupMemberNameArg);
|
|
removeFromGroupGroupNameArg = new StringArgument("groupName",
|
OPTION_SHORT_GROUPNAME, OPTION_LONG_GROUPNAME, true, true,
|
OPTION_VALUE_GROUPNAME,
|
MSGID_ADMIN_ARG_GROUPNAME_DESCRIPTION);
|
removeFromGroupSubCmd.addArgument(removeFromGroupGroupNameArg);
|
|
|
// list-members
|
listMembersSubCmd = new SubCommand(argParser,SubCommandNameEnum.LIST_MEMBERS
|
.toString(), MSGID_ADMIN_SUBCMD_LIST_MEMBERS_DESCRIPTION);
|
subCommands.add(listMembersSubCmd);
|
|
listMembersGroupNameArg = new StringArgument("groupName",
|
OPTION_SHORT_GROUPNAME, OPTION_LONG_GROUPNAME, true, true,
|
OPTION_VALUE_GROUPNAME,
|
MSGID_ADMIN_ARG_GROUPNAME_DESCRIPTION);
|
listMembersSubCmd.addArgument(listMembersGroupNameArg);
|
|
// list-membership
|
listMembershipSubCmd = new SubCommand(argParser,
|
SubCommandNameEnum.LIST_MEMBERSHIP.toString(),
|
MSGID_ADMIN_SUBCMD_LIST_MEMBERSHIP_DESCRIPTION);
|
subCommands.add(listMembershipSubCmd);
|
|
listMembershipMemberNameArg = new StringArgument("memberName",
|
OPTION_SHORT_MEMBERNAME, OPTION_LONG_MEMBERNAME, true, true,
|
OPTION_VALUE_MEMBERNAME,
|
MSGID_ADMIN_ARG_MEMBERNAME_DESCRIPTION);
|
listMembershipSubCmd.addArgument(listMembershipMemberNameArg);
|
|
// Create association between ADSContext enum and display field
|
attributeDisplayName = new HashMap<ServerGroupProperty, String>();
|
attributeDisplayName.put(ServerGroupProperty.UID, OPTION_LONG_GROUPNAME);
|
attributeDisplayName.put(ServerGroupProperty.DESCRIPTION,
|
OPTION_LONG_DESCRIPTION);
|
attributeDisplayName.put(ServerGroupProperty.MEMBERS,
|
OPTION_LONG_MEMBERNAME);
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
public boolean isSubCommand(SubCommand subCmd)
|
{
|
return SubCommandNameEnum.isSubCommand(subCmd.getName());
|
}
|
|
|
/**
|
* {@inheritDoc}
|
*/
|
public ReturnCode performSubCommand(SubCommand subCmd, OutputStream outStream,
|
OutputStream errStream)
|
throws ADSContextException, ArgumentException
|
{
|
ADSContext adsCtx = null ;
|
InitialLdapContext ctx = null ;
|
|
ReturnCode returnCode = ReturnCode.ERROR_UNEXPECTED;
|
try
|
{
|
// -----------------------
|
// create-group subcommand
|
// -----------------------
|
if (subCmd.getName().equals(createGroupSubCmd.getName()))
|
{
|
String groupId = createGroupGroupNameArg.getValue();
|
HashMap<ServerGroupProperty, Object> serverGroupProperties =
|
new HashMap<ServerGroupProperty, Object>();
|
|
// get the GROUP_NAME
|
serverGroupProperties.put(ServerGroupProperty.UID, groupId);
|
|
// get the Description
|
if (createGroupDescriptionArg.isPresent())
|
{
|
serverGroupProperties.put(ServerGroupProperty.DESCRIPTION,
|
createGroupDescriptionArg.getValue());
|
}
|
|
// Create the group
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
adsCtx.createServerGroup(serverGroupProperties);
|
returnCode = ReturnCode.SUCCESSFUL;
|
}
|
// -----------------------
|
// delete-group subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(deleteGroupSubCmd.getName()))
|
{
|
String groupId = deleteGroupGroupNameArg.getValue();
|
HashMap<ServerGroupProperty, Object> serverGroupProperties =
|
new HashMap<ServerGroupProperty, Object>();
|
|
// get the GROUP_ID
|
serverGroupProperties.put(ServerGroupProperty.UID, groupId);
|
|
// Delete the group
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
adsCtx.deleteServerGroup(serverGroupProperties);
|
returnCode = ReturnCode.SUCCESSFUL;
|
}
|
// -----------------------
|
// list-groups subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(listGroupSubCmd.getName()))
|
{
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
|
Set<Map<ServerGroupProperty, Object>> result = adsCtx
|
.readServerGroupRegistry();
|
StringBuffer buffer = new StringBuffer();
|
|
// if not verbose mode, print group name (1 per line)
|
if (! verboseArg.isPresent())
|
{
|
for (Map<ServerGroupProperty, Object> groupProps : result)
|
{
|
// Get the group name
|
buffer.append(groupProps.get(ServerGroupProperty.UID));
|
buffer.append(EOL);
|
}
|
}
|
else
|
{
|
// Look for the max group identifier length
|
int uidLength = 0 ;
|
for (ServerGroupProperty sgp : ServerGroupProperty.values())
|
{
|
int cur = attributeDisplayName.get(sgp).toString().length();
|
if (cur > uidLength)
|
{
|
uidLength = cur;
|
}
|
}
|
uidLength++;
|
|
for (Map<ServerGroupProperty, Object> groupProps : result)
|
{
|
// Get the group name
|
buffer.append(attributeDisplayName.get(ServerGroupProperty.UID));
|
// add space
|
int curLen = attributeDisplayName.get(ServerGroupProperty.UID)
|
.length();
|
for (int i = curLen; i < uidLength; i++)
|
{
|
buffer.append(" ");
|
}
|
buffer.append(": ");
|
buffer.append(groupProps.get(ServerGroupProperty.UID));
|
buffer.append(EOL);
|
|
// Write other props
|
for (ServerGroupProperty propName : ServerGroupProperty.values())
|
{
|
if (propName.compareTo(ServerGroupProperty.UID) == 0)
|
{
|
// We have already displayed the group Id
|
continue;
|
}
|
buffer.append(attributeDisplayName.get(propName));
|
// add space
|
curLen = attributeDisplayName.get(propName).length();
|
for (int i = curLen; i < uidLength; i++)
|
{
|
buffer.append(" ");
|
}
|
buffer.append(": ");
|
|
if (propName.compareTo(ServerGroupProperty.MEMBERS) == 0)
|
{
|
Set atts = (Set) groupProps.get(propName);
|
if (atts != null)
|
{
|
boolean indent = false;
|
for (Object att : atts)
|
{
|
if (indent)
|
{
|
buffer.append(EOL);
|
for (int i = 0; i < uidLength + 2; i++)
|
{
|
buffer.append(" ");
|
}
|
}
|
else
|
{
|
indent = true;
|
}
|
buffer.append(att.toString().substring(3));
|
}
|
}
|
}
|
else
|
{
|
if (groupProps.get(propName) != null)
|
{
|
buffer.append(groupProps.get(propName));
|
}
|
}
|
buffer.append(EOL);
|
}
|
buffer.append(EOL);
|
}
|
}
|
try
|
{
|
outStream.write(buffer.toString().getBytes());
|
}
|
catch (IOException e)
|
{
|
}
|
returnCode = ReturnCode.SUCCESSFUL;
|
}
|
// -----------------------
|
// modify-group subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(modifyGroupSubCmd.getName()))
|
{
|
String groupId = modifyGroupGroupNameArg.getValue();
|
HashMap<ServerGroupProperty, Object> serverGroupProperties =
|
new HashMap<ServerGroupProperty, Object>();
|
HashSet<ServerGroupProperty> serverGroupPropertiesToRemove =
|
new HashSet<ServerGroupProperty>();
|
|
Boolean updateRequired = false;
|
Boolean removeRequired = false;
|
// get the GROUP_ID
|
if (modifyGroupGroupIdArg.isPresent())
|
{
|
// rename the entry !
|
serverGroupProperties.put(ServerGroupProperty.UID,
|
modifyGroupGroupIdArg.getValue());
|
updateRequired = true;
|
}
|
else
|
{
|
serverGroupProperties.put(ServerGroupProperty.UID, groupId) ;
|
}
|
|
|
// get the Description
|
if (modifyGroupDescriptionArg.isPresent())
|
{
|
String newDesc = modifyGroupDescriptionArg.getValue();
|
if (newDesc.length() == 0)
|
{
|
serverGroupPropertiesToRemove.add(ServerGroupProperty.DESCRIPTION);
|
removeRequired = true;
|
}
|
else
|
{
|
serverGroupProperties.put(ServerGroupProperty.DESCRIPTION,
|
modifyGroupDescriptionArg.getValue());
|
updateRequired = true;
|
}
|
}
|
|
|
// Update the server group
|
if ( ! (updateRequired || removeRequired ) )
|
{
|
returnCode = ReturnCode.SUCCESSFUL_NOP;
|
}
|
|
// We need to perform an update
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
|
if (updateRequired)
|
{
|
adsCtx.updateServerGroup(groupId, serverGroupProperties);
|
}
|
if (removeRequired)
|
{
|
adsCtx.removeServerGroupProp(groupId,
|
serverGroupPropertiesToRemove);
|
}
|
|
returnCode = ReturnCode.SUCCESSFUL;
|
}
|
// -----------------------
|
// add-to-group subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(addToGroupSubCmd.getName()))
|
{
|
String groupId = addToGroupGroupNameArg.getValue();
|
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
|
// Check if the server is registered inside to ADS
|
Set<Map<ServerProperty, Object>> serverList = adsCtx
|
.readServerRegistry();
|
boolean found = false ;
|
Map<ServerProperty, Object> foundServerProperties = null ;
|
for (Map<ServerProperty, Object> serverProperties : serverList)
|
{
|
String serverId = ADSContext
|
.getServerIdFromServerProperties(serverProperties);
|
if (addToGoupMemberNameArg.getValue().equals(serverId))
|
{
|
found = true;
|
foundServerProperties = serverProperties ;
|
break;
|
}
|
}
|
if ( !found )
|
{
|
return ReturnCode.SERVER_NOT_REGISTERED;
|
}
|
|
// Add the server inside the group
|
returnCode = addServerTogroup(adsCtx, groupId, foundServerProperties);
|
}
|
// -----------------------
|
// remove-from-group subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(removeFromGroupSubCmd.getName()))
|
{
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
|
returnCode = removeServerFromGroup(adsCtx,
|
removeFromGroupGroupNameArg.getValue(),
|
removeFromGoupMemberNameArg.getValue());
|
}
|
// -----------------------
|
// list-members subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(listMembersSubCmd.getName()))
|
{
|
String groupId = listMembersGroupNameArg.getValue();
|
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
|
// get the current member list
|
Set<String> memberList = adsCtx.getServerGroupMemberList(groupId);
|
if (memberList == null)
|
{
|
returnCode = ReturnCode.SUCCESSFUL;
|
}
|
StringBuffer buffer = new StringBuffer();
|
for (String member : memberList)
|
{
|
buffer.append(member.substring(3));
|
buffer.append(EOL);
|
}
|
try
|
{
|
outStream.write(buffer.toString().getBytes());
|
}
|
catch (IOException e)
|
{
|
}
|
|
return ReturnCode.SUCCESSFUL;
|
}
|
// -----------------------
|
// list-membership subcommand
|
// -----------------------
|
else if (subCmd.getName().equals(listMembershipSubCmd.getName()))
|
{
|
|
ctx = argParser.getContext(outStream, errStream);
|
if (ctx == null)
|
{
|
return ReturnCode.CANNOT_CONNECT_TO_ADS;
|
}
|
adsCtx = new ADSContext(ctx) ;
|
|
Set<Map<ServerGroupProperty, Object>> result = adsCtx
|
.readServerGroupRegistry();
|
String MemberId = listMembershipMemberNameArg.getValue();
|
|
StringBuffer buffer = new StringBuffer();
|
for (Map<ServerGroupProperty, Object> groupProps : result)
|
{
|
// Get the group name;
|
String groupId = groupProps.get(ServerGroupProperty.UID).toString();
|
|
// look for memeber list attribute
|
for (ServerGroupProperty propName : groupProps.keySet())
|
{
|
if (propName.compareTo(ServerGroupProperty.MEMBERS) != 0)
|
{
|
continue;
|
}
|
// Check if the member list contains the member-id
|
Set atts = (Set) groupProps.get(propName);
|
for (Object att : atts)
|
{
|
if (att.toString().substring(3).toLowerCase().equals(
|
MemberId.toLowerCase()))
|
{
|
buffer.append(groupId);
|
buffer.append(EOL);
|
break;
|
}
|
}
|
break;
|
}
|
}
|
try
|
{
|
outStream.write(buffer.toString().getBytes());
|
}
|
catch (IOException e)
|
{
|
}
|
returnCode = ReturnCode.SUCCESSFUL;
|
}
|
else
|
{
|
// Should never occurs: If we are here, it means that the code to
|
// handle to subcommand is not yet written.
|
returnCode = ReturnCode.ERROR_UNEXPECTED;
|
}
|
}
|
catch (ADSContextException e)
|
{
|
if (ctx != null)
|
{
|
try
|
{
|
ctx.close();
|
}
|
catch (NamingException x)
|
{
|
}
|
}
|
throw e;
|
}
|
|
// Close the connection, if needed
|
if (ctx != null)
|
{
|
try
|
{
|
ctx.close();
|
}
|
catch (NamingException x)
|
{
|
}
|
}
|
|
// return part
|
return returnCode;
|
|
}
|
|
/**
|
* Remove a server from a group.
|
*
|
* @param adsCtx
|
* The ADS context to use.
|
* @param groupId
|
* The group identifier from which a server has to be
|
* remove.
|
* @param serverId
|
* The server identifier to be removed.
|
* @return The return code.
|
* @throws ADSContextException
|
* If there is a problem with any of the parameters used
|
* to create this argument.
|
*/
|
private ReturnCode removeServerFromGroup(ADSContext adsCtx, String groupId,
|
String serverId)
|
throws ADSContextException
|
{
|
ReturnCode returnCode = ReturnCode.SUCCESSFUL;
|
|
// get the current group member list
|
Set<String> memberList = adsCtx.getServerGroupMemberList(groupId);
|
if (memberList == null)
|
{
|
returnCode = ReturnCode.NOT_YET_REGISTERED;
|
}
|
String memberToRemove = "cn="
|
+ Rdn.escapeValue(serverId);
|
if (!memberList.contains(memberToRemove))
|
{
|
returnCode = ReturnCode.NOT_YET_REGISTERED;
|
}
|
|
memberList.remove(memberToRemove);
|
HashMap<ServerGroupProperty, Object> serverGroupProperties =
|
new HashMap<ServerGroupProperty, Object>();
|
serverGroupProperties.put(ServerGroupProperty.MEMBERS, memberList);
|
|
// Update the server group
|
adsCtx.updateServerGroup(groupId, serverGroupProperties);
|
|
// Update the server property "GROUPS"
|
Set<Map<ServerProperty,Object>> serverList = adsCtx.readServerRegistry() ;
|
boolean found = false;
|
Map<ServerProperty,Object> serverProperties = null ;
|
for (Map<ServerProperty,Object> elm : serverList)
|
{
|
if (serverId.equals(elm.get(ServerProperty.ID)))
|
{
|
found = true ;
|
serverProperties = elm ;
|
break ;
|
}
|
}
|
if ( ! found )
|
{
|
return returnCode.SERVER_NOT_REGISTERED ;
|
}
|
|
Set rawGroupList = (Set) serverProperties.get(ServerProperty.GROUPS);
|
Set<String> groupList = new HashSet<String>();
|
if (rawGroupList != null)
|
{
|
for (Object elm :rawGroupList.toArray() )
|
{
|
if (groupId.equals(elm))
|
{
|
continue ;
|
}
|
groupList.add(elm.toString());
|
}
|
}
|
serverProperties.put(ServerProperty.GROUPS, groupList);
|
adsCtx.updateServer(serverProperties, null);
|
|
return returnCode;
|
}
|
|
/**
|
* Add a server inside a group.
|
*
|
* @param adsCtx
|
* The ADS context to use.
|
* @param groupId
|
* The group identifier in which a server has to be added.
|
* @param map
|
* The properties of the server that have to be added to the
|
* group.
|
* @return the return code.
|
* @throws ADSContextException
|
* If there is a problem with any of the parameters used
|
* to create this argument.
|
*/
|
static ReturnCode addServerTogroup(ADSContext adsCtx, String groupId,
|
Map<ServerProperty, Object> map) throws ADSContextException
|
{
|
ReturnCode returnCode = ReturnCode.SUCCESSFUL;
|
String serverId = (String) map.get(ServerProperty.ID);
|
|
// Add the server inside the group
|
HashMap<ServerGroupProperty, Object> serverGroupProperties =
|
new HashMap<ServerGroupProperty, Object>();
|
Set<String> memberList = adsCtx.getServerGroupMemberList(groupId);
|
if (memberList == null)
|
{
|
memberList = new HashSet<String>();
|
}
|
String newMember = "cn="
|
+ Rdn.escapeValue(serverId);
|
if (memberList.contains(newMember))
|
{
|
returnCode = ReturnCode.ALREADY_REGISTERED;
|
}
|
memberList.add(newMember);
|
serverGroupProperties.put(ServerGroupProperty.MEMBERS, memberList);
|
|
adsCtx.updateServerGroup(groupId, serverGroupProperties);
|
|
|
// Update the server property "GROUPS"
|
Set rawGroupList = (Set) map.get(ServerProperty.GROUPS);
|
Set<String> groupList = new HashSet<String>();
|
if (rawGroupList != null)
|
{
|
for (Object elm :rawGroupList.toArray() )
|
{
|
groupList.add(elm.toString());
|
}
|
}
|
groupList.add(groupId) ;
|
map.put(ServerProperty.GROUPS, groupList);
|
adsCtx.updateServer(map, null);
|
|
return returnCode;
|
}
|
}
|
