| /* |
| * 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); |
| } |
| |
| } |