| | |
| | | import org.slf4j.Logger; |
| | | import org.slf4j.LoggerFactory; |
| | | |
| | | import ro.fortsoft.pf4j.ExtensionPoint; |
| | | |
| | | import com.gitblit.models.UserModel; |
| | | import com.gitblit.utils.StringUtils; |
| | | import com.gitblit.utils.cli.SubcommandHandler; |
| | |
| | | import com.google.common.base.Strings; |
| | | import com.google.common.collect.Maps; |
| | | |
| | | public abstract class DispatchCommand extends BaseCommand { |
| | | public abstract class DispatchCommand extends BaseCommand implements ExtensionPoint { |
| | | |
| | | private Logger log = LoggerFactory.getLogger(getClass()); |
| | | |
| | |
| | | private List<String> args = new ArrayList<String>(); |
| | | |
| | | private final Set<Class<? extends BaseCommand>> commands; |
| | | private final Map<String, DispatchCommand> dispatchers; |
| | | private final List<BaseCommand> instantiated; |
| | | private Map<String, Class<? extends BaseCommand>> map; |
| | | private Map<String, BaseCommand> dispatchers; |
| | | |
| | | protected DispatchCommand() { |
| | | commands = new HashSet<Class<? extends BaseCommand>>(); |
| | | dispatchers = Maps.newHashMap(); |
| | | instantiated = new ArrayList<BaseCommand>(); |
| | | } |
| | | |
| | | @Override |
| | | public void destroy() { |
| | | super.destroy(); |
| | | commands.clear(); |
| | | map = null; |
| | | |
| | | for (BaseCommand command : instantiated) { |
| | | command.destroy(); |
| | | } |
| | | for (DispatchCommand dispatcher : dispatchers.values()) { |
| | | dispatcher.destroy(); |
| | | } |
| | | } |
| | | |
| | | protected void registerDispatcher(UserModel user, Class<? extends DispatchCommand> cmd) { |
| | | if (!cmd.isAnnotationPresent(CommandMetaData.class)) { |
| | | throw new RuntimeException(MessageFormat.format("{0} must be annotated with {1}!", cmd.getName(), |
| | | try { |
| | | DispatchCommand dispatcher = cmd.newInstance(); |
| | | registerDispatcher(user, dispatcher); |
| | | } catch (Exception e) { |
| | | log.error("failed to instantiate {}", cmd.getName()); |
| | | } |
| | | } |
| | | |
| | | protected void registerDispatcher(UserModel user, DispatchCommand dispatcher) { |
| | | Class<? extends DispatchCommand> dispatcherClass = dispatcher.getClass(); |
| | | if (!dispatcherClass.isAnnotationPresent(CommandMetaData.class)) { |
| | | throw new RuntimeException(MessageFormat.format("{0} must be annotated with {1}!", dispatcher.getName(), |
| | | CommandMetaData.class.getName())); |
| | | } |
| | | if (dispatchers == null) { |
| | | dispatchers = Maps.newHashMap(); |
| | | } |
| | | |
| | | CommandMetaData meta = cmd.getAnnotation(CommandMetaData.class); |
| | | CommandMetaData meta = dispatcherClass.getAnnotation(CommandMetaData.class); |
| | | if (meta.admin() && !user.canAdmin()) { |
| | | log.debug(MessageFormat.format("excluding admin dispatch command {0} for {1}", meta.name(), user.username)); |
| | | log.debug(MessageFormat.format("excluding admin dispatcher {0} for {1}", meta.name(), user.username)); |
| | | return; |
| | | } |
| | | |
| | | log.debug("registering {} dispatcher", meta.name()); |
| | | try { |
| | | DispatchCommand dispatcher = cmd.newInstance(); |
| | | dispatcher.registerCommands(user); |
| | | dispatchers.put(meta.name(), dispatcher); |
| | | } catch (Exception e) { |
| | |
| | | } |
| | | } |
| | | |
| | | protected abstract void registerCommands(UserModel user); |
| | | |
| | | protected abstract void registerCommands(UserModel user); |
| | | |
| | | /** |
| | | * Registers a command as long as the user is permitted to execute it. |
| | |
| | | CommandMetaData meta = cmd.getAnnotation(CommandMetaData.class); |
| | | map.put(meta.name(), cmd); |
| | | } |
| | | if (dispatchers != null) { |
| | | for (Map.Entry<String, BaseCommand> entry : dispatchers.entrySet()) { |
| | | map.put(entry.getKey(), entry.getValue().getClass()); |
| | | } |
| | | |
| | | for (Map.Entry<String, DispatchCommand> entry : dispatchers.entrySet()) { |
| | | map.put(entry.getKey(), entry.getValue().getClass()); |
| | | } |
| | | } |
| | | return map; |
| | |
| | | BaseCommand cmd = null; |
| | | try { |
| | | cmd = c.newInstance(); |
| | | instantiated.add(cmd); |
| | | } catch (Exception e) { |
| | | throw new UnloggedFailure(1, MessageFormat.format("Failed to instantiate {0} command", commandName)); |
| | | } |
| | |
| | | for (String name : m.keySet()) { |
| | | Class<? extends BaseCommand> c = m.get(name); |
| | | CommandMetaData meta = c.getAnnotation(CommandMetaData.class); |
| | | if (meta != null) { |
| | | if (meta.hidden()) { |
| | | continue; |
| | | } |
| | | if (meta.hidden()) { |
| | | continue; |
| | | } |
| | | |
| | | maxLength = Math.max(maxLength, name.length()); |
| | |
| | | usage.append("COMMAND --help' for more information.\n"); |
| | | usage.append("\n"); |
| | | return usage.toString(); |
| | | } |
| | | |
| | | protected void provideStateTo(final BaseCommand cmd) { |
| | | if (cmd instanceof BaseCommand) { |
| | | cmd.setContext(ctx); |
| | | } |
| | | cmd.setInputStream(in); |
| | | cmd.setOutputStream(out); |
| | | cmd.setErrorStream(err); |
| | | cmd.setExitCallback(exit); |
| | | } |
| | | } |