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