Reverted back to ListDataProvider. Fixed race condition when creating my shared tree...
[pithos] / web_client / src / gr / grnet / pithos / web / client / foldertree / FolderTreeViewModel.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.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.HashSet;
47 import java.util.Iterator;
48 import java.util.Map;
49
50 import com.google.gwt.cell.client.AbstractCell;
51 import com.google.gwt.cell.client.Cell;
52 import com.google.gwt.core.client.GWT;
53 import com.google.gwt.core.client.Scheduler;
54 import com.google.gwt.event.dom.client.ContextMenuEvent;
55 import com.google.gwt.http.client.Response;
56 import com.google.gwt.http.client.URL;
57 import com.google.gwt.safehtml.shared.SafeHtml;
58 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
59 import com.google.gwt.user.client.Command;
60 import com.google.gwt.user.client.ui.AbstractImagePrototype;
61 import com.google.gwt.view.client.AsyncDataProvider;
62 import com.google.gwt.view.client.HasData;
63 import com.google.gwt.view.client.ListDataProvider;
64 import com.google.gwt.view.client.SingleSelectionModel;
65 import com.google.gwt.view.client.TreeViewModel;
66
67 public class FolderTreeViewModel implements TreeViewModel {
68
69         protected Pithos app;
70     
71     private Cell<Folder> folderCell = new AbstractCell<Folder>(ContextMenuEvent.getType().getName()) {
72
73        @Override
74         public void render(@SuppressWarnings("unused") Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
75             String html;
76             SafeHtml name;
77                 if (folder.isHome()) {
78                         html = AbstractImagePrototype.create(FolderTreeView.images.home()).getHTML();
79                         name = Templates.INSTANCE.nameSpan("Home");
80                 }
81                 else if (folder.isTrash()) {
82                         html = AbstractImagePrototype.create(FolderTreeView.images.trash()).getHTML();
83                         name = Templates.INSTANCE.nameSpan("Trash");
84                 }
85             else {
86                 html = AbstractImagePrototype.create(FolderTreeView.images.folderYellow()).getHTML();
87                         name = Templates.INSTANCE.nameSpan(folder.getName());
88             }
89             safeHtmlBuilder.appendHtmlConstant(html);
90             safeHtmlBuilder.append(name);
91         }
92
93         @Override
94         public void onBrowserEvent(@SuppressWarnings("unused") Cell.Context context, @SuppressWarnings("unused") com.google.gwt.dom.client.Element parent, Folder folder, com.google.gwt.dom.client.NativeEvent event, @SuppressWarnings("unused") com.google.gwt.cell.client.ValueUpdater<Folder> valueUpdater) {
95             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
96                 FolderTreeViewModel.this.selectionModel.setSelected(folder, true);
97                 FolderContextMenu menu = new FolderContextMenu(app, FolderTreeView.images, app.getSelectedTree(), folder);
98                 menu.setPopupPosition(event.getClientX(), event.getClientY());
99                 menu.show();
100             }
101         }
102     };
103
104     protected ListDataProvider<Folder> rootDataProvider = new ListDataProvider<Folder>();
105
106     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
107
108     protected SingleSelectionModel<Folder> selectionModel;
109
110     public FolderTreeViewModel(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             Folder f = new Folder("Loading ...");
119             rootDataProvider.getList().add(f);
120             return new DefaultNodeInfo<Folder>(rootDataProvider, folderCell, selectionModel, null);
121         }
122                 final Folder f = (Folder) value;
123                 if (dataProviderMap.get(f) == null) {
124                     dataProviderMap.put(f, new ListDataProvider<Folder>());
125                 }
126                 final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
127                 dataProvider.getList().clear();
128                 dataProvider.getList().addAll(f.getSubfolders());
129                 fetchFolder(f, dataProvider, false, null);
130                 return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
131     }
132
133     @Override
134     public boolean isLeaf(Object o) {
135         if (o instanceof Folder) {
136             Folder f = (Folder) o;
137             return f.getSubfolders().isEmpty();
138         }
139         return false;
140     }
141
142     protected void fetchFolder(final Iterator<Folder> iter, final Command callback) {
143         if (iter.hasNext()) {
144             final Folder f = iter.next();
145
146             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
147             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
148                 @Override
149                 public void onSuccess(@SuppressWarnings("unused") Folder _result) {
150                     fetchFolder(iter, callback);
151                 }
152
153                 @Override
154                 public void onError(Throwable t) {
155                     GWT.log("Error getting folder", t);
156                     if (t instanceof RestException)
157                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
158                     else
159                         app.displayError("System error fetching folder: " + t.getMessage());
160                 }
161
162                                 @Override
163                                 protected void onUnauthorized(Response response) {
164                                         app.sessionExpired();
165                                 }
166             };
167             getFolder.setHeader("X-Auth-Token", app.getToken());
168             Scheduler.get().scheduleDeferred(getFolder);
169         }
170         else if (callback != null)
171             callback.execute();
172     }
173
174     public void initialize(final AccountResource account, final Command callback) {
175         Iterator<Folder> iter = account.getContainers().iterator();
176         fetchFolder(iter, new Command() {
177             @Override
178             public void execute() {
179                 rootDataProvider.getList().clear();
180                 rootDataProvider.getList().addAll(account.getContainers());
181                 selectionModel.setSelected(rootDataProvider.getList().get(0), true);
182                 if (callback != null)
183                         callback.execute();
184             }
185         });
186     }
187
188     public Folder getSelection() {
189         return selectionModel.getSelectedObject();
190     }
191
192     public void updateFolder(final Folder folder, boolean showfiles, final Command callback) {
193         if (dataProviderMap.get(folder) == null) {
194             dataProviderMap.put(folder, new ListDataProvider<Folder>());
195         }
196         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
197         fetchFolder(folder, dataProvider, showfiles, new Command() {
198                         
199                         @Override
200                         public void execute() {
201                                 app.getFolderTreeView().openFolder(folder);
202                                 if (callback != null)
203                                         callback.execute();
204                         }
205                 });
206     }
207
208     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles, final Command callback) {
209         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + URL.encodeQueryString(f.getPrefix());
210         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), f.getOwner(), path, f) {
211             @Override
212             public void onSuccess(final Folder _result) {
213                 if (showfiles)
214                     app.showFiles(_result);
215                 Iterator<Folder> iter = _result.getSubfolders().iterator();
216                 fetchFolder(iter, new Command() {
217                     @Override
218                     public void execute() {
219                         dataProvider.getList().clear();
220                         dataProvider.getList().addAll(_result.getSubfolders());
221 //                        app.getFolderTreeView().updateChildren(f);
222                         if (callback != null)
223                                 callback.execute();
224                     }
225                 });
226             }
227
228             @Override
229             public void onError(Throwable t) {
230                 GWT.log("Error getting folder", t);
231                 if (t instanceof RestException)
232                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
233                 else
234                     app.displayError("System error fetching folder: " + t.getMessage());
235             }
236
237                         @Override
238                         protected void onUnauthorized(Response response) {
239                                 app.sessionExpired();
240                         }
241         };
242         getFolder.setHeader("X-Auth-Token", app.getToken());
243         Scheduler.get().scheduleDeferred(getFolder);
244     }
245 }