1/*
2 * Copyright (c) 1997, 2016, 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
26package com.sun.tools.internal.xjc;
27
28import java.io.File;
29import java.io.IOException;
30import java.io.OutputStream;
31import java.io.Writer;
32
33import com.sun.codemodel.internal.CodeWriter;
34import com.sun.codemodel.internal.JPackage;
35import com.sun.codemodel.internal.writer.FilterCodeWriter;
36
37/**
38 * {@link CodeWriter} that reports progress to {@link XJCListener}.
39 */
40final class ProgressCodeWriter extends FilterCodeWriter {
41
42    private int current;
43    private final int totalFileCount;
44
45    public ProgressCodeWriter( CodeWriter output, XJCListener progress, int totalFileCount ) {
46        super(output);
47        this.progress = progress;
48        this.totalFileCount = totalFileCount;
49        if(progress==null)
50            throw new IllegalArgumentException();
51    }
52
53    private final XJCListener progress;
54
55    @Override
56    public Writer openSource(JPackage pkg, String fileName) throws IOException {
57        report(pkg,fileName);
58        return super.openSource(pkg, fileName);
59    }
60
61    @Override
62    public OutputStream openBinary(JPackage pkg, String fileName) throws IOException {
63        report(pkg,fileName);
64        return super.openBinary(pkg,fileName);
65    }
66
67    /**
68     * Report progress to {@link XJCListener}.
69     * @param pkg The package of file being written. Value of {@code null} means that file has no package.
70     * @param fileName The file name being written. Value can't be {@code null}.
71     */
72    private void report(final JPackage pkg, final String fileName) {
73        if (fileName == null) {
74            throw new IllegalArgumentException("File name is null");
75        }
76
77        final String pkgName;
78        final String fileNameOut;
79        if (pkg != null && (pkgName = pkg.name().replace('.', File.separatorChar)).length() > 0 ) {
80            final StringBuilder sb = new StringBuilder(fileName.length() + pkgName.length() + 1);
81            sb.append(pkgName);
82            sb.append(File.separatorChar);
83            sb.append(fileName);
84            fileNameOut = sb.toString();
85        } else {
86            fileNameOut = fileName;
87        }
88
89        if(progress.isCanceled())
90            throw new AbortException();
91        progress.generatedFile(fileNameOut, current++, totalFileCount);
92    }
93}
94