JAL-715 allow headless construction for testing
[jalview.git] / src / jalview / ws / rest / RestClient.java
1 /*
2  * Jalview - A Sequence Alignment Editor and Viewer (Version 2.8)
3  * Copyright (C) 2012 J Procter, AM Waterhouse, LM Lui, J Engelhardt, G Barton, M Clamp, S Searle
4  * 
5  * This file is part of Jalview.
6  * 
7  * Jalview is free software: you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License 
9  * as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
10  *  
11  * Jalview is distributed in the hope that it will be useful, but 
12  * WITHOUT ANY WARRANTY; without even the implied warranty 
13  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
14  * PURPOSE.  See the GNU General Public License for more details.
15  * 
16  * You should have received a copy of the GNU General Public License along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
17  */
18 package jalview.ws.rest;
19
20 import java.awt.event.ActionEvent;
21 import java.awt.event.ActionListener;
22 import java.util.Hashtable;
23 import java.util.Vector;
24
25 import javax.swing.JMenu;
26 import javax.swing.JMenuItem;
27 import javax.swing.JOptionPane;
28 import javax.swing.event.MenuEvent;
29 import javax.swing.event.MenuListener;
30
31 import jalview.bin.Cache;
32 import jalview.datamodel.AlignmentView;
33 import jalview.gui.AlignFrame;
34 import jalview.gui.AlignViewport;
35 import jalview.gui.AlignmentPanel;
36 import jalview.gui.Desktop;
37 import jalview.gui.WebserviceInfo;
38 import jalview.io.packed.DataProvider.JvDataType;
39 import jalview.ws.WSClient;
40 import jalview.ws.WSClientI;
41 import jalview.ws.WSMenuEntryProviderI;
42
43 /**
44  * @author JimP
45  * 
46  */
47 public class RestClient extends WSClient implements WSClientI,
48         WSMenuEntryProviderI
49 {
50   RestServiceDescription service;
51
52   public RestClient(RestServiceDescription rsd)
53   {
54     service = rsd;
55   }
56
57   /**
58    * parent alignframe for this job
59    */
60   AlignFrame af;
61
62   /**
63    * alignment view which provides data for job.
64    */
65   AlignViewport av;
66
67   /**
68    * get the alignFrame for the associated input data if it exists.
69    * 
70    * @return
71    */
72   protected AlignFrame recoverAlignFrameForView()
73   {
74     return jalview.gui.Desktop.getAlignFrameFor(av);
75   }
76
77   public RestClient(RestServiceDescription service2, AlignFrame alignFrame)
78   {
79     this(service2, alignFrame, false);
80   }
81   boolean headless = false;
82   public RestClient(RestServiceDescription service2, AlignFrame alignFrame, boolean nogui)
83   {
84     service = service2;
85     af = alignFrame;
86     av = alignFrame.getViewport();
87     headless = nogui;
88     constructJob();
89   }
90
91   public void setWebserviceInfo(boolean headless)
92   {
93     WebServiceJobTitle = service.details.Action + " using "
94             + service.details.Name;
95     WebServiceName = service.details.Name;
96     WebServiceReference = "No reference - go to url for more info";
97     if (service.details.description != null)
98     {
99       WebServiceReference = service.details.description;
100     }
101     if (!headless)
102     {
103       wsInfo = new WebserviceInfo(WebServiceJobTitle, WebServiceName + "\n"
104               + WebServiceReference);
105       wsInfo.setRenderAsHtml(true);
106     }
107
108   }
109
110   @Override
111   public boolean isCancellable()
112   {
113     // TODO define process for cancelling rsbws jobs
114     return false;
115   }
116
117   @Override
118   public boolean canMergeResults()
119   {
120     // TODO process service definition to identify if the results might be
121     // mergeable
122     // TODO: change comparison for annotation merge
123     return false;
124   }
125
126   @Override
127   public void cancelJob()
128   {
129     System.err.println("Cannot cancel this job type: " + service);
130   }
131
132   @Override
133   public void attachWSMenuEntry(final JMenu wsmenu,
134           final AlignFrame alignFrame)
135   {
136     JMenuItem submit = new JMenuItem(service.details.Name);
137     submit.setToolTipText(service.details.Action + " using "
138             + service.details.Name);
139     submit.addActionListener(new ActionListener()
140     {
141
142       @Override
143       public void actionPerformed(ActionEvent e)
144       {
145         new RestClient(service, alignFrame);
146       }
147
148     });
149     wsmenu.add(submit);
150     // TODO: menu listener should enable/disable entry depending upon selection
151     // state of the alignment
152     wsmenu.addMenuListener(new MenuListener()
153     {
154
155       @Override
156       public void menuSelected(MenuEvent e)
157       {
158         // TODO Auto-generated method stub
159
160       }
161
162       @Override
163       public void menuDeselected(MenuEvent e)
164       {
165         // TODO Auto-generated method stub
166
167       }
168
169       @Override
170       public void menuCanceled(MenuEvent e)
171       {
172         // TODO Auto-generated method stub
173
174       }
175
176     });
177
178   }
179
180   /**
181    * record of initial undoredo hash for the alignFrame providing data for this
182    * job.
183    */
184   long[] undoredo = null;
185
186   /**
187    * Compare the original input data to the data currently presented to the
188    * user. // LOGIC: compare undo/redo - if same, merge regardless (coping with
189    * any changes in hidden columns as normal) // if different undo/redo then
190    * compare region that was submitted // if same, then merge as before, if
191    * different then prompt user to open a new window.
192    * 
193    * @return
194    */
195   protected boolean isAlignmentModified()
196   {
197     if (undoredo == null || av == null || av.getAlignment() == null)
198     {
199       // always return modified if we don't have access to live GUI elements
200       // anymore.
201       return true;
202     }
203     if (av.isUndoRedoHashModified(undoredo))
204     {
205       // alignment has been modified in some way.
206       return true;
207     }
208     // TODO: look deeper into modification of selection state, etc that may
209     // affect RestJobThread.realiseResults(boolean merge);
210     return false;
211
212   }
213
214   /**
215    * TODO: combine to form a dataset+alignment+annotation context
216    */
217   AlignmentView _input;
218
219   /**
220    * input data context
221    */
222   jalview.io.packed.JalviewDataset jds;
223
224   /**
225    * informative name for results
226    */
227   public String viewTitle;
228
229   protected void constructJob()
230   {
231     service.setInvolvesFlags();
232     // record all aspects of alignment view so we can merge back or recreate
233     // later
234     undoredo = av.getUndoRedoHash();
235     /**
236      * delete ? Vector sgs = av.getAlignment().getGroups(); if (sgs!=null) {
237      * _sgs = new SequenceGroup[sgs.size()]; sgs.copyInto(_sgs); } else { _sgs =
238      * new SequenceGroup[0]; }
239      */
240     boolean selExists = (av.getSelectionGroup() != null)
241             && (av.getSelectionGroup().getSize() > 1);
242     // TODO: JAL-715: refactor to alignViewport methods and revise to full
243     // focus+context+dataset input data staging model
244     if (selExists)
245     {
246       if (service.partitiondata)
247       {
248         if (av.getAlignment().getGroups() != null
249                 && av.getAlignment().getGroups().size() > 0)
250         {
251           // intersect groups with selected region
252           _input = new AlignmentView(av.getAlignment(),
253                   av.getColumnSelection(), av.getSelectionGroup(),
254                   av.hasHiddenColumns(), true, true);
255           viewTitle = "selected "
256                   + (av.hasHiddenColumns() ? "visible" : "")
257                   + " region of " + af.getTitle();
258         }
259         else
260         {
261           // use selected region to partition alignment
262           _input = new AlignmentView(av.getAlignment(),
263                   av.getColumnSelection(), av.getSelectionGroup(),
264                   av.hasHiddenColumns(), false, true);
265         }
266         viewTitle = "select and unselected "
267                 + (av.hasHiddenColumns() ? "visible" : "")
268                 + " regions from " + af.getTitle();
269       }
270       else
271       {
272         // just take selected region intersection
273         _input = new AlignmentView(av.getAlignment(),
274                 av.getColumnSelection(), av.getSelectionGroup(),
275                 av.hasHiddenColumns(), true, true);
276         viewTitle = "selected " + (av.hasHiddenColumns() ? "visible" : "")
277                 + " region of " + af.getTitle();
278       }
279     }
280     else
281     {
282       // standard alignment view without selection present
283       _input = new AlignmentView(av.getAlignment(),
284               av.getColumnSelection(), null, av.hasHiddenColumns(), false,
285               true);
286       viewTitle = "" + (av.hasHiddenColumns() ? "visible region of " : "")
287               + af.getTitle();
288     }
289
290     RestJobThread jobsthread = new RestJobThread(this);
291
292     if (jobsthread.isValid())
293     {
294       setWebserviceInfo(headless);
295       if (!headless) {
296         wsInfo.setthisService(this);
297         jobsthread.setWebServiceInfo(wsInfo);
298       }
299       jobsthread.start();
300     }
301     else
302     {
303       // TODO: try to tell the user why the job couldn't be started.
304       JOptionPane
305               .showMessageDialog(
306                       Desktop.desktop,
307                       (jobsthread.hasWarnings() ? jobsthread.getWarnings()
308                               : "The Job couldn't be started. Please check your input, and the Jalview console for any warning messages."),
309                       "Unable to start web service analysis",
310                       JOptionPane.WARNING_MESSAGE);
311     }
312   }
313
314   public static RestClient makeShmmrRestClient()
315   {
316     String action = "Analysis", description = "Sequence Harmony and Multi-Relief (Brandt et al. 2010)", name = "Multi-Harmony";
317     Hashtable<String, InputType> iparams = new Hashtable<String, InputType>();
318     jalview.ws.rest.params.JobConstant toolp;
319     // toolp = new jalview.ws.rest.JobConstant("tool","jalview");
320     // iparams.put(toolp.token, toolp);
321     // toolp = new jalview.ws.rest.params.JobConstant("mbjob[method]","shmr");
322     // iparams.put(toolp.token, toolp);
323     // toolp = new
324     // jalview.ws.rest.params.JobConstant("mbjob[description]","step 1");
325     // iparams.put(toolp.token, toolp);
326     // toolp = new jalview.ws.rest.params.JobConstant("start_search","1");
327     // iparams.put(toolp.token, toolp);
328     // toolp = new jalview.ws.rest.params.JobConstant("blast","0");
329     // iparams.put(toolp.token, toolp);
330
331     jalview.ws.rest.params.Alignment aliinput = new jalview.ws.rest.params.Alignment();
332     // SHMR server has a 65K limit for content pasted into the 'ali' parameter,
333     // so we always upload our files.
334     aliinput.token = "ali_file";
335     aliinput.writeAsFile = true;
336     iparams.put(aliinput.token, aliinput);
337     jalview.ws.rest.params.SeqGroupIndexVector sgroups = new jalview.ws.rest.params.SeqGroupIndexVector();
338     sgroups.setMinsize(2);
339     sgroups.min = 2;// need at least two group defined to make a partition
340     iparams.put("groups", sgroups);
341     sgroups.token = "groups";
342     sgroups.sep = " ";
343     RestServiceDescription shmrService = new RestServiceDescription(
344             action,
345             description,
346             name,
347             "http://zeus.few.vu.nl/programs/shmrwww/index.php?tool=jalview",// ?tool=jalview&mbjob[method]=shmr&mbjob[description]=step1",
348             "?tool=jalview", iparams, true, false, '-');
349     // a priori knowledge of the data returned from the service
350     shmrService.addResultDatatype(JvDataType.ANNOTATION);
351     return new RestClient(shmrService);
352   }
353
354   public AlignmentPanel recoverAlignPanelForView()
355   {
356     AlignmentPanel[] aps = Desktop
357             .getAlignmentPanels(av.getSequenceSetId());
358     for (AlignmentPanel alp : aps)
359     {
360       if (alp.av == av)
361       {
362         return alp;
363       }
364     }
365     return null;
366   }
367
368   public boolean isShowResultsInNewView()
369   {
370     // TODO make this a property of the service
371     return true;
372   }
373
374   protected static Vector<String> services = null;
375
376   public static final String RSBS_SERVICES = "RSBS_SERVICES";
377
378   public static RestClient[] getRestClients()
379   {
380     if (services == null)
381     {
382       services = new Vector<String>();
383       try
384       {
385         for (RestServiceDescription descr : RestServiceDescription
386                 .parseDescriptions(jalview.bin.Cache.getDefault(
387                         RSBS_SERVICES,
388                         makeShmmrRestClient().service.toString())))
389         {
390           services.add(descr.toString());
391         }
392       } catch (Exception ex)
393       {
394         System.err
395                 .println("Serious - RSBS descriptions in user preferences are corrupt!");
396         ex.printStackTrace();
397       }
398
399     }
400     RestClient[] lst = new RestClient[services.size()];
401     int i = 0;
402     for (String svc : services)
403     {
404       lst[i++] = new RestClient(new RestServiceDescription(svc));
405     }
406     return lst;
407   }
408
409   public static void main(String args[])
410   {
411     try
412     {
413       RestClient[] clients = getRestClients();
414       System.out.println("Got " + clients.length + " clients.");
415       int i = 0;
416       Vector<String> urls = new Vector<String>();
417       for (RestClient cl : clients)
418       {
419         System.out.println("" + (++i) + ": " + cl.service.toString());
420         urls.add(cl.service.toString());
421       }
422       setRsbsServices(urls);
423       if (clients.length != getRestClients().length)
424       {
425         System.err
426                 .println("Failed. Differing numbers of clients when stringified and parsed again.");
427       }
428
429     } catch (Throwable x)
430     {
431       System.err.println("Failed. Unexpected exception.");
432       x.printStackTrace();
433     }
434   }
435
436   public String getAction()
437   {
438     return service.details.Action;
439   }
440
441   public RestServiceDescription getRestDescription()
442   {
443     return service;
444   }
445
446   public static Vector<String> getRsbsDescriptions()
447   {
448     Vector<String> rsbsDescrs = new Vector<String>();
449     for (RestClient rsbs : getRestClients())
450     {
451       rsbsDescrs.add(rsbs.getRestDescription().toString());
452     }
453     return rsbsDescrs;
454   }
455
456   public static void setRsbsServices(Vector<String> rsbsUrls)
457   {
458     if (rsbsUrls != null)
459     {
460       // TODO: consider validating services ?
461       services = new Vector<String>(rsbsUrls);
462       StringBuffer sprop = new StringBuffer();
463       for (String s : services)
464       {
465         sprop.append(s);
466       }
467       Cache.setProperty(RSBS_SERVICES, sprop.toString());
468     }
469     else
470     {
471       Cache.removeProperty(RSBS_SERVICES);
472     }
473   }
474
475 }