123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- /*
- * Copyright (c) 2006-2014 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.commandparser.commands.global;
-
- import com.dmdirc.FrameContainer;
- import com.dmdirc.actions.Action;
- import com.dmdirc.actions.ActionFactory;
- import com.dmdirc.actions.ActionManager;
- import com.dmdirc.actions.wrappers.Alias;
- import com.dmdirc.actions.wrappers.AliasWrapper;
- import com.dmdirc.commandparser.BaseCommandInfo;
- import com.dmdirc.commandparser.CommandArguments;
- import com.dmdirc.commandparser.CommandInfo;
- import com.dmdirc.commandparser.CommandType;
- import com.dmdirc.commandparser.commands.Command;
- import com.dmdirc.commandparser.commands.IntelligentCommand;
- import com.dmdirc.commandparser.commands.context.CommandContext;
- import com.dmdirc.interfaces.CommandController;
- import com.dmdirc.ui.input.AdditionalTabTargets;
- import com.dmdirc.ui.input.TabCompleter;
-
- import javax.inject.Inject;
-
- /**
- * The alias command allows users to create aliases on-the-fly.
- */
- public class AliasCommand extends Command implements IntelligentCommand {
-
- /** A command info object for this command. */
- public static final CommandInfo INFO = new BaseCommandInfo("alias",
- "alias [--remove] <name> [command] - creates or removes the specified alias",
- CommandType.TYPE_GLOBAL);
- /** Factory to use when creating aliases. */
- private final ActionFactory actionFactory;
- /** Wrapper to use to modify aliases. */
- private final AliasWrapper aliasWrapper;
-
- /**
- * Creates a new instance of {@link AliasCommand}.
- *
- * @param controller The controller that owns this command.
- * @param actionFactory The factory to use when creating new aliases.
- * @param aliasWrapper The wrapper to use to modify aliases.
- */
- @Inject
- public AliasCommand(
- final CommandController controller,
- final ActionFactory actionFactory,
- final AliasWrapper aliasWrapper) {
- super(controller);
- this.actionFactory = actionFactory;
- this.aliasWrapper = aliasWrapper;
- }
-
- /** {@inheritDoc} */
- @Override
- public void execute(final FrameContainer origin,
- final CommandArguments args, final CommandContext context) {
- if (args.getArguments().length < 2) {
- showUsage(origin, args.isSilent(), "alias", "[--remove] <name> [command]");
- return;
- }
-
- if (args.getArguments()[0].equalsIgnoreCase("--remove")) {
- final String name = args.getArguments()[1].charAt(0)
- == getController().getCommandChar()
- ? args.getArguments()[1].substring(1) : args.getArguments()[1];
-
- if (doRemove(name)) {
- sendLine(origin, args.isSilent(), FORMAT_OUTPUT, "Alias '"
- + name + "' removed.");
- } else {
- sendLine(origin, args.isSilent(), FORMAT_ERROR, "Alias '"
- + name + "' not found.");
- }
-
- return;
- }
-
- final String name = args.getArguments()[0].charAt(0)
- == getController().getCommandChar()
- ? args.getArguments()[0].substring(1) : args.getArguments()[0];
-
- for (Action alias : aliasWrapper) {
- if (aliasWrapper.getCommandName(alias).substring(1).equalsIgnoreCase(name)) {
- sendLine(origin, args.isSilent(), FORMAT_ERROR, "Alias '" + name
- + "' already exists.");
- return;
- }
- }
-
- final Alias myAlias = new Alias(actionFactory, name);
- myAlias.setResponse(new String[]{args.getArgumentsAsString(1)});
- myAlias.createAction().save();
-
- sendLine(origin, args.isSilent(), FORMAT_OUTPUT, "Alias '" + name
- + "' created.");
- }
-
- /**
- * Removes the alias with the specified name.
- *
- * @param name The name of the alias to remove
- *
- * @return True if the alias was deleted, false otherwise
- */
- private boolean doRemove(final String name) {
- for (Action alias : aliasWrapper) {
- if (aliasWrapper.getCommandName(alias).substring(1).equalsIgnoreCase(name)) {
- alias.delete();
- ActionManager.getActionManager().removeAction(alias);
-
- return true;
- }
- }
-
- return false;
- }
-
- /** {@inheritDoc} */
- @Override
- public AdditionalTabTargets getSuggestions(final int arg,
- final IntelligentCommandContext context) {
- final AdditionalTabTargets res = new AdditionalTabTargets().excludeAll();
-
- if (arg == 0) {
- res.add("--remove");
- } else if (arg == 1 && context.getPreviousArgs().get(0).equals("--remove")) {
- for (Action alias : aliasWrapper) {
- res.add(aliasWrapper.getCommandName(alias));
- }
- } else if (arg >= 1 && !context.getPreviousArgs().get(0).equals("--remove")) {
- return TabCompleter.getIntelligentResults(arg, context, 1);
- }
-
- return res;
- }
-
- }
|