1 /*
   2  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package com.sun.tools.internal.xjc.api.util;
  27 
  28 import javax.tools.StandardLocation;
  29 import java.io.IOException;
  30 import java.io.OutputStream;
  31 import java.io.Writer;
  32 
  33 import com.sun.codemodel.internal.CodeWriter;
  34 import com.sun.codemodel.internal.JPackage;
  35 import javax.annotation.processing.Filer;
  36 
  37 import static javax.tools.StandardLocation.CLASS_PATH;
  38 import static javax.tools.StandardLocation.SOURCE_PATH;
  39 
  40 /**
  41  * {@link CodeWriter} that generates source code to {@link Filer}.
  42  *
  43  * @author Kohsuke Kawaguchi
  44  */
  45 public final class FilerCodeWriter extends CodeWriter {
  46 
  47     private final Filer filer;
  48 
  49     public FilerCodeWriter(Filer filer) {
  50         this.filer = filer;
  51     }
  52 
  53     public OutputStream openBinary(JPackage pkg, String fileName) throws IOException {
  54         StandardLocation loc;
  55         if(fileName.endsWith(".java")) {
  56             // Annotation Processing doesn't do the proper Unicode escaping on Java source files,
  57             // so we can't rely on Filer.createSourceFile.
  58             loc = SOURCE_PATH;
  59         } else {
  60             // put non-Java files directly to the output folder
  61             loc = CLASS_PATH;
  62         }
  63         return filer.createResource(loc, pkg.name(), fileName).openOutputStream();
  64     }
  65 
  66     public Writer openSource(JPackage pkg, String fileName) throws IOException {
  67         String name;
  68         if(pkg.isUnnamed())
  69             name = fileName;
  70         else
  71             name = pkg.name()+'.'+fileName;
  72 
  73         name = name.substring(0,name.length()-5);   // strip ".java"
  74 
  75         return filer.createSourceFile(name).openWriter();
  76     }
  77 
  78     public void close() {
  79         ; // noop
  80     }
  81 }