FileDownloader.java

Go to the documentation of this file.

package cedar.jetweb.servlet;

import java.util.HashMap;

import javax.servlet.*;
import javax.servlet.http.*;
import java.io.*;

import cedar.jetweb.*;
import cedar.jetweb.model.*;
import cedar.jetweb.html.*;
import cedar.jetweb.generator.Generator;

import org.apache.tools.tar.*;
import org.apache.tools.ant.taskdefs.Tar.TarFileSet;
import org.apache.tools.ant.taskdefs.Tar;


public class FileDownloader extends HttpServlet {

    private String destinationDirName;
    private File destinationDir;

    private PrintWriter out;
    private int paperId;

    public void init(ServletConfig config) throws ServletException {
    super.init(config);
    // Set up configuration with servlet mode on. 
    JetWebConfig.init();
    }

    protected void doPost(HttpServletRequest request, 
              HttpServletResponse response)
    throws ServletException, java.io.IOException {
        download(request,response);
    }

    protected void doGet(HttpServletRequest request, 
              HttpServletResponse response)
    throws ServletException, java.io.IOException {
        download(request,response);
    }


    public void download(HttpServletRequest request, HttpServletResponse res) {
      
    String tarRoot;
    File   tarDir;

    // Could test on the servlet request here and make different tarballs as required.

    try{

        if (true) {
        tarRoot = "HZSteerCards";
        tarDir = makeCards(request,tarRoot);
        }

        // Set the headers.
        res.setContentType("application/x-download");
        res.setHeader("Content-Disposition", "attachment; filename=" + tarRoot+".tar");     

        // Send the file.
        OutputStream out = res.getOutputStream();
        TarOutputStream tos = new TarOutputStream(out);

        TarEntry te = new TarEntry(tarDir);
        for (TarEntry newte : te.getDirectoryEntries()){
        for (TarEntry newte2 : newte.getDirectoryEntries()){

            if (!newte2.isDirectory()) {
            TarEntry go = new TarEntry(tarRoot+"/"
                           +newte2.getFile().getParentFile().getName()
                           +"/"+newte2.getFile().getName());
            go.setSize(newte2.getFile().length());
            tos.putNextEntry(go);
            //System.out.println("Loop1 "+go.getName());
            FileInputStream fIn = new FileInputStream(newte2.getFile());
            byte[] buffer = new byte[8 * 1024];
            int count = 0;
            System.out.println(go.getName());
            do {
                tos.write(buffer, 0, count);
                count = fIn.read(buffer, 0, buffer.length);
            } while (count != -1);
            tos.closeEntry();           
            }
        }
        }

        tos.close();

    } catch (Exception e) {
        try {
        res.setContentType("text/html");
        res.setHeader("Content-Disposition","");        
        e.printStackTrace(System.out);
        e.printStackTrace(res.getWriter());
        HTMLErrorWriter.write(res.getWriter(),e);
        } catch (IOException io) {
        System.out.println(io);
        }
    }
    }

    // Build a tarball of card files based upon the request.
    private static File makeCards(HttpServletRequest request, String tarRoot) 
    throws IOException, JetWebException {

    File workDir = File.createTempFile(tarRoot,"");
    workDir.delete();
    workDir.mkdir();
    
    // loop over all known generators
    for (String name : Generator.allNames()){
        
        // make a directory called name-version
        for (String version : Generator.allVersions(name)){
        
        File genDir = new File(workDir.getPath(),name+"-"+version);
        genDir.mkdir();
        
        Model model = new Model();
        model.getProtonPDFList().add(JetWebConfig.defaultProtonPDF);
        model.getPhotonPDFList().add(JetWebConfig.defaultPhotonPDF);
        
        Generator gen = Generator.Maker(name,version);
        gen.setSoftDefaults();
        
        // loop over all known valid processtypes and write a card file
        for (MCProcessType proc: MCProcessType.getAll(true)){
            String cardName = genDir.getPath()+"/"+proc+".input";
            
            BufferedWriter cards = 
            new BufferedWriter(new FileWriter(cardName) );
            gen.writeSteering(cards,model,proc);
            cards.newLine();
            cards.close();
        }
        }
    }
    // end loops
    return workDir;
    }
}


                

Generated Wed Jan 17 09:14:27 GMT 2007