Pithos_GSS » GSS

/*

 * Copyright 2009 Electronic Business Systems Ltd.

 *

 * This file is part of GSS.

 *

 * GSS is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * GSS is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with GSS.  If not, see <http://www.gnu.org/licenses/>.

 */

package gr.ebs.gss.client;

import gr.ebs.gss.client.rest.PostCommand;

import gr.ebs.gss.client.rest.RestCommand;

import gr.ebs.gss.client.rest.RestException;

import gr.ebs.gss.client.rest.resource.FileResource;

import gr.ebs.gss.client.rest.resource.FolderResource;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import com.google.gwt.core.client.GWT;

import com.google.gwt.event.dom.client.ClickEvent;

import com.google.gwt.event.dom.client.ClickHandler;

import com.google.gwt.gears.client.Factory;

import com.google.gwt.gears.client.desktop.Desktop;

import com.google.gwt.gears.client.desktop.File;

import com.google.gwt.gears.client.desktop.OpenFilesHandler;

import com.google.gwt.gears.client.httprequest.HttpRequest;

import com.google.gwt.gears.client.httprequest.ProgressEvent;

import com.google.gwt.gears.client.httprequest.ProgressHandler;

import com.google.gwt.gears.client.httprequest.RequestCallback;

import com.google.gwt.http.client.URL;

import com.google.gwt.json.client.JSONObject;

import com.google.gwt.json.client.JSONString;

import com.google.gwt.user.client.Command;

import com.google.gwt.user.client.DeferredCommand;

import com.google.gwt.user.client.ui.Button;

import com.google.gwt.user.client.ui.FlexTable;

import com.google.gwt.user.client.ui.HTML;

import com.google.gwt.user.client.ui.HasHorizontalAlignment;

import com.google.gwt.user.client.ui.HorizontalPanel;

import com.google.gwt.user.client.ui.VerticalPanel;

/**

 * The 'File upload' dialog box implementation with Google Gears support.

 */

public class FileUploadGearsDialog extends FileUploadDialog implements Updateable {

	protected final Factory factory = Factory.getInstance();

	/**

	 * The array of files to upload.

	 */

	private File[] fileObjects;

	/**

	 * A list of files to upload, created from files array. Used to signal

	 * finished state when empty.

	 */

	protected List<File> selectedFiles = new ArrayList<File>();

	/**

	 * The list of progress bars for individual files.

	 */

	protected List<ProgressBar> progressBars = new ArrayList<ProgressBar>();

	private Button browse;

	private Button submit;

	private FlexTable generalTable;

	private Map<String, FileResource> toRename;

	protected List<HttpRequest> requests = new ArrayList<HttpRequest>();

	/**

	 * The widget's constructor.

	 */

	public FileUploadGearsDialog() {

		// Set the dialog's caption.

		setText("File upload");

		setAnimationEnabled(true);

		// Create a panel to hold all of the dialog widgets.

		VerticalPanel panel = new VerticalPanel();

		final HTML info = new HTML("You may select one or more files to upload.");

		info.addStyleName("gss-uploadNote");

		panel.add(info);

		// Add an informative label with the folder name.

		Object selection = GSS.get().getFolders().getCurrent().getUserObject();

		folder = (FolderResource) selection;

		browse = new Button("Browse...");

		HorizontalPanel fileUploadPanel = new HorizontalPanel();

		fileUploadPanel.add(browse);

		generalTable = new FlexTable();

		generalTable.setText(0, 0, "Folder");

		generalTable.setText(1, 0, "File");

		generalTable.setText(0, 1, folder.getName());

		generalTable.setWidget(1, 1, fileUploadPanel);

		generalTable.getCellFormatter().setStyleName(0, 0, "props-labels");

		generalTable.getCellFormatter().setStyleName(1, 0, "props-labels");

		generalTable.getCellFormatter().setStyleName(0, 1, "props-values");

		generalTable.getCellFormatter().setStyleName(1, 1, "props-values");

		generalTable.setCellSpacing(4);

		panel.add(generalTable);

		// Create a panel to hold the buttons.

		HorizontalPanel buttons = new HorizontalPanel();

		submit = new Button("Upload");

		submit.addClickHandler(new ClickHandler() {

			@Override

			public void onClick(ClickEvent event) {

				prepareAndSubmit();

			}

		});

		submit.setEnabled(false);

		buttons.add(submit);

		buttons.setCellHorizontalAlignment(submit, HasHorizontalAlignment.ALIGN_CENTER);

		// Create the 'Cancel' button, along with a listener that hides the

		// dialog when the button is clicked.

		Button cancel = new Button("Cancel", new ClickHandler() {

			@Override

			public void onClick(ClickEvent event) {

				cancelUpload();

			}

		});

		buttons.add(cancel);

		buttons.setCellHorizontalAlignment(cancel, HasHorizontalAlignment.ALIGN_CENTER);

		buttons.setSpacing(8);

		buttons.addStyleName("gss-DialogBox");

		browse.addClickHandler(new ClickHandler() {

			@Override

			public void onClick(ClickEvent event) {

				Desktop desktop = factory.createDesktop();

				desktop.openFiles(new OpenFilesHandler() {

					@Override

					public void onOpenFiles(OpenFilesEvent ofevent) {

						fileObjects = ofevent.getFiles();

						selectedFiles.addAll(Arrays.asList(fileObjects));

						for (int i = 0; i< selectedFiles.size(); i++) {

							generalTable.setText(i+1, 0, "File");

							generalTable.setText(i+1, 1, selectedFiles.get(i).getName());

							ProgressBar progress = new ProgressBar(20, 0);

							generalTable.setWidget(i+1, 2, progress);

							progressBars.add(progress);

							generalTable.getCellFormatter().setStyleName(i+1, 0, "props-labels");

							generalTable.getCellFormatter().setStyleName(i+1, 1, "props-values");

						}

						submit.setEnabled(true);

					}

				});

			}

		});

		panel.add(buttons);

		panel.setCellHorizontalAlignment(buttons, HasHorizontalAlignment.ALIGN_CENTER);

		panel.addStyleName("gss-DialogBox");

		addStyleName("gss-DialogBox");

		setWidget(panel);

	}

