Examples with NodeServiceCommand org.structr.core.graph.NodeServiceCommand used on opensource projects

Example 1 with NodeServiceCommand

use of org.structr.core.graph.NodeServiceCommand in project structr by structr.

the class InitConsoleCommand method run.

@Override
public void run(final SecurityContext securityContext, final List<String> parameters, final Writable writable) throws FrameworkException, IOException {
    final Principal user = securityContext.getUser(false);
    final int paramCount = parameters.size();
    if (user != null && user.isAdmin()) {
        NodeServiceCommand command = null;
        boolean allNodes = true;
        boolean allRels = true;
        boolean isCreateLabels = false;
        boolean isIndex = false;
        boolean isIds = false;
        boolean hasCommand = false;
        boolean hasFor = false;
        boolean error = false;
        String type = null;
        String typeKey = null;
        String mode = null;
        // parse parameters
        for (int i = 1; i < paramCount && !error; i++) {
            final String param = getParameter(parameters, i);
            switch(param) {
                case "index":
                    if (hasCommand) {
                        writable.println("Syntax error, too many parameters.");
                        error = true;
                    } else {
                        command = StructrApp.getInstance(securityContext).command(BulkRebuildIndexCommand.class);
                        command.setLogBuffer(writable);
                        isIndex = true;
                        hasCommand = true;
                    }
                    break;
                case "ids":
                    if (hasCommand) {
                        writable.println("Syntax error, too many parameters.");
                        error = true;
                    } else {
                        command = StructrApp.getInstance(securityContext).command(BulkSetUuidCommand.class);
                        command.setLogBuffer(writable);
                        isIds = true;
                        hasCommand = true;
                    }
                    break;
                case "labels":
                    if (hasCommand) {
                        writable.println("Syntax error, too many parameters.");
                        error = true;
                    } else {
                        if ("relsOnly".equals(mode)) {
                            writable.println("Cannot set labels on relationships.");
                            error = true;
                        } else {
                            command = StructrApp.getInstance(securityContext).command(BulkCreateLabelsCommand.class);
                            command.setLogBuffer(writable);
                            isCreateLabels = true;
                            hasCommand = true;
                        }
                    }
                    break;
                case "node":
                    if (hasCommand) {
                        if (isIndex) {
                            writable.println("Index type must be specified before the 'index' keyword.");
                            error = true;
                        } else if (isIds) {
                            writable.println("Entity type must be specified before the 'ids' keyword.");
                            error = true;
                        }
                    } else {
                        mode = "nodesOnly";
                        typeKey = "type";
                        allRels = false;
                    }
                    break;
                case "rel":
                case "relationship":
                    if (hasCommand) {
                        if (isIndex) {
                            writable.println("Index type must be specified before the 'index' keyword.");
                            error = true;
                        } else if (isIds) {
                            writable.println("Entity type must be specified before the 'ids' keyword.");
                            error = true;
                        }
                    } else {
                        if (isCreateLabels) {
                            writable.println("Cannot set labels on relationships.");
                            error = true;
                        }
                        mode = "relsOnly";
                        typeKey = "relType";
                        allNodes = false;
                    }
                    break;
                case "for":
                    if (!hasCommand) {
                        writable.println("Unknown init mode 'for'.");
                        error = true;
                    }
                    hasFor = true;
                    break;
                default:
                    // specify node or rel type
                    if (hasCommand && hasFor) {
                        // prevent too many parameters from being accepted
                        if (StringUtils.isNotBlank(type)) {
                            writable.println("Syntax error, too many parameters.");
                            error = true;
                            break;
                        }
                        type = param;
                        // only set type key if not already set, default is "type" not "relType"
                        if (typeKey == null) {
                            typeKey = "type";
                        }
                    } else {
                        if (!hasCommand) {
                            writable.println("Unknown init mode '" + param + "'.");
                            error = true;
                        } else {
                            writable.println("Syntax error, please specify something like 'init node index for User'.");
                            error = true;
                        }
                    }
                    break;
            }
            // break early on errors
            if (error) {
                break;
            }
        }
        if (!error && !hasCommand) {
            writable.println("Please specify what to initialize.");
            error = true;
        }
        if (!error && hasCommand && hasFor && StringUtils.isEmpty(type)) {
            writable.println("Missing type specification, please specify something like 'init node index for User'.");
            error = true;
        }
        if (!error) {
            if (command instanceof MaintenanceCommand) {
                final Map<String, Object> data = toMap("mode", mode, typeKey, type);
                if (type == null) {
                    data.put("allNodes", allNodes);
                    data.put("allRels", allRels);
                }
                ((MaintenanceCommand) command).execute(data);
            } else if (command != null) {
                writable.println("Cannot execute command '" + command.getClass().getSimpleName() + "', wrong type.");
            } else {
                writable.println("Cannot execute null command.");
            }
        }
    } else {
        writable.println("You must be admin user to use this command.");
    }
}