jaxws/src/jdk.xml.bind/share/classes/com/sun/codemodel/internal/writer/FileCodeWriter.java
changeset 25871 b80b84e87032
parent 22678 ac1ea46be942
child 42124 640a383428fb
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jaxws/src/jdk.xml.bind/share/classes/com/sun/codemodel/internal/writer/FileCodeWriter.java	Sun Aug 17 15:52:15 2014 +0100
@@ -0,0 +1,111 @@
+/*
+ * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.  Oracle designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Oracle in the LICENSE file that accompanied this code.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+package com.sun.codemodel.internal.writer;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.HashSet;
+import java.util.Set;
+
+import com.sun.codemodel.internal.CodeWriter;
+import com.sun.codemodel.internal.JPackage;
+
+/**
+ * Writes all the source files under the specified file folder.
+ *
+ * @author
+ *      Kohsuke Kawaguchi (kohsuke.kawaguchi@sun.com)
+ */
+public class FileCodeWriter extends CodeWriter {
+
+    /** The target directory to put source code. */
+    private final File target;
+
+    /** specify whether or not to mark the generated files read-only */
+    private final boolean readOnly;
+
+    /** Files that shall be marked as read only. */
+    private final Set<File> readonlyFiles = new HashSet<File>();
+
+    public FileCodeWriter( File target ) throws IOException {
+        this(target,false);
+    }
+
+    public FileCodeWriter( File target, String encoding ) throws IOException {
+        this(target,false, encoding);
+    }
+
+    public FileCodeWriter( File target, boolean readOnly ) throws IOException {
+        this(target, readOnly, null);
+    }
+
+    public FileCodeWriter( File target, boolean readOnly, String encoding ) throws IOException {
+        this.target = target;
+        this.readOnly = readOnly;
+        this.encoding = encoding;
+        if(!target.exists() || !target.isDirectory())
+            throw new IOException(target + ": non-existent directory");
+    }
+
+    public OutputStream openBinary(JPackage pkg, String fileName) throws IOException {
+        return new FileOutputStream(getFile(pkg,fileName));
+    }
+
+    protected File getFile(JPackage pkg, String fileName ) throws IOException {
+        File dir;
+        if(pkg.isUnnamed())
+            dir = target;
+        else
+            dir = new File(target, toDirName(pkg));
+
+        if(!dir.exists())   dir.mkdirs();
+
+        File fn = new File(dir,fileName);
+
+        if (fn.exists()) {
+            if (!fn.delete())
+                throw new IOException(fn + ": Can't delete previous version");
+        }
+
+
+        if(readOnly)        readonlyFiles.add(fn);
+        return fn;
+    }
+
+    public void close() throws IOException {
+        // mark files as read-onnly if necessary
+        for (File f : readonlyFiles)
+            f.setReadOnly();
+    }
+
+    /** Converts a package name to the directory name. */
+    private static String toDirName( JPackage pkg ) {
+        return pkg.name().replace('.',File.separatorChar);
+    }
+
+}