org.ow2.petals.cli.command.Set Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of petals-cli-base Show documentation
There is a newer version: 1.2.0
/**
 * Copyright (c) 2010-2012 EBM WebSourcing, 2012-2016 Linagora
 * 
 * This program/library is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 2.1 of the License, or (at your
 * option) any later version.
 * 
 * This program/library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
 * for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program/library; If not, see http://www.gnu.org/licenses/
 * for the GNU Lesser General Public License version 2.1.
 */
package org.ow2.petals.cli.command;

import java.util.Arrays;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.ow2.petals.cli.api.command.AbstractCommand;
import org.ow2.petals.cli.api.command.exception.CommandException;
import org.ow2.petals.cli.api.command.exception.CommandTooManyArgumentsException;
import org.ow2.petals.cli.shell.AbstractShell;

/**
 * This command sets the value of a property.
 * @author Sebastien Andre - EBM WebSourcing
 */
public class Set extends AbstractCommand {

    public static final Pattern REGEX_DECLARATION = Pattern
            .compile("^\\s*([-\\. \\w]+)\\s*[^\\\\][=:]?\\s*(.*)$");

    public Set() {
        super();
        this.setUsage("key [=:] value");
        this.setDescription("Assign a value to the system property named key");
    }

    /**
     * {@inheritDoc}
     * 
     * 
     * Dedicated processing to the command 'set': No connection is required to
     * execute it.
     * 
     */
    @Override
    public boolean isConnectionRequired() {
        return false;
    }

    @Override
    public void doExecute(final String[] args) throws CommandException {
        if (!this.checkArguments(args, 0, 3)) {
            throw new CommandTooManyArgumentsException(this, args);
        }

        if (args.length == 0) {
            this.printVariables();

        } else {
            final StringBuilder sb = new StringBuilder();
            for (final String arg : args) {
                sb.append(arg);
            }

            this.parseDeclaration(sb.toString());
        }
    }

    private void parseDeclaration(final String expr) throws CommandException {

        final Matcher m = REGEX_DECLARATION.matcher(expr);
        if (m.matches()) {
            System.setProperty(m.group(1), m.group(2));
        } else {
            throw new CommandException(this, "invalid syntax in declaration");
        }

    }

    private void printVariables() {
        final StringBuilder sb = new StringBuilder();

        final String[] propertyNames = System.getProperties().keySet().toArray(new String[0]);
        Arrays.sort(propertyNames);
        for (final String propertyName : propertyNames) {
            sb.append(propertyName).append('=').append(System.getProperty(propertyName))
                    .append(System.getProperty("line.separator"));
        }

        final String[] environmentVariables = System.getenv().keySet().toArray(new String[0]);
        Arrays.sort(environmentVariables);
        for (final String environmentVariable : environmentVariables) {
            sb.append(AbstractShell.ENV_PREFIX).append(environmentVariable).append('=')
                    .append(System.getenv(environmentVariable)).append(System.getProperty("line.separator"));
        }

        this.getShell().getPrintStream().print(sb.toString());
    }
}