FileUploader.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 java.util.Enumeration;
import java.util.Set;

import cedar.jetweb.*;
import cedar.jetweb.html.*;
import cedar.jetweb.model.paper.Paper;
import cedar.jetweb.model.plots.RealPlot;
import cedar.jetweb.model.Model;
import cedar.jetweb.model.plots.PlotSelection;
import cedar.jetweb.model.RunSeries;
import cedar.jetweb.job.LogFile;

import com.oreilly.servlet.multipart.*;

import cedar.jetweb.db.DBConnectionManager;
import cedar.jetweb.db.DBObjectManager;
import cedar.jetweb.db.DBConfig;
import cedar.jetweb.db.DBGeneratorManager;
import cedar.jetweb.generator.Generator;
import cedar.jetweb.generator.GeneratorWrapper;
import cedar.jetweb.xml.JetWebGeneratorReader;
import cedar.jetweb.xml.JetWebLogReader;


public class FileUploader 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 {
        upload(request,response);
    }

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


    public void upload(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, java.io.IOException {
      
    System.out.println("Called upload");
    //PrintWriter out = response.getWriter();

    out = response.getWriter();

    boolean isXML       = false;
    boolean isPaperText = false;
    boolean isDefaults = false;
    boolean isModel = false;
    int maxPlotNum = 0;
    String message = "";
    response.setContentType("text/html");

    try {
        MultipartParser mp = 
        new MultipartParser(request, 10*1024*1024); // 10MB
        Part part;

        HashMap<String,String> parameters = new HashMap<String,String>();
        
        while ((part = mp.readNextPart()) != null) {
        String name = part.getName();
        String plotNum = "";

        // Handle parameter parts. Determine the directory the file
        // is to be written into, the paper ID, and whether this is an XML
        // upload or not.
        if (part.isParam()) {
            ParamPart paramPart = (ParamPart) part;
            String value        = paramPart.getStringValue();
            if(name.equals("MODELXML")){
            isModel = true;
            }else if (name.equals("DEFAULTSUPLOAD")){
            isDefaults = true;
            }else if (name.equals("PLOTUPLOAD")) {
            isXML = true;
            } else if (name.equals("PAPERTEXTUPLOAD")) {
            isPaperText = true;
            } else if (name.equals("DIRECTORY")) {
            destinationDirName = value;
            } else if (name.equals("PAPERID")) {
            try {
                paperId= Integer.parseInt(value);       
            } catch (Exception e) {
                HTMLErrorWriter.write(out,e);
            }
            } else {
            parameters.put(name,value);
            }

        }

        if(part.isFile()){
            FilePart filePart = (FilePart) part;
            
            destinationDir = new File(destinationDirName);
            if (filePart.getFileName()!=null){
            System.out.println
                ("Directory is "+destinationDirName+". File is "
                 +filePart.getFileName());
            
            if (! destinationDir.exists() ) {
                System.out.println("ERROR: Specified dir = "
                           +destinationDirName
                           +" DOES NOT EXIST.");
            }
            
            if (isXML){
                
                try {
                message = doPlotXMLUpload(filePart,parameters);
                } catch (JetWebException jwe){
                HTMLErrorWriter.write(out,jwe);
                return;
                }
            } else if(isDefaults){
                try{
                Generator gen = 
                    doDefaultsXMLUpload(filePart, parameters);
                //HTMLGeneratorWriter.installedMessage(out,gen);
                }catch (JetWebException jwe){
                HTMLErrorWriter.write(out,jwe);
                }
                return;
            }else if(isModel){
                try{
                doModelUpload(filePart, request);
                }catch (JetWebException err){
                HTMLErrorWriter.write(out, err);
                }
                return;
            }else {
                
                JetWebException jwe = new 
                JetWebException("Unknown file upload request."," ");
                HTMLErrorWriter.write(out,jwe);
                return;
            }
            }
        

        }
        }
    
        System.out.println("Paper ID="+paperId);
        
        // When we get here, we have all the parameters 
        // and so can start altering the plots.
        try {
        
        Paper paper = new Paper(paperId);       
        ServletUtils.deletePlots(parameters,paper);
        ServletUtils.modifyPlots(parameters,paper);
        ServletUtils.addPlot(parameters,paper,"NEW");
        System.out.println("back from ServeletUtils.addPlot()");
        HTMLPaperWriter pm = new HTMLPaperWriter
            (out, JetWebConfig.mainServletPath);
        System.out.println("Created new HTMLPaperWriter");

        pm.writePaperForm(paper," ");

        System.out.println("written blank message");

        pm.writePlotsForm(paper,message);
        System.out.println("written message " + message);

        pm.close();
        System.out.println("closed HTMLPaperWriter");

        return;
        
        } catch (JetWebException jwe){
        HTMLErrorWriter.write(out,jwe);
        }
    }catch (IOException lEx) {
        //this.getServletContext().log("ERROR: reading or saving file"+lEx.toString());
    }
    }


    private String doPlotXMLUpload(FilePart filePart,
                   HashMap<String,String> parameters) 
    throws JetWebException, IOException {

    String message   = "";
    String fileName = filePart.getFileName();

    String fileParm = filePart.getName();
    String plotNum  = fileParm.substring(fileParm.indexOf("-")+1);

    if (paperId != 0){

        System.out.println("Plot number "+plotNum);
        System.out.println("File name " + fileName );
        System.out.println("HZsteer plot id " + parameters.get("HZSTEERPLOTID"));

        System.out.println(parameters);

        if (fileName.equals("plot"+plotNum+".xml")) {

        long size = filePart.writeTo(destinationDir);
            
        System.out.println("replacing "+fileName);
        message = message+"<br>Successfully replaced file: "+fileName;
        parameters.put("UPDATED"+plotNum,"done");
        } else if(fileName.startsWith("n0") && fileName.endsWith(".aida") &&
              parameters.get("HZSTEERPLOTID")!=null &&
              !parameters.get("HZSTEERPLOTID").equals("")){
        //Then it's a file from hzsteer output.

        String plotDir = JetWebConfig.rootDirectory + "/" + 
            JetWebConfig.plotCacheDirName;

        System.out.println("found HZSteer output - writing to " +
                   plotDir);

        filePart.writeTo(new File(plotDir));

        //we save the file name in the parameters so that 
        //ServletUtils.addPlot can find it later.

        parameters.put("HZSTEERPLOT", 
                   plotDir + "/" + 
                   fileName.substring(fileName.lastIndexOf("/")+1));

        }else{
        System.out.println("Invalid file name");
        message = message+"XML file not replaced due to invalid name: "+fileName;
        }

    } else {
        JetWebException jwe = new 
        JetWebException("Paper ID is zero.","="+paperId);
        HTMLErrorWriter.write(out,jwe);
    }

    return message;
        
    }

    private Generator doDefaultsXMLUpload(FilePart filePart,
                   HashMap<String,String> parameters) 
    throws JetWebException, IOException {

    System.out.println("Called the xml defaults uploader");

    String fileName = filePart.getFileName();

    filePart.writeTo(destinationDir);

    GeneratorWrapper wrapper = null;

    System.out.println("opening xml file "+destinationDir+"/"+fileName);

    try{
        JetWebGeneratorReader reader = 
        new JetWebGeneratorReader(destinationDir+"/"+fileName);
        wrapper = reader.getGeneratorWrapper();
        
    }catch(Exception e){
        throw new JetWebException(e);
        }

    DBConnectionManager connManager = DBConfig.getDBConnectionManager();
    DBObjectManager objManager = new DBObjectManager();
    objManager.passDBConnection(connManager);
    String tableName =wrapper.toString()+"_defaults";

    Generator gen = wrapper.generator();
    String genName = gen.toString();
    int ii = genName.indexOf(".");

    while(ii>-1){
        genName = genName.substring(0,ii)+genName.substring(ii+1);
        ii = genName.indexOf(".");
    }

    ii = genName.indexOf("-");
    
    String genVersion = genName.substring(ii+1);
    genName = genName.substring(0, ii);

    Set<String> knownVersions = DBGeneratorManager.getVersions(genName);

    if(!knownVersions.contains(genVersion)){


        objManager.writeObject(wrapper, tableName, DBConfig.JETWEB);
                
        // Now add the row to the generators table.
        System.out.println("using generator id = "+gen.getGeneratorId());
        DBGeneratorManager.add(wrapper.generator());
        
        System.out.println("used generator id = "+gen.getGeneratorId());

        HTMLGeneratorWriter.installedMessage(out,wrapper.generator());
        

    }else{
        HTMLGeneratorWriter.alreadyInstalledMessage
        (out, wrapper.generator());
    }
    return wrapper.generator();
    }

    private Model doModelUpload(FilePart filePart, 
                HttpServletRequest request)
    throws JetWebException, IOException{

    String fileName = filePart.getFileName();
    filePart.writeTo(destinationDir);

    try{
        JetWebLogReader reader = 
        new JetWebLogReader(destinationDir+"/"+fileName);
        LogFile log = new LogFile(destinationDir+"/"+fileName);
        reader.fillLog(log);
        RunSeries rs = log.getRunSeries();
        PlotSelection selection = ServletUtils.makePlotSelection(request);
        // make this check only the sum plots!
        if (selection.fitIsEmpty()) { selection.setFitDefault(); }
        // make this check only the sum plots!
        if (selection.sumIsEmpty()) { selection.setSumDefault(); }

        ServletUtils.performResultSearch
        (rs, out, JetWebConfig.searchServletPath, selection, request);

    }catch(Exception e){
        throw new JetWebException(e);
    }

    return new Model();
    }
}

Generated Wed Jan 17 09:14:27 GMT 2007