command line client fix create object using hashmaps
[pithos] / web_client / src / gr / grnet / pithos / web / client / mysharedtree / MysharedTreeViewModel.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.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.mysharedtree.MysharedTreeView.Templates;
43 import gr.grnet.pithos.web.client.rest.GetRequest;
44 import gr.grnet.pithos.web.client.rest.RestException;
45
46 import java.util.HashMap;
47 import java.util.HashSet;
48 import java.util.Iterator;
49 import java.util.List;
50 import java.util.Map;
51 import java.util.Set;
52
53 import com.google.gwt.cell.client.AbstractCell;
54 import com.google.gwt.cell.client.Cell;
55 import com.google.gwt.cell.client.TextCell;
56 import com.google.gwt.cell.client.ValueUpdater;
57 import com.google.gwt.core.client.GWT;
58 import com.google.gwt.core.client.Scheduler;
59 import com.google.gwt.event.dom.client.ContextMenuEvent;
60 import com.google.gwt.safehtml.shared.SafeHtml;
61 import com.google.gwt.safehtml.shared.SafeHtmlBuilder;
62 import com.google.gwt.text.shared.SafeHtmlRenderer;
63 import com.google.gwt.user.client.Command;
64 import com.google.gwt.user.client.ui.AbstractImagePrototype;
65 import com.google.gwt.view.client.ListDataProvider;
66 import com.google.gwt.view.client.SelectionChangeEvent;
67 import com.google.gwt.view.client.SelectionChangeEvent.Handler;
68 import com.google.gwt.view.client.SingleSelectionModel;
69 import com.google.gwt.view.client.TreeViewModel;
70
71 public class MysharedTreeViewModel implements TreeViewModel {
72
73     protected Pithos app;
74
75     private Cell<Folder> folderCell = new AbstractCell<Folder>(ContextMenuEvent.getType().getName()) {
76
77        @Override
78         public void render(@SuppressWarnings("unused") Context context, Folder folder, SafeHtmlBuilder safeHtmlBuilder) {
79             String html = AbstractImagePrototype.create(MysharedTreeView.images.folderYellow()).getHTML();
80             safeHtmlBuilder.appendHtmlConstant(html);
81             safeHtmlBuilder.append(Templates.INSTANCE.nameSpan(folder.getName()));
82         }
83
84         @Override
85         public void onBrowserEvent(@SuppressWarnings("unused") Context context, @SuppressWarnings("unused") com.google.gwt.dom.client.Element parent, Folder folder, com.google.gwt.dom.client.NativeEvent event, @SuppressWarnings("unused") ValueUpdater<Folder> valueUpdater) {
86             if (event.getType().equals(ContextMenuEvent.getType().getName())) {
87                 MysharedTreeViewModel.this.selectionModel.setSelected(folder, true);
88                 FolderContextMenu menu = new FolderContextMenu(app, MysharedTreeView.images, folder);
89                 menu.setPopupPosition(event.getClientX(), event.getClientY());
90                 menu.show();
91             }
92         }
93     };
94
95     protected ListDataProvider<Folder> firstLevelDataProvider = new ListDataProvider<Folder>();
96
97     private Map<Folder, ListDataProvider<Folder>> dataProviderMap = new HashMap<Folder, ListDataProvider<Folder>>();
98     
99     protected Set<File> sharedFiles = new HashSet<File>();
100
101     protected SingleSelectionModel<Folder> selectionModel;
102
103     public MysharedTreeViewModel(Pithos _app, SingleSelectionModel<Folder> selectionModel) {
104         app = _app;
105         this.selectionModel = selectionModel;
106     }
107
108     @Override
109     public <T> NodeInfo<?> getNodeInfo(T value) {
110         if (value == null) {
111             ListDataProvider<String> rootDataProvider = new ListDataProvider<String>();
112             rootDataProvider.getList().add("My Shared");
113             final SingleSelectionModel<String> selectionModel2 = new SingleSelectionModel<String>();
114             selectionModel2.addSelectionChangeHandler(new Handler() {
115
116                 @Override
117                 public void onSelectionChange(@SuppressWarnings("unused") SelectionChangeEvent event) {
118                     if (selectionModel2.getSelectedObject() != null) {
119                         app.deselectOthers(selectionModel2);
120                         fetchSharedFiles();
121                     }
122                 }
123             });
124             app.addSelectionModel(selectionModel2);
125             return new DefaultNodeInfo<String>(rootDataProvider, new TextCell(new SafeHtmlRenderer<String>() {
126                 @Override
127                 public SafeHtml render(String object) {
128                     SafeHtmlBuilder builder = new SafeHtmlBuilder();
129                     render(object, builder);
130                     return builder.toSafeHtml();
131                 }
132
133                 @Override
134                 public void render(String object, SafeHtmlBuilder builder) {
135                     String html = AbstractImagePrototype.create(MysharedTreeView.images.myShared()).getHTML();
136                     builder.appendHtmlConstant(html);
137                     builder.append(MysharedTreeView.Templates.INSTANCE.nameSpan(object));
138                 }
139             }),  selectionModel2, null);
140         }
141         else if (value instanceof String) {
142                 fetchSharedContainers();
143             return new DefaultNodeInfo<Folder>(firstLevelDataProvider, folderCell, selectionModel, null);
144         }
145         else {
146             final Folder f = (Folder) value;
147             if (dataProviderMap.get(f) == null) {
148                 dataProviderMap.put(f, new ListDataProvider<Folder>());
149             }
150             final ListDataProvider<Folder> dataProvider = dataProviderMap.get(f);
151             fetchFolder(f, dataProvider, false);
152             return new DefaultNodeInfo<Folder>(dataProvider, folderCell, selectionModel, null);
153         }
154     }
155
156         protected void fetchSharedFiles() {
157         final List<Folder> containers = app.getAccount().getContainers();
158         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
159         Iterator<Folder> iter = containers.iterator();
160         fetchFolder(iter, tempProvider, new Command() {
161                         
162                         @Override
163                         public void execute() {
164                                 firstLevelDataProvider.getList().clear();
165                                 firstLevelDataProvider.getList().addAll(tempProvider.getList());
166                                 app.showFiles(sharedFiles);
167                         }
168                 });
169         }
170
171         private void fetchSharedContainers() {
172         final List<Folder> containers = app.getAccount().getContainers();
173         final ListDataProvider<Folder> tempProvider = new ListDataProvider<Folder>();
174         Iterator<Folder> iter = containers.iterator();
175         fetchFolder(iter, tempProvider, new Command() {
176                         
177                         @Override
178                         public void execute() {
179                                 firstLevelDataProvider.getList().clear();
180                                 firstLevelDataProvider.getList().addAll(tempProvider.getList());
181                         }
182                 });
183         }
184
185         @Override
186     public boolean isLeaf(Object o) {
187         if (o instanceof Folder) {
188             Folder f = (Folder) o;
189             return f.getSubfolders().isEmpty();
190         }
191         return false;
192     }
193
194     protected void fetchFolder(final Iterator<Folder> iter, final ListDataProvider<Folder> dataProvider, final Command callback) {
195         if (iter.hasNext()) {
196             final Folder f = iter.next();
197
198             String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + f.getPrefix();
199             GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), app.getUsername(), path, f) {
200                 @Override
201                 public void onSuccess(Folder _result) {
202                         if (!_result.isShared()) {
203                                 for (File file : _result.getFiles()) {
204                                         if (file.isShared())
205                                                 sharedFiles.add(file);
206                                 }
207                                 Iterator<Folder> iter2 = _result.getSubfolders().iterator();
208                                 fetchFolder(iter2, dataProvider, new Command() {
209                                                         
210                                                         @Override
211                                                         public void execute() {
212                                             fetchFolder(iter, dataProvider, callback);
213                                                         }
214                                                 });
215                         }
216                         else {
217                                 dataProvider.getList().add(_result);
218                             fetchFolder(iter, dataProvider, callback);
219                         }
220                 }
221
222                 @Override
223                 public void onError(Throwable t) {
224                     GWT.log("Error getting folder", t);
225                     if (t instanceof RestException)
226                         app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
227                     else
228                         app.displayError("System error fetching folder: " + t.getMessage());
229                 }
230             };
231             getFolder.setHeader("X-Auth-Token", app.getToken());
232             Scheduler.get().scheduleDeferred(getFolder);
233         }
234         else if (callback != null)
235             callback.execute();
236     }
237
238     public Folder getSelection() {
239         return selectionModel.getSelectedObject();
240     }
241
242     public void updateFolder(Folder folder, boolean showfiles) {
243         if (dataProviderMap.get(folder) == null) {
244             dataProviderMap.put(folder, new ListDataProvider<Folder>());
245         }
246         final ListDataProvider<Folder> dataProvider = dataProviderMap.get(folder);
247         fetchFolder(folder, dataProvider, showfiles);
248     }
249
250     public void fetchFolder(final Folder f, final ListDataProvider<Folder> dataProvider, final boolean showfiles) {
251         String path = "/" + f.getContainer() + "?format=json&delimiter=/&prefix=" + f.getPrefix();
252         GetRequest<Folder> getFolder = new GetRequest<Folder>(Folder.class, app.getApiPath(), app.getUsername(), path, f) {
253             @Override
254             public void onSuccess(final Folder _result) {
255                 if (showfiles)
256                     app.showFiles(_result);
257                 Iterator<Folder> iter = _result.getSubfolders().iterator();
258                 fetchFolder(iter, dataProvider, new Command() {
259                     @Override
260                     public void execute() {
261                         dataProvider.getList().clear();
262                                 dataProvider.getList().addAll(_result.getSubfolders());
263                         app.getMySharedTreeView().updateChildren(f);
264                     }
265                 });
266             }
267
268             @Override
269             public void onError(Throwable t) {
270                 GWT.log("Error getting folder", t);
271                 if (t instanceof RestException)
272                     app.displayError("Error getting folder: " + ((RestException) t).getHttpStatusText());
273                 else
274                     app.displayError("System error fetching folder: " + t.getMessage());
275             }
276         };
277         getFolder.setHeader("X-Auth-Token", app.getToken());
278         Scheduler.get().scheduleDeferred(getFolder);
279     }
280 }