package org.webslinger.containers;

 import java.io.File;

 import java.io.IOException;

 import java.io.PrintStream;

 import java.net.URL;

 import java.net.URLClassLoader;

 import java.util.Collection;

 import java.util.Hashtable;

 import java.util.List;

 import gnu.getopt.LongOpt;

 import gnu.getopt.Getopt;

 import com.meterware.servletunit.InvocationContext;

 import com.meterware.servletunit.ServletRunner;

 import com.meterware.servletunit.ServletUnitClient;

 import com.meterware.httpunit.ClientProperties;

 import com.meterware.httpunit.GetMethodWebRequest;

 import com.meterware.httpunit.WebResponse;

 import org.webslinger.servlet.WebslingerServlet;

 import org.webslinger.servlet.WebslingerServletContextFactoryHardcoded;

 import org.webslinger.launcher.AbstractGnuoptContainer;

 import org.webslinger.launcher.Container;

 public class RunContainer extends AbstractGnuoptContainer {

     private String mountPath;

     private String hostName = "localhost";

     private ServletRunner runner = new ServletRunner();

     private ServletUnitClient suc = runner.newClient();

     private ClientProperties cp = suc.getClientProperties();

     private int count;

     private String path;

     public RunContainer() {

     }

     public void setMountPath(String mountPath) {

         this.mountPath = mountPath;

     }

     public String getMountPath() {

         return mountPath;

     }

     public void setHostName(String hostName) {

         this.hostName = hostName;

     }

     public String getHostName() {

         return hostName;

     }

     public void setCount(int count) {

         this.count = count;

     }

     public int getCount() {

         return count;

     }

     public void setPath(String path) {

         this.path = path;

     }

     public String getPath() {

         return path;

     }

     public void showHelp(PrintStream out) throws IOException {

         out.println("run help");

     }

     protected Getopt getGetopt(String[] args) {

         LongOpt[] longOpts = new LongOpt[] {

             new LongOpt("mount-path", LongOpt.REQUIRED_ARGUMENT, null, 'm'),

             new LongOpt("count", LongOpt.REQUIRED_ARGUMENT, null, 'c'),

             new LongOpt("hostname", LongOpt.REQUIRED_ARGUMENT, null, 'H'),

         };

         return new Getopt("webslinger=run", args, "H:m:c:", longOpts);

     }

     protected int processOneArg(int c, Getopt g) {

         switch (c) {

             case 'H':

                 setHostName(g.getOptarg());

                 break;

             case 'm':

                 setMountPath(g.getOptarg());

                 break;

             case 'c':

                 setCount(Integer.parseInt(g.getOptarg()));

                 return -1;

             default:

         }

         return 0;

     }

     protected boolean processRest(List<String> rest) {

         setPath(rest.remove(0));

         return true;

     }

     public void init() throws Exception {

         Hashtable<String, String> params = new Hashtable<String, String>();

         params.put("mount-path", getMountPath());

         params.put(WebslingerServlet.class.getName() + ".WebslingerServletContextFactory", WebslingerServletContextFactoryHardcoded.class.getName());

         runner.registerServlet("/*", WebslingerServlet.class.getName(), params);

         cp.setAcceptCookies(false);

         cp.setAcceptGzip(false);

         cp.setAutoRedirect(false);

         cp.setAutoRefresh(false);

         cp.setIframeSupported(false);

     }

     public void run() throws Exception {

         GetMethodWebRequest gmwr = new GetMethodWebRequest("http://" + getHostName() + "/" + getPath());

         WebResponse res = suc.getResponse(gmwr);

         if (res.getContentType().startsWith("text/")) {

             System.out.println(res.getText());

         }

         if (getCount() != 0) {

             int count = getCount();

             long start = System.currentTimeMillis();

             for (int i = 0; i < count; i++) {

                 suc.newInvocation(gmwr).service();

             }

             long end = System.currentTimeMillis();

             System.err.println("count=" + count + ", time=" + (end - start) + ", pages/sec=" + ((float) count / (end - start) * 1000));

         }

     }

 }