com.sun.xml.rpc.processor.util.IndentingWriter Maven / Gradle / Ivy
/*
* Copyright (c) 1997, 2018 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the
* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
* version 2 with the GNU Classpath Exception, which is available at
* https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
*/
package com.sun.xml.rpc.processor.util;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.Writer;
import java.nio.charset.Charset;
import java.nio.charset.CharsetEncoder;
import java.text.MessageFormat;
import com.sun.xml.rpc.processor.generator.GeneratorException;
/**
*
* @author JAX-RPC Development Team
*/
public class IndentingWriter extends BufferedWriter {
private boolean beginningOfLine = true;
private int currentIndent = 0;
private int indentStep = 4;
public IndentingWriter(Writer out) {
super(out);
}
public IndentingWriter(Writer out,int step) {
this(out);
if (indentStep < 0) {
throw new IllegalArgumentException("negative indent step");
}
indentStep = step;
}
public void write(int c) throws IOException {
checkWrite();
super.write(c);
}
public void write(char[] cbuf, int off, int len) throws IOException {
if (len > 0) {
checkWrite();
}
super.write(cbuf, off, len);
}
public void write(String s, int off, int len) throws IOException {
if (len > 0) {
checkWrite();
}
super.write(s, off, len);
}
public void newLine() throws IOException {
super.newLine();
beginningOfLine = true;
}
protected void checkWrite() throws IOException {
if (beginningOfLine) {
beginningOfLine = false;
int i = currentIndent;
while (i > 0) {
super.write(' ');
-- i;
}
}
}
protected void indentIn() {
currentIndent += indentStep;
}
protected void indentOut() {
currentIndent -= indentStep;
if (currentIndent < 0) {
currentIndent = 0;
}
}
public void pI() {
indentIn();
}
public void pO() {
indentOut();
}
public void pI(int levels) {
for (int i = 0; i < levels; ++i) {
indentIn();
}
}
public void pO(int levels) {
for (int i = 0; i < levels; ++i) {
indentOut();
}
}
public void p(String s) throws IOException {
int tabCount = 0;
for (int i = 0; i < s.length(); ++i) {
if (s.charAt(i) == '\t') {
++tabCount;
indentIn();
}
}
String printStr = s.substring(tabCount);
boolean canEncode = true;
//bug fix: 4839636
try{
if(!canEncode(printStr)) {
canEncode = false;
}
} catch (Throwable t) {
// there was some exception, what should we do?
// lets ignore it for now and proceed with the code generation!
}
if(!canEncode) {
throw new GeneratorException(
"generator.indentingwriter.charset.cantencode", printStr);
}
write(printStr);
while (tabCount-- > 0) {
indentOut();
}
}
/**
* Check if encode can handle the chars in this string.
*
*/
protected boolean canEncode(String s) {
final CharsetEncoder encoder =
Charset.forName(System.getProperty("file.encoding")).newEncoder();
char[] chars = s.toCharArray();
for (int i=0; i