| /* Copyright 2004 The Apache Software Foundation |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.xmlbeans.impl.util; |
| |
| import org.apache.xmlbeans.Filer; |
| import org.apache.xmlbeans.impl.repackage.Repackager; |
| |
| import java.io.*; |
| import java.nio.charset.Charset; |
| import java.nio.charset.CharsetEncoder; |
| import java.nio.charset.CodingErrorAction; |
| import java.nio.charset.StandardCharsets; |
| import java.nio.file.Files; |
| import java.util.ArrayList; |
| import java.util.HashSet; |
| import java.util.List; |
| import java.util.Set; |
| |
| /** |
| * This implementation of Filer writes to disk. |
| */ |
| public class FilerImpl implements Filer { |
| private final File classdir; |
| private final File srcdir; |
| private final Repackager repackager; |
| private final boolean verbose; |
| private final List<File> sourceFiles; |
| private final boolean incrSrcGen; |
| private Set<String> seenTypes; |
| private static final Charset CHARSET; |
| |
| static { |
| Charset temp = null; |
| try { |
| temp = Charset.forName(System.getProperty("file.encoding")); |
| } catch (Exception ignored) { |
| } |
| CHARSET = temp; |
| } |
| |
| public FilerImpl(File classdir, File srcdir, Repackager repackager, boolean verbose, boolean incrSrcGen) { |
| this.classdir = classdir; |
| this.srcdir = srcdir; |
| this.repackager = repackager; |
| this.verbose = verbose; |
| this.sourceFiles = new ArrayList<>(); |
| this.incrSrcGen = incrSrcGen; |
| if (this.incrSrcGen) { |
| seenTypes = new HashSet<>(); |
| } |
| } |
| |
| /** |
| * Creates a new schema binary file (.xsb) and returns a stream for writing to it. |
| * |
| * @param typename fully qualified type name |
| * @return a stream to write the type to |
| */ |
| public OutputStream createBinaryFile(String typename) throws IOException { |
| if (verbose) { |
| System.err.println("created binary: " + typename); |
| } |
| // KHK: for now the typename will already be a relative filename for the binary |
| //String filename = typename.replace('.', File.separatorChar) + ".xsb"; |
| File source = new File(classdir, typename); |
| source.getParentFile().mkdirs(); |
| |
| return new FileOutputStream(source); |
| } |
| |
| /** |
| * Creates a new binding source file (.java) and returns a writer for it. |
| * |
| * @param typename fully qualified type name |
| * @return a stream to write the type to |
| */ |
| public Writer createSourceFile(String typename) throws IOException { |
| if (incrSrcGen) { |
| seenTypes.add(typename); |
| } |
| |
| if (typename.indexOf('$') > 0) { |
| typename = |
| typename.substring(0, typename.lastIndexOf('.')) + "." + |
| typename.substring(typename.indexOf('$') + 1); |
| } |
| |
| String filename = typename.replace('.', File.separatorChar) + ".java"; |
| |
| File sourcefile = new File(srcdir, filename); |
| sourcefile.getParentFile().mkdirs(); |
| if (verbose) { |
| System.err.println("created source: " + sourcefile.getAbsolutePath()); |
| } |
| |
| sourceFiles.add(sourcefile); |
| |
| if (incrSrcGen && sourcefile.exists()) { |
| // Generate the file in a buffer and then compare it to the |
| // file already on disk |
| return new IncrFileWriter(sourcefile, repackager); |
| } else { |
| return repackager == null ? |
| writerForFile(sourcefile) : |
| new RepackagingWriter(sourcefile, repackager); |
| } |
| } |
| |
| public List<File> getSourceFiles() { |
| return new ArrayList<>(sourceFiles); |
| } |
| |
| public Repackager getRepackager() { |
| return repackager; |
| } |
| |
| private static Writer writerForFile(File f) throws IOException { |
| if (CHARSET == null) { |
| return Files.newBufferedWriter(f.toPath(), StandardCharsets.ISO_8859_1); |
| } |
| |
| FileOutputStream fileStream = new FileOutputStream(f); |
| CharsetEncoder ce = CHARSET.newEncoder(); |
| ce.onUnmappableCharacter(CodingErrorAction.REPORT); |
| return new OutputStreamWriter(fileStream, ce); |
| } |
| |
| static class IncrFileWriter extends StringWriter { |
| private final File _file; |
| private final Repackager _repackager; |
| |
| public IncrFileWriter(File file, Repackager repackager) { |
| _file = file; |
| _repackager = repackager; |
| } |
| |
| public void close() throws IOException { |
| super.close(); |
| |
| // This is where all the real work happens |
| StringBuffer sb = _repackager != null ? |
| _repackager.repackage(getBuffer()) : |
| getBuffer(); |
| String str = sb.toString(); |
| List<String> diffs = new ArrayList<>(); |
| |
| try (StringReader sReader = new StringReader(str); |
| Reader fReader = Files.newBufferedReader(_file.toPath(), StandardCharsets.ISO_8859_1)) { |
| Diff.readersAsText(sReader, "<generated>", fReader, _file.getName(), diffs); |
| } |
| |
| if (diffs.size() > 0) { |
| // Diffs encountered, replace the file on disk with text from the buffer |
| try (Writer fw = writerForFile(_file)) { |
| fw.write(str); |
| } |
| } |
| } |
| } |
| |
| static class RepackagingWriter extends StringWriter { |
| public RepackagingWriter(File file, Repackager repackager) { |
| _file = file; |
| _repackager = repackager; |
| } |
| |
| public void close() throws IOException { |
| super.close(); |
| |
| try (Writer fw = writerForFile(_file)) { |
| fw.write(_repackager.repackage(getBuffer()).toString()); |
| } |
| } |
| |
| private final File _file; |
| private final Repackager _repackager; |
| } |
| } |