protected static NodeOrRelationship getThingById(GraphDatabaseShellServer server, TypedId typedId)
throws ShellException {
NodeOrRelationship result = null;
if (typedId.isNode()) {
try {
result = NodeOrRelationship.wrap(server.getDb().getNodeById(typedId.getId()));
} catch (NotFoundException e) {
throw new ShellException("Node " + typedId.getId() + " not found");
}
} else {
try {
result = NodeOrRelationship.wrap(server.getDb().getRelationshipById(typedId.getId()));
} catch (NotFoundException e) {
throw new ShellException("Relationship " + typedId.getId() + " not found");
}
}
return result;
}
protected void cdTo(Session session, Node node) throws RemoteException, ShellException {
List<TypedId> wd = readCurrentWorkingDir(session);
try {
NodeOrRelationship current = getCurrent(session);
wd.add(getCurrent(session).getTypedId());
} catch (ShellException e) { // OK not found then
}
writeCurrentWorkingDir(wd, session);
setCurrent(session, NodeOrRelationship.wrap(node));
}
@Override
protected String exec(AppCommandParser parser, Session session, Output out)
throws ShellException, RemoteException {
List<TypedId> paths = readCurrentWorkingDir(session);
NodeOrRelationship current = getCurrent(session);
NodeOrRelationship newThing = null;
if (parser.arguments().isEmpty()) {
newThing = NodeOrRelationship.wrap(getServer().getDb().getReferenceNode());
paths.clear();
} else {
String arg = parser.arguments().get(0);
TypedId newId = current.getTypedId();
if (arg.equals("..")) {
if (paths.size() > 0) {
newId = paths.remove(paths.size() - 1);
}
} else if (arg.equals(".")) {
} else if (arg.equals(START_ALIAS) || arg.equals(END_ALIAS)) {
newId = getStartOrEnd(current, arg);
paths.add(current.getTypedId());
} else {
long suppliedId = -1;
try {
suppliedId = Long.parseLong(arg);
} catch (NumberFormatException e) {
suppliedId = findNodeWithTitle(current.asNode(), arg, session);
if (suppliedId == -1) {
throw new ShellException("No connected node with title '" + arg + "'");
}
}
newId =
parser.options().containsKey("r")
? new TypedId(NodeOrRelationship.TYPE_RELATIONSHIP, suppliedId)
: new TypedId(NodeOrRelationship.TYPE_NODE, suppliedId);
if (newId.equals(current.getTypedId())) {
throw new ShellException("Can't cd to where you stand");
}
boolean absolute = parser.options().containsKey("a");
if (!absolute && !this.isConnected(current, newId)) {
throw new ShellException(
getDisplayName(getServer(), session, newId, false)
+ " isn't connected to the current primitive,"
+ " use -a to force it to go there anyway");
}
paths.add(current.getTypedId());
}
newThing = this.getThingById(newId);
}
setCurrent(session, newThing);
writeCurrentWorkingDir(paths, session);
return null;
}
/**
* @param server the {@link GraphDatabaseShellServer} to run at.
* @param session the {@link Session} used by the client.
* @param node the {@link Node} to get a display string for.
* @return a display string for {@code node}.
*/
public static String getDisplayName(
GraphDatabaseShellServer server, Session session, Node node, boolean checkForMe)
throws ShellException {
if (checkForMe && isCurrent(session, NodeOrRelationship.wrap(node))) {
return getDisplayNameForCurrent(server, session);
}
String title = findTitle(server, session, node);
StringBuilder result = new StringBuilder("(");
result.append((title != null ? title + "," : ""));
result.append(node.getId());
result.append(")");
return result.toString();
}
private TypedId getStartOrEnd(NodeOrRelationship current, String arg) throws ShellException {
if (!current.isRelationship()) {
throw new ShellException("Only allowed on relationships");
}
Node newNode = null;
if (arg.equals(START_ALIAS)) {
newNode = current.asRelationship().getStartNode();
} else if (arg.equals(END_ALIAS)) {
newNode = current.asRelationship().getEndNode();
} else {
throw new ShellException("Unknown alias '" + arg + "'");
}
return NodeOrRelationship.wrap(newNode).getTypedId();
}
/**
* @param server the {@link GraphDatabaseShellServer} to get the current node/relationship from.
* @param session the {@link Session} used by the client.
* @return the current node/relationship the client stands on at the moment.
* @throws ShellException if some error occured.
*/
public static NodeOrRelationship getCurrent(GraphDatabaseShellServer server, Session session)
throws ShellException {
String currentThing = (String) session.get(CURRENT_KEY);
NodeOrRelationship result = null;
if (currentThing == null) {
try {
result = NodeOrRelationship.wrap(server.getDb().getReferenceNode());
} catch (NotFoundException e) {
throw new ShellException("Reference node not found");
}
setCurrent(session, result);
} else {
TypedId typedId = new TypedId(currentThing);
result = getThingById(server, typedId);
}
return result;
}
/**
* @param server the {@link GraphDatabaseShellServer} to run at.
* @param session the {@link Session} used by the client.
* @param relationship the {@link Relationship} to get a display name for.
* @param verbose whether or not to include the relationship id as well.
* @return a display string for the {@code relationship}.
*/
public static String getDisplayName(
GraphDatabaseShellServer server,
Session session,
Relationship relationship,
boolean verbose,
boolean checkForMe)
throws ShellException {
if (checkForMe && isCurrent(session, NodeOrRelationship.wrap(relationship))) {
return getDisplayNameForCurrent(server, session);
}
StringBuilder result = new StringBuilder("[");
result.append(":" + relationship.getType().name());
result.append(verbose ? "," + relationship.getId() : "");
result.append("]");
return result.toString();
}
@Override
protected Continuation exec(AppCommandParser parser, Session session, Output out)
throws ShellException, RemoteException {
assertCurrentIsNode(session);
Node node = this.getCurrent(session).asNode();
boolean caseInsensitiveFilters = parser.options().containsKey("i");
boolean looseFilters = parser.options().containsKey("l");
boolean quiet = parser.options().containsKey("q");
// Order
TraversalDescription description = Traversal.description();
String order = parser.options().get("o");
if (order != null) {
description = description.order(parseOrder(order));
}
// Relationship types / expander
String relationshipTypes = parser.options().get("r");
if (relationshipTypes != null) {
Map<String, Object> types = parseFilter(relationshipTypes, out);
description =
description.expand(
toExpander(getServer().getDb(), null, types, caseInsensitiveFilters, looseFilters));
}
// Uniqueness
String uniqueness = parser.options().get("u");
if (uniqueness != null) {
description = description.uniqueness(parseUniqueness(uniqueness));
}
// Depth limit
String depthLimit = parser.options().get("d");
if (depthLimit != null) {
description = description.evaluator(toDepth(parseInt(depthLimit)));
}
String filterString = parser.options().get("f");
Map<String, Object> filterMap = filterString != null ? parseFilter(filterString, out) : null;
String commandToRun = parser.options().get("c");
Collection<String> commandsToRun = new ArrayList<String>();
if (commandToRun != null) {
commandsToRun.addAll(Arrays.asList(commandToRun.split(Pattern.quote("&&"))));
}
for (Path path : description.traverse(node)) {
boolean hit = false;
if (filterMap == null) {
hit = true;
} else {
Node endNode = path.endNode();
Map<String, Boolean> matchPerFilterKey = new HashMap<String, Boolean>();
for (String key : endNode.getPropertyKeys()) {
for (Map.Entry<String, Object> filterEntry : filterMap.entrySet()) {
String filterKey = filterEntry.getKey();
if (matchPerFilterKey.containsKey(filterKey)) {
continue;
}
if (matches(
newPattern(filterKey, caseInsensitiveFilters),
key,
caseInsensitiveFilters,
looseFilters)) {
Object value = endNode.getProperty(key);
String filterPattern =
filterEntry.getValue() != null ? filterEntry.getValue().toString() : null;
if (matches(
newPattern(filterPattern, caseInsensitiveFilters),
value.toString(),
caseInsensitiveFilters,
looseFilters)) {
matchPerFilterKey.put(filterKey, true);
}
}
}
}
if (matchPerFilterKey.size() == filterMap.size()) {
hit = true;
}
}
if (hit) {
if (commandsToRun.isEmpty()) {
printPath(path, quiet, session, out);
} else {
printAndInterpretTemplateLines(
commandsToRun,
false,
true,
NodeOrRelationship.wrap(path.endNode()),
getServer(),
session,
out);
}
}
}
return Continuation.INPUT_COMPLETE;
}
