org.crsh.lang.java.ProducerCommandImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crash.shell Show documentation
The Shell module
There is a newer version: 1.3.2
/*
 * Copyright (C) 2012 eXo Platform SAS.
 *
 * This 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 software 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 software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.crsh.lang.java;

import org.crsh.cli.impl.invocation.CommandInvoker;
import org.crsh.cli.impl.invocation.InvocationException;
import org.crsh.cli.impl.invocation.InvocationMatch;
import org.crsh.cli.impl.lang.Instance;
import org.crsh.command.BaseCommand;
import org.crsh.command.CommandContext;
import org.crsh.command.InvocationContext;
import org.crsh.command.SyntaxException;
import org.crsh.console.KeyHandler;
import org.crsh.shell.impl.command.InvocationContextImpl;
import org.crsh.shell.impl.command.spi.CommandCreationException;

import java.io.IOException;
import java.lang.reflect.UndeclaredThrowableException;

/**
* @author Julien Viet
*/
class ProducerCommandImpl extends CommandImpl {

  private final CommandInvoker, ?> invoker;
  private final Class producedType;

  public ProducerCommandImpl(ShellCommandImpl shellCommand, CommandInvoker, ?> invoker, Class
 producedType) {
    super(shellCommand);

    //
    this.invoker = invoker;
    this.producedType = producedType;
  }

  @Override
  public InvocationMatch getMatch() {
    return invoker.getMatch();
  }

  @Override
  public Class
 getProducedType() {
    return producedType;
  }

  @Override
  public Class getConsumedType() {
    return Void.class;
  }

  @Override
  BaseInvoker getInvoker(T command) throws CommandCreationException {

    //
    return new BaseInvoker(command) {

      /** . */
      private InvocationContext
 invocationContext;

      public Class
 getProducedType() {
        return producedType;
      }

      public Class getConsumedType() {
        return Void.class;
      }

      public void open(CommandContext consumer) {
        // Java is fine with that but not intellij....
        CommandContext
 consumer2 = (CommandContext
)consumer;
        open2(consumer2);
      }

      public void open2(final CommandContext
 consumer) {
        invocationContext = new InvocationContextImpl(consumer);
        command.pushContext(invocationContext);
        command.unmatched = invoker.getMatch().getRest();
      }


      @Override
      public KeyHandler getKeyHandler() {
        if (command instanceof KeyHandler) {
          return (KeyHandler)command;
        } else {
          return null;
        }
      }

      public void provide(Void element) throws IOException {
        // Drop everything
      }

      public void flush() throws IOException {
      }

      public void close() throws IOException, UndeclaredThrowableException {

        //
        Object ret;
        try {
          ret = invoker.invoke(this);
        }
        catch (org.crsh.cli.SyntaxException e) {
          throw new SyntaxException(e.getMessage());
        } catch (InvocationException e) {
          throw command.toScript(e.getCause());
        }

        //
        if (ret != null && producedType.isInstance(ret)) {
          P produced = producedType.cast(ret);
          invocationContext.provide(produced);
        }

        //
        invocationContext.flush();
        invocationContext.close();
        command.unmatched = null;
        invocationContext = null;
      }
    };
  }
}