216f9fdae746e88c37ec940097046ffe07b935e4
[pithos-web-client] / 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.ArrayList;
47 import java.util.HashMap;
48 import java.util.HashSet;
49 import java.util.Iterator;
50 import java.util.List;
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 MysharedTreeViewModel implements TreeViewModel {
75
76     protected Pithos app;
77
78     private Cell<Folder> folderCell = new AbstractCell<Folder>() {
79
80        @Override
81         public void render(Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
82             String html = AbstractImagePrototype.create(MysharedTreeView.images.folderYellow()).getHTML();
83             safeHtmlBuilder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
84             safeHtmlBuilder.append(Templates.INSTANCE.nameSpan(folder.getName()));
85         }
86
87         @Override
88         public void onBrowserEvent(Context context, com.google.gwt.dom.client.Element parent, Folder folder, com.google.gwt.dom.client.NativeEvent event, ValueUpdater<Folder> valueUpdater) {
89             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
90                 MysharedTreeViewModel.this.selectionModel.setSelected(folder, true);
91                 FolderContextMenu menu = new FolderContextMenu(app, MysharedTreeView.images, app.getSelectedTree(), folder);
92                 menu.setPopupPosition(event.getClientX(), event.getClientY());
93                 menu.show();
94             }
95         }
96     };
97
98     protected ListDataProvider<Folder> firstLevelDataProvider = new ListDataProvider<Folder>();
99
100     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
101     
102     protected Set<File> sharedFiles = new HashSet<File>();
103
104     protected SingleSelectionModel<Folder> selectionModel;
105
106     public MysharedTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
107         app = _app;
108         this.selectionModel = selectionModel;
109     }
110
111     @Override
112     public <T> NodeInfo<?> getNodeInfo(T value) {
113         if (value == null) {
114             ListDataProvider<String> rootDataProvider = new ListDataProvider<String>();
115             rootDataProvider.getList().add("My Shared");
116             final SingleSelectionModel<String> selectionModel2 = new SingleSelectionModel<String>();
117             selectionModel2.addSelectionChangeHandler(new Handler() {
118
119                 @Override
120                 public void onSelectionChange(SelectionChangeEvent event) {
121                     if (selectionModel2.getSelectedObject() != null) {
122                         app.deselectOthers(app.getMySharedTreeView(), selectionModel2);
123                         app.applyPermissions(null);
124                         fetchSharedFiles(true);
125                     }
126                     app.showRelevantToolbarButtons();
127                 }
128             });
129             app.addSelectionModel(selectionModel2);
130             return new DefaultNodeInfo<String>(rootDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
131                 @Override
132                 public SafeHtml render(String object) {
133                     SafeHtmlBuilder builder = new SafeHtmlBuilder();
134                     render(object, builder);
135                     return builder.toSafeHtml();
136                 }
137
138                 @Override
139                 public void render(String object, SafeHtmlBuilder builder) {
140                     String html = AbstractImagePrototype.create(MysharedTreeView.images.myShared()).getHTML();
141                     builder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
142                     builder.append(MysharedTreeView.Templates.INSTANCE.nameSpan(object));
143                 }
144             }),  selectionModel2, null);
145         }
146         else if (value instanceof String) {
147                 fetchSharedContainers(null);
148             return new DefaultNodeInfo<Folder>(firstLevelDataProvider, folderCell, selectionModel, null);
149         }
150         else {
151             final Folder f = (Folder) value;
152             if (dataProviderMap.get(f) == null) {
153                 dataProviderMap.put(f, new ListDataProvider<Folder>());
154             }
155             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
156             fetchFolder(f, dataProvider, false);
157             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
158         }
159     }
160
161         protected void fetchSharedFiles(final boolean showFiles) {
162                 sharedFiles.clear();
163         final List<Folder> containers = app.getAccount().getContainers();
164         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
165         Iterator<Folder> iter = containers.iterator();
166         fetchFolder(iter, tempProvider, new Command() {
167                         
168                         @Override
169                         public void execute() {
170                                 firstLevelDataProvider.getList().clear();
171                                 firstLevelDataProvider.getList().addAll(tempProvider.getList());
172                                 if (showFiles)
173                                         app.showFiles(sharedFiles);
174                         }
175                 });
176         }
177
178         private void fetchSharedContainers(final Command callback) {
179         final List<Folder> containers = app.getAccount().getContainers();
180         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
181         Iterator<Folder> iter = containers.listIterator();
182         fetchFolder(iter, tempProvider, new Command() {
183                         
184                         @Override
185                         public void execute() {
186                                 firstLevelDataProvider.getList().clear();
187                                 firstLevelDataProvider.getList().addAll(tempProvider.getList());
188                                 if (callback != null)
189                                         callback.execute();
190                         }
191                 });
192         }
193
194         @Override
195     public boolean isLeaf(Object o) {
196                 if (o == null)
197                         return false;
198                 else if (o instanceof Folder) {
199             Folder f = (Folder) o;
200             return f.getSubfolders().isEmpty();
201         }
202                 else {
203                         return firstLevelDataProvider.getList().isEmpty();
204                 }
205     }
206         
207         private native void log(String msg) /*-{
208                 $wnd.console.log(msg);
209         }-*/;
210
211     protected void fetchFolder(final Iterator<Folder> iter, final ListDataProvider<Folder> dataProvider, final Command callback) {
212         if (iter.hasNext()) {
213             final Folder f = iter.next();
214
215             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
216             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
217                 @Override
218                 public void onSuccess(Folder _result) {
219                         if (!_result.isShared()) {
220                                 for (File file : _result.getFiles()) {
221                                         if (file.isShared())
222                                                 sharedFiles.add(file);
223                                 }
224                                 Iterator<Folder> iter2 = new ArrayList<Folder>(result.getSubfolders()).listIterator();
225                                 fetchFolder(iter2, dataProvider, new Command() {
226                                                         
227                                                         @Override
228                                                         public void execute() {
229                                             fetchFolder(iter, dataProvider, callback);
230                                                         }
231                                                 });
232                         }
233                         else {
234                                 dataProvider.getList().add(_result);
235                             fetchFolder(iter, dataProvider, callback);
236                         }
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             getFolder.setHeader("X-Auth-Token", app.getToken());
255             Scheduler.get().scheduleDeferred(getFolder);
256         }
257         else if (callback != null)
258             callback.execute();
259     }
260
261     public Folder getSelection() {
262         return selectionModel.getSelectedObject();
263     }
264
265     public void updateFolder(Folder folder, boolean showfiles) {
266         if (dataProviderMap.get(folder) == null) {
267             dataProviderMap.put(folder, new ListDataProvider<Folder>());
268         }
269         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
270         fetchFolder(folder, dataProvider, showfiles);
271     }
272
273     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles) {
274         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
275         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
276             @Override
277             public void onSuccess(final Folder _result) {
278                 if (showfiles)
279                     app.showFiles(_result);
280                 Iterator<Folder> iter = _result.getSubfolders().iterator();
281                 fetchFolder(iter, dataProvider, new Command() {
282                     @Override
283                     public void execute() {
284                         dataProvider.getList().clear();
285                                 dataProvider.getList().addAll(_result.getSubfolders());
286                         app.getMySharedTreeView().updateChildren(f);
287                     }
288                 });
289             }
290
291             @Override
292             public void onError(Throwable t) {
293                 GWT.log("Error getting folder", t);
294                                 app.setError(t);
295                 if (t instanceof RestException)
296                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
297                 else
298                     app.displayError("System error fetching folder: " + t.getMessage());
299             }
300
301                         @Override
302                         protected void onUnauthorized(Response response) {
303                                 app.sessionExpired();
304                         }
305         };
306         getFolder.setHeader("X-Auth-Token", app.getToken());
307         Scheduler.get().scheduleDeferred(getFolder);
308     }
309
310         public void initialize(Command callback) {
311                 fetchSharedContainers(callback);
312         }
313 }