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