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    package org.crsh.util;
020    
021    import java.io.ByteArrayOutputStream;
022    import java.io.IOException;
023    import java.io.InputStream;
024    
025    public class BytesOutputStream extends ByteArrayOutputStream {
026    
027      /** . */
028      private boolean closed;
029    
030      public BytesOutputStream() {
031      }
032    
033      public BytesOutputStream(int size) {
034        super(size);
035      }
036    
037      public InputStream getInputStream() {
038        if (!closed) {
039          throw new IllegalStateException("Not yet closed");
040        }
041        return new InputStream() {
042    
043          /** . */
044          int read = 0;
045    
046          @Override
047          public int read() throws IOException {
048            if (read < count) {
049              return buf[read++];
050            } else {
051              return -1;
052            }
053          }
054        };
055      }
056    
057      @Override
058      public void close() throws IOException {
059        super.close();
060    
061        //
062        closed = true;
063      }
064    }