/**

  * Copyright 2010 The Kuali Foundation Licensed under the

  * Educational Community 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.osedu.org/licenses/ECL-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.kuali.student.common.ws.handler;

 import java.io.PrintStream;

 import java.util.Set;

 import javax.xml.namespace.QName;

 import javax.xml.ws.handler.MessageContext;

 import javax.xml.ws.handler.soap.SOAPHandler;

 import javax.xml.ws.handler.soap.SOAPMessageContext;

 import org.apache.log4j.Logger;

 public class DebugHandler implements SOAPHandler<SOAPMessageContext> {

     final Logger logger = Logger.getLogger(DebugHandler.class);

         private PrintStream err;

         private PrintStream out;

         public DebugHandler()        {

                 this(System.out, System.err);

         }

         public DebugHandler(PrintStream ps)        {

                 this(ps, ps);

         }

         public DebugHandler(PrintStream out, PrintStream err)        {

                 this.out = out;

                 this.err = err;

         }

         @Override

         public Set<QName> getHeaders() {

                 // does not handle any headers

                 return null;

         }

         @Override

         public void close(MessageContext context) {

                 // no cleanup needed

         }

         @Override

         public boolean handleFault(SOAPMessageContext context) {

                 return this.printMessage(context, this.err);

         }

         @Override

         public boolean handleMessage(SOAPMessageContext context) {

                 return this.printMessage(context, this.out);

         }

         private boolean printMessage(SOAPMessageContext context, PrintStream ps)        {

                 String message = "DebugHandler ";

                 Boolean out = (Boolean)context.get(SOAPMessageContext.MESSAGE_OUTBOUND_PROPERTY);

                 if(out.booleanValue())

                         message += "Out: ";

                 else

                         message += "In: ";

                 ps.print(message);

                 try        {

                         context.getMessage().writeTo(ps);

                         ps.println();

                 }

                 catch(Exception ex)        {

                     logger.error(this.err, ex);

                 }

                 return true;

         }

 }