Java tutorial
/******************************************************************************* * Copyright (c) 2003, 2011 IBM Corporation and others. * All rights reserved. This program and the accompanying materials * are made available under the terms of the Eclipse Public License v1.0 * which accompanies this distribution, and is available at * http://www.eclipse.org/legal/epl-v10.html * * Contributors: * IBM Corporation - Initial API and implementation * * 2005-11-25 Arthur Ryman, ryman@ca.ibm.com * - fixed bug 118102: set xmlTagMissing correctly each time setContent() is called *******************************************************************************/ package org.eclipse.wst.internet.monitor.ui.internal.viewers; import java.io.*; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.transform.*; import javax.xml.transform.dom.DOMSource; import javax.xml.transform.stream.StreamResult; import org.eclipse.jface.resource.JFaceResources; import org.eclipse.swt.SWT; import org.eclipse.swt.custom.StackLayout; import org.eclipse.swt.widgets.Composite; import org.eclipse.swt.widgets.Display; import org.eclipse.swt.widgets.Label; import org.eclipse.swt.widgets.Text; import org.eclipse.ui.PlatformUI; import org.eclipse.wst.internet.monitor.ui.internal.ContextIds; import org.eclipse.wst.internet.monitor.ui.internal.Messages; import org.eclipse.wst.internet.monitor.ui.internal.MonitorUIPlugin; import org.eclipse.wst.internet.monitor.ui.internal.provisional.ContentViewer; import org.w3c.dom.*; import org.xml.sax.*; /** * XML Viewer. */ public class XMLViewer extends ContentViewer { protected Composite viewerComp; protected StackLayout layout; protected Text messageText; protected Label messageLabel; protected boolean xmlTagMissing = false; protected boolean setEncoding = false; protected boolean missingEncoding = false; protected String originalEncoding; protected byte[] content; /** (non-Javadoc) * @see ContentViewer#setContent(byte[]) */ public void setContent(byte[] b) { content = b; String out = ""; if (b != null) { b = MonitorUIPlugin.unzip(b); out = MonitorUIPlugin.parse(b); } String lineSeparator = System.getProperty("line.separator"); int ls = lineSeparator.length(); if (out.length() > ls) { while (out.substring(0, ls).indexOf(lineSeparator) >= 0) out = out.substring(ls, out.length()); } String out_temp = out.toLowerCase(); xmlTagMissing = !out_temp.startsWith("<?xml"); if (out.length() > 0) { String finalMsg = null; try { byte[] b1 = createDocument(out); finalMsg = new String(b1).trim(); } catch (Exception e) { // case: error parsing messageText.setVisible(false); layout.topControl = messageLabel; messageLabel.setVisible(true); messageLabel.setText(Messages.xmlViewInvalid); return; } if (xmlTagMissing && (finalMsg.toLowerCase().startsWith("<?xml version=\"1.0\" encoding=\"utf-8\"?>") || finalMsg.toLowerCase() .startsWith("<?xml version=\"1.0\" encoding=\"utf-8\" standalone=\"no\"?>") || finalMsg.toLowerCase() .startsWith("<?xml version=\"1.0\" encoding=\"utf-8\" standalone=\"yes\"?>"))) { int x = finalMsg.indexOf(">") + 1; //remove <?xml version="1.0" encoding="UTF-8"?> String Msg = finalMsg.substring(x); //remove starting newlines while (Msg.substring(0, ls).indexOf(lineSeparator) >= 0) { Msg = Msg.substring(ls, Msg.length()); } finalMsg = Msg; messageText.setText(finalMsg); } else if (setEncoding) { // change back to original encoding int begin = finalMsg.toLowerCase().indexOf("utf-8"); // location of opening " if (begin >= 0) { int last = begin + 5; // location of closing " String first_half = finalMsg.substring(0, begin); String second_half = finalMsg.substring(last); finalMsg = first_half + originalEncoding + second_half; } messageText.setText(finalMsg); } else if (missingEncoding) { // remove encoding completely int begin = finalMsg.toLowerCase().indexOf("encoding=\"utf-8\""); //location of opening " int last = begin + 16; //location of closing " String first_half = finalMsg.substring(0, begin); String second_half = finalMsg.substring(last); finalMsg = first_half + second_half; messageText.setText(finalMsg); } else messageText.setText(finalMsg); } else messageText.setText(out); messageLabel.setVisible(false); layout.topControl = messageText; messageText.setVisible(true); } /** * @see ContentViewer#getContent() */ public byte[] getContent() { return content; } /** (non-Javadoc) * @see ContentViewer#init(Composite) */ public void init(Composite parent) { viewerComp = new Composite(parent, SWT.NONE); layout = new StackLayout(); layout.marginHeight = 0; layout.marginWidth = 0; viewerComp.setLayout(layout); messageText = new Text(viewerComp, SWT.BORDER | SWT.MULTI | SWT.READ_ONLY | SWT.H_SCROLL | SWT.V_SCROLL); Display display = viewerComp.getDisplay(); messageText.setBackground(display.getSystemColor(SWT.COLOR_LIST_BACKGROUND)); messageText.setForeground(display.getSystemColor(SWT.COLOR_LIST_FOREGROUND)); messageText.setFont(JFaceResources.getTextFont()); messageText.setVisible(true); PlatformUI.getWorkbench().getHelpSystem().setHelp(messageText, ContextIds.VIEW_RESPONSE); messageLabel = new Label(viewerComp, SWT.NONE); messageLabel.setVisible(false); layout.topControl = messageText; } /* (non-Javadoc) * @#createDocument(String) */ protected byte[] createDocument(String str) throws IOException { try { DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance(); try { factory.setAttribute("http://apache.org/xml/features/allow-java-encodings", new Boolean(true)); factory.setAttribute("http://apache.org/xml/features/continue-after-fatal-error", new Boolean(true)); } catch (Exception e) { // ignore } DocumentBuilder parser = factory.newDocumentBuilder(); if (xmlTagMissing) { str = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>" + str; } else { String str_temp = str.toLowerCase(); // if encoding present, then save original encoding and change to UTF-8 int ind = str_temp.indexOf("encoding="); if (ind >= 0) { String temp1 = str.substring(ind); int beginIndex = temp1.indexOf("\"") + 1; String temp2 = temp1.substring(beginIndex); int endIndex = temp2.indexOf("\""); originalEncoding = temp2.substring(0, endIndex); if (!"utf-8".equals(originalEncoding)) setEncoding = true; } else if (ind < 0) { //if no encoding at all,then no changes to be made setEncoding = false; missingEncoding = true; } } byte[] parseArray = str.getBytes(); Document document = parser.parse(new InputSource(new ByteArrayInputStream(parseArray))); return getContents(document); } catch (Exception e) { throw new IOException("Invalid XML"); } } protected byte[] getContents(Document document) throws IOException { ByteArrayOutputStream out = new ByteArrayOutputStream(); Result result = new StreamResult(out); Source source = new DOMSource(document); try { TransformerFactory tf = TransformerFactory.newInstance(); try { tf.setAttribute("indent-number", new Integer(2)); } catch (Exception e) { e.printStackTrace(); // ignore - fails on JDK 1.4 } Transformer transformer = tf.newTransformer(); transformer.setOutputProperty(OutputKeys.INDENT, "yes"); //$NON-NLS-1$ transformer.setOutputProperty(OutputKeys.METHOD, "xml"); //$NON-NLS-1$ transformer.setOutputProperty(OutputKeys.STANDALONE, "no"); //$NON-NLS-1$ transformer.transform(source, result); } catch (TransformerConfigurationException e) { throw (IOException) (new IOException().initCause(e)); } catch (TransformerException e) { throw (IOException) (new IOException().initCause(e)); } return out.toByteArray(); } /** (non-Javadoc) * @see ContentViewer#dispose() */ public void dispose() { viewerComp.dispose(); viewerComp = null; content = null; } }