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.lang.groovy.command;
021    
022    import org.crsh.command.CommandInvoker;
023    import org.crsh.command.InvocationContext;
024    import org.crsh.command.ScriptException;
025    import org.crsh.io.Consumer;
026    import org.crsh.text.Chunk;
027    import org.crsh.shell.ScreenContext;
028    import org.crsh.text.RenderPrintWriter;
029    
030    import java.io.IOException;
031    import java.util.Map;
032    
033    class InnerInvocationContext<P> implements InvocationContext<P> {
034    
035      /** . */
036      final InvocationContext<?> outter;
037    
038      /** . */
039      final Consumer<P> consumer;
040    
041      /** . */
042      private RenderPrintWriter writer;
043    
044      /** . */
045      private final boolean piped;
046    
047      InnerInvocationContext(
048        InvocationContext<?> outter,
049        Consumer<P> consumer,
050        boolean piped) {
051    
052        //
053        this.outter = outter;
054        this.consumer = consumer;
055        this.piped = piped;
056      }
057    
058      public boolean isPiped() {
059        return piped;
060      }
061    
062      public CommandInvoker<?, ?> resolve(String s) throws ScriptException, IOException {
063        return outter.resolve(s);
064      }
065    
066      public int getWidth() {
067        return outter.getWidth();
068      }
069    
070      public int getHeight() {
071        return outter.getHeight();
072      }
073    
074      public boolean takeAlternateBuffer() throws IOException {
075        return outter.takeAlternateBuffer();
076      }
077    
078      public boolean releaseAlternateBuffer() throws IOException {
079        return outter.releaseAlternateBuffer();
080      }
081    
082      public String getProperty(String propertyName) {
083        return outter.getProperty(propertyName);
084      }
085    
086      public String readLine(String msg, boolean echo) {
087        return outter.readLine(msg, echo);
088      }
089    
090      public RenderPrintWriter getWriter() {
091        if (writer == null) {
092          writer = new RenderPrintWriter(new ScreenContext<Chunk>() {
093            public Class<Chunk> getConsumedType() {
094              return Chunk.class;
095            }
096            public int getWidth() {
097              return outter.getWidth();
098            }
099            public int getHeight() {
100              return outter.getHeight();
101            }
102            public void provide(Chunk element) throws IOException {
103              Class<P> consumedType = consumer.getConsumedType();
104              if (consumedType.isInstance(element)) {
105                P p = consumedType.cast(element);
106                consumer.provide(p);
107              }
108            }
109            public void flush() throws IOException {
110              consumer.flush();
111            }
112          });
113        }
114        return writer;
115      }
116    
117      public Class<P> getConsumedType() {
118        return consumer.getConsumedType();
119      }
120    
121      public void provide(P element) throws IOException {
122        consumer.provide(element);
123      }
124    
125      public void flush() throws IOException {
126        consumer.flush();
127      }
128    
129      public void close() throws IOException {
130        // Nothing to do
131      }
132    
133      public Map<String, Object> getSession() {
134        return outter.getSession();
135      }
136    
137      public Map<String, Object> getAttributes() {
138        return outter.getAttributes();
139      }
140    }