code.grnet.gr Git - pithos-web-client/blob - src/gr/grnet/pithos/web/client/mysharedtree/MysharedTreeViewModel.java

   1 /*
   2  * Copyright 2011 GRNET S.A. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or
   5  * without modification, are permitted provided that the following
   6  * conditions are met:
   7  *
   8  *   1. Redistributions of source code must retain the above
   9  *      copyright notice, this list of conditions and the following
  10  *      disclaimer.
  11  *
  12  *   2. Redistributions in binary form must reproduce the above
  13  *      copyright notice, this list of conditions and the following
  14  *      disclaimer in the documentation and/or other materials
  15  *      provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  24  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  *
  30  * The views and conclusions contained in the software and
  31  * documentation are those of the authors and should not be
  32  * interpreted as representing official policies, either expressed
  33  * or implied, of GRNET S.A.
  34  */
  35
  36 package gr.grnet.pithos.web.client.mysharedtree;
  37
  38 import gr.grnet.pithos.web.client.FolderContextMenu;
  39 import gr.grnet.pithos.web.client.Pithos;
  40 import gr.grnet.pithos.web.client.foldertree.File;
  41 import gr.grnet.pithos.web.client.foldertree.Folder;
  42 import gr.grnet.pithos.web.client.mysharedtree.MysharedTreeView.Templates;
  43 import gr.grnet.pithos.web.client.rest.GetRequest;
  44 import gr.grnet.pithos.web.client.rest.RestException;
  45
  46 import java.util.HashMap;
  47 import java.util.HashSet;
  48 import java.util.Iterator;
  49 import java.util.List;
  50 import java.util.Map;
  51 import java.util.Set;
  52
  53 import com.google.gwt.cell.client.AbstractCell;
  54 import com.google.gwt.cell.client.Cell;
  55 import com.google.gwt.cell.client.TextCell;
  56 import com.google.gwt.cell.client.ValueUpdater;
  57 import com.google.gwt.core.client.GWT;
  58 import com.google.gwt.core.client.Scheduler;
  59 import com.google.gwt.event.dom.client.ContextMenuEvent;
  60 import com.google.gwt.http.client.Response;
  61 import com.google.gwt.http.client.URL;
  62 import com.google.gwt.safehtml.shared.SafeHtml;
  63 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
  64 import com.google.gwt.text.shared.SafeHtmlRenderer;
  65 import com.google.gwt.user.client.Command;
  66 import com.google.gwt.user.client.ui.AbstractImagePrototype;
  67 import com.google.gwt.view.client.ListDataProvider;
  68 import com.google.gwt.view.client.SelectionChangeEvent;
  69 import com.google.gwt.view.client.SelectionChangeEvent.Handler;
  70 import com.google.gwt.view.client.SingleSelectionModel;
  71 import com.google.gwt.view.client.TreeViewModel;
  72
  73 public class MysharedTreeViewModel implements TreeViewModel {
  74
  75     protected Pithos app;
  76
  77     private Cell<Folder> folderCell = new AbstractCell<Folder>() {
  78
  79        @Override
  80         public void render(@SuppressWarnings("unused") Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
  81             String html = AbstractImagePrototype.create(MysharedTreeView.images.folderYellow()).getHTML();
  82             safeHtmlBuilder.appendHtmlConstant(html);
  83             safeHtmlBuilder.append(Templates.INSTANCE.nameSpan(folder.getName()));
  84         }
  85
  86         @Override
  87         public void onBrowserEvent(@SuppressWarnings("unused") Context context, @SuppressWarnings("unused") com.google.gwt.dom.client.Element parent, Folder folder, com.google.gwt.dom.client.NativeEvent event, @SuppressWarnings("unused") ValueUpdater<Folder> valueUpdater) {
  88             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
  89                 MysharedTreeViewModel.this.selectionModel.setSelected(folder, true);
  90                 FolderContextMenu menu = new FolderContextMenu(app, MysharedTreeView.images, app.getSelectedTree(), folder);
  91                 menu.setPopupPosition(event.getClientX(), event.getClientY());
  92                 menu.show();
  93             }
  94         }
  95     };
  96
  97     protected ListDataProvider<Folder> firstLevelDataProvider = new ListDataProvider<Folder>();
  98
  99     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
 100
 101     protected Set<File> sharedFiles = new HashSet<File>();
 102
 103     protected SingleSelectionModel<Folder> selectionModel;
 104
 105     public MysharedTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
 106         app = _app;
 107         this.selectionModel = selectionModel;
 108     }
 109
 110     @Override
 111     public <T> NodeInfo<?> getNodeInfo(T value) {
 112         if (value == null) {
 113             ListDataProvider<String> rootDataProvider = new ListDataProvider<String>();
 114             rootDataProvider.getList().add("My Shared");
 115             final SingleSelectionModel<String> selectionModel2 = new SingleSelectionModel<String>();
 116             selectionModel2.addSelectionChangeHandler(new Handler() {
 117
 118                 @Override
 119                 public void onSelectionChange(@SuppressWarnings("unused") SelectionChangeEvent event) {
 120                     if (selectionModel2.getSelectedObject() != null) {
 121                         app.deselectOthers(app.getMySharedTreeView(), selectionModel2);
 122                         app.applyPermissions(null);
 123                         fetchSharedFiles();
 124                     }
 125                 }
 126             });
 127             app.addSelectionModel(selectionModel2);
 128             return new DefaultNodeInfo<String>(rootDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
 129                 @Override
 130                 public SafeHtml render(String object) {
 131                     SafeHtmlBuilder builder = new SafeHtmlBuilder();
 132                     render(object, builder);
 133                     return builder.toSafeHtml();
 134                 }
 135
 136                 @Override
 137                 public void render(String object, SafeHtmlBuilder builder) {
 138                     String html = AbstractImagePrototype.create(MysharedTreeView.images.myShared()).getHTML();
 139                     builder.appendHtmlConstant(html);
 140                     builder.append(MysharedTreeView.Templates.INSTANCE.nameSpan(object));
 141                 }
 142             }),  selectionModel2, null);
 143         }
 144         else if (value instanceof String) {
 145                 fetchSharedContainers(null);
 146             return new DefaultNodeInfo<Folder>(firstLevelDataProvider, folderCell, selectionModel, null);
 147         }
 148         else {
 149             final Folder f = (Folder) value;
 150             if (dataProviderMap.get(f) == null) {
 151                 dataProviderMap.put(f, new ListDataProvider<Folder>());
 152             }
 153             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
 154             fetchFolder(f, dataProvider, false);
 155             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
 156         }
 157     }
 158
 159         protected void fetchSharedFiles() {
 160         final List<Folder> containers = app.getAccount().getContainers();
 161         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
 162         Iterator<Folder> iter = containers.iterator();
 163         fetchFolder(iter, tempProvider, new Command() {
 164
 165                         @Override
 166                         public void execute() {
 167                                 firstLevelDataProvider.getList().clear();
 168                                 firstLevelDataProvider.getList().addAll(tempProvider.getList());
 169                                 app.showFiles(sharedFiles);
 170                         }
 171                 });
 172         }
 173
 174         private void fetchSharedContainers(final Command callback) {
 175         final List<Folder> containers = app.getAccount().getContainers();
 176         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
 177         Iterator<Folder> iter = containers.iterator();
 178         fetchFolder(iter, tempProvider, new Command() {
 179
 180                         @Override
 181                         public void execute() {
 182                                 firstLevelDataProvider.getList().clear();
 183                                 firstLevelDataProvider.getList().addAll(tempProvider.getList());
 184                                 if (callback != null)
 185                                         callback.execute();
 186                         }
 187                 });
 188         }
 189
 190         @Override
 191     public boolean isLeaf(Object o) {
 192                 if (o == null)
 193                         return false;
 194                 else if (o instanceof Folder) {
 195             Folder f = (Folder) o;
 196             return f.getSubfolders().isEmpty();
 197         }
 198                 else {
 199                         return firstLevelDataProvider.getList().isEmpty();
 200                 }
 201     }
 202
 203     protected void fetchFolder(final Iterator<Folder> iter, final ListDataProvider<Folder> dataProvider, final Command callback) {
 204         if (iter.hasNext()) {
 205             final Folder f = iter.next();
 206
 207             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
 208             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
 209                 @Override
 210                 public void onSuccess(Folder _result) {
 211                         if (!_result.isShared()) {
 212                                 for (File file : _result.getFiles()) {
 213                                         if (file.isShared())
 214                                                 sharedFiles.add(file);
 215                                 }
 216                                 Iterator<Folder> iter2 = _result.getSubfolders().iterator();
 217                                 fetchFolder(iter2, dataProvider, new Command() {
 218
 219                                                         @Override
 220                                                         public void execute() {
 221                                             fetchFolder(iter, dataProvider, callback);
 222                                                         }
 223                                                 });
 224                         }
 225                         else {
 226                                 dataProvider.getList().add(_result);
 227                             fetchFolder(iter, dataProvider, callback);
 228                         }
 229                 }
 230
 231                 @Override
 232                 public void onError(Throwable t) {
 233                     GWT.log("Error getting folder", t);
 234                     if (t instanceof RestException)
 235                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
 236                     else
 237                         app.displayError("System error fetching folder: " + t.getMessage());
 238                 }
 239
 240                                 @Override
 241                                 protected void onUnauthorized(Response response) {
 242                                         app.sessionExpired();
 243                                 }
 244             };
 245             getFolder.setHeader("X-Auth-Token", app.getToken());
 246             Scheduler.get().scheduleDeferred(getFolder);
 247         }
 248         else if (callback != null)
 249             callback.execute();
 250     }
 251
 252     public Folder getSelection() {
 253         return selectionModel.getSelectedObject();
 254     }
 255
 256     public void updateFolder(Folder folder, boolean showfiles) {
 257         if (dataProviderMap.get(folder) == null) {
 258             dataProviderMap.put(folder, new ListDataProvider<Folder>());
 259         }
 260         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
 261         fetchFolder(folder, dataProvider, showfiles);
 262     }
 263
 264     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles) {
 265         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
 266         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
 267             @Override
 268             public void onSuccess(final Folder _result) {
 269                 if (showfiles)
 270                     app.showFiles(_result);
 271                 Iterator<Folder> iter = _result.getSubfolders().iterator();
 272                 fetchFolder(iter, dataProvider, new Command() {
 273                     @Override
 274                     public void execute() {
 275                         dataProvider.getList().clear();
 276                                 dataProvider.getList().addAll(_result.getSubfolders());
 277                         app.getMySharedTreeView().updateChildren(f);
 278                     }
 279                 });
 280             }
 281
 282             @Override
 283             public void onError(Throwable t) {
 284                 GWT.log("Error getting folder", t);
 285                 if (t instanceof RestException)
 286                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
 287                 else
 288                     app.displayError("System error fetching folder: " + t.getMessage());
 289             }
 290
 291                         @Override
 292                         protected void onUnauthorized(Response response) {
 293                                 app.sessionExpired();
 294                         }
 295         };
 296         getFolder.setHeader("X-Auth-Token", app.getToken());
 297         Scheduler.get().scheduleDeferred(getFolder);
 298     }
 299
 300         public void initialize(Command callback) {
 301                 fetchSharedContainers(callback);
 302         }
 303 }