001 /*
002 * Copyright (C) 2012 eXo Platform SAS.
003 *
004 * This is free software; you can redistribute it and/or modify it
005 * under the terms of the GNU Lesser General Public License as
006 * published by the Free Software Foundation; either version 2.1 of
007 * the License, or (at your option) any later version.
008 *
009 * This software is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012 * Lesser General Public License for more details.
013 *
014 * You should have received a copy of the GNU Lesser General Public
015 * License along with this software; if not, write to the Free
016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
018 */
019
020 package org.crsh.shell.impl.async;
021
022 import org.crsh.cli.impl.completion.CompletionMatch;
023 import org.crsh.shell.Shell;
024
025 import java.io.Closeable;
026 import java.util.Collections;
027 import java.util.HashSet;
028 import java.util.Set;
029 import java.util.concurrent.ExecutorService;
030
031 public class AsyncShell implements Shell, Closeable {
032
033 /** . */
034 final Shell shell;
035
036 /** . */
037 private AsyncProcess current;
038
039 /** . */
040 final ExecutorService executor;
041
042 /** . */
043 boolean closed;
044
045 /** . */
046 final Object lock = new Object();
047
048 /** . */
049 final Set<AsyncProcess> processes;
050
051 public AsyncShell(ExecutorService executor, Shell shell) {
052 this.shell = shell;
053 this.current = null;
054 this.executor = executor;
055 this.closed = false;
056 this.processes = Collections.synchronizedSet(new HashSet<AsyncProcess>());
057 }
058
059 public void close() {
060
061 AsyncProcess[] toCancel = null;
062 synchronized (lock) {
063 if (closed) {
064 toCancel = null;
065 } else {
066 closed = true;
067 toCancel = processes.toArray(new AsyncProcess[processes.size()]);
068 }
069 }
070
071 // Cancel all process
072 if (toCancel != null) {
073 for (AsyncProcess process : toCancel) {
074 process.cancel();
075 }
076 }
077 }
078
079 // Shell implementation **********************************************************************************************
080
081 public String getWelcome() {
082 return shell.getWelcome();
083 }
084
085 public String getPrompt() {
086 return shell.getPrompt();
087 }
088
089 public CompletionMatch complete(String prefix) {
090 return shell.complete(prefix);
091 }
092
093 public AsyncProcess createProcess(String request) {
094 synchronized (lock) {
095 if (closed) {
096 throw new IllegalStateException();
097 }
098 }
099 return new AsyncProcess(this, request);
100 }
101 }