Metacat

package edu.ucsb.nceas.metacat.harvesterClient;

import java.io.*;

import java.util.*;

import javax.servlet.ServletConfig;

import javax.servlet.ServletContext;

import javax.servlet.ServletException;

import javax.servlet.ServletInputStream;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

import javax.servlet.http.HttpUtils;

import javax.servlet.ServletOutputStream;

import com.oreilly.servlet.multipart.FilePart;

import com.oreilly.servlet.multipart.MultipartParser;

import com.oreilly.servlet.multipart.ParamPart;

import com.oreilly.servlet.multipart.Part;

import edu.ucsb.nceas.metacat.client.*;

import edu.ucsb.nceas.utilities.IOUtil;

/**

 *  MetUpload implements a Harvester servlet to upload a single file to Metacat

 */

public class MetUpload extends HttpServlet {

    private Metacat m;

    private String  insertXML = "";

    private String  metacatUrl;

    private String  userName = System.getProperty("user.name");

    private String  tmpDirPath = "/tmp/" + userName;

    private File    tmpDir = new File(tmpDirPath);

    /**

     *  Service requests made to the Harvester MetUpload servlet

     *

     *  @param  req   The request

     *  @param  res   The response

     *  @throws IOException

     */

    public void service(HttpServletRequest req,

                        HttpServletResponse res) throws IOException {

        int             strLen;

        File            tmpFile;

        FilePart        fPart = null;

        FileReader      fr;

        Hashtable       formElements = new Hashtable();

        HttpSession     sess;

        MultipartParser parser;

        ParamPart       pPart;

        Part            aPart;

        PrintWriter     out = null;

        String          docid = "";

        String          tmpFileName = "";

        String          tmpFilePath = "";

        String          fieldName;

        String          fieldValue;

        String          metacatResponse;

        String          password = null;

        String          revision;

        String          updateResult;

        String          username = null;

        StringReader    sr;

        if (userName.equals("harvest")) {

            metacatUrl = "http://knb.lternet.edu:8888/knb/servlet/metacat";

        }

        else {

            metacatUrl = "http://knb.lternet.edu:8088/knb/servlet/metacat";

        }

        try {

            out = res.getWriter();

            parser = new MultipartParser(req, 1024 * 1024);

            while ((aPart = parser.readNextPart()) != null) {

                if (aPart.isParam() == true) {

                    pPart = (ParamPart) aPart;

                    fieldName = pPart.getName();

                    fieldValue = pPart.getStringValue();

                    if (fieldName != null) {

                        if (fieldValue == null) {

                            fieldValue = "";

                        }

                        formElements.put(fieldName, fieldValue);

                        if (fieldName.equals("docid")) {

                            docid = fieldValue;

                        }

                    }

                }

                else if (aPart.isFile() == true) {

                    fPart = (FilePart)aPart;

                    fieldName = fPart.getName();

                    tmpFileName = fPart.getFileName();

                    if (tmpFileName != null) {

                        // Create the temporary directory if it doesn't exist

                        try {

                            if (!tmpDir.exists()) {

                                tmpDir.mkdirs();

                            }

                        } catch (SecurityException se) {

                            out.println("Can't create temporary directory: " +

                                        tmpDir.getPath());

                            out.println(se.getMessage());

                        }

                        // Write the image to a file

                        tmpFile = new File(tmpDirPath, tmpFileName);

                        fPart.writeTo(tmpFile);

                        tmpFilePath = tmpDirPath + "/" + tmpFileName;

                        formElements.put(fieldName, tmpFileName);

                        break;

                    }

                }

            }

            sess = req.getSession(true);

            username = (String) sess.getAttribute("Musername");

            password = (String) sess.getAttribute("Mpassword");

            try {

                fr = new FileReader(tmpFilePath);

                insertXML = IOUtil.getAsString(fr, true);

            } catch (IOException ioe) {

                out.println("Error reading file: " + tmpFilePath);

                out.println(ioe.getMessage());

            }

            try {

                m = MetacatFactory.createMetacatConnection(metacatUrl);

            } catch (MetacatInaccessibleException mie) {

                out.println("Metacat connection failed.");

                out.println(mie.getMessage());

            }

            try {

                m.login(username, password);

                strLen = docid.length();

                revision = docid.substring((strLen - 2), strLen);

                sr = new StringReader(insertXML);

                if (revision.equals(".1")) {

                    metacatResponse = m.insert(docid, sr, null);

                }

                else {

                    metacatResponse = m.update(docid, sr, null);

                }

                out.println(metacatResponse);

            } catch (Exception e) {

				out.println("General exception:\n" + e.getMessage());

            }

            // Clean up the temporary file

            if (!tmpFileName.equals("")) {

                tmpFile = new File(tmpDirPath, tmpFileName);

                tmpFile.delete();

            }

        } catch (Exception e) {

            out.println("An error occurred while attempting to upload.");

            e.printStackTrace(out);

        }

        out.flush();

    }

}