95d471ba7a92146bd70966f3e0a7b99320f7f2b7
[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 com.google.gwt.cell.client.AbstractCell;
39 import com.google.gwt.cell.client.Cell;
40 import com.google.gwt.cell.client.ValueUpdater;
41 import com.google.gwt.core.client.GWT;
42 import com.google.gwt.core.client.Scheduler;
43 import com.google.gwt.core.client.Scheduler.ScheduledCommand;
44 import com.google.gwt.event.dom.client.ContextMenuEvent;
45 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
46 import com.google.gwt.user.client.Command;
47 import com.google.gwt.user.client.ui.AbstractImagePrototype;
48 import com.google.gwt.view.client.CellPreviewEvent;
49 import com.google.gwt.view.client.CellPreviewEvent.Handler;
50 import com.google.gwt.view.client.ListDataProvider;
51 import com.google.gwt.view.client.ProvidesKey;
52 import com.google.gwt.view.client.SingleSelectionModel;
53 import com.google.gwt.view.client.TreeViewModel;
54 import gr.grnet.pithos.web.client.FolderContextMenu;
55 import gr.grnet.pithos.web.client.Pithos;
56 import gr.grnet.pithos.web.client.foldertree.FolderTreeView.Templates;
57 import gr.grnet.pithos.web.client.rest.GetRequest;
58 import gr.grnet.pithos.web.client.rest.RestException;
59 import java.util.HashMap;
60 import java.util.Iterator;
61 import java.util.Map;
62 import java.util.Set;
63
64 public class FolderTreeViewModel implements TreeViewModel {
65
66     private 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             if (folder.isTrash())
74                 html = AbstractImagePrototype.create(FolderTreeView.images.emptyTrash()).getHTML();
75             else
76                 html = AbstractImagePrototype.create(FolderTreeView.images.folderYellow()).getHTML();
77             safeHtmlBuilder.appendHtmlConstant(html);
78             safeHtmlBuilder.append(Templates.INSTANCE.nameSpan(folder.getName()));
79         }
80
81         @Override
82         public void onBrowserEvent(Cell.Context context, com.google.gwt.dom.client.Element parent, Folder folder, com.google.gwt.dom.client.NativeEvent event, com.google.gwt.cell.client.ValueUpdater<Folder> valueUpdater) {
83             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
84                 FolderTreeViewModel.this.selectionModel.setSelected(folder, true);
85                 if (!folder.isTrash()) {
86                     FolderContextMenu menu = new FolderContextMenu(app, FolderTreeView.images, folder);
87                     menu.setPopupPosition(event.getClientX(), event.getClientY());
88                     menu.show();
89                 }
90             }
91         }
92     };
93
94     private ListDataProvider<Folder> rootDataProvider = new ListDataProvider<Folder>();
95
96     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
97
98     private SingleSelectionModel<Folder> selectionModel;
99
100     public FolderTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
101         app = _app;
102         this.selectionModel = selectionModel;
103     }
104
105     @Override
106     public <T> NodeInfo<?> getNodeInfo(T value) {
107         if (value == null) {
108             Folder f = new Folder("Loading ...");
109             rootDataProvider.getList().add(f);
110             return new DefaultNodeInfo<Folder>(rootDataProvider, folderCell, selectionModel, null);
111         }
112         else {
113             final Folder f = (Folder) value;
114             if (dataProviderMap.get(f) == null) {
115                 dataProviderMap.put(f, new ListDataProvider<Folder>(new ProvidesKey<Folder>() {
116                     @Override
117                     public Object getKey(Folder item) {
118                         return item;
119                     }
120                 }));
121             }
122             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
123             fetchFolder(f, dataProvider, false);
124             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
125         }
126     }
127
128     @Override
129     public boolean isLeaf(Object o) {
130         if (o instanceof Folder) {
131             Folder f = (Folder) o;
132             return f.getSubfolders().isEmpty();
133         }
134         return false;
135     }
136
137     private void fetchFolder(final Iterator<Folder> iter, final Command callback) {
138         if (iter.hasNext()) {
139             final Folder f = iter.next();
140
141             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + f.getPrefix();
142             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), app.getUsername(), path, f) {
143                 @Override
144                 public void onSuccess(Folder result) {
145                     fetchFolder(iter, callback);
146                 }
147
148                 @Override
149                 public void onError(Throwable t) {
150                     GWT.log("Error getting folder", t);
151                     if (t instanceof RestException)
152                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
153                     else
154                         app.displayError("System error fetching folder: " + t.getMessage());
155                 }
156             };
157             getFolder.setHeader("X-Auth-Token", app.getToken());
158             Scheduler.get().scheduleDeferred(getFolder);
159         }
160         else if (callback != null)
161             callback.execute();
162     }
163
164     public void initialize(final AccountResource account) {
165         Iterator<Folder> iter = account.getContainers().iterator();
166         fetchFolder(iter, new Command() {
167             @Override
168             public void execute() {
169                 rootDataProvider.getList().clear();
170                 rootDataProvider.getList().addAll(account.getContainers());
171                 selectionModel.setSelected(rootDataProvider.getList().get(0), true);
172                 Folder f = new Folder("Trash");
173                 f.setTrash(true);
174                 f.setContainer("trash");
175                 rootDataProvider.getList().add(f);
176                 app.updateTags();
177             }
178         });
179     }
180
181     public Folder getSelection() {
182         return selectionModel.getSelectedObject();
183     }
184
185     public void updateFolder(Folder folder, boolean showfiles) {
186         if (dataProviderMap.get(folder) == null) {
187             dataProviderMap.put(folder, new ListDataProvider<Folder>());
188         }
189         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
190         if (!folder.isTrash())
191             fetchFolder(folder, dataProvider, showfiles);
192         else
193             app.showFiles(folder);
194     }
195
196     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles) {
197         Scheduler.get().scheduleDeferred(new ScheduledCommand() {
198             @Override
199             public void execute() {
200                 String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + f.getPrefix();
201                 GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), app.getUsername(), path, f) {
202                     @Override
203                     public void onSuccess(final Folder result) {
204                         if (showfiles)
205                             app.showFiles(result);
206                         Iterator<Folder> iter = result.getSubfolders().iterator();
207                         fetchFolder(iter, new Command() {
208                             @Override
209                             public void execute() {
210                                 dataProvider.getList().clear();
211                                 dataProvider.getList().addAll(result.getSubfolders());
212                                 app.getFolderTreeView().updateChildren(f);
213                             }
214                         });
215                     }
216
217                     @Override
218                     public void onError(Throwable t) {
219                         GWT.log("Error getting folder", t);
220                         if (t instanceof RestException)
221                             app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
222                         else
223                             app.displayError("System error fetching folder: " + t.getMessage());
224                     }
225                 };
226                 getFolder.setHeader("X-Auth-Token", app.getToken());
227                 Scheduler.get().scheduleDeferred(getFolder);
228             }
229         });
230     }
231 }