	/**

	 * Cancels the file upload.

	 */

	private void cancelUpload() {

		for (HttpRequest request: requests)

			request.abort();

		hide();

	}

	/**

	 * Check whether the specified file name exists in the selected folder.

	 */

	private boolean canContinue(File file) {

		String fileName = getFilename(file.getName());

		if (getFileForName(fileName) == null)

			// For file creation, check to see if the file already exists.

			for (FileResource fileRes : files)

				if (!fileRes.isDeleted() && fileRes.getName().equals(fileName))

					return false;

		return true;

	}

	@Override

	public void prepareAndSubmit() {

		GSS app = GSS.get();

		if (selectedFiles.size() == 0) {

			app.displayError("You must select a file!");

			hide();

			return;

		}

		for (File file: selectedFiles)

			if (!canContinue(file)) {

				app.displayError("The file name " + file.getName() +

							" already exists in this folder");

				hide();

				return;

			}

		submit.setEnabled(false);

		browse.setVisible(false);

		List<String> toUpdate = new ArrayList<String>();

		toRename = new HashMap<String, FileResource>();

		for (File file: selectedFiles) {

			String fname = getFilename(file.getName());

			if (getFileForName(fname) == null) {

				// We are going to create a file, so we check to see if there is a

				// trashed file with the same name.

				FileResource same = null;

				for (FileResource fres : folder.getFiles())

					if (fres.isDeleted() && fres.getName().equals(fname))

						same = fres;

				// In that case add it to the list of files to rename.

				if (same != null)

					toRename.put(getBackupFilename(fname), same);

			} else

				// If we are updating a file add it to the list of files to update.

				toUpdate.add(fname);

		}

		if (!toUpdate.isEmpty()) {

			StringBuffer sb = new StringBuffer();

			for (String name: toUpdate)

				sb.append(name).append("<br/>");

			// We are going to update existing files, so show a confirmation dialog.

			ConfirmationDialog confirm = new ConfirmationDialog("Are you sure " +

					"you want to update the following files?<br/><i>" + sb +

					"</i>", "Update") {

				@Override

				public void cancel() {

					hide();

				}

				@Override

				public void confirm() {

					confirmRename();

				}

			};

			confirm.center();

		} else

			confirmRename();

	}

	/**

	 * Confirm the renames of synonymous files already in the trash.

	 */

	private void confirmRename() {

		if (!toRename.isEmpty()) {

			StringBuffer sb = new StringBuffer();

			for (FileResource file: toRename.values())

				sb.append(file.getName()).append("<br/>");

			ConfirmationDialog confirm = new ConfirmationDialog("Files " +

					"with the following names already exist in the trash. If" +

					" you continue,<br/>the trashed files will be renamed " +

					"automatically for you:<br/><i>" + sb + "</i>", "Continue") {

				@Override

				public void cancel() {

					hide();

				}

				@Override

				public void confirm() {

					updateTrashedFiles();

				}

			};

			confirm.center();

		} else

			uploadFiles();

	}

