0da9572b664bc720e90a19b611e0fe66ed80655d
[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.Folder;
43 import gr.grnet.pithos.web.client.othersharedtree.OtherSharedTreeView.Templates;
44 import gr.grnet.pithos.web.client.rest.GetRequest;
45 import gr.grnet.pithos.web.client.rest.RestException;
46
47 import java.util.HashMap;
48 import java.util.Iterator;
49 import java.util.Map;
50
51 import com.google.gwt.cell.client.AbstractCell;
52 import com.google.gwt.cell.client.Cell;
53 import com.google.gwt.cell.client.TextCell;
54 import com.google.gwt.cell.client.ValueUpdater;
55 import com.google.gwt.core.client.GWT;
56 import com.google.gwt.core.client.Scheduler;
57 import com.google.gwt.event.dom.client.ContextMenuEvent;
58 import com.google.gwt.http.client.Response;
59 import com.google.gwt.http.client.URL;
60 import com.google.gwt.safehtml.shared.SafeHtml;
61 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
62 import com.google.gwt.text.shared.SafeHtmlRenderer;
63 import com.google.gwt.user.client.Command;
64 import com.google.gwt.user.client.ui.AbstractImagePrototype;
65 import com.google.gwt.view.client.ListDataProvider;
66 import com.google.gwt.view.client.SingleSelectionModel;
67 import com.google.gwt.view.client.TreeViewModel;
68
69 public class OtherSharedTreeViewModel implements TreeViewModel {
70
71     protected Pithos app;
72
73     String dummy = "No shares by others yet";
74
75     private Cell<Folder> folderCell = new AbstractCell<Folder>(ContextMenuEvent.getType().getName()) {
76
77        @Override
78         public void render(Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
79             String html = AbstractImagePrototype.create(OtherSharedTreeView.images.folderYellow()).getHTML();
80                 safeHtmlBuilder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
81             safeHtmlBuilder.append(Templates.INSTANCE.nameSpan(folder.getName()));
82         }
83
84         @Override
85         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) {
86             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
87                 final int x = event.getClientX();
88                 final int y = event.getClientY();
89                 OtherSharedTreeViewModel.this.selectionModel.setSelected(folder, true);
90                 app.scheduleFolderHeadCommand(folder, new Command() {
91                                         
92                                         @Override
93                                         public void execute() {
94                                 FolderContextMenu menu = new FolderContextMenu(app, OtherSharedTreeView.images, app.getSelectedTree(), folder);
95                                 menu.setPopupPosition(x, y);
96                                 menu.show();
97                                         }
98                                 });
99             }
100         }
101     };
102
103     protected ListDataProvider<String> userLevelDataProvider = new ListDataProvider<String>();
104
105     protected Map<String, ListDataProvider<Folder>> userDataProviderMap = new HashMap<String, ListDataProvider<Folder>>();
106     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
107     
108     protected SingleSelectionModel<Folder> selectionModel;
109
110     public OtherSharedTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
111         app = _app;
112         this.selectionModel = selectionModel;
113     }
114
115     @Override
116     public <T> NodeInfo<?> getNodeInfo(T value) {
117         if (value == null) {
118                 fetchSharingUsers(null);
119             return new DefaultNodeInfo<String>(userLevelDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
120
121                                 @Override
122                                 public SafeHtml render(String object) {
123                     SafeHtmlBuilder builder = new SafeHtmlBuilder();
124                     render(object, builder);
125                     return builder.toSafeHtml();
126                                 }
127
128                                 @Override
129                                 public void render(String object, SafeHtmlBuilder builder) {
130                     if (!object.equals(dummy)) {
131                         String html = AbstractImagePrototype.create(OtherSharedTreeView.images.myShared()).getHTML();
132                         builder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
133                     }
134                     builder.append(OtherSharedTreeView.Templates.INSTANCE.nameSpan(object));
135                                 }
136                         }), null, null);
137         }
138         else if (value instanceof String) {
139                         String username = (String) value;
140                         if (userDataProviderMap.get(username) == null) {
141                                 userDataProviderMap.put(username, new ListDataProvider<Folder>());
142                         }
143                         final ListDataProvider<Folder> dataProvider = userDataProviderMap.get(username);
144                         fetchSharedContainers(username, dataProvider, null);
145                         return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
146         }
147         else {
148             final Folder f = (Folder) value;
149             if (dataProviderMap.get(f) == null) {
150                 dataProviderMap.put(f, new ListDataProvider<Folder>());
151             }
152             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
153             fetchFolder(f, dataProvider, false, null);
154             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
155         }
156     }
157
158     private void fetchSharingUsers(final Command callback) {
159         GetRequest<SharingUsers> getSharingUsers = new GetRequest<SharingUsers>(SharingUsers.class, app.getApiPath(), "", "?format=json") {
160             @Override
161             public void onSuccess(final SharingUsers _result) {
162                 userLevelDataProvider.getList().clear();
163                 userLevelDataProvider.getList().addAll(_result.getUsers());
164                 if (userLevelDataProvider.getList().isEmpty())
165                         userLevelDataProvider.getList().add(dummy);
166                 Iterator<String> iter = _result.getUsers().iterator();
167                 fetchSharedContainers(iter, callback);
168             }
169
170             @Override
171             public void onError(Throwable t) {
172                 GWT.log("Error getting folder", t);
173                                 app.setError(t);
174                 if (t instanceof RestException)
175                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
176                 else
177                     app.displayError("System error fetching folder: " + t.getMessage());
178             }
179
180                         @Override
181                         protected void onUnauthorized(Response response) {
182                                 app.sessionExpired();
183                         }
184         };
185         getSharingUsers.setHeader("X-Auth-Token", app.getUserToken());
186         Scheduler.get().scheduleDeferred(getSharingUsers);
187         }
188
189         protected void fetchSharedContainers(final Iterator<String> iter, final Command callback) {
190                 if (iter.hasNext()) {
191                         String username = iter.next();
192                         if (userDataProviderMap.get(username) == null) {
193                                 userDataProviderMap.put(username, new ListDataProvider<Folder>());
194                         }
195                         final ListDataProvider<Folder> dataProvider = userDataProviderMap.get(username);
196                         fetchSharedContainers(username, dataProvider, new Command() {
197                                 
198                                 @Override
199                                 public void execute() {
200                                         fetchSharedContainers(iter, callback);
201                                         
202                                 }
203                         });
204                 }
205                 else
206                         if (callback != null)
207                                 callback.execute();
208         }
209
210         @Override
211     public boolean isLeaf(Object o) {
212                 if (o == null)
213                         return userLevelDataProvider.getList().isEmpty();
214                 else if (o instanceof Folder) {
215             Folder f = (Folder) o;
216             return f.getSubfolders().isEmpty();
217         }
218                 else {
219                         ListDataProvider<Folder> dp = userDataProviderMap.get(o);
220                         if (dp != null)
221                                 return dp.getList().isEmpty();
222                         return true;
223                 }
224     }
225
226         private void fetchSharedContainers(final String username, final ListDataProvider<Folder> dataProvider, final Command callback) {
227                 GetRequest<AccountResource> getUserSharedContainers = new GetRequest<AccountResource>(AccountResource.class, app.getApiPath(), username, "?format=json") {
228
229                         @Override
230                         public void onSuccess(AccountResource _result) {
231                         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
232                                 Iterator<Folder> iter = _result.getContainers().iterator();
233                                 fetchFolder(username, iter, tempProvider, new Command() {
234                                         
235                                         @Override
236                                         public void execute() {
237                                                 dataProvider.getList().clear();
238                                                 dataProvider.getList().addAll(tempProvider.getList());
239                                                 if (callback != null)
240                                                         callback.execute();
241                                         }
242                                 });
243                         }
244
245                         @Override
246                         public void onError(Throwable t) {
247                 GWT.log("Error getting account", t);
248                                 app.setError(t);
249                 if (t instanceof RestException)
250                     app.displayError("Error getting account: " + ((RestException) t).getHttpStatusText());
251                 else
252                     app.displayError("System error fetching user data: " + t.getMessage());
253                         }
254
255                         @Override
256                         protected void onUnauthorized(Response response) {
257                                 app.sessionExpired();
258                         }
259                 };
260                 getUserSharedContainers.setHeader("X-Auth-Token", app.getUserToken());
261                 Scheduler.get().scheduleDeferred(getUserSharedContainers);
262         }
263
264         protected void fetchFolder(final String username, final Iterator<Folder> iter, final ListDataProvider<Folder> dataProvider, final Command callback) {
265         if (iter.hasNext()) {
266             final Folder f = iter.next();
267
268             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
269             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), username, path, f) {
270                 @Override
271                 public void onSuccess(Folder _result) {
272                         dataProvider.getList().add(_result);
273                     fetchFolder(username, iter, dataProvider, callback);
274                 }
275
276                 @Override
277                 public void onError(Throwable t) {
278                     GWT.log("Error getting folder", t);
279                                         app.setError(t);
280                     if (t instanceof RestException)
281                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
282                     else
283                         app.displayError("System error fetching folder: " + t.getMessage());
284                 }
285
286                                 @Override
287                                 protected void onUnauthorized(Response response) {
288                                         app.sessionExpired();
289                                 }
290             };
291             getFolder.setHeader("X-Auth-Token", app.getUserToken());
292             Scheduler.get().scheduleDeferred(getFolder);
293         }
294         else if (callback != null)
295             callback.execute();
296     }
297
298     public Folder getSelection() {
299         return selectionModel.getSelectedObject();
300     }
301
302     public void updateFolder(Folder folder, boolean showfiles, Command callback) {
303         if (dataProviderMap.get(folder) == null) {
304             dataProviderMap.put(folder, new ListDataProvider<Folder>());
305         }
306         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
307         fetchFolder(folder, dataProvider, showfiles, callback);
308     }
309
310     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles, final Command callback) {
311         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
312         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwnerID(), path, f) {
313             @Override
314             public void onSuccess(final Folder _result) {
315                 if (showfiles)
316                     app.showFiles(_result);
317                 Iterator<Folder> iter = _result.getSubfolders().iterator();
318                 fetchFolder(_result.getOwnerID(), iter, dataProvider, new Command() {
319                     @Override
320                     public void execute() {
321                         dataProvider.getList().clear();
322                         dataProvider.getList().addAll(_result.getSubfolders());
323                         app.getOtherSharedTreeView().updateChildren(f);
324                         if (callback != null)
325                                 callback.execute();
326                     }
327                 });
328             }
329
330             @Override
331             public void onError(Throwable t) {
332                 GWT.log("Error getting folder", t);
333                                 app.setError(t);
334                 if (t instanceof RestException)
335                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
336                 else
337                     app.displayError("System error fetching folder: " + t.getMessage());
338             }
339
340                         @Override
341                         protected void onUnauthorized(Response response) {
342                                 app.sessionExpired();
343                         }
344         };
345         getFolder.setHeader("X-Auth-Token", app.getUserToken());
346         Scheduler.get().scheduleDeferred(getFolder);
347     }
348     
349     public void initialize(Command callback) {
350         fetchSharingUsers(callback);
351     }
352 }