Fixed refresh issues with shared folders and files
[pithos-web-client] / src / gr / grnet / pithos / web / client / othersharedtree / OtherSharedTreeViewModel.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.othersharedtree;
37
38 import gr.grnet.pithos.web.client.FolderContextMenu;
39 import gr.grnet.pithos.web.client.Pithos;
40 import gr.grnet.pithos.web.client.SharingUsers;
41 import gr.grnet.pithos.web.client.foldertree.AccountResource;
42 import gr.grnet.pithos.web.client.foldertree.File;
43 import gr.grnet.pithos.web.client.foldertree.Folder;
44 import gr.grnet.pithos.web.client.othersharedtree.OtherSharedTreeView.Templates;
45 import gr.grnet.pithos.web.client.rest.GetRequest;
46 import gr.grnet.pithos.web.client.rest.RestException;
47
48 import java.util.HashMap;
49 import java.util.HashSet;
50 import java.util.Iterator;
51 import java.util.Map;
52 import java.util.Set;
53
54 import com.google.gwt.cell.client.AbstractCell;
55 import com.google.gwt.cell.client.Cell;
56 import com.google.gwt.cell.client.TextCell;
57 import com.google.gwt.cell.client.ValueUpdater;
58 import com.google.gwt.core.client.GWT;
59 import com.google.gwt.core.client.Scheduler;
60 import com.google.gwt.event.dom.client.ContextMenuEvent;
61 import com.google.gwt.http.client.Response;
62 import com.google.gwt.http.client.URL;
63 import com.google.gwt.safehtml.shared.SafeHtml;
64 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
65 import com.google.gwt.text.shared.SafeHtmlRenderer;
66 import com.google.gwt.user.client.Command;
67 import com.google.gwt.user.client.ui.AbstractImagePrototype;
68 import com.google.gwt.view.client.ListDataProvider;
69 import com.google.gwt.view.client.SelectionChangeEvent;
70 import com.google.gwt.view.client.SelectionChangeEvent.Handler;
71 import com.google.gwt.view.client.SingleSelectionModel;
72 import com.google.gwt.view.client.TreeViewModel;
73
74 public class OtherSharedTreeViewModel implements TreeViewModel {
75
76         private static final String treeTitle = "Other 's shared";
77     protected Pithos app;
78
79     private Cell<Folder> folderCell = new AbstractCell<Folder>(ContextMenuEvent.getType().getName()) {
80
81        @Override
82         public void render(@SuppressWarnings("unused") Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
83             String html = AbstractImagePrototype.create(OtherSharedTreeView.images.folderYellow()).getHTML();
84             safeHtmlBuilder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
85             safeHtmlBuilder.append(Templates.INSTANCE.nameSpan(folder.getName()));
86         }
87
88         @Override
89         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) {
90             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
91                 OtherSharedTreeViewModel.this.selectionModel.setSelected(folder, true);
92                 FolderContextMenu menu = new FolderContextMenu(app, OtherSharedTreeView.images, app.getSelectedTree(), folder);
93                 menu.setPopupPosition(event.getClientX(), event.getClientY());
94                 menu.show();
95             }
96         }
97     };
98
99     private ListDataProvider<String> rootDataProvider = new ListDataProvider<String>();
100     protected ListDataProvider<String> userLevelDataProvider = new ListDataProvider<String>();
101
102     protected Map<String, ListDataProvider<Folder>> userDataProviderMap = new HashMap<String, ListDataProvider<Folder>>();
103     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
104     
105     protected Map<String, Set<File>> sharedFiles = new HashMap<String, Set<File>>();
106
107     protected SingleSelectionModel<Folder> selectionModel;
108
109     public OtherSharedTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
110         app = _app;
111         this.selectionModel = selectionModel;
112     }
113
114     @Override
115     public <T> NodeInfo<?> getNodeInfo(T value) {
116         if (value == null) {
117             rootDataProvider.getList().add(treeTitle);
118             final SingleSelectionModel<String> selectionModel2 = new SingleSelectionModel<String>();
119             selectionModel2.addSelectionChangeHandler(new Handler() {
120
121                 @Override
122                 public void onSelectionChange(@SuppressWarnings("unused") SelectionChangeEvent event) {
123                     if (selectionModel2.getSelectedObject() != null) {
124                         app.deselectOthers(app.getOtherSharedTreeView(), selectionModel2);
125                         app.applyPermissions(null);
126                         app.showFiles(new HashSet<File>());
127                     }
128                 }
129             });
130             app.addSelectionModel(selectionModel2);
131             return new DefaultNodeInfo<String>(rootDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
132                 @Override
133                 public SafeHtml render(String object) {
134                     SafeHtmlBuilder builder = new SafeHtmlBuilder();
135                     render(object, builder);
136                     return builder.toSafeHtml();
137                 }
138
139                 @Override
140                 public void render(String object, SafeHtmlBuilder builder) {
141                     String html = AbstractImagePrototype.create(OtherSharedTreeView.images.othersShared()).getHTML();
142                     builder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
143                     builder.append(OtherSharedTreeView.Templates.INSTANCE.nameSpan(object));
144                 }
145             }),  selectionModel2, null);
146         }
147         else if (value instanceof String) {
148                 if (value.equals(treeTitle)) {
149                         fetchSharingUsers(null);
150                     final SingleSelectionModel<String> selectionModel3 = new SingleSelectionModel<String>();
151                     selectionModel3.addSelectionChangeHandler(new Handler() {
152
153                         @Override
154                         public void onSelectionChange(@SuppressWarnings("unused") SelectionChangeEvent event) {
155                             if (selectionModel3.getSelectedObject() != null) {
156                                 app.deselectOthers(app.getOtherSharedTreeView(), selectionModel3);
157                                 app.applyPermissions(null);
158                                 String username = selectionModel3.getSelectedObject();
159                                         if (userDataProviderMap.get(username) == null) {
160                                                 userDataProviderMap.put(username, new ListDataProvider<Folder>());
161                                         }
162                                 fetchSharedFiles(username, userDataProviderMap.get(username));
163                             }
164                         }
165                     });
166                     app.addSelectionModel(selectionModel3);
167                     return new DefaultNodeInfo<String>(userLevelDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
168
169                                         @Override
170                                         public SafeHtml render(String object) {
171                             SafeHtmlBuilder builder = new SafeHtmlBuilder();
172                             render(object, builder);
173                             return builder.toSafeHtml();
174                                         }
175
176                                         @Override
177                                         public void render(String object, SafeHtmlBuilder builder) {
178                             String html = AbstractImagePrototype.create(OtherSharedTreeView.images.user()).getHTML();
179                             builder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
180                             builder.append(OtherSharedTreeView.Templates.INSTANCE.nameSpan(object));
181                                         }
182                                 }), selectionModel3, null);
183                 }
184                         String username = (String) value;
185                         if (userDataProviderMap.get(username) == null) {
186                                 userDataProviderMap.put(username, new ListDataProvider<Folder>());
187                         }
188                         final ListDataProvider<Folder> dataProvider = userDataProviderMap.get(username);
189                         fetchSharedContainers(username, dataProvider, null);
190                         return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
191         }
192         else {
193             final Folder f = (Folder) value;
194             if (dataProviderMap.get(f) == null) {
195                 dataProviderMap.put(f, new ListDataProvider<Folder>());
196             }
197             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
198             fetchFolder(f, dataProvider, false);
199             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
200         }
201     }
202
203     private void fetchSharingUsers(final Command callback) {
204         GetRequest<SharingUsers> getSharingUsers = new GetRequest<SharingUsers>(SharingUsers.class, app.getApiPath(), "", "?format=json") {
205             @Override
206             public void onSuccess(final SharingUsers _result) {
207                 userLevelDataProvider.getList().clear();
208                 userLevelDataProvider.getList().addAll(_result.getUsers());
209                 for (String name : _result.getUsers()) {
210                         sharedFiles.put(name, new HashSet<File>());
211                 }
212                 Iterator<String> iter = _result.getUsers().iterator();
213                 fetchSharedContainers(iter, callback);
214             }
215
216             @Override
217             public void onError(Throwable t) {
218                 GWT.log("Error getting folder", t);
219                                 app.setError(t);
220                 if (t instanceof RestException)
221                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
222                 else
223                     app.displayError("System error fetching folder: " + t.getMessage());
224             }
225
226                         @Override
227                         protected void onUnauthorized(Response response) {
228                                 app.sessionExpired();
229                         }
230         };
231         getSharingUsers.setHeader("X-Auth-Token", app.getToken());
232         Scheduler.get().scheduleDeferred(getSharingUsers);
233         }
234
235         protected void fetchSharedContainers(final Iterator<String> iter, final Command callback) {
236                 if (iter.hasNext()) {
237                         String username = iter.next();
238                         if (userDataProviderMap.get(username) == null) {
239                                 userDataProviderMap.put(username, new ListDataProvider<Folder>());
240                         }
241                         final ListDataProvider<Folder> dataProvider = userDataProviderMap.get(username);
242                         fetchSharedContainers(username, dataProvider, new Command() {
243                                 
244                                 @Override
245                                 public void execute() {
246                                         fetchSharedContainers(iter, callback);
247                                         
248                                 }
249                         });
250                 }
251                 else
252                         if (callback != null)
253                                 callback.execute();
254         }
255
256         @Override
257     public boolean isLeaf(Object o) {
258                 if (o == null)
259                         return false;
260                 else if (o instanceof Folder) {
261             Folder f = (Folder) o;
262             return f.getSubfolders().isEmpty();
263         }
264                 else {
265                         if (o.equals(treeTitle))
266                                 return userLevelDataProvider.getList().isEmpty();
267                         ListDataProvider<Folder> dp = userDataProviderMap.get(o);
268                         if (dp != null)
269                                 return dp.getList().isEmpty();
270                         return true;
271                 }
272     }
273
274         private void fetchSharedContainers(final String username, final ListDataProvider<Folder> dataProvider, final Command callback) {
275                 GetRequest<AccountResource> getUserSharedContainers = new GetRequest<AccountResource>(AccountResource.class, app.getApiPath(), username, "?format=json") {
276
277                         @Override
278                         public void onSuccess(AccountResource _result) {
279                         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
280                                 Iterator<Folder> iter = _result.getContainers().iterator();
281                                 fetchFolder(username, iter, tempProvider, new Command() {
282                                         
283                                         @Override
284                                         public void execute() {
285                                                 dataProvider.getList().clear();
286                                                 dataProvider.getList().addAll(tempProvider.getList());
287                                                 if (callback != null)
288                                                         callback.execute();
289                                         }
290                                 });
291                         }
292
293                         @Override
294                         public void onError(Throwable t) {
295                 GWT.log("Error getting account", t);
296                                 app.setError(t);
297                 if (t instanceof RestException)
298                     app.displayError("Error getting account: " + ((RestException) t).getHttpStatusText());
299                 else
300                     app.displayError("System error fetching user data: " + t.getMessage());
301                         }
302
303                         @Override
304                         protected void onUnauthorized(Response response) {
305                                 app.sessionExpired();
306                         }
307                 };
308                 getUserSharedContainers.setHeader("X-Auth-Token", app.getToken());
309                 Scheduler.get().scheduleDeferred(getUserSharedContainers);
310         }
311
312         protected void fetchSharedFiles(final String username, final ListDataProvider<Folder> dataProvider) {
313                 GetRequest<AccountResource> getUserSharedContainers = new GetRequest<AccountResource>(AccountResource.class, app.getApiPath(), username, "?format=json") {
314
315                         @Override
316                         public void onSuccess(AccountResource _result) {
317                         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
318                                 Iterator<Folder> iter = _result.getContainers().iterator();
319                                 sharedFiles.get(username).clear();
320                                 fetchFolder(username, iter, tempProvider, new Command() {
321                                         
322                                         @Override
323                                         public void execute() {
324                                                 dataProvider.getList().clear();
325                                                 dataProvider.getList().addAll(tempProvider.getList());
326                                                 app.showFiles(sharedFiles.get(username));
327                                         }
328                                 });
329                         }
330
331                         @Override
332                         public void onError(Throwable t) {
333                 GWT.log("Error getting account", t);
334                                 app.setError(t);
335                 if (t instanceof RestException)
336                     app.displayError("Error getting account: " + ((RestException) t).getHttpStatusText());
337                 else
338                     app.displayError("System error fetching user data: " + t.getMessage());
339                         }
340
341                         @Override
342                         protected void onUnauthorized(Response response) {
343                                 app.sessionExpired();
344                         }
345                 };
346                 getUserSharedContainers.setHeader("X-Auth-Token", app.getToken());
347                 Scheduler.get().scheduleDeferred(getUserSharedContainers);
348         }
349
350         protected void fetchFolder(final String username, final Iterator<Folder> iter, final ListDataProvider<Folder> dataProvider, final Command callback) {
351         if (iter.hasNext()) {
352             final Folder f = iter.next();
353
354             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
355             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), username, path, f) {
356                 @Override
357                 public void onSuccess(Folder _result) {
358                         if (!_result.isShared()) {
359                                 for (File file : _result.getFiles()) {
360                                         if (file.isShared())
361                                                 sharedFiles.get(username).add(file);
362                                 }
363                                 Iterator<Folder> iter2 = _result.getSubfolders().iterator();
364                                 fetchFolder(username, iter2, dataProvider, new Command() {
365                                                         
366                                                         @Override
367                                                         public void execute() {
368                                             fetchFolder(username, iter, dataProvider, callback);
369                                                         }
370                                                 });
371                         }
372                         else {
373                                 dataProvider.getList().add(_result);
374                             fetchFolder(username, iter, dataProvider, callback);
375                         }
376                 }
377
378                 @Override
379                 public void onError(Throwable t) {
380                     GWT.log("Error getting folder", t);
381                                         app.setError(t);
382                     if (t instanceof RestException)
383                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
384                     else
385                         app.displayError("System error fetching folder: " + t.getMessage());
386                 }
387
388                                 @Override
389                                 protected void onUnauthorized(Response response) {
390                                         app.sessionExpired();
391                                 }
392             };
393             getFolder.setHeader("X-Auth-Token", app.getToken());
394             Scheduler.get().scheduleDeferred(getFolder);
395         }
396         else if (callback != null)
397             callback.execute();
398     }
399
400     public Folder getSelection() {
401         return selectionModel.getSelectedObject();
402     }
403
404     public void updateFolder(Folder folder, boolean showfiles) {
405         if (dataProviderMap.get(folder) == null) {
406             dataProviderMap.put(folder, new ListDataProvider<Folder>());
407         }
408         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
409         fetchFolder(folder, dataProvider, showfiles);
410     }
411
412     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles) {
413         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
414         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
415             @Override
416             public void onSuccess(final Folder _result) {
417                 GWT.log(String.valueOf(_result.getFiles().size()));
418                 if (showfiles)
419                     app.showFiles(_result);
420                 Iterator<Folder> iter = _result.getSubfolders().iterator();
421                 fetchFolder(_result.getOwner(), iter, dataProvider, new Command() {
422                     @Override
423                     public void execute() {
424                         dataProvider.getList().clear();
425                         dataProvider.getList().addAll(_result.getSubfolders());
426                         app.getOtherSharedTreeView().updateChildren(f);
427                     }
428                 });
429             }
430
431             @Override
432             public void onError(Throwable t) {
433                 GWT.log("Error getting folder", t);
434                                 app.setError(t);
435                 if (t instanceof RestException)
436                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
437                 else
438                     app.displayError("System error fetching folder: " + t.getMessage());
439             }
440
441                         @Override
442                         protected void onUnauthorized(Response response) {
443                                 app.sessionExpired();
444                         }
445         };
446         getFolder.setHeader("X-Auth-Token", app.getToken());
447         Scheduler.get().scheduleDeferred(getFolder);
448     }
449     
450     public void initialize(Command callback) {
451         fetchSharingUsers(callback);
452     }
453 }