fb4d31f225378b5741ea7980a7d34ddac80f766e
[pithos-web-client] / src / gr / grnet / pithos / web / client / foldertree / FolderTreeViewModel.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.foldertree;
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.FolderTreeView.Templates;
41 import gr.grnet.pithos.web.client.rest.GetRequest;
42 import gr.grnet.pithos.web.client.rest.RestException;
43
44 import java.util.ArrayList;
45 import java.util.HashMap;
46 import java.util.Iterator;
47 import java.util.Map;
48
49 import com.google.gwt.cell.client.AbstractCell;
50 import com.google.gwt.cell.client.Cell;
51 import com.google.gwt.core.client.GWT;
52 import com.google.gwt.core.client.Scheduler;
53 import com.google.gwt.event.dom.client.ContextMenuEvent;
54 import com.google.gwt.http.client.Response;
55 import com.google.gwt.http.client.URL;
56 import com.google.gwt.safehtml.shared.SafeHtml;
57 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
58 import com.google.gwt.user.client.Command;
59 import com.google.gwt.user.client.ui.AbstractImagePrototype;
60 import com.google.gwt.view.client.ListDataProvider;
61 import com.google.gwt.view.client.SingleSelectionModel;
62 import com.google.gwt.view.client.TreeViewModel;
63
64 public class FolderTreeViewModel implements TreeViewModel {
65
66         protected Pithos app;
67     
68     private Cell<Folder> folderCell = new AbstractCell<Folder>(ContextMenuEvent.getType().getName()) {
69
70        @Override
71         public void render(Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
72             String html;
73             SafeHtml name;
74                 if (folder.isHome()) {
75                         html = AbstractImagePrototype.create(FolderTreeView.images.folderYellow()).getHTML();
76                         name = Templates.INSTANCE.nameSpan(folder.getName());
77                 }
78                 else if (folder.isTrash()) {
79                         html = AbstractImagePrototype.create(FolderTreeView.images.trash()).getHTML();
80                         name = Templates.INSTANCE.nameSpan(folder.getName());
81                 }
82             else {
83                 html = AbstractImagePrototype.create(folder.isShared() ? FolderTreeView.images.myShared() : FolderTreeView.images.folderYellow()).getHTML();
84                         name = Templates.INSTANCE.nameSpan(folder.getName());
85             }
86             safeHtmlBuilder.appendHtmlConstant(html).appendHtmlConstant("&nbsp;");
87             safeHtmlBuilder.append(name);
88         }
89
90         @Override
91         public void onBrowserEvent(Cell.Context context, com.google.gwt.dom.client.Element parent, final Folder folder, com.google.gwt.dom.client.NativeEvent event, com.google.gwt.cell.client.ValueUpdater<Folder> valueUpdater) {
92             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
93                 final int x = event.getClientX();
94                 final int y = event.getClientY();
95                 FolderTreeViewModel.this.selectionModel.setSelected(folder, true);
96                 app.scheduleFolderHeadCommand(folder, new Command() {
97                                         
98                                         @Override
99                                         public void execute() {
100                                 FolderContextMenu menu = new FolderContextMenu(app, FolderTreeView.images, app.getSelectedTree(), folder);
101                                 menu.setPopupPosition(x, y);
102                                 menu.show();
103                                         }
104                                 });
105             }
106         }
107     };
108
109     protected ListDataProvider<Folder> rootDataProvider = new ListDataProvider<Folder>();
110
111     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
112
113     protected SingleSelectionModel<Folder> selectionModel;
114
115     public FolderTreeViewModel(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             Folder f = new Folder("Loading ...");
124             rootDataProvider.getList().add(f);
125             return new DefaultNodeInfo<Folder>(rootDataProvider, folderCell, selectionModel, null);
126         }
127                 final Folder f = (Folder) value;
128                 if (dataProviderMap.get(f) == null) {
129                     dataProviderMap.put(f, new ListDataProvider<Folder>());
130                 }
131                 final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
132                 //This prevents the loading indicator
133                 fetchFolder(f, dataProvider, false, null);
134                 return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
135     }
136
137     @Override
138     public boolean isLeaf(Object o) {
139         if (o instanceof Folder) {
140             Folder f = (Folder) o;
141             return f.getSubfolders().isEmpty();
142         }
143         return false;
144     }
145
146         private native void log(String msg) /*-{
147         $wnd.console.log(msg);
148         }-*/;
149
150     protected void fetchFolder(final Iterator<Folder> iter, final Command callback) {
151         if (iter.hasNext()) {
152             final Folder f = iter.next();
153
154             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
155             GetRequest<Folder> getFolder = new GetRequest<Folder>(app, Folder.class, app.getApiPath(), f.getOwnerID(), path, f) {
156                 @Override
157                 public void onSuccess(Folder _result) {
158                     fetchFolder(iter, callback);
159                 }
160
161                 @Override
162                 public void onError(Throwable t) {
163                         if (retries >= MAX_RETRIES) {
164                             GWT.log("Error getting folder", t);
165                                                 app.setError(t);
166                             if (t instanceof RestException)
167                                 app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
168                             else
169                                 app.displayError("System error fetching folder: " + t.getMessage());
170                         }
171                         else {//retry
172                                 GWT.log("Retry " + retries);
173                                 Scheduler.get().scheduleDeferred(this);
174                         }
175                 }
176
177                                 @Override
178                                 protected void onUnauthorized(Response response) {
179                                         if (retries >= MAX_RETRIES)
180                                                 app.sessionExpired();
181                         else //retry
182                                 Scheduler.get().scheduleDeferred(this);
183                                 }
184             };
185             getFolder.setHeader("X-Auth-Token", app.getUserToken());
186             Scheduler.get().scheduleDeferred(getFolder);
187         }
188         else if (callback != null)
189             callback.execute();
190     }
191
192     public void initialize(final AccountResource account, final Command callback) {
193         Iterator<Folder> iter = account.getContainers().listIterator();
194         fetchFolder(iter, new Command() {
195             @Override
196             public void execute() {
197                 rootDataProvider.getList().clear();
198                 Folder t = null;
199                 for (Folder c : account.getContainers()) {
200                         if (c.isHome())
201                         rootDataProvider.getList().add(0, c); //Pithos is always first
202                         else if (c.isTrash())
203                                 t = c; //Keep trash for adding in the end
204                         else
205                                 rootDataProvider.getList().add(c);
206                 }
207                 if (t != null)
208                         rootDataProvider.getList().add(t);
209                 selectionModel.setSelected(rootDataProvider.getList().get(0), true);
210                 if (callback != null)
211                         callback.execute();
212             }
213         });
214     }
215
216     public Folder getSelection() {
217         return selectionModel.getSelectedObject();
218     }
219
220     public void updateFolder(final Folder folder, boolean showfiles, final Command callback, final boolean openParent) {
221         if (dataProviderMap.get(folder) == null) {
222             dataProviderMap.put(folder, new ListDataProvider<Folder>());
223         }
224         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
225         fetchFolder(folder, dataProvider, showfiles, new Command() {
226                         
227                         @Override
228                         public void execute() {
229                                 if (openParent)
230                                         app.getFolderTreeView().openFolder(folder);
231                                 if (callback != null)
232                                         callback.execute();
233                         }
234                 });
235     }
236
237     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles, final Command callback) {
238         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
239         GetRequest<Folder> getFolder = new GetRequest<Folder>(app, Folder.class, app.getApiPath(), f.getOwnerID(), path, f) {
240             @Override
241             public void onSuccess(final Folder _result) {
242                 if (showfiles)
243                     app.showFiles(_result);
244                 int defaultSize = app.getFolderTreeView().tree.getDefaultNodeSize();
245                 int size = _result.getSubfolders().size();
246                 if (size > defaultSize)
247                         app.getFolderTreeView().tree.setDefaultNodeSize(size);
248                 Iterator<Folder> iter = new ArrayList<Folder>(_result.getSubfolders()).listIterator();
249                 fetchFolder(iter, new Command() {
250                     @Override
251                     public void execute() {
252                         dataProvider.getList().clear();
253                         dataProvider.getList().addAll(_result.getSubfolders());
254                         if (callback != null)
255                                 callback.execute();
256                     }
257                 });
258             }
259
260             @Override
261             public void onError(Throwable t) {
262                 if (retries >= MAX_RETRIES) {
263                         GWT.log("Error getting folder", t);
264                                         app.setError(t);
265                         if (t instanceof RestException)
266                             app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
267                         else
268                             app.displayError("System error fetching folder: " + t.getMessage());
269                 }
270                 else {//retry
271                         GWT.log("Retry " + retries);
272                         Scheduler.get().scheduleDeferred(this);
273                 }
274             }
275
276                         @Override
277                         protected void onUnauthorized(Response response) {
278                                 if (retries >= MAX_RETRIES)
279                                         app.sessionExpired();
280                 else //retry
281                         Scheduler.get().scheduleDeferred(this);
282                         }
283         };
284         getFolder.setHeader("X-Auth-Token", app.getUserToken());
285         Scheduler.get().scheduleDeferred(getFolder);
286     }
287 }