jaxws/src/share/classes/com/sun/codemodel/internal/writer/FileCodeWriter.java
changeset 8 474761f14bca
child 2678 57cf2a1c1a05
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jaxws/src/share/classes/com/sun/codemodel/internal/writer/FileCodeWriter.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,103 @@
+/*
+ * Copyright 2006 Sun Microsystems, Inc.  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.  Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.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, boolean readOnly ) throws IOException {
+        this.target = target;
+        this.readOnly = readOnly;
+        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);
+    }
+
+}