123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- /*
- * Copyright (c) 2006-2015 DMDirc Developers
- *
- * Permission is hereby granted, free of charge, to any person obtaining a copy
- * of this software and associated documentation files (the "Software"), to deal
- * in the Software without restriction, including without limitation the rights
- * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- * copies of the Software, and to permit persons to whom the Software is
- * furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in
- * all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- * SOFTWARE.
- */
-
- package com.dmdirc.addons.exec;
-
- import com.dmdirc.DMDircMBassador;
- import com.dmdirc.commandparser.BaseCommandInfo;
- import com.dmdirc.commandparser.CommandArguments;
- import com.dmdirc.commandparser.CommandType;
- import com.dmdirc.commandparser.commands.Command;
- import com.dmdirc.commandparser.commands.context.CommandContext;
- import com.dmdirc.events.UserErrorEvent;
- import com.dmdirc.interfaces.CommandController;
- import com.dmdirc.interfaces.WindowModel;
- import com.dmdirc.logger.ErrorLevel;
- import com.dmdirc.util.CommandUtils;
- import com.dmdirc.util.io.StreamUtils;
-
- import com.google.common.io.CharStreams;
-
- import java.io.File;
- import java.io.IOException;
- import java.io.InputStreamReader;
- import java.util.List;
-
- import javax.annotation.Nonnull;
- import javax.inject.Inject;
-
- /**
- * A command which allows users execute scripts.
- */
- public class ExecCommand extends Command {
-
- /** A command info object for this command. */
- public static final BaseCommandInfo INFO = new BaseCommandInfo("exec",
- "exec <command> [<parameters>] - executes an external program "
- + "and displays the output", CommandType.TYPE_GLOBAL);
- /** Event bus to post errors to. */
- private final DMDircMBassador eventBus;
-
- /**
- * Creates a new instance of this command.
- *
- * @param controller The controller to use for command information.
- * @param eventBus The event bus to post errors to
- */
- @Inject
- public ExecCommand(final CommandController controller, final DMDircMBassador eventBus) {
- super(controller);
- this.eventBus = eventBus;
- }
-
- @Override
- public void execute(@Nonnull final WindowModel origin,
- final CommandArguments args, final CommandContext context) {
- final String[] commandArray = CommandUtils.parseArguments(
- args.getArgumentsAsString());
-
- try {
- // This checks the command to execute has correct quotes
- // (if necessary). Without this /exec "command arg1 arg2 would error.
- if (commandArray.length == 0) {
- sendLine(origin, args.isSilent(), FORMAT_ERROR,
- "Could not execute: Invalid file name provided");
- } else if (!new File(commandArray[0]).exists()) {
- sendLine(origin, args.isSilent(), FORMAT_ERROR,
- "Could not execute: " + commandArray[0] + " does not exist.");
- } else {
- final Process p = Runtime.getRuntime().exec(commandArray);
- if (args.isSilent()) {
- StreamUtils.readStream(p.getInputStream());
- StreamUtils.readStream(p.getErrorStream());
- } else {
- final List<String> execOutput = CharStreams.readLines(
- new InputStreamReader(p.getInputStream()));
- final List<String> errorOutput = CharStreams.readLines(
- new InputStreamReader(p.getErrorStream()));
- for (String line : execOutput) {
- sendLine(origin, args.isSilent(), FORMAT_OUTPUT, line);
- }
- for (String line : errorOutput) {
- sendLine(origin, args.isSilent(), FORMAT_ERROR, line);
- }
- }
- }
- } catch (IOException ex) {
- eventBus.publishAsync(new UserErrorEvent(ErrorLevel.LOW, ex,
- "Unable to run application: " + ex.getMessage(), ""));
- }
- }
-
- }
|