package org.jboss.forge.addon.shell.commands;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import javax.inject.Inject;
import org.jboss.aesh.terminal.TerminalSize;
import org.jboss.aesh.util.Parser;
import org.jboss.forge.addon.resource.FileResource;
import org.jboss.forge.addon.resource.Resource;
import org.jboss.forge.addon.resource.ResourceFactory;
import org.jboss.forge.addon.shell.Shell;
import org.jboss.forge.addon.shell.ui.AbstractShellCommand;
import org.jboss.forge.addon.shell.ui.ShellContext;
import org.jboss.forge.addon.shell.util.PathspecParser;
import org.jboss.forge.addon.ui.context.UIBuilder;
import org.jboss.forge.addon.ui.hints.InputType;
import org.jboss.forge.addon.ui.input.UIInputMany;
import org.jboss.forge.addon.ui.metadata.WithAttributes;
import org.jboss.forge.addon.ui.result.Result;
import org.jboss.forge.addon.ui.result.Results;
import org.jboss.forge.addon.ui.util.Metadata;
/**
* @author <a href="mailto:stale.pedersen@jboss.org">Ståle W. Pedersen</a>
*/
public class LsCommand extends AbstractShellCommand
{
@Inject
ResourceFactory resourceFactory;
@Inject
@WithAttributes(label = "Arguments", type = InputType.DIRECTORY_PICKER)
private UIInputMany<String> arguments;
@Override
public Metadata getMetadata()
{
return super.getMetadata().name("ls").description("List files");
}
@Override
public void initializeUI(UIBuilder builder) throws Exception
{
builder.add(arguments);
}
@Override
public Result execute(ShellContext context) throws Exception
{
Shell shell = context.getProvider();
FileResource<?> currentResource = shell.getCurrentResource();
Iterator<String> it = arguments.getValue() == null ? Collections.<String> emptyList().iterator() : arguments
.getValue()
.iterator();
final Result result;
final FileResource<?> newResource = (it.hasNext()) ? new PathspecParser(
resourceFactory, currentResource, it.next()).resolve().get(0).reify(FileResource.class)
: currentResource;
if (!newResource.exists())
{
result = Results.fail(newResource.getName() + ": No such file or directory");
}
else
{
result = Results.success(listMany(newResource.listResources(), context));
}
return result;
}
private String listMany(Iterable<Resource<?>> files, ShellContext context)
{
TerminalSize terminalSize = context.getProvider().getConsole().getTerminalSize();
List<String> display = new ArrayList<String>();
if (files != null)
{
for (Resource<?> file : files)
{
display.add(file.getName());
}
}
return Parser.formatDisplayList(display, terminalSize.getHeight(), terminalSize.getWidth());
}
}