Changed SharedByMe tree to always display shared files under their full subtree
[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.AccountResource;
41 import gr.grnet.pithos.web.client.foldertree.File;
42 import gr.grnet.pithos.web.client.foldertree.Folder;
43 import gr.grnet.pithos.web.client.foldertree.FolderTreeView;
44 import gr.grnet.pithos.web.client.mysharedtree.MysharedTreeView.Templates;
45 import gr.grnet.pithos.web.client.rest.GetRequest;
46 import gr.grnet.pithos.web.client.rest.RestException;
47
48 import java.util.ArrayList;
49 import java.util.HashMap;
50 import java.util.HashSet;
51 import java.util.Iterator;
52 import java.util.List;
53 import java.util.Map;
54 import java.util.Set;
55
56 import com.google.gwt.cell.client.AbstractCell;
57 import com.google.gwt.cell.client.Cell;
58 import com.google.gwt.cell.client.TextCell;
59 import com.google.gwt.cell.client.ValueUpdater;
60 import com.google.gwt.core.client.GWT;
61 import com.google.gwt.core.client.Scheduler;
62 import com.google.gwt.event.dom.client.ContextMenuEvent;
63 import com.google.gwt.http.client.Response;
64 import com.google.gwt.http.client.URL;
65 import com.google.gwt.safehtml.shared.SafeHtml;
66 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
67 import com.google.gwt.text.shared.SafeHtmlRenderer;
68 import com.google.gwt.user.client.Command;
69 import com.google.gwt.user.client.ui.AbstractImagePrototype;
70 import com.google.gwt.view.client.ListDataProvider;
71 import com.google.gwt.view.client.SelectionChangeEvent;
72 import com.google.gwt.view.client.SelectionChangeEvent.Handler;
73 import com.google.gwt.view.client.SingleSelectionModel;
74 import com.google.gwt.view.client.TreeViewModel;
75
76 public class MysharedTreeViewModel implements TreeViewModel {
77
78     protected Pithos app;
79
80     private Cell<Folder> folderCell = new AbstractCell<Folder>() {
81
82        @Override
83         public void render(Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
84             String html = AbstractImagePrototype.create(MysharedTreeView.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                 MysharedTreeViewModel.this.selectionModel.setSelected(folder, true);
95                 app.scheduleFolderHeadCommand(folder, new Command() {
96                                         
97                                         @Override
98                                         public void execute() {
99                                 FolderContextMenu menu = new FolderContextMenu(app, MysharedTreeView.images, app.getSelectedTree(), folder);
100                                 menu.setPopupPosition(x, y);
101                                 menu.show();
102                                         }
103                                 });
104             }
105         }
106     };
107
108     protected ListDataProvider<Folder> firstLevelDataProvider = new ListDataProvider<Folder>();
109
110     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
111     
112     protected SingleSelectionModel<Folder> selectionModel;
113
114     public MysharedTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
115         app = _app;
116         this.selectionModel = selectionModel;
117     }
118
119     @Override
120     public <T> NodeInfo<?> getNodeInfo(T value) {
121         if (value == null) {
122             ListDataProvider<String> rootDataProvider = new ListDataProvider<String>();
123             rootDataProvider.getList().add("Shared by me");
124             return new DefaultNodeInfo<String>(rootDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
125                 @Override
126                 public SafeHtml render(String object) {
127                     SafeHtmlBuilder builder = new SafeHtmlBuilder();
128                     render(object, builder);
129                     return builder.toSafeHtml();
130                 }
131
132                 @Override
133                 public void render(String object, SafeHtmlBuilder builder) {
134                     String html = AbstractImagePrototype.create(MysharedTreeView.images.myShared()).getHTML();
135                     builder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
136                     builder.append(MysharedTreeView.Templates.INSTANCE.nameSpan(object));
137                 }
138             }),  null, null);
139         }
140         else if (value instanceof String) {
141                 fetchSharedContainers(null);
142             return new DefaultNodeInfo<Folder>(firstLevelDataProvider, folderCell, selectionModel, null);
143         }
144         else {
145             final Folder f = (Folder) value;
146             if (dataProviderMap.get(f) == null) {
147                 dataProviderMap.put(f, new ListDataProvider<Folder>());
148             }
149             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
150             fetchFolder(f, dataProvider, false);
151             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
152         }
153     }
154
155         private void fetchSharedContainers(final Command callback) {
156         String path = "?format=json&shared=";
157         GetRequest<AccountResource> getAccount = new GetRequest<AccountResource>(AccountResource.class, app.getApiPath(), app.getUsername(), path) {
158             @Override
159             public void onSuccess(final AccountResource _result) {
160                 Iterator<Folder> iter = _result.getContainers().listIterator();
161                 fetchFolder(iter, new Command() {
162                                         
163                                         @Override
164                                         public void execute() {
165                                                 firstLevelDataProvider.getList().clear();
166                                 Folder t = null;
167                                 for (Folder c : _result.getContainers()) {
168                                         if (c.isHome())
169                                                 firstLevelDataProvider.getList().add(0, c); //Pithos is always first
170                                         else if (!c.isTrash())
171                                                 firstLevelDataProvider.getList().add(c);
172                                 }
173                                                 if (callback != null)
174                                                         callback.execute();
175                                         }
176                                 });
177             }
178
179             @Override
180             public void onError(Throwable t) {
181                 GWT.log("Error getting account", t);
182                                 app.setError(t);
183                 if (t instanceof RestException)
184                     app.displayError("Error getting account: " + ((RestException) t).getHttpStatusText());
185                 else
186                     app.displayError("System error fetching user data: " + t.getMessage());
187             }
188
189                         @Override
190                         protected void onUnauthorized(Response response) {
191                                 app.sessionExpired();
192                         }
193         };
194         getAccount.setHeader("X-Auth-Token", app.getToken());
195         Scheduler.get().scheduleDeferred(getAccount);
196         }
197
198         @Override
199     public boolean isLeaf(Object o) {
200                 if (o == null)
201                         return false;
202                 else if (o instanceof Folder) {
203             Folder f = (Folder) o;
204             return f.getSubfolders().isEmpty();
205         }
206                 else {
207                         return firstLevelDataProvider.getList().isEmpty();
208                 }
209     }
210         
211         private native void log(String msg) /*-{
212                 $wnd.console.log(msg);
213         }-*/;
214
215     protected void fetchFolder(final Iterator<Folder> iter, final Command callback) {
216         if (iter.hasNext()) {
217             final Folder f = iter.next();
218
219             String path = "/" + f.getContainer() + "?format=json&shared=&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
220             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
221                 @Override
222                 public void onSuccess(Folder _result) {
223                     fetchFolder(iter, callback);
224                 }
225
226                 @Override
227                 public void onError(Throwable t) {
228                     GWT.log("Error getting folder", t);
229                                         app.setError(t);
230                     if (t instanceof RestException)
231                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
232                     else
233                         app.displayError("System error fetching folder: " + t.getMessage());
234                 }
235
236                                 @Override
237                                 protected void onUnauthorized(Response response) {
238                                         app.sessionExpired();
239                                 }
240             };
241             getFolder.setHeader("X-Auth-Token", app.getToken());
242             Scheduler.get().scheduleDeferred(getFolder);
243         }
244         else if (callback != null)
245             callback.execute();
246     }
247
248     public Folder getSelection() {
249         return selectionModel.getSelectedObject();
250     }
251
252     public void updateFolder(Folder folder, boolean showfiles) {
253         if (dataProviderMap.get(folder) == null) {
254             dataProviderMap.put(folder, new ListDataProvider<Folder>());
255         }
256         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
257         fetchFolder(folder, dataProvider, showfiles);
258     }
259
260     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles) {
261         String path = "/" + f.getContainer() + "?format=json&shared=&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
262         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
263             @Override
264             public void onSuccess(final Folder _result) {
265                 if (showfiles)
266                     app.showFiles(_result);
267                 Iterator<Folder> iter = new ArrayList<Folder>(_result.getSubfolders()).listIterator();
268                 fetchFolder(iter, new Command() {
269                     @Override
270                     public void execute() {
271                         dataProvider.getList().clear();
272                                 dataProvider.getList().addAll(_result.getSubfolders());
273                         app.getMySharedTreeView().updateChildren(f);
274                     }
275                 });
276             }
277
278             @Override
279             public void onError(Throwable t) {
280                 GWT.log("Error getting folder", t);
281                                 app.setError(t);
282                 if (t instanceof RestException)
283                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
284                 else
285                     app.displayError("System error fetching folder: " + t.getMessage());
286             }
287
288                         @Override
289                         protected void onUnauthorized(Response response) {
290                                 app.sessionExpired();
291                         }
292         };
293         getFolder.setHeader("X-Auth-Token", app.getToken());
294         Scheduler.get().scheduleDeferred(getFolder);
295     }
296
297         public void initialize(Command callback) {
298                 fetchSharedContainers(callback);
299         }
300 }