fileaddactionservlet.java

来自「web版的SVN客户端」· Java 代码 · 共 160 行

/*
 * Copyright (c) 2004, 2005 Polarion Software, All rights reserved.
 * Email: [e-mail]
 *
 * This program and the accompanying materials are made available under the
 * terms of the Apache License, Version 2.0 (the "License"). You may not use
 * this file except in compliance with the License. Copy of the License is
 * located in the file LICENSE.txt in the project distribution. You may also
 * obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
 *
 *
 * POLARION SOFTWARE MAKES NO REPRESENTATIONS OR WARRANTIES
 * ABOUT THE SUITABILITY OF THE SOFTWARE, EITHER EXPRESSED OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. POLARION SOFTWARE
 * SHALL NOT BE LIABLE FOR ANY DAMAGES SUFFERED BY LICENSEE AS A RESULT
 * OF USING, MODIFYING OR DISTRIBUTING THIS SOFTWARE OR ITS DERIVATIVES.
 */
package org.polarion.svnwebclient.web.servlet;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.polarion.svnwebclient.SVNWebClientException;
import org.polarion.svnwebclient.configuration.ConfigurationProvider;
import org.polarion.svnwebclient.data.AuthenticationException;
import org.polarion.svnwebclient.data.DataProviderException;
import org.polarion.svnwebclient.data.IDataProvider;
import org.polarion.svnwebclient.data.model.DataChangedElement;
import org.polarion.svnwebclient.data.model.DataRepositoryElement;
import org.polarion.svnwebclient.util.FileUtil;
import org.polarion.svnwebclient.util.Uploader;
import org.polarion.svnwebclient.web.AttributeStorage;
import org.polarion.svnwebclient.web.controller.ChangeConfirmation;
import org.polarion.svnwebclient.web.model.Navigation;
import org.polarion.svnwebclient.web.model.data.ChangeResult;
import org.polarion.svnwebclient.web.resource.Links;
import org.polarion.svnwebclient.web.support.FormParameters;
import org.polarion.svnwebclient.web.support.RequestHandler;
import org.w3c.util.UUID;

/**
 * 
 * @author <A HREF="mailto:[e-mail]">Polarion Software </A>
 */
public class FileAddActionServlet extends AbstractServlet {
    protected static final long CURRENT_REVISION = -1;    

    private static final long serialVersionUID = -744812068956415395L;    
    protected static String DESTINATION_DIRECTORY = "destinationDirectory";
    
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        this.execute(request, response);
    }
    
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        this.execute(request, response);
    }
    
    protected void executeSVNOperation(IDataProvider dataProvider) throws SVNWebClientException {
        ChangeResult changeResult = null;        
        String temporaryDirectory = ConfigurationProvider.getInstance().getTempDirectory();
        String destinationDirectory = temporaryDirectory + "/" + UUID.getUUID();
        Map parameters = null;
        String name = null;
        String comment = null;
        try {
        	parameters = Uploader.doPost(this.state.getRequest(), this.state.getResponse(), destinationDirectory, temporaryDirectory);
        	name = (String) parameters.get(FormParameters.FILE_NAME);
            comment = (String) parameters.get(FormParameters.COMMENT);
            
            AttributeStorage handler = AttributeStorage.getInstance();
            HttpSession session  = this.state.getRequest().getSession();
            handler.addParameter(session, FormParameters.FILE_NAME, name);
            handler.addParameter(session, FormParameters.COMMENT, comment);
            handler.addParameter(session, FileAddActionServlet.DESTINATION_DIRECTORY, destinationDirectory);
                       
        } catch (Exception e) {
        	AttributeStorage  stotage = AttributeStorage.getInstance();
        	HttpSession session  = this.state.getRequest().getSession();
        	name = (String) stotage.getParameter(session, FormParameters.FILE_NAME);
        	comment =  (String) stotage.getParameter(session, FormParameters.COMMENT);
        	destinationDirectory = (String) stotage.getParameter(session,FileAddActionServlet.DESTINATION_DIRECTORY);
        }
        try {

            long revision = dataProvider.getHeadRevision();
            DataRepositoryElement dataRepositoryElement = null;

            String fileUrl = this.requestHandler.getUrl();
            if (!fileUrl.endsWith("/")) {
                fileUrl += "/";
            }
            fileUrl += name;
            
            dataRepositoryElement = dataProvider.getInfo(fileUrl, revision);
            
            Navigation navigation = new Navigation(this.requestHandler.getUrl(), this.requestHandler.getLocation(),FileAddActionServlet.CURRENT_REVISION, false);
            
            if (dataRepositoryElement != null) {
                List elements = new ArrayList();
                ChangeResult.Element element = new ChangeResult.Element();
                element.setAuthor(dataRepositoryElement.getAuthor());
                element.setComment(dataRepositoryElement.getComment());
                element.setDate(dataRepositoryElement.getDate());
                element.setDirectory(false);
                element.setName(dataRepositoryElement.getName());
                element.setRevision(dataRepositoryElement.getRevision());
                elements.add(element);
                String message = "File " + name + " already exists in specified location";
                changeResult = new ChangeResult(false, message, elements, navigation);
            } else {
                DataChangedElement changedElement = null;
                changedElement = dataProvider.addFile(this.requestHandler.getUrl(), destinationDirectory + "/" + name, comment);
                
                List elements = new ArrayList();
                ChangeResult.Element element = new ChangeResult.Element();
                element.setAuthor(changedElement.getAuthor());
                element.setComment(changedElement.getComment());
                element.setDate(changedElement.getDate());
                element.setDirectory(false);
                element.setName(changedElement.getName());
                element.setRevision(changedElement.getRevision());
                elements.add(element);
                String message = "File " + name + " was succesfully added";
                changeResult = new ChangeResult(true, message, elements, navigation);            
            }
        } catch (AuthenticationException ae) {
        	throw ae;
        } catch (DataProviderException e) {
        	AttributeStorage.getInstance().cleanSession(this.state.getRequest().getSession());
            FileUtil.deleteDirectory(new File(destinationDirectory));    
            throw e;
        }
                
        AttributeStorage.getInstance().cleanSession(this.state.getRequest().getSession());
        FileUtil.deleteDirectory(new File(destinationDirectory));    
        this.state.getRequest().getSession().setAttribute(ChangeConfirmation.CHANGE_RESULT, changeResult);
        RequestDispatcher dispatcher = this.state.getRequest().getRequestDispatcher(Links.CHANGE_CONFIRMATION);
        try {
            dispatcher.forward(this.state.getRequest(), this.state.getResponse());
        } catch (Exception e) {
            throw new SVNWebClientException(e);
        }
    }

    protected RequestHandler getRequestHandler(HttpServletRequest request) {
        return new RequestHandler(request);
    }
}