/*

  * Copyright 1999-2001,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.commons.latka.jelly;

 import java.io.IOException;

 import java.util.HashMap;

 import java.util.Map;

 import org.apache.commons.jelly.JellyTagException;

 import org.apache.commons.jelly.TagSupport;

 import org.apache.commons.jelly.XMLOutput;

 import org.apache.commons.latka.DefaultLatkaEventInfo;

 import org.apache.commons.latka.Latka;

 import org.apache.commons.latka.LatkaException;

 import org.apache.commons.latka.XMLReporter;

 import org.apache.commons.latka.event.LatkaEventInfo;

 import org.apache.commons.latka.event.LatkaEventListener;

 import org.apache.commons.latka.event.SuiteCompletedEvent;

 /**

  *

  * @author  Morgan Delagrange

  */

 public class SuiteTag extends TagSupport {

     protected String _defaultHost = null;

     protected int _defaultPort = -1;

     protected String _defaultProxyHost = null;

     protected int _defaultProxyPort = -1;

     protected String _label = null;

     protected SuiteSettings _settings = null;

     protected Map _sessionCache = new HashMap();

     /**

      *  Wraps Latka tests, provides some defaults for host, port etc.

      *

      * @param xmlOutput a place to write output

      * @throws JellyTagException if the suite fails, either as a result of test failures

      *    or from problems executing tags, generating reports, etc.

      */

     public void doTag(XMLOutput xmlOutput) throws JellyTagException {

         // if an enclosing tag does not specify a listener, provide a default

         boolean defaultListener = false;

         LatkaEventListener listener = null;

         LatkaEventInfo eventInfo = null;

         {

             listener = 

                 JellyUtils.getInstance().getLatkaEventListener(getContext());

             if (listener == null) {

                 listener = new XMLReporter();

                 defaultListener = true;

             }

             eventInfo = new DefaultLatkaEventInfo(listener);

             JellyUtils.getInstance().setLatkaEventInfo(getContext(), eventInfo);

         }

         _settings = 

             new SuiteSettings(_defaultHost, _defaultPort, _defaultProxyHost, 

                               _defaultProxyPort);

         invokeBody(xmlOutput);

         eventInfo.suiteCompleted(new SuiteCompletedEvent());

         JellyUtils.getInstance().removeLatkaEventInfo(getContext());

         if (defaultListener == true) {

             try {

                 Latka latka = new Latka();

                 String transformedReport = 

                   latka.transformXML(((XMLReporter) listener).getDocumentAsString());

                 System.out.println(transformedReport);

             } catch (LatkaException e) {

                 throw new JellyTagException("could not generate latka report",e);

             } catch (IOException e) {

                 throw new JellyTagException("could not generate latka report",e);

             }

             if (eventInfo.didSuiteSucceed() == false) {

                 throw new JellyTagException("SUITE FAILED");

             }

         }

     }

     public SuiteSettings getSuiteSettings() {

         return _settings;

     }

     public Map getSessionCache() {

         return _sessionCache;

     }

     /**

      * Setter for defaultHost

      * 

      * @param defaultHost

      *               defaultHost for all requests

      */

     public void setDefaultHost(String defaultHost) {

         _defaultHost = defaultHost;

     }

     /**

      * Setter for defaultPort

      * 

      * @param defaultPort

      *               defaultPort for all requests

      */

     public void setDefaultPort(int defaultPort) {

         _defaultPort = defaultPort;

     }

     /**

      * Setter for defaultProxyHost

      * 

      * @param defaultHost

      *               defaultProxyHost for all requests

      */

     public void setDefaultProxyHost(String defaultHost) {

         _defaultProxyHost = defaultHost;

     }

     /**

      * Setter for defaultProxyPort

      * 

      * @param defaultPort

      *               defaultProxyPort for all requests

      */

     public void setDefaultProxyPort(int defaultPort) {

         _defaultProxyPort = defaultPort;

     }

     /**

      * Set the label for this suite

      * 

      * @param label  suite label

      */

     public void setLabel(String label) {

         _label = label;

     }

 }