	/**

	 * Rename the conflicting trashed files with the supplied new names.

	 */

	private void updateTrashedFiles() {

		for (final String name: toRename.keySet()) {

			JSONObject json = new JSONObject();

			json.put("name", new JSONString(name));

			PostCommand cf = new PostCommand(toRename.get(name).getUri() + "?update=", json.toString(), 200) {

				@Override

				public void onComplete() {

					toRename.remove(name);

					uploadFiles();

				}

				@Override

				public void onError(Throwable t) {

					GSS app = GSS.get();

					GWT.log("", t);

					if (t instanceof RestException) {

						int statusCode = ((RestException) t).getHttpStatusCode();

						if (statusCode == 405)

							app.displayError("You don't have the necessary permissions");

						else if (statusCode == 404)

							app.displayError("User in permissions does not exist");

						else if (statusCode == 409)

							app.displayError("A file with the same name already exists");

						else if (statusCode == 413)

							app.displayError("Your quota has been exceeded");

						else

							app.displayError("Unable to modify file:" + ((RestException) t).getHttpStatusText());

					} else

						app.displayError("System error modifying file:" + t.getMessage());

				}

			};

			DeferredCommand.addCommand(cf);

		}

	}

	/**

	 * Checks if the renaming step for already trashed files is complete and

	 * starts file uploads.

	 */

	private void uploadFiles() {

		if (!toRename.isEmpty()) return;

		for (int i = 0; i< fileObjects.length; i++) {

			final int index = i;

			DeferredCommand.addCommand(new Command() {

				@Override

				public void execute() {

					doSend(fileObjects[index], index);

				}

			});

		}

	}

	/**

	 * Perform the HTTP request to upload the specified file.

	 */

	protected void doSend(final File file, final int index) {

		final GSS app = GSS.get();

		HttpRequest request = factory.createHttpRequest();

		requests.add(request);

		String method = "PUT";

		String path;

		final String filename = getFilename(file.getName());

		path = folder.getUri();

		if (!path.endsWith("/"))

			path = path + "/";

		path = path + encode(filename);

		String token = app.getToken();

		String resource = path.substring(app.getApiPath().length()-1, path.length());

		String date = RestCommand.getDate();

		String sig = RestCommand.calculateSig(method, date, resource, RestCommand.base64decode(token));

		request.open(method, path);

		request.setRequestHeader("X-GSS-Date", date);

		request.setRequestHeader("Authorization", app.getCurrentUserResource().getUsername() + " " + sig);

		request.setRequestHeader("Accept", "application/json; charset=utf-8");

		request.setCallback(new RequestCallback() {

			@Override

			public void onResponseReceived(HttpRequest req) {

				int state = req.getReadyState();

				if (state != 4) return;

				switch(req.getStatus()) {

					case 201: // Created falls through to updated.

					case 204:

						selectedFiles.remove(file);

						finish();

						break;

					case 403:

						SessionExpiredDialog dlg = new SessionExpiredDialog();

						dlg.center();

						break;

					case 405:

						app.displayError("You don't have permission to " +

								"upload file " + filename);

						break;

					case 409:

						app.displayError("A folder with the name " + filename +

								" already exists at this level");

						break;

					case 413:

						app.displayError("There is not enough free space " +

								"available for uploading " + filename);

						break;

					default:

						app.displayError("Error uploading file " + filename +

									": " + req.getStatus());

				}

			}

		});

		request.getUpload().setProgressHandler(new ProgressHandler() {

			@Override

			public void onProgress(ProgressEvent event) {

				double pcnt = (double) event.getLoaded() / event.getTotal();

				progressBars.get(index).setProgress((int) Math.floor(pcnt * 100));

			}

		});

		request.send(file.getBlob());

	}

	/**

	 * Perform the final actions after the files are uploaded.

	 */

	protected void finish() {

		if (!selectedFiles.isEmpty()) return;

		hide();

		GSS.get().showFileList(true);

		GSS.get().getStatusPanel().updateStats();

	}

	/**

	 * Same as URL.encode, but also encode apostrophe since browsers aren't

	 * consistent about it (FF encodes, IE does not).

	 */

	protected String encode(String decodedURL) {

		String retv = decodedURL.replaceAll("@", "_"); // Replace bad character

		retv = URL.encodeComponent(retv);

		retv = retv.replaceAll("'", "%27");

		return retv;

	}

}