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 */
019package org.crsh.lang.impl.java;
020
021import javax.tools.SimpleJavaFileObject;
022import java.io.ByteArrayOutputStream;
023import java.io.IOException;
024import java.io.OutputStream;
025import java.net.URI;
026import java.net.URISyntaxException;
027
028/** @author Julien Viet */
029class JavaClassFileObject extends SimpleJavaFileObject {
030
031  /** . */
032  private ByteArrayOutputStream baos;
033
034  /** . */
035  private final String className;
036
037  JavaClassFileObject(String className) throws URISyntaxException {
038    super(new URI("whatever", null, '/' + className.replace('.', '/') + ".class", null), Kind.CLASS);
039
040    //
041    this.className = className;
042  }
043
044  String getClassName() {
045    return className;
046  }
047
048  public byte[] getBytes() {
049    return baos != null ? baos.toByteArray() : null;
050  }
051
052  @Override
053  public OutputStream openOutputStream() throws IOException {
054    if (baos != null) {
055      throw new IOException("Already open");
056    } else {
057      return baos = new ByteArrayOutputStream();
058    }
059  }
